Browse Source

优化报告短链

hsun 1 month ago
parent
commit
310044c883
1 changed files with 5 additions and 3 deletions
  1. 5 3
      services/report_v2.go

+ 5 - 3
services/report_v2.go

@@ -1509,6 +1509,7 @@ func GetReportShareUrlToken(req models.ReportShartUrlReq, adminId int) (linkToke
 	if linkToken != "" && utils.Rc.IsExist(utils.CACHE_REPORT_SHARE_ORIGIN_Url+linkToken) {
 		return
 	}
+	var tokenKey string
 
 	var ok bool
 	// 冲突检测
@@ -1522,7 +1523,8 @@ func GetReportShareUrlToken(req models.ReportShartUrlReq, adminId int) (linkToke
 		// 拼上报告标题
 		linkToken = fmt.Sprintf("%s %s", linkToken, req.Title)
 
-		ok = utils.Rc.IsExist(utils.CACHE_REPORT_SHARE_ORIGIN_Url + linkToken)
+		tokenKey = fmt.Sprint(utils.CACHE_REPORT_SHARE_ORIGIN_Url, utils.MD5(linkToken))
+		ok = utils.Rc.IsExist(tokenKey)
 		if !ok {
 			break
 		}
@@ -1533,7 +1535,7 @@ func GetReportShareUrlToken(req models.ReportShartUrlReq, adminId int) (linkToke
 		if err != nil {
 			return
 		}
-		err = utils.Rc.Put(utils.CACHE_REPORT_SHARE_ORIGIN_Url+linkToken, req.Url, time.Until(after))
+		err = utils.Rc.Put(tokenKey, req.Url, time.Until(after))
 		if err != nil {
 			return
 		}
@@ -1545,7 +1547,7 @@ func GetReportShareUrlToken(req models.ReportShartUrlReq, adminId int) (linkToke
 }
 
 func TransfromToOriginUrl(linkToken string) (originLink string, msg string, err error) {
-	cacheLinkKey := utils.CACHE_REPORT_SHARE_ORIGIN_Url + linkToken
+	cacheLinkKey := fmt.Sprint(utils.CACHE_REPORT_SHARE_ORIGIN_Url, utils.MD5(linkToken))
 	originLink, err = utils.Rc.RedisString(cacheLinkKey)
 	if err != nil {
 		if err == redis.Nil {