zhangqian
2024-07-04 c950586b718ae6fc198bedf424609a4ac94cb5d1
service/more_units.go
@@ -1,6 +1,7 @@
package service
import (
   "errors"
   "github.com/shopspring/decimal"
   "github.com/xuri/excelize/v2"
   "strconv"
@@ -10,8 +11,11 @@
func CreateMoreUnit(amount decimal.Decimal, units []models.UnitItems) []models.UnitItems {
   moreValueArr := make([]models.UnitItems, 0, len(units))
   for _, unitItem := range units {
      if unitItem.Amount.IsZero() {
         continue
      }
      moreValueArr = append(moreValueArr, models.UnitItems{
         Amount:   amount.Mul(unitItem.Amount),
         Amount:   amount.Div(unitItem.Amount),
         Unit:     unitItem.Unit,
         Floating: unitItem.Floating,
      })
@@ -33,3 +37,54 @@
   }
   return
}
func SetExcelHeader(headers []interface{}, f *excelize.File) (err error) {
   // 设置表头
   var i int
   for _, h := range headers {
      if v, ok := h.(string); ok {
         err = f.SetCellValue("Sheet1", getColumnAlphabet(i)+"1", v)
         if err != nil {
            return
         }
         i++
      } else if childHeaders, ok := h.(map[string][]string); ok {
         for title, list := range childHeaders {
            err = f.SetCellValue("Sheet1", getColumnAlphabet(i)+"1", title)
            if err != nil {
               return err
            }
            err = f.MergeCell("Sheet1", getColumnAlphabet(i)+"1", getColumnAlphabet(i-1+len(list))+"1") // 合并单元格
            for _, t := range list {
               err = f.SetCellValue("Sheet1", getColumnAlphabet(i)+"2", t)
               if err != nil {
                  return err
               }
               i++
            }
         }
      } else {
         return errors.New("unsupported header value")
      }
   }
   return nil
}
func SetHeaderStyle(f *excelize.File) (styleInt int, err error) {
   // 设置表头样式
   style := &excelize.Style{
      Border: nil,
      Fill: excelize.Fill{
         Type:    "pattern",
         Pattern: 1,
         Shading: 0,
      },
      Font: &excelize.Font{
         Bold: true,
      },
      Alignment: &excelize.Alignment{
         Horizontal: "center",
      },
   }
   return f.NewStyle(style)
}