go使用excelize導出xls

package excel

import (
	"bytes"
	"fmt"
	"github.com/kataras/iris/v12"
	"github.com/xuri/excelize/v2"
	"net/url"
	"reflect"
	"strconv"
)

// DownExcelFile 下載excel文件
func DownExcelFile(ctx iris.Context, fileName string, bt *bytes.Buffer) {
	//設置文件類型
	ctx.Header("Content-Type", "application/vnd.ms-excel;charset=utf8")
	//設置文件名稱
	ctx.Header("Content-Disposition", "attachment; filename="+url.QueryEscape(fileName))
	_, _ = ctx.Write(bt.Bytes())
}

// WriteExcel 寫入excel
func WriteExcel(data []interface{}, sheetName string) (*bytes.Buffer, error) {

	f := excelize.NewFile()
	f.SetSheetName("Sheet1", sheetName)
	rowNum := 1
	//保存內容
	for i, u1 := range data {
		//讀取結構體
		p1 := reflect.TypeOf(u1)
		if i == 0 {
			fmt.Println(p1.String())
			//設置表頭
			header := make([]string, 0)
			for j := 0; j < p1.NumField(); j++ {
				key := p1.Field(j)
				fmt.Println("name=", key.Name, ",tag=", key.Tag.Get("xlsx"))
				xlsxTag := key.Tag.Get("xlsx")
				if xlsxTag != "" {
					header = append(header, xlsxTag)
				}
			}
			f.SetSheetRow(sheetName, "A1", &header)

		}
		//讀取值
		v1 := reflect.ValueOf(u1)
		sp1 := make([]interface{}, 0)
		for l := 0; l < p1.NumField(); l++ {
			key := p1.Field(l)
			xlsxTag := key.Tag.Get("xlsx")
			if xlsxTag != "" {
				val := v1.Field(l).Interface()
				sp1 = append(sp1, val)
			}
		}
		rowNum++
		f.SetSheetRow(sheetName, "A"+strconv.Itoa(rowNum), &sp1)
	}
	return f.WriteToBuffer()
}

// WriteExcelHeader 設置excel表頭
func WriteExcelHeader(data interface{}, sheetName string) (*bytes.Buffer, error) {

	f := excelize.NewFile()
	f.SetSheetName("Sheet1", sheetName)
	p1 := reflect.TypeOf(data)
	//設置表頭
	header := make([]string, 0)
	for j := 0; j < p1.NumField(); j++ {
		key := p1.Field(j)
		fmt.Println("name=", key.Name, ",tag=", key.Tag.Get("xlsx"))
		xlsxTag := key.Tag.Get("xlsx")
		if xlsxTag != "" {
			header = append(header, xlsxTag)
		}
	}
	f.SetSheetRow(sheetName, "A1", &header)
	return f.WriteToBuffer()
}