提交kit/utils

Signed-off-by: lixxxww <941403820@qq.com>

kit/utils/excel.go

@@ -0,0 +1,46 @@
+package utils
+
+import (
+	"fmt"
+	"github.com/xuri/excelize/v2"
+	"reflect"
+)
+
+func ExportExcel(head []string, datas [][]any, filePath string) error {
+	excel := excelize.NewFile()
+	excel.SetSheetRow("Sheet1", "A1", &head)
+	for i, data := range datas {
+		axis := fmt.Sprintf("A%d", i+2)
+		excel.SetSheetRow("Sheet1", axis, &data)
+	}
+	excel.SaveAs(filePath)
+	return nil
+}
+
+func GetFileName(path, filename string) string {
+	return path + filename
+}
+
+func InterfaceToExcel(data any, fileName string) {
+	heads := make([]string, 0)
+	datas := make([][]any, 0)
+	v := reflect.ValueOf(data)
+	max := int64(v.Len())
+	for i := 0; i < int(max); i++ {
+		u := v.Index(i).Interface()
+		var key = reflect.TypeOf(u)
+		var val = reflect.ValueOf(u)
+		num := key.NumField()
+		if i == 0 {
+			for i := 0; i < num; i++ {
+				heads = append(heads, key.Field(i).Name)
+			}
+		}
+		field := make([]any, 0)
+		for i := 0; i < num; i++ {
+			field = append(field, val.Field(i).Interface())
+		}
+		datas = append(datas, field)
+	}
+	ExportExcel(heads, datas, fileName)
+}