Excel中的一些概念

一個(gè)excel文件中可以包含多個(gè)sheet，一個(gè)sheet可以理解成一個(gè)表格

表格的每一行稱為Row

表格的每一行中的任意一個(gè)單元格稱為Cell

使用tealeg操作Excel

安裝tealeg

go get github.com/tealeg/xlsx

使用tealeg新建一個(gè)表格

tealeg提供了傻瓜式API，主要流程為創(chuàng)建一個(gè)Sheet，在Sheet中添加Row，然后在Row中添加每個(gè)單元格的值，最終持久化到磁盤。

func TestTealeg(t *testing.T) { excel := xlsx.NewFile()
 // 新建一個(gè)sheet sheet, err := excel.AddSheet("Sheet1") if err != nil {  t.Fatal(err) }
 // 創(chuàng)建首行 headerRow := sheet.AddRow()
 // 設(shè)置行高 headerRow.SetHeightCM(0.5)
 // 填充行中的單元格 headerRow.AddCell().Value = "姓名" headerRow.AddCell().Value = "年齡"
 valList := [][]string{  {"張三", "13"},  {"李四", "14"},  {"王五", "15"}, }
 for _, line := range valList {  row := sheet.AddRow()  row.SetHeightCM(0.5)  for _, v := range line {   row.AddCell().Value = v  } }
 // 持久化到磁盤 if err := excel.Save("username.xlsx"); err != nil {  t.Fatal(err) }}

執(zhí)行這個(gè)Test用例后，在項(xiàng)目的當(dāng)前文件夾中會(huì)出現(xiàn)一個(gè)username.xlsx的表格，內(nèi)容和前面介紹的excel概念中使用的表格是一個(gè)。

Go結(jié)合反射將結(jié)構(gòu)體轉(zhuǎn)換成Excel

大概思路是在Go的結(jié)構(gòu)體中每個(gè)屬性打上一個(gè)excel標(biāo)簽

利用反射獲取標(biāo)簽中的內(nèi)容，作為表格的Header

利用反射獲取Go結(jié)構(gòu)體中的屬性的值，組成一個(gè)map，key為從標(biāo)簽中excel反射獲取的值，val為結(jié)構(gòu)體屬性具體的值，map的偽代碼如下：map[excelTag]strucVal 類型為map[string]string。

如果一個(gè)array或者slice中的結(jié)構(gòu)體需要轉(zhuǎn)換成excel，那么只需要將每個(gè)元素轉(zhuǎn)換成第 3 步提到的map作為一行，組成一個(gè)[]map[excelTag]strucVal然后遍歷這個(gè)切片，追加到表格中即可。

反射獲取每個(gè)Struct中的Tag

func getStructTagList(v interface{}, tag string) []string { var resList []string if v == nil {  return resList } var item interface{} switch reflect.TypeOf(v).Kind() { case reflect.Slice, reflect.Array:  values := reflect.ValueOf(v)  if values.Len() == 0 {   return resList  }  item = values.Index(0).Interface() case reflect.Struct:  item = reflect.ValueOf(v).Interface() default:  panic(fmt.Sprintf("type %v not support", reflect.TypeOf(v).Kind())) } typeOf := reflect.TypeOf(item) fieldNum := typeOf.NumField() for i := 0; i < fieldNum; i++ {  resList = append(resList, typeOf.Field(i).Tag.Get(tag)) } return resList}

通過反射將結(jié)構(gòu)體的值轉(zhuǎn)換成map[excelTag]strucVal

func getTagValMap(v interface{}, tag string) map[string]string { resMap := make(map[string]string) if v == nil {  return resMap } typeOf := reflect.TypeOf(v) fieldNum := typeOf.NumField() for i := 0; i < fieldNum; i++ {  structField := typeOf.Field(i)  tagValue := structField.Tag.Get(tag)  val := reflect.ValueOf(v).FieldByName(structField.Name)  resMap[tagValue] = fmt.Sprintf("%v", val.Interface()) } return resMap}

利用反射將一個(gè)Silce，Array或者Struct轉(zhuǎn)換成[]map[excelTag]strucVal

func struct2MapTagList(v interface{}, tag string) []map[string]string { var resList []map[string]string switch reflect.TypeOf(v).Kind() { case reflect.Slice, reflect.Array:  values := reflect.ValueOf(v)  for i := 0; i < values.Len(); i++ {   resList = append(resList, getTagValMap(values.Index(i).Interface(), tag))  }  break case reflect.Struct:  val := reflect.ValueOf(v).Interface()  resList = append(resList, getTagValMap(val, tag))  break default:  panic(fmt.Sprintf("type %v not support", reflect.TypeOf(v).Kind())) } return resList}

通過tealeg將[]map[excelTag]strucVal轉(zhuǎn)換成Excel

通過上面兩步，已經(jīng)可以將一個(gè)結(jié)構(gòu)體，切片或者list轉(zhuǎn)換成了一個(gè)[]map[excelTag]strucVal類型的切片，下面我們只需要調(diào)用tealeg轉(zhuǎn)換成excel

func Struct2Xlsx(v interface{}) (*xlsx.File, error) { var tag = "excel" tagList := getStructTagList(v, tag) mapTagList := struct2MapTagList(v, tag) excelFile := xlsx.NewFile() sheet, err := excelFile.AddSheet("Sheet1") if err != nil {  return nil, err } headerRow := sheet.AddRow() for _, tagVal := range tagList {  headerRow.SetHeightCM(0.5)  headerRow.AddCell().Value = tagVal } for _, mapTagVal := range mapTagList {  row := sheet.AddRow()  for _, tagVal := range tagList {   row.SetHeightCM(0.5)   row.AddCell().Value = mapTagVal[tagVal]  } } return excelFile, nil}

運(yùn)行測(cè)試用例驗(yàn)證

type Username struct { Name string `excel:"姓名"` Age  int    `excel:"年齡"`}
func TestStruct2Excet(t *testing.T) { var data = []Username{  {   Name: "張三",   Age:  13,  },  {   Name: "李四",   Age:  14,  },  {   Name: "王五",   Age:  15,  }, } excel, err := Struct2Xlsx(data) if err != nil {  t.Fatal(err) } if err := excel.Save("username.xlsx"); err != nil {  t.Fatal(err) }}

測(cè)試運(yùn)行成功后，會(huì)在項(xiàng)目目錄創(chuàng)建一個(gè)username.xlsx的文件，這個(gè)文件就是我們的結(jié)構(gòu)體轉(zhuǎn)換成excel的結(jié)果