fix：报告详情新增AuthToken校验

controllers/report_share.go

@@ -4,6 +4,7 @@ import (
 	"encoding/json"
 	"eta/eta_report/models"
 	"eta/eta_report/services"
+	"eta/eta_report/utils"
 	"fmt"
 	"html"
 	"strings"
@@ -35,6 +36,7 @@ func (this *ReportShareController) Detail() {
 		br.ErrMsg = "参数错误，reportCode 为空"
 		return
 	}
+
 	report, err := models.GetReportByCode(reportCode)
 	if err != nil {
 		br.Msg = "该报告已删除"
@@ -164,16 +166,32 @@ func (this *ReportShareController) Detail() {
 		return
 	}
 
+	var hasAuth bool
+	authToken := this.GetString("AuthToken")
+	if authToken != "" {
+		key := fmt.Sprint(utils.CACHE_REPORT_AUTH, authToken)
+		redisReportId, err := utils.Rc.GetUInt64(key)
+		if err == nil && int(redisReportId) == report.Id {
+			hasAuth = true
+		}
+	}
+
 	// 报告内图表授权
 	if v, ok := conf[models.BusinessConfIsOpenChartExpired]; ok {
 		if v == `true` {
-			tokenMap := make(map[string]string)
-			report.Content = services.HandleReportContent(report.Content, "add", tokenMap)
-			//report.ContentStruct = services.HandleReportContent(report.ContentStruct, "add", tokenMap)
+			resp.IsOpenChartExpired = true
 
-			for _, v := range reportChapters {
-				v.Content = services.HandleReportContent(v.Content, "add", tokenMap)
+			// 有权限才会添加token
+			if hasAuth {
+				tokenMap := make(map[string]string)
+				report.Content = services.HandleReportContent(report.Content, "add", tokenMap)
+				//report.ContentStruct = services.HandleReportContent(report.ContentStruct, "add", tokenMap)
+
+				for _, v := range reportChapters {
+					v.Content = services.HandleReportContent(v.Content, "add", tokenMap)
+				}
 			}
+
 		}
 	}
 

models/report.go

@@ -66,14 +66,15 @@ func GetReportByCode(reportCode string) (item *Report, err error) {
 }
 
 type ReportShareDetailResp struct {
-	Report           *ReportItem `description:"报告"`
-	Disclaimer       string      `description:"免责声明"`
-	H5ShareName      string      `description:"研报分享抬头"`
-	H5ReportShareImg string      `description:"研报分享图片"`
-	WatermarkChart   string      `description:"图表是否需要水印"`
-	WatermarkReport  string      `description:"报告是否需要水印"`
-	Hz               int
-	ReportLogo       string `description:"报告logo"`
+	Report             *ReportItem `description:"报告"`
+	Disclaimer         string      `description:"免责声明"`
+	H5ShareName        string      `description:"研报分享抬头"`
+	H5ReportShareImg   string      `description:"研报分享图片"`
+	WatermarkChart     string      `description:"图表是否需要水印"`
+	WatermarkReport    string      `description:"报告是否需要水印"`
+	Hz                 int
+	ReportLogo         string `description:"报告logo"`
+	IsOpenChartExpired bool   `description:"是否开启图表有效期鉴权/报告禁止复制"`
 }
 
 type ReportItem struct {

utils/constants.go

@@ -121,3 +121,7 @@ const (
 	DbNameAI          = "eta_ai"
 	DbNameWeekly      = "weekly_report"
 )
+
+const (
+	CACHE_REPORT_AUTH = "eta:report:auth:" //报告图表数据授权
+)

utils/redis.go

@@ -7,6 +7,8 @@ import (
 
 type RedisClient interface {
 	Get(key string) interface{}
+	GetStr(key string) string
+	GetUInt64(key string) (uint64, error)
 	RedisBytes(key string) (data []byte, err error)
 	RedisString(key string) (data string, err error)
 	RedisInt(key string) (data int, err error)

utils/redis/cluster_redis.go

@@ -86,6 +86,25 @@ func (rc *ClusterRedisClient) Get(key string) interface{} {
 	return data
 }
 
+// GetStr
+// @Description: 根据key获取字符串数据
+// @receiver rc
+// @param key
+// @return string
+func (rc *ClusterRedisClient) GetStr(key string) string {
+	return rc.redisClient.Get(context.TODO(), key).Val()
+}
+
+// GetUInt64
+// @Description: 根据key获取uint64数据
+// @receiver rc
+// @param key
+// @return int
+// @return error
+func (rc *ClusterRedisClient) GetUInt64(key string) (uint64, error) {
+	return rc.redisClient.Get(context.TODO(), key).Uint64()
+}
+
 // RedisBytes
 // @Description: 根据key获取字节编码数据
 // @receiver rc

utils/redis/standalone_redis.go

@@ -78,6 +78,25 @@ func (rc *StandaloneRedisClient) Get(key string) interface{} {
 	return data
 }
 
+// GetStr
+// @Description: 根据key获取字符串数据
+// @receiver rc
+// @param key
+// @return string
+func (rc *StandaloneRedisClient) GetStr(key string) string {
+	return rc.redisClient.Get(context.TODO(), key).Val()
+}
+
+// GetUInt64
+// @Description: 根据key获取uint64数据
+// @receiver rc
+// @param key
+// @return int
+// @return error
+func (rc *StandaloneRedisClient) GetUInt64(key string) (uint64, error) {
+	return rc.redisClient.Get(context.TODO(), key).Uint64()
+}
+
 // RedisBytes
 // @Description: 根据key获取字节编码数据
 // @receiver rc