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

---
 service/more_units.go |   24 ++++++++++++++++++++++++
 1 files changed, 24 insertions(+), 0 deletions(-)

diff --git a/service/more_units.go b/service/more_units.go
index 6982f4e..20c8d84 100644
--- a/service/more_units.go
+++ b/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
+}

--
Gitblit v1.8.0