发布代码生成、更新20+表单组件，优化数据字典，gf版本更新到2.3.1

server/utility/excel/excel.go

@@ -7,42 +7,63 @@
 package excel
 
 import (
+	"context"
 	"fmt"
+	"github.com/gogf/gf/v2/errors/gcode"
+	"github.com/gogf/gf/v2/errors/gerror"
 	"github.com/gogf/gf/v2/net/ghttp"
+	"github.com/gogf/gf/v2/os/gctx"
 	"github.com/gogf/gf/v2/util/gconv"
 	"github.com/xuri/excelize/v2"
+	"hotgo/internal/library/contexts"
+	"hotgo/internal/model"
 	"net/url"
 	"reflect"
+	"time"
 )
 
-func ExportByStruct(w *ghttp.ResponseWriter, titleList []string, data []interface{}, fileName string, sheetName string) error {
+var (
+	// 单元格表头
+	char = []string{"", "A", "B", "C", "D", "E", "F", "G", "H", "I", "J", "K", "L", "M", "N", "O", "P", "Q", "R", "S", "T", "U", "V", "W", "X", "Y", "Z"}
+	// 默认行样式
+	defaultRowStyle = `{"font":{"color":"#666666","size":13,"family":"arial"},"alignment":{"vertical":"center","horizontal":"center"}}`
+)
 
+// ExportByStructs 导出切片结构体
+func ExportByStructs(ctx context.Context, tags []string, list interface{}, fileName string, sheetName string) (err error) {
 	f := excelize.NewFile()
 	f.SetSheetName("Sheet1", sheetName)
+	_ = f.SetRowHeight("Sheet1", 1, 30)
 	header := make([]string, 0)
-	for _, v := range titleList {
+	for _, v := range tags {
 		header = append(header, v)
 	}
 
-	rowStyleID, _ := f.NewStyle(`{"font":{"color":"#666666","size":13,"family":"arial"},"alignment":{"vertical":"center","horizontal":"center"}}`)
+	rowStyleID, _ := f.NewStyle(defaultRowStyle)
+	if err != nil {
+		return
+	}
 	_ = f.SetSheetRow(sheetName, "A1", &header)
-	_ = f.SetRowHeight("Sheet1", 1, 30)
-	length := len(titleList)
-	headStyle := letter(length)
-	var lastRow string
-	var widthRow string
+
+	var (
+		length    = len(tags)
+		headStyle = letter(length)
+		lastRow   string
+		widthRow  string
+	)
+
 	for k, v := range headStyle {
 		if k == length-1 {
 			lastRow = fmt.Sprintf("%s1", v)
 			widthRow = v
 		}
 	}
-	if err := f.SetColWidth(sheetName, "A", widthRow, 30); err != nil {
+	if err = f.SetColWidth(sheetName, "A", widthRow, 30); err != nil {
 		return err
 	}
 
-	rowNum := 1
-	for _, v := range data {
+	var rowNum = 1
+	for _, v := range gconv.Interfaces(list) {
 		t := reflect.TypeOf(v)
 		value := reflect.ValueOf(v)
 		row := make([]interface{}, 0)
@@ -51,32 +72,60 @@ func ExportByStruct(w *ghttp.ResponseWriter, titleList []string, data []interfac
 			row = append(row, val)
 		}
 		rowNum++
-		err := f.SetSheetRow(sheetName, "A"+gconv.String(rowNum), &row)
-		_ = f.SetCellStyle(sheetName, fmt.Sprintf("A%d", rowNum), fmt.Sprintf("%s", lastRow), rowStyleID)
-		if err != nil {
+		if err = f.SetSheetRow(sheetName, "A"+gconv.String(rowNum), &row); err != nil {
+			return err
+		}
+		if err = f.SetCellStyle(sheetName, fmt.Sprintf("A%d", rowNum), fmt.Sprintf("%s", lastRow), rowStyleID); err != nil {
 			return err
 		}
 	}
-	disposition := fmt.Sprintf("attachment; filename=%s.xlsx", url.QueryEscape(fileName))
 
+	// 强转类型
+	writer := ghttp.RequestFromCtx(ctx).Response.Writer
+	w, ok := interface{}(writer).(*ghttp.ResponseWriter)
+	if !ok {
+		return gerror.New("Response.Writer uninitialized!")
+	}
 	w.Header().Set("Content-Type", "application/octet-stream")
-	w.Header().Set("Content-Disposition", disposition)
+	w.Header().Set("Content-Disposition", fmt.Sprintf("attachment; filename=%s.xlsx", url.QueryEscape(fileName)))
 	w.Header().Set("Content-Transfer-Encoding", "binary")
 	w.Header().Set("Access-Control-Expose-Headers", "Content-Disposition")
 
-	err := f.Write(w)
-	if err != nil {
+	if err := f.Write(w); err != nil {
 		return err
 	}
 
+	// 加入到上下文
+	contexts.SetResponse(ctx, &model.Response{
+		Code:      gcode.CodeOK.Code(),
+		Message:   "export successfully!",
+		Timestamp: time.Now().Unix(),
+		TraceID:   gctx.CtxId(ctx),
+	})
+
 	return nil
 }
 
-// Letter 遍历a-z
+// letter 生成完整的表头
 func letter(length int) []string {
 	var str []string
 	for i := 0; i < length; i++ {
-		str = append(str, string(rune('A'+i)))
+		str = append(str, numToChars(i))
 	}
 	return str
 }
+
+// numToChars 将数字转换为具体的表格表头名称
+func numToChars(num int) string {
+	var cols string
+	v := num
+	for v > 0 {
+		k := v % 26
+		if k == 0 {
+			k = 26
+		}
+		v = (v - k) / 26
+		cols = char[k] + cols
+	}
+	return cols
+}