init...
This commit is contained in:
@@ -0,0 +1,143 @@
|
||||
package excels
|
||||
|
||||
import (
|
||||
"github.com/druidcaesa/gotool"
|
||||
"github.com/xuri/excelize/v2"
|
||||
"reflect"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
// 获取title头list
|
||||
func getTitle(list []map[string]string) []string {
|
||||
titleList := make([]string, 0)
|
||||
for _, item := range list {
|
||||
for _, s := range item {
|
||||
titleList = append(titleList, s)
|
||||
}
|
||||
}
|
||||
return titleList
|
||||
}
|
||||
|
||||
// ExportExcel 导出excel
|
||||
func ExportExcel(list []interface{}, title string) (error, *excelize.File) {
|
||||
// 获取标题
|
||||
headerList, expList := ExcelCreate(list[0])
|
||||
headers := getTitle(headerList)
|
||||
// 默认存在第一个工作簿是 Sheet1 首字母要大写, 否则会报错。
|
||||
// 如果想额外的创建工作簿, 可以使用, sheet2 := file.NewSheet("Sheet2"), 工作簿的名称不区分大小写。
|
||||
// 如果有多个工作簿, 可以使用 file.SetActiveSheet(index) 来指定打开文件时focus到哪个工作簿
|
||||
sheet1 := "Sheet1"
|
||||
files := excelize.NewFile()
|
||||
character := string(65 + len(headers) - 1)
|
||||
/* -------------------- 第一行大标题 -------------------- */
|
||||
// 设置行高
|
||||
err := files.SetRowHeight(sheet1, 1, 25)
|
||||
if err != nil {
|
||||
return err, nil
|
||||
}
|
||||
// 合并单元格
|
||||
err = files.MergeCell(sheet1, "A1", character+"1")
|
||||
if err != nil {
|
||||
return err, nil
|
||||
}
|
||||
// 设置单元格样式：对齐；字体, 大小；单元格边框
|
||||
styleTitle, _ := files.NewStyle(`{"alignment":{"horizontal":"center","vertical":"center"},"font":{"bold":true,"italic":false,"family":"Calibri","size":16,"color":"#000000"},"border":[{"type":"left","color":"#3FAD08","style":0},{"type":"top","color":"#3FAD08","style":0},{"type":"bottom","color":"#3FAD08","style":2},{"type":"right","color":"#3FAD08","style":0}]}`)
|
||||
err = files.SetCellStyle(sheet1, "A1", character+"1", styleTitle)
|
||||
if err != nil {
|
||||
return err, nil
|
||||
}
|
||||
|
||||
err = files.SetCellValue(sheet1, "A1", title)
|
||||
if err != nil {
|
||||
return err, nil
|
||||
}
|
||||
|
||||
/* -------------------- 字段标题 -------------------- */
|
||||
styleHeader, _ := files.NewStyle(`{"alignment":{"horizontal":"center","vertical":"center"},"font":{"bold":false,"italic":false,"family":"Calibri","size":10,"color":"#000000"}}`)
|
||||
err = files.SetCellStyle(sheet1, "A2", character+"2", styleHeader)
|
||||
if err != nil {
|
||||
return err, nil
|
||||
}
|
||||
for k, v := range headers {
|
||||
err = files.SetCellValue(sheet1, string(65+k)+"2", v)
|
||||
if err != nil {
|
||||
return err, nil
|
||||
}
|
||||
}
|
||||
// 设置最后一列宽度
|
||||
err = files.SetColWidth(sheet1, "C", character, 20)
|
||||
if err != nil {
|
||||
return err, nil
|
||||
}
|
||||
// 冻结窗口：冻结第一行和第二行
|
||||
err = files.SetPanes(sheet1, `{"freeze":true,"split":false,"x_split":0,"y_split":2}`)
|
||||
if err != nil {
|
||||
return err, nil
|
||||
}
|
||||
///* -------------------- 填充行数据 -------------------- */
|
||||
line := 3
|
||||
for _, v := range list {
|
||||
var num = 0
|
||||
lineChr := strconv.Itoa(line)
|
||||
// 设置样式
|
||||
err = files.SetCellStyle(sheet1, "A"+lineChr, character+lineChr, styleHeader)
|
||||
if err != nil {
|
||||
return err, nil
|
||||
}
|
||||
// 反射获取数据和类型
|
||||
getValue := reflect.ValueOf(v)
|
||||
getType := reflect.TypeOf(v)
|
||||
n := getValue.NumField()
|
||||
for i := 0; i < n; i++ {
|
||||
val := getValue.Field(i)
|
||||
name := getType.Field(i).Name
|
||||
if !getIsTitle(name, headerList) {
|
||||
continue
|
||||
}
|
||||
err = files.SetCellValue(sheet1, string(65+num)+lineChr, getExp(name, expList, val.Interface()))
|
||||
if err != nil {
|
||||
return err, nil
|
||||
}
|
||||
num++
|
||||
}
|
||||
line++
|
||||
}
|
||||
return nil, files
|
||||
}
|
||||
|
||||
func getIsTitle(name string, headerList []map[string]string) bool {
|
||||
flag := false
|
||||
for _, m := range headerList {
|
||||
if gotool.StrUtils.HasNotEmpty(m[name]) {
|
||||
flag = true
|
||||
}
|
||||
}
|
||||
return flag
|
||||
}
|
||||
|
||||
func getExp(name string, expList []map[string][]map[string]string, value interface{}) interface{} {
|
||||
for _, m := range expList {
|
||||
if len(m[name]) > 0 {
|
||||
maps := m[name]
|
||||
for _, m2 := range maps {
|
||||
if gotool.StrUtils.HasNotEmpty(m2[interfaceToString(value)]) {
|
||||
value = m2[interfaceToString(value)]
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
return value
|
||||
}
|
||||
|
||||
func interfaceToString(inter interface{}) string {
|
||||
switch inter.(type) {
|
||||
case string:
|
||||
return inter.(string)
|
||||
case int:
|
||||
return strconv.Itoa(inter.(int))
|
||||
case int64:
|
||||
return strconv.FormatInt(inter.(int64), 10)
|
||||
default:
|
||||
return inter.(string)
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,84 @@
|
||||
package excels
|
||||
|
||||
import (
|
||||
"reflect"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// 定义tagName名称
|
||||
const tagName = "excel"
|
||||
|
||||
// Excels 定义Excel接口
|
||||
type Excels interface {
|
||||
Excels(interface{}) (bool, string)
|
||||
}
|
||||
|
||||
// DefaultExcels 默认Excel配置
|
||||
type DefaultExcels struct {
|
||||
}
|
||||
|
||||
// Excels 实现接口
|
||||
func (e DefaultExcels) Excels(val interface{}) (bool, string) {
|
||||
return true, ""
|
||||
}
|
||||
|
||||
// `excel:"name='',readConverterExp=''"`
|
||||
// 获取标题头
|
||||
func getExcelFromTagTitle(proName string, tag string) map[string]string {
|
||||
args := strings.SplitN(tag, ",", 2)
|
||||
m := make(map[string]string)
|
||||
for i := 0; i < len(args); i++ {
|
||||
contains := strings.Contains(args[i], "name")
|
||||
if contains {
|
||||
m[proName] = strings.Split(args[i], "=")[1]
|
||||
break
|
||||
}
|
||||
}
|
||||
return m
|
||||
}
|
||||
|
||||
// `excel:"name='',readConverterExp='Y=1,N=2'"`
|
||||
func getReadConverterExp(proName string, tag string) map[string][]map[string]string {
|
||||
list := make(map[string][]map[string]string, 0)
|
||||
args := strings.SplitN(tag, ",", 2)
|
||||
for i := 0; i < len(args); i++ {
|
||||
if strings.Contains(args[i], "format") {
|
||||
maps := make([]map[string]string, 0)
|
||||
converter := strings.SplitN(args[i], "=", 2)[1]
|
||||
values := strings.Split(converter, ",")
|
||||
for j := 0; j < len(values); j++ {
|
||||
split := strings.Split(values[j], "=")
|
||||
m := make(map[string]string)
|
||||
m[split[0]] = split[1]
|
||||
maps = append(maps, m)
|
||||
}
|
||||
list[proName] = maps
|
||||
break
|
||||
}
|
||||
}
|
||||
return list
|
||||
}
|
||||
|
||||
// ExcelCreate 获取name名称和readConverterExp集合
|
||||
func ExcelCreate(s interface{}) ([]map[string]string, []map[string][]map[string]string) {
|
||||
titleList := make([]map[string]string, 0)
|
||||
expList := make([]map[string][]map[string]string, 0)
|
||||
// 获取值域
|
||||
v := reflect.ValueOf(s)
|
||||
t := reflect.TypeOf(s)
|
||||
var title map[string]string
|
||||
var exp map[string][]map[string]string
|
||||
// 遍历反射树形
|
||||
for i := 0; i < t.NumField(); i++ {
|
||||
tag := v.Type().Field(i).Tag.Get(tagName)
|
||||
if tag == "" || tag == "-" {
|
||||
continue
|
||||
}
|
||||
name := t.Field(i).Name
|
||||
title = getExcelFromTagTitle(name, tag)
|
||||
exp = getReadConverterExp(name, tag)
|
||||
titleList = append(titleList, title)
|
||||
expList = append(expList, exp)
|
||||
}
|
||||
return titleList, expList
|
||||
}
|
||||
Reference in New Issue
Block a user