视频播放链接优化

Author: TruthHun88

controllers/DocumentController.go

@@ -300,7 +300,7 @@ func (this *DocumentController) Read() {
 					poster, _ := sel.Attr("poster")
 					src, _ := sel.Attr("src")
 					if !(strings.HasPrefix(src, "https://") || strings.HasPrefix(src, "http://")) {
-						sign, _ := utils.GenerateMediaSign(src, time.Duration(utils.MediaDuration)*time.Second)
+						sign, _ := utils.GenerateMediaSign(src, time.Now().UnixNano(), time.Duration(utils.MediaDuration)*time.Second)
 						if strings.Contains(src, "?") {
 							src = src + "&sign=" + sign
 						} else {

controllers/api/CommonController.go

@@ -949,7 +949,7 @@ func (this *CommonController) handleReleaseV3(release, bookIdentify string) inte
 						}
 					}
 				} else {
-					if sign, err := utils.GenerateMediaSign(src, time.Duration(utils.MediaDuration)); err == nil {
+					if sign, err := utils.GenerateMediaSign(src, time.Now().UnixNano(), time.Duration(utils.MediaDuration)); err == nil {
 						if strings.Contains(src, "?") {
 							src = src + "&sign=" + sign
 						} else {

utils/jwt.go

@@ -12,7 +12,8 @@ import (
 
 // MediaClaims Media Claims
 type MediaClaims struct {
-	Path string
+	Path     string
+	UnixNano int64
 	jwt.StandardClaims
 }
 
@@ -52,7 +53,7 @@ func IsSignUsed(sign string) bool {
 }
 
 // GenerateSign 生成token
-func GenerateMediaSign(path string, expire ...time.Duration) (sign string, err error) {
+func GenerateMediaSign(path string, unixNano int64, expire ...time.Duration) (sign string, err error) {
 	path = strings.TrimLeft(path, "/")
 	// 默认过期时间为一个月
 	expireDuration := time.Now().Add(30 * 24 * time.Hour)
@@ -62,6 +63,7 @@ func GenerateMediaSign(path string, expire ...time.Duration) (sign string, err e
 
 	customClaims := &MediaClaims{
 		path,
+		unixNano,
 		jwt.StandardClaims{
 			ExpiresAt: expireDuration.Unix(),
 		},