今天小编给大家分享一下Go如何将结构体转换成Excel的相关知识点,内容详细,逻辑清晰,相信大部分人都还太了解这方面的知识,所以分享这篇文章给大家参考一下,希望大家阅读完这篇文章后有所收获,下面我们一起来了解一下吧。
Excel中的一些概念
一个excel文件中可以包含多个sheet
,一个sheet
可以理解成一个表格
表格的每一行称为 Row
表格的每一行中的任意一个单元格称为 Cell
使用tealeg
操作Excel
安装tealeg
go get github.com/tealeg/xlsx
使用tealeg
新建一个表格
tealeg提供了傻瓜式API,主要流程为创建一个Sheet,在Sheet中添加Row,然后在Row中添加每个单元格的值,最终持久化到磁盘。
func TestTealeg(t *testing.T) {excel := xlsx.NewFile()// 新建一个sheetsheet, err := excel.AddSheet("Sheet1")if err != nil {t.Fatal(err)}// 创建首行headerRow := sheet.AddRow()// 设置行高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)}}
执行这个Test用例后,在项目的当前文件夹中会出现一个username.xlsx
的表格,内容和前面介绍的excel概念中使用的表格是一个。
Go结合反射将结构体转换成Excel
大概思路是在Go的结构体中每个属性打上一个
excel
标签利用反射获取标签中的内容,作为表格的Header
利用反射获取Go结构体中的属性的值,组成一个map,key为从标签中
excel
反射获取的值,val为结构体属性具体的值,map的伪代码如下:map[excelTag]strucVal
类型为map[string]string
。如果一个array或者slice中的结构体需要转换成excel,那么只需要将每个元素转换成第 3 步提到的map作为一行,组成一个
[]map[excelTag]strucVal
然后遍历这个切片,追加到表格中即可。
反射获取每个Struct中的Tag
func getStructTagList(v interface{}, tag string) []string {var resList []stringif 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}
通过反射将结构体的值转换成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}
利用反射将一个Silce,Array或者Struct转换成[]map[excelTag]strucVal
func struct2MapTagList(v interface{}, tag string) []map[string]string {var resList []map[string]stringswitch 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))}breakcase reflect.Struct:val := reflect.ValueOf(v).Interface()resList = append(resList, getTagValMap(val, tag))breakdefault:panic(fmt.Sprintf("type %v not support", reflect.TypeOf(v).Kind()))}return resList}
通过tealeg将[]map[excelTag]strucVal
转换成Excel
通过上面两步,已经可以将一个结构体,切片或者list转换成了一个[]map[excelTag]strucVal
类型的切片,下面我们只需要调用tealeg
转换成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}
运行测试用例验证
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)}}
测试运行成功后,会在项目目录创建一个username.xlsx
的文件,这个文件就是我们的结构体转换成excel的结果
以上就是“Go如何将结构体转换成Excel”这篇文章的所有内容,感谢各位的阅读!相信大家阅读完这篇文章都有很大的收获,小编每天都会为大家更新不同的知识,如果还想学习更多的知识,请关注编程网行业资讯频道。