global/response/response.go: add download xlsx method

Author: 汪先森

server/global/response/response.go

@@ -1,8 +1,10 @@
 package response
 
 import (
+	"github.com/360EntSecGroup-Skylar/excelize"
 	"github.com/gin-gonic/gin"
 	"net/http"
+	"strconv"
 )
 
 type Response struct {
@@ -41,6 +43,20 @@ func OkDetailed(data interface{}, message string, c *gin.Context) {
 	Result(SUCCESS, data, message, c)
 }
 
+func OkWithXlsx(list []interface{}, title []string, fileName string, c *gin.Context) {
+	file := excelize.NewFile()
+	file.SetSheetRow("Sheet1", "A1", &title)
+	for i, v := range list {
+		// 第一行被title占用
+		lint := strconv.Itoa(i + 2)
+		file.SetSheetRow("Sheet1", "A"+lint, v)
+	}
+	c.Header("Content-Type", "application/octet-stream")
+	c.Header("Content-Disposition", "attachment; filename="+fileName)
+	c.Header("Content-Transfer-Encoding", "binary")
+	_ = file.Write(c.Writer)
+}
+
 func Fail(c *gin.Context) {
 	Result(ERROR, map[string]interface{}{}, "操作失败", c)
 }

server/go.mod

@@ -3,6 +3,7 @@ module gin-vue-admin
 go 1.12
 
 require (
+	github.com/360EntSecGroup-Skylar/excelize v1.4.1 // indirect
 	github.com/alecthomas/template v0.0.0-20190718012654-fb15b899a751
 	github.com/casbin/casbin v1.9.1
 	github.com/casbin/casbin/v2 v2.11.0