Merge branch 'master' of http://101.43.70.124:3000/williamet/cas_tt_cloud_backend

lib/excel.go

@@ -0,0 +1,142 @@
+package lib
+
+import (
+	"SciencesServer/utils"
+	"errors"
+	"fmt"
+	"github.com/tealeg/xlsx"
+)
+
+type Excel struct {
+	File *xlsx.File
+
+	name    string
+	sheet   string
+	title   []string
+	content *ExcelContent
+	//Style
+}
+
+type ExcelContent struct {
+	Kind    ExcelContentKind
+	Content []interface{}
+}
+
+type ExcelContentKind int
+
+const (
+	ExcelContentKindForStruct ExcelContentKind = iota + 1
+	ExcelContentKindForSlice
+	ExcelContentKindForText
+)
+
+type OptionExcel func(excel *Excel)
+
+func WithExcelName(name string) OptionExcel {
+	return func(excel *Excel) {
+		excel.name = name
+	}
+}
+
+func WithExcelSheet(sheet string) OptionExcel {
+	return func(excel *Excel) {
+		excel.sheet = sheet
+	}
+}
+
+func WithExcelTitle(title []string) OptionExcel {
+	return func(excel *Excel) {
+		excel.title = title
+	}
+}
+
+func WithExcelContent(content *ExcelContent) OptionExcel {
+	return func(excel *Excel) {
+		excel.content = content
+	}
+}
+
+func WithExcelStyle() OptionExcel {
+	return func(excel *Excel) {
+
+	}
+}
+
+// Export 导出
+func (this *Excel) Export() error {
+	this.File = xlsx.NewFile()
+	sheet, _ := this.File.AddSheet(this.sheet)
+	row := sheet.AddRow()
+	cell := new(xlsx.Cell)
+
+	style := &xlsx.Style{
+		//Fill:   *xlsx.NewFill("solid", "EFEFDE", "EFEFDE"),
+		Border: xlsx.Border{RightColor: "FF"},
+	}
+	if this.title != nil && len(this.title) > 0 {
+		for _, v := range this.title {
+			cell = row.AddCell()
+			cell.Value = v
+			cell.SetStyle(style)
+		}
+	}
+	if this.content != nil {
+		for _, v := range this.content.Content {
+			row = sheet.AddRow()
+
+			if this.content.Kind == ExcelContentKindForStruct {
+				row.WriteStruct(v, -1)
+				continue
+			}
+			if this.content.Kind == ExcelContentKindForSlice {
+				row.WriteSlice(v, -1)
+				continue
+			}
+			for i := 0; i < len(this.title); i++ {
+				cell = row.AddCell()
+				// TODO：存在不同类型，请自行加判断
+				switch v.(type) {
+				case []string:
+					cell.Value = v.([]string)[i]
+					break
+				case []int:
+					cell.Value = fmt.Sprintf("%v", v.([]int)[i])
+					break
+				case []interface{}:
+					cell.Value = fmt.Sprintf("%v", v.([]interface{})[i])
+					break
+				default:
+					return errors.New("未知数据类型")
+				}
+			}
+		}
+	}
+	return nil
+}
+
+// Save 保存
+func (this *Excel) Save(path string) error {
+	return this.File.Save(this.name + ".xlsx")
+	isExist, err := utils.PathExists(path)
+
+	if err != nil {
+		return err
+	}
+	if !isExist {
+		if err = utils.MkdirAll(path); err != nil {
+			return err
+		}
+	}
+	fmt.Printf("%s/%s.xlsx", path, this.name)
+	return nil
+	//return this.Save(fmt.Sprintf("%s\%s.xlsx", path, this.name))
+}
+
+func NewExcel(options ...OptionExcel) *Excel {
+	out := new(Excel)
+
+	for _, opt := range options {
+		opt(out)
+	}
+	return out
+}

lib/excel_test.go

@@ -0,0 +1,26 @@
+package lib
+
+import "testing"
+
+func TestNewExcel(t *testing.T) {
+	content := make([]interface{}, 0)
+
+	for i := 0; i < 2; i++ {
+		content = append(content, []interface{}{
+			"戴恒顺", i + 1 + 1,
+		})
+	}
+	excel := NewExcel(WithExcelName("测试"), WithExcelSheet("专家信息"),
+		WithExcelTitle([]string{"姓名", "年龄"}), WithExcelContent(&ExcelContent{
+			Kind:    0,
+			Content: content,
+		}))
+	err := excel.Export()
+
+	if err != nil {
+		t.Log(err)
+		return
+	}
+	err = excel.Save("./lib")
+	t.Log(err)
+}