feat(video): add user id info to video link

GuGoOrg · Aug 21, 2023 · 38d7953 · 38d7953
1 parent 4d42635
commit 38d7953
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/src/services/feed/handler.go b/src/services/feed/handler.go
@@ -202,7 +202,7 @@ func queryDetailed(
 		// fill play url
 		go func(i int, v *models.Video) {
 			defer wg.Done()
-			playUrl, localErr := file.GetLink(ctx, v.FileName)
+			playUrl, localErr := file.GetLink(ctx, v.FileName, v.UserId)
 			if localErr != nil {
 				logger.WithFields(logrus.Fields{
 					"video_id":  v.ID,
@@ -218,7 +218,7 @@ func queryDetailed(
 		// fill cover url
 		go func(i int, v *models.Video) {
 			defer wg.Done()
-			coverUrl, localErr := file.GetLink(ctx, v.CoverName)
+			coverUrl, localErr := file.GetLink(ctx, v.CoverName, v.UserId)
 			if localErr != nil {
 				logger.WithFields(logrus.Fields{
 					"video_id":   v.ID,

diff --git a/src/storage/file/provider.go b/src/storage/file/provider.go
@@ -3,6 +3,7 @@ package file
 import (
 	"GuGoTik/src/constant/config"
 	"context"
+	"fmt"
 	"io"
 )
 
@@ -31,6 +32,8 @@ func GetLocalPath(ctx context.Context, fileName string) string {
 	return client.GetLocalPath(ctx, fileName)
 }
 
-func GetLink(ctx context.Context, fileName string) (link string, err error) {
-	return client.GetLink(ctx, fileName)
+func GetLink(ctx context.Context, fileName string, userId uint32) (link string, err error) {
+	originLink, err := client.GetLink(ctx, fileName)
+	link = fmt.Sprintf("%s?user_id=%d", originLink, userId)
+	return
 }