抽离设置表头为公共方法

zhangqian

2024-07-31 3bbe212120c1199e574087f63afe8f384ffe16f5

 service/more_units.go

@@ -140,3 +140,27 @@
   }
   return
}

func SetExcelHeaders(headers []interface{}, f *excelize.File) (lastColumnNumber int, err error) {
   // 设置表头
   i := 1
   for _, h := range headers {
      if v, ok := h.(string); ok {
         f.SetCellValue("Sheet1", getColumnAlphabet(i)+"1", v)
         i++
      } else if childHeaders, ok := h.(map[string][]string); ok {
         for title, list := range childHeaders {
            f.SetCellValue("Sheet1", getColumnAlphabet(i)+"1", title)
            err = f.MergeCell("Sheet1", getColumnAlphabet(i)+"1", getColumnAlphabet(i-1+len(list))+"1") // 合并单元格
            if err != nil {
               return 1, err
            }
            for _, t := range list {
               f.SetCellValue("Sheet1", getColumnAlphabet(i)+"2", t)
               i++
            }
         }
      }
   }
   return i, nil
}