Go结合反射将结构体转换成Excel的过程详解
作者:范闲 发布时间:2024-05-05 09:26:49
标签:Go,结构体,转换,Exce
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()
// 新建一个sheet
sheet, 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 []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
}
通过反射将结构体的值转换成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]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
转换成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的结果
来源:https://juejin.cn/post/7106508948299055112
0
投稿
猜你喜欢
- 原作者:Nik Piepenbreier翻译&内容补充:费弗里原文地址:https://towardsdatascience.com
- 支持聚合函数的方法:提到聚合函数,首先我们要知道的就是这些聚合函数是不能在django中单独使用的,要想在django中使用这些聚合函数,就
- 一、基于json模块的存储、读取数据names_writer.pyimport jsonnames = ['joker',&
- 程序出错的时候,我们往往需要根据异常信息来找到具体出错的代码。简单地用print打印异常信息并不能很好地追溯出错的代码:# -*- codi
- 我的环境,Windows10,Python3.6.3查询了很多有关资料,发现都是Python2版本操作Word文件的,所以就写了这篇短小的文
- 1. 读取CSV文件 csv.reader()该方法的作用相当于就是通过 ',' 分割csv格式的数据,并将分割好的每行数据
- python调用mysql数据库通常通过mysqldb模块,简单说下如何调用1.安装驱动目前有两个MySQL的驱动,我们可以选择其中一个进行
- 协同过滤在 用户 —— 物品(user - item)的数据关系下很容易收集到一些偏好信息(preference),比如评分。利用这些分散的
- 前言ThinkPHP,是为了简化企业级应用开发和敏捷WEB应用开发而诞生的开源轻量级PHP框架。随着框架代码量的增加,一些潜在的威胁也逐渐暴
- 本节内容0、列表常用功能汇总1、定义列表2、访问列表中的元素3、切片4、追加5、插入6、修改7、拷贝8、删除9、扩展10、统计11、翻转12
- swagger介绍Swagger本质上是一种用于描述使用JSON表示的RESTful API的接口描述语言。Swagger与一组开源软件工具
- 1.Beautiful Soup库简介Beautiful Soup 简称 BS4(其中 4 表示版本号)是一个 Python 中常用的页面解
- 本文介绍了python 支持向量机非线性回归SVR模型,废话不多说,具体如下:import numpy as npimport matplo
- 1. 得到安全字符串,在查询中使用,过滤单引号。Function Get_SafeStr(str) &nb
- 一、insert语句insert用于向表中输入数据,其具体的语法结构如下。INSERT INTO 表名称 VALUES (值1, 值2,..
- <?php $search = array ("'<script[^>]*?>.*?</sc
- 只要为pycharm配置anaconda中的解析器即可。操作步骤:1. 在pycharm菜单栏中,点击File-Settings-Proje
- title: 利用Django实现一个能与用户交互的初级框架author: Sun-Winddate: September 1, 2021D
- 安装包 PyCharm笔者使用PyCharm2018.3.2,请根据机器是64位还是32位来选择对应的PyCharm版本。(相信绝大部分人都
- 闭包与defer1.闭包闭包 : 一个函数与其相关的引用环境组合的一个实体,其实可以理解为面向对象中类中的属性与方法。如代码块中,函数fun