From 3bbe212120c1199e574087f63afe8f384ffe16f5 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期三, 31 七月 2024 15:01:21 +0800
Subject: [PATCH] 抽离设置表头为公共方法

---
 service/location_forms.go |   48 +++++++++++-------------------------------------
 1 files changed, 11 insertions(+), 37 deletions(-)

diff --git a/service/location_forms.go b/service/location_forms.go
index 262fa59..28d686d 100644
--- a/service/location_forms.go
+++ b/service/location_forms.go
@@ -104,46 +104,20 @@
 	unitData, unitIndexMap := GetAllUnits()
 	// 鑷畾涔夎〃澶�
 	headers := []interface{}{"浣嶇疆", "浜у搧", "浜у搧绫诲埆", map[string][]string{"鍦ㄥ簱鏁伴噺": unitData}, "浠峰��"}
-
-	// 璁剧疆琛ㄥご
-	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") // 鍚堝苟鍗曞厓鏍�
-				for _, t := range list {
-					f.SetCellValue("Sheet1", getColumnAlphabet(i)+"2", t)
-					i++
-				}
-			}
-		}
-	}
-
-	// 璁剧疆琛ㄥご鏍峰紡
-	style := &excelize.Style{
-		Border: nil,
-		Fill: excelize.Fill{
-			Type:    "pattern",
-			Pattern: 1,
-			Shading: 0,
-		},
-		Font: &excelize.Font{
-			Bold: true,
-		},
-		Alignment: &excelize.Alignment{
-			Horizontal: "center",
-		},
-	}
-	titleStyle, err := f.NewStyle(style)
+	lastColumnNumber, err := SetExcelHeaders(headers, f)
 	if err != nil {
-		return
+		return "", err
+	}
+	f.MergeCell("Sheet1", "A1", "A2") // 鍚堝苟鍗曞厓鏍�
+	f.MergeCell("Sheet1", "B1", "B2") // 鍚堝苟鍗曞厓鏍�
+	f.MergeCell("Sheet1", "C1", "C2") // 鍚堝苟鍗曞厓鏍�
+
+	titleStyle, err := SetHeaderStyle(f)
+	if err != nil {
+		return "", err
 	}
 
-	lastColumn := getColumnAlphabet(i)
+	lastColumn := getColumnAlphabet(lastColumnNumber)
 	f.SetCellStyle("Sheet1", "A1", lastColumn+"2", titleStyle)
 	// 璁剧疆鍒楀
 	f.SetColWidth("Sheet1", "A", "C", 30)

--
Gitblit v1.8.0