GoLang+gin框架導出Excel

文章目錄

• • 使用模塊
  • 代碼實現

使用模塊

github.com/tealeg/xlsx

模塊地址

官方文檔地址

---

代碼實現

• 核心代碼func ToExcel、ResponseXls
• 篇幅原因這里只貼出核心代碼

import (
	"bytes"
	"fmt"
	"io"
	"net/http"
	"time"

	"github.com/gin-gonic/gin"
	"github.com/tealeg/xlsx"
)

type DownloadRoleInfoBo struct {
	Name        string    `json:"name"`
	Level       int       `json:"level"`
	Description string    `json:"description"`
	CreateTime  time.Time `json:"createTime"`
}

// API處理器函數
func DownRolesHandler(c *gin.Context) {
	var roleData []DownloadRoleInfoBo
	
	// 略過向 roleData 添加數據過程
	
	var res []interface{}
	for _, role := range roleData {
		res = append(res, &DownloadRoleInfoBo{
			Name:        role.Name,
			Level:       role.Level,
			Description: role.Description,
			CreateTime:  role.CreateTime,
		})
	}
	content := ToExcel([]string{`角色名稱`, `角色級別`, `描述`, `創建日期`}, res)
	ResponseXls(c, content, "角色數據")
}

// 生成io.ReadSeeker 參數 titleList 為Excel表頭，dataList 為數據
func ToExcel(titleList []string, dataList []interface{}) (content io.ReadSeeker) {
	// 生成一個新的文件
	file := xlsx.NewFile()
	// 添加sheet頁
	sheet, _ := file.AddSheet("Sheet1")
	// 插入表頭
	titleRow := sheet.AddRow()
	for _, v := range titleList {
		cell := titleRow.AddCell()
		cell.Value = v
	}
	// 插入內容
	for _, v := range dataList {
		row := sheet.AddRow()
		row.WriteStruct(v, -1)
	}

	var buffer bytes.Buffer
	_ = file.Write(&buffer)
	content = bytes.NewReader(buffer.Bytes())
	return
}

// 向前端返回Excel文件
// 參數 content 為上面生成的io.ReadSeeker， fileTag 為返回前端的文件名
func ResponseXls(c *gin.Context, content io.ReadSeeker, fileTag string) {
	fileName := fmt.Sprintf("%s%s%s.xlsx", NowTime(), `-`, fileTag)
	c.Writer.Header().Add("Content-Disposition", fmt.Sprintf(`attachment; filename="%s"`, fileName))
	c.Writer.Header().Add("Content-Type", "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet")
	http.ServeContent(c.Writer, c.Request, fileName, time.Now(), content)
}