liulei/jx-callback
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

- colWithList in Obj2Excel

Browse Source
This commit is contained in:
gazebo
2019-07-16 18:44:48 +08:00
parent 70b8444bd8
commit 13409e141f
1 changed files with 21 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
business/jxutils/excel/excel.go
View File

@@ -46,10 +46,14 @@ func Obj2Excel(sheetList []*Obj2ExcelSheetConfig) []byte {
excelFile.NewSheet(sheetConfig.Title) excelFile.NewSheet(sheetConfig.Title)
} }
isMemberStruct := typeInfo.Kind() == reflect.Struct isMemberStruct := typeInfo.Kind() == reflect.Struct
colWithList := make([]float64, len(sheetConfig.CaptionList))
if isMemberStruct { if isMemberStruct {
var indexSlice [][]int var indexSlice [][]int
name2IndexMap := utils.GetStructNameIndex(typeInfo, "json") name2IndexMap := utils.GetStructNameIndex(typeInfo, "json")
for col, name := range sheetConfig.CaptionList { for col, name := range sheetConfig.CaptionList {
if colWith := float64(len(name)); colWith > colWithList[col] {
colWithList[col] = colWith
}
if _, ok := name2IndexMap[name]; !ok { if _, ok := name2IndexMap[name]; !ok {
panic(fmt.Sprintf("col:%s不能找到相应的数据", name)) panic(fmt.Sprintf("col:%s不能找到相应的数据", name))
} }
@@ -58,7 +62,11 @@ func Obj2Excel(sheetList []*Obj2ExcelSheetConfig) []byte {
} }
for i := 0; i < valueInfo.Len(); i++ { for i := 0; i < valueInfo.Len(); i++ {
for col, index := range indexSlice { for col, index := range indexSlice {
excelFile.SetCellStr(sheetConfig.Title, genAxis(i+1, col), fmt.Sprint(reflect.Indirect(valueInfo.Index(i)).FieldByIndex(index).Interface())) str := fmt.Sprint(reflect.Indirect(valueInfo.Index(i)).FieldByIndex(index).Interface())
// if colWith := float64(len(str)); colWith > colWithList[col] {
// colWithList[col] = colWith
// }
excelFile.SetCellStr(sheetConfig.Title, genAxis(i+1, col), str)
} }
} }
} else { } else {
@@ -71,6 +79,9 @@ func Obj2Excel(sheetList []*Obj2ExcelSheetConfig) []byte {
} }
} }
for col, name := range sheetConfig.CaptionList { for col, name := range sheetConfig.CaptionList {
if colWith := float64(len(name)); colWith > colWithList[col] {
colWithList[col] = colWith
}
if name2IndexMap != nil { if name2IndexMap != nil {
if _, ok := name2IndexMap[name]; !ok { if _, ok := name2IndexMap[name]; !ok {
panic(fmt.Sprintf("col:%s不能找到相应的数据", name)) panic(fmt.Sprintf("col:%s不能找到相应的数据", name))
@@ -81,10 +92,18 @@ func Obj2Excel(sheetList []*Obj2ExcelSheetConfig) []byte {
for i := 0; i < valueInfo.Len(); i++ { for i := 0; i < valueInfo.Len(); i++ {
mapData := valueInfo.Index(i).Interface().(map[string]interface{}) mapData := valueInfo.Index(i).Interface().(map[string]interface{})
for col, name := range sheetConfig.CaptionList { for col, name := range sheetConfig.CaptionList {
excelFile.SetCellStr(sheetConfig.Title, genAxis(i+1, col), fmt.Sprint(mapData[name])) str := fmt.Sprint(mapData[name])
// if colWith := float64(len(str)); colWith > colWithList[col] {
// colWithList[col] = colWith
// }
excelFile.SetCellStr(sheetConfig.Title, genAxis(i+1, col), str)
} }
} }
} }
// for col := range sheetConfig.CaptionList {
// colName, _ := excelize.ColumnNumberToName(col + 1)
// excelFile.SetColWidth(sheetConfig.Title, colName, colName, colWithList[col])
// }
} }
buf := &bytes.Buffer{} buf := &bytes.Buffer{}
excelFile.Write(buf) excelFile.Write(buf)