fix: relation isfollow cache

src/services/feed/handler.go

@@ -19,6 +19,7 @@ import (
 	"context"
 	"encoding/json"
 	"errors"
+	"fmt"
 	amqp "github.com/rabbitmq/amqp091-go"
 	"gorm.io/gorm"
 	"strconv"
@@ -361,7 +362,7 @@ func (s FeedServiceImpl) QueryVideoExisted(ctx context.Context, req *feed.VideoE
 	logging.SetSpanWithHostname(span)
 	logger := logging.LogService("FeedService.QueryVideoExisted").WithContext(ctx)
 	var video models.Video
-	_, err = cached.GetWithFunc(ctx, "VideoExistedCached", func(ctx context.Context, key string) (string, error) {
+	_, err = cached.GetWithFunc(ctx, fmt.Sprintf("VideoExistedCached-%d", req.VideoId), func(ctx context.Context, key string) (string, error) {
 		row := database.Client.WithContext(ctx).Where("id = ?", req.VideoId).First(&video)
 		if row.Error != nil {
 			return "false", row.Error

src/services/publish/handler.go

@@ -359,6 +359,8 @@ func (a PublishServiceImpl) CreateVideo(ctx context.Context, request *publish.Cr
 		}
 	}
 
+	countStringKey := fmt.Sprintf("VideoCount-%d", request.ActorId)
+	cached.TagDelete(ctx, countStringKey)
 	resp = &publish.CreateVideoResponse{
 		StatusCode: strings.ServiceOKCode,
 		StatusMsg:  strings.ServiceOK,

src/services/relation/handler.go

@@ -13,10 +13,12 @@ import (
 	grpc2 "GuGoTik/src/utils/grpc"
 	"GuGoTik/src/utils/logging"
 	"context"
+	"errors"
 	"fmt"
 	"github.com/go-redis/redis_rate/v10"
 	"github.com/sirupsen/logrus"
 	"go.opentelemetry.io/otel/trace"
+	"gorm.io/gorm"
 	"strconv"
 	"sync"
 	"time"
@@ -206,6 +208,7 @@ func (r RelationServiceImpl) Follow(ctx context.Context, request *relation.Relat
 		logging.SetSpanError(span, err)
 		return
 	}
+	cached.TagDelete(ctx, fmt.Sprintf("IsFollowedCache-%d-%d", request.UserId, request.ActorId))
 	resp = &relation.RelationActionResponse{
 		StatusCode: strings.ServiceOKCode,
 		StatusMsg:  strings.ServiceOK,
@@ -343,7 +346,7 @@ func (r RelationServiceImpl) Unfollow(ctx context.Context, request *relation.Rel
 		logging.SetSpanError(span, err)
 		return
 	}
-
+	cached.TagDelete(ctx, fmt.Sprintf("IsFollowedCache-%d-%d", request.UserId, request.ActorId))
 	resp = &relation.RelationActionResponse{
 		StatusCode: strings.ServiceOKCode,
 		StatusMsg:  strings.ServiceOK,
@@ -740,15 +743,21 @@ func (r RelationServiceImpl) IsFollow(ctx context.Context, request *relation.IsF
 	logging.SetSpanWithHostname(span)
 	logger := logging.LogService("RelationService.isFollow").WithContext(ctx)
 
-	var count int64
-	result := database.Client.WithContext(ctx).
-		Model(&models.Relation{}).
-		Where("user_id = ? AND actor_id = ?", request.UserId, request.ActorId).
-		Count(&count)
+	res, err := cached.GetWithFunc(ctx, fmt.Sprintf("IsFollowedCache-%d-%d", request.UserId, request.ActorId), func(ctx context.Context, key string) (string, error) {
+		var count int64
+		row := database.Client.WithContext(ctx).
+			Model(&models.Relation{}).
+			Where("user_id = ? AND actor_id = ?", request.UserId, request.ActorId).
+			Count(&count)
+		if row.Error != nil && !errors.Is(row.Error, gorm.ErrRecordNotFound) {
+			return "false", row.Error
+		}
+		return strconv.FormatInt(count, 10), nil
+	})
 
-	if result.Error != nil {
+	if err != nil {
 		logger.WithFields(logrus.Fields{
-			"err":     result.Error,
+			"err":     err,
 			"ActorId": request.ActorId,
 			"UserId":  request.UserId,
 		}).Errorf("IsFollowService failed")
@@ -765,7 +774,7 @@ func (r RelationServiceImpl) IsFollow(ctx context.Context, request *relation.IsF
 	resp = &relation.IsFollowResponse{
 		StatusCode: strings.ServiceOKCode,
 		StatusMsg:  strings.ServiceOK,
-		Result:     count > 0,
+		Result:     res != "0",
 	}
 	return
 }