From e718c2568e9712ab40dc4d2d4d20cf3635dc4c34 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期五, 07 六月 2024 14:10:49 +0800
Subject: [PATCH] 月度统计报表下载接口

---
 controllers/report_forms_controller.go |   59 ++++++++++++++++++++++++++++++++++++++++++++++-------------
 1 files changed, 46 insertions(+), 13 deletions(-)

diff --git a/controllers/report_forms_controller.go b/controllers/report_forms_controller.go
index 6d7b0ce..17288a7 100644
--- a/controllers/report_forms_controller.go
+++ b/controllers/report_forms_controller.go
@@ -8,7 +8,6 @@
 	"wms/constvar"
 	"wms/extend/code"
 	"wms/extend/util"
-	"wms/models"
 	"wms/pkg/logx"
 	"wms/request"
 	"wms/service"
@@ -83,11 +82,6 @@
 	if err != nil {
 		logx.Errorf("DownloadInventoryForms Export err:%v", err)
 		util.ResponseFormat(c, code.InternalError, "瀵煎嚭鏁版嵁鍒版枃浠跺け璐�")
-		return
-	}
-
-	if err != nil {
-		util.ResponseFormat(c, code.RequestParamError, "瀵煎嚭澶辫触")
 		return
 	}
 
@@ -306,11 +300,6 @@
 		return
 	}
 
-	if err != nil {
-		util.ResponseFormat(c, code.RequestParamError, "瀵煎嚭澶辫触")
-		return
-	}
-
 	fileContentDisposition := "attachment;filename=\"" + url.QueryEscape(filename) + "\""
 	c.Header("Content-Type", "application/xlsx")
 	c.Header("Content-Disposition", fileContentDisposition)
@@ -333,13 +322,57 @@
 		return
 	}
 
-	list, total, err := models.NewMonthStatsSearch().SetPage(params.Page, params.PageSize).SetKeyword(params.Keyword).SetDate(params.Date).Find()
+	monthFormsService := service.NewMonthFormsService()
+	total, err := monthFormsService.Count(params)
 	if err != nil {
+		logx.Errorf("MonthStats count err:%v", err)
+		util.ResponseFormat(c, code.InternalError, "鏌ヨ鎬绘暟澶辫触")
+		return
+	}
+	result, err := monthFormsService.Query(params)
+	if err != nil {
+		logx.Errorf("MonthStats query err:%v", err)
 		util.ResponseFormat(c, code.InternalError, "鏌ヨ澶辫触")
 		return
 	}
 
-	util.ResponseFormatList(c, code.Success, list, int(total))
+	util.ResponseFormatList(c, code.Success, result, int(total))
+}
+
+// DownloadMonthStats
+// @Tags      鎶ヨ〃
+// @Summary   涓嬭浇鏈堝害缁熻搴撳瓨鎶ヨ〃
+// @Produce   application/json
+// @Param     object  body  request.GetMonthStats true  "鏌ヨ鍙傛暟"
+// @Param     Authorization	header string true "token"
+// @Success   200 {object} util.ResponseList{data=[]models.MonthStats}	"鎴愬姛"
+// @Router    /api-wms/v1/forms/downloadMonthStats [post]
+func (slf ReportFormsController) DownloadMonthStats(c *gin.Context) {
+	var params request.GetMonthStats
+	if err := c.BindJSON(&params); err != nil {
+		util.ResponseFormat(c, code.RequestParamError, "鍙傛暟瑙ｆ瀽澶辫触锛屾暟鎹被鍨嬮敊璇�")
+		return
+	}
+
+	monthFormsService := service.NewMonthFormsService()
+	list, err := monthFormsService.FetchAll(params)
+	if err != nil {
+		logx.Errorf("DownloadMonthStats FetchAll err:%v", err)
+		util.ResponseFormat(c, code.InternalError, "鏌ヨ澶辫触")
+		return
+	}
+	filename, err := monthFormsService.Export(list)
+	if err != nil {
+		logx.Errorf("DownloadMonthStats Export err:%v", err)
+		util.ResponseFormat(c, code.InternalError, "瀵煎嚭鏁版嵁鍒版枃浠跺け璐�")
+		return
+	}
+
+	fileContentDisposition := "attachment;filename=\"" + url.QueryEscape(filename) + "\""
+	c.Header("Content-Type", "application/xlsx")
+	c.Header("Content-Disposition", fileContentDisposition)
+	c.File(filename)
+	defer os.Remove(filename)
 }
 
 // DoMonthStats

--
Gitblit v1.8.0