feat:报告数据加密

controllers/base_common.go

@@ -4,7 +4,9 @@ import (
 	"encoding/json"
 	"fmt"
 	"github.com/beego/beego/v2/server/web"
+	"hongze/hongze_open_api/services"
 	"hongze/hongze_open_api/utils"
+	"net/http"
 	"net/url"
 )
 
@@ -42,8 +44,23 @@ func (c BaseCommon) Result() {
 
 	//不将errMsg暴露给用户
 	c.Response.ErrMsg = c.Response.Msg
-	c.Controller.Data["json"] = c.Response
-	c.Controller.ServeJSON()
+	//c.Controller.Data["json"] = c.Response
+	//c.Controller.ServeJSON()
+
+	// 将处理后的数据返回给前端
+	content, err = json.Marshal(c.Response)
+	if err != nil {
+		http.Error(c.Ctx.Output.Context.ResponseWriter, err.Error(), http.StatusInternalServerError)
+		return
+	}
+	//content = []byte(utils.StringsToJSON(string(content)))
+	// 数据加密
+	if services.CheckEncryption(c.Ctx.Request.URL.Path) {
+		content = utils.DesBase64Encrypt(content)
+	}
+	c.Ctx.Output.Header("Content-Type", "application/json; charset=utf-8")
+	c.Ctx.Output.Body(content)
+
 	c.StopRun()
 }
 

services/verify.go

@@ -0,0 +1,20 @@
+package services
+
+// AfterHandlerUrlMap 结束后待处理的url
+var EncryptionPathMap = map[string]int{
+	"/api/report/getReportInfo":           1, //报告详情接口
+	"/api/report/getReportChapterInfo":    1, //报告章节详情接口
+	"/api/report/getReportInfo/v2":        1, //报告详情接口V2
+	"/api/report/getReportChapterInfo/v2": 1, //报告章节详情接口V2
+	"/api/report/getTickerData":           1, //表格数据
+}
+
+// CheckEncryption 校验需要是否加密
+func CheckEncryption(urlPath string) (isEncryption bool) {
+	//if utils.RunMode == "release" {
+	if _, ok := EncryptionPathMap[urlPath]; ok {
+		isEncryption = true
+	}
+	//}
+	return
+}