140 lines
3.8 KiB
Go
140 lines
3.8 KiB
Go
package social
|
||
|
||
import (
|
||
"context"
|
||
"errors"
|
||
"feedsystem_video_go/internal/account"
|
||
"feedsystem_video_go/internal/middleware/rabbitmq"
|
||
rediscache "feedsystem_video_go/internal/middleware/redis"
|
||
"log"
|
||
)
|
||
|
||
type SocialService struct {
|
||
repo *SocialRepository
|
||
accountrepo *account.AccountRepository
|
||
socialMQ *rabbitmq.SocialMQ
|
||
cache *rediscache.Client
|
||
}
|
||
|
||
func NewSocialService(repo *SocialRepository, accountrepo *account.AccountRepository, socialMQ *rabbitmq.SocialMQ, cache *rediscache.Client) *SocialService {
|
||
return &SocialService{repo: repo, accountrepo: accountrepo, socialMQ: socialMQ, cache: cache}
|
||
}
|
||
|
||
func (s *SocialService) Follow(ctx context.Context, social *Social) error {
|
||
_, err := s.accountrepo.FindByID(ctx, social.FollowerID)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
_, err = s.accountrepo.FindByID(ctx, social.VloggerID)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if social.FollowerID == social.VloggerID {
|
||
return errors.New("can not follow self")
|
||
}
|
||
isFollowed, err := s.repo.IsFollowed(ctx, social)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if isFollowed {
|
||
return errors.New("already followed")
|
||
}
|
||
|
||
// 先写 DB，确保数据持久化
|
||
if err := s.repo.Follow(ctx, social); err != nil {
|
||
return err
|
||
}
|
||
|
||
// DB 成功后，失效该用户的关注列表缓存
|
||
s.invalidateFollowingFeedCache(context.Background(), social.FollowerID)
|
||
|
||
// 最后发 MQ（用于通知），失败只记日志不影响业务
|
||
if s.socialMQ != nil {
|
||
if err := s.socialMQ.Follow(ctx, social.FollowerID, social.VloggerID); err != nil {
|
||
log.Printf("social MQ Follow 发布失败: %v", err)
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func (s *SocialService) Unfollow(ctx context.Context, social *Social) error {
|
||
_, err := s.accountrepo.FindByID(ctx, social.FollowerID)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
_, err = s.accountrepo.FindByID(ctx, social.VloggerID)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
isFollowed, err := s.repo.IsFollowed(ctx, social)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if !isFollowed {
|
||
return errors.New("not followed")
|
||
}
|
||
|
||
// 先写 DB
|
||
if err := s.repo.Unfollow(ctx, social); err != nil {
|
||
return err
|
||
}
|
||
|
||
// 失效缓存
|
||
s.invalidateFollowingFeedCache(context.Background(), social.FollowerID)
|
||
|
||
// 最后发 MQ
|
||
if s.socialMQ != nil {
|
||
if err := s.socialMQ.UnFollow(ctx, social.FollowerID, social.VloggerID); err != nil {
|
||
log.Printf("social MQ UnFollow 发布失败: %v", err)
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// 失效关注列表缓存，确保下次查询时能获取到最新的关注列表
|
||
func (s *SocialService) invalidateFollowingFeedCache(ctx context.Context, accountID uint) {
|
||
if s.cache == nil {
|
||
return
|
||
}
|
||
pattern := s.cache.Key("feed:listByFollowing:*:accountID=%d:*", accountID)
|
||
if err := s.cache.DelByPattern(ctx, pattern); err != nil {
|
||
log.Printf("失效 Following 缓存失败: accountID=%d, err=%v", accountID, err)
|
||
}
|
||
}
|
||
|
||
func (s *SocialService) GetAllFollowers(ctx context.Context, VloggerID uint) ([]*account.Account, error) {
|
||
_, err := s.accountrepo.FindByID(ctx, VloggerID)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
return s.repo.GetAllFollowers(ctx, VloggerID)
|
||
}
|
||
|
||
func (s *SocialService) GetAllVloggers(ctx context.Context, FollowerID uint) ([]*account.Account, error) {
|
||
_, err := s.accountrepo.FindByID(ctx, FollowerID)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
return s.repo.GetAllVloggers(ctx, FollowerID)
|
||
}
|
||
|
||
func (s *SocialService) CountFollowers(ctx context.Context, vloggerID uint) (int64, error) {
|
||
return s.repo.CountFollowers(ctx, vloggerID)
|
||
}
|
||
|
||
func (s *SocialService) CountVloggers(ctx context.Context, followerID uint) (int64, error) {
|
||
return s.repo.CountVloggers(ctx, followerID)
|
||
}
|
||
|
||
func (s *SocialService) IsFollowed(ctx context.Context, social *Social) (bool, error) {
|
||
_, err := s.accountrepo.FindByID(ctx, social.FollowerID)
|
||
if err != nil {
|
||
return false, err
|
||
}
|
||
_, err = s.accountrepo.FindByID(ctx, social.VloggerID)
|
||
if err != nil {
|
||
return false, err
|
||
}
|
||
return s.repo.IsFollowed(ctx, social)
|
||
}
|