lan
4c0177149a
This removes verbose trace output in handlers/services and keeps only actionable error-level logs.
566 lines
17 KiB
Go
566 lines
17 KiB
Go
package service
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"strings"
|
||
|
||
"carrot_bbs/internal/cache"
|
||
"carrot_bbs/internal/model"
|
||
"carrot_bbs/internal/pkg/utils"
|
||
"carrot_bbs/internal/repository"
|
||
)
|
||
|
||
// UserService 用户服务
|
||
type UserService struct {
|
||
userRepo *repository.UserRepository
|
||
systemMessageService SystemMessageService
|
||
emailCodeService EmailCodeService
|
||
}
|
||
|
||
// NewUserService 创建用户服务
|
||
func NewUserService(
|
||
userRepo *repository.UserRepository,
|
||
systemMessageService SystemMessageService,
|
||
emailService EmailService,
|
||
cacheBackend cache.Cache,
|
||
) *UserService {
|
||
return &UserService{
|
||
userRepo: userRepo,
|
||
systemMessageService: systemMessageService,
|
||
emailCodeService: NewEmailCodeService(emailService, cacheBackend),
|
||
}
|
||
}
|
||
|
||
// SendRegisterCode 发送注册验证码
|
||
func (s *UserService) SendRegisterCode(ctx context.Context, email string) error {
|
||
user, err := s.userRepo.GetByEmail(email)
|
||
if err == nil && user != nil {
|
||
return ErrEmailExists
|
||
}
|
||
return s.emailCodeService.SendCode(ctx, CodePurposeRegister, email)
|
||
}
|
||
|
||
// SendPasswordResetCode 发送找回密码验证码
|
||
func (s *UserService) SendPasswordResetCode(ctx context.Context, email string) error {
|
||
user, err := s.userRepo.GetByEmail(email)
|
||
if err != nil || user == nil {
|
||
return ErrUserNotFound
|
||
}
|
||
return s.emailCodeService.SendCode(ctx, CodePurposePasswordReset, email)
|
||
}
|
||
|
||
// SendCurrentUserEmailVerifyCode 发送当前用户邮箱验证验证码
|
||
func (s *UserService) SendCurrentUserEmailVerifyCode(ctx context.Context, userID, email string) error {
|
||
user, err := s.userRepo.GetByID(userID)
|
||
if err != nil || user == nil {
|
||
return ErrUserNotFound
|
||
}
|
||
|
||
targetEmail := strings.TrimSpace(email)
|
||
if targetEmail == "" && user.Email != nil {
|
||
targetEmail = strings.TrimSpace(*user.Email)
|
||
}
|
||
if targetEmail == "" || !utils.ValidateEmail(targetEmail) {
|
||
return ErrInvalidEmail
|
||
}
|
||
|
||
if user.EmailVerified && user.Email != nil && strings.EqualFold(strings.TrimSpace(*user.Email), targetEmail) {
|
||
return ErrEmailAlreadyVerified
|
||
}
|
||
|
||
existingUser, queryErr := s.userRepo.GetByEmail(targetEmail)
|
||
if queryErr == nil && existingUser != nil && existingUser.ID != userID {
|
||
return ErrEmailExists
|
||
}
|
||
|
||
return s.emailCodeService.SendCode(ctx, CodePurposeEmailVerify, targetEmail)
|
||
}
|
||
|
||
// VerifyCurrentUserEmail 验证当前用户邮箱
|
||
func (s *UserService) VerifyCurrentUserEmail(ctx context.Context, userID, email, verificationCode string) error {
|
||
user, err := s.userRepo.GetByID(userID)
|
||
if err != nil || user == nil {
|
||
return ErrUserNotFound
|
||
}
|
||
|
||
targetEmail := strings.TrimSpace(email)
|
||
if targetEmail == "" && user.Email != nil {
|
||
targetEmail = strings.TrimSpace(*user.Email)
|
||
}
|
||
if targetEmail == "" || !utils.ValidateEmail(targetEmail) {
|
||
return ErrInvalidEmail
|
||
}
|
||
|
||
if err := s.emailCodeService.VerifyCode(CodePurposeEmailVerify, targetEmail, verificationCode); err != nil {
|
||
return err
|
||
}
|
||
|
||
existingUser, queryErr := s.userRepo.GetByEmail(targetEmail)
|
||
if queryErr == nil && existingUser != nil && existingUser.ID != userID {
|
||
return ErrEmailExists
|
||
}
|
||
|
||
user.Email = &targetEmail
|
||
user.EmailVerified = true
|
||
return s.userRepo.Update(user)
|
||
}
|
||
|
||
// SendChangePasswordCode 发送修改密码验证码
|
||
func (s *UserService) SendChangePasswordCode(ctx context.Context, userID string) error {
|
||
user, err := s.userRepo.GetByID(userID)
|
||
if err != nil || user == nil {
|
||
return ErrUserNotFound
|
||
}
|
||
if user.Email == nil || strings.TrimSpace(*user.Email) == "" {
|
||
return ErrEmailNotBound
|
||
}
|
||
return s.emailCodeService.SendCode(ctx, CodePurposeChangePassword, *user.Email)
|
||
}
|
||
|
||
// Register 用户注册
|
||
func (s *UserService) Register(ctx context.Context, username, email, password, nickname, phone, verificationCode string) (*model.User, error) {
|
||
// 验证用户名
|
||
if !utils.ValidateUsername(username) {
|
||
return nil, ErrInvalidUsername
|
||
}
|
||
|
||
// 注册必须提供邮箱并完成验证码校验
|
||
if email == "" || !utils.ValidateEmail(email) {
|
||
return nil, ErrInvalidEmail
|
||
}
|
||
if err := s.emailCodeService.VerifyCode(CodePurposeRegister, email, verificationCode); err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
// 验证密码
|
||
if !utils.ValidatePassword(password) {
|
||
return nil, ErrWeakPassword
|
||
}
|
||
|
||
// 验证手机号（如果提供）
|
||
if phone != "" && !utils.ValidatePhone(phone) {
|
||
return nil, ErrInvalidPhone
|
||
}
|
||
|
||
// 检查用户名是否已存在
|
||
existingUser, err := s.userRepo.GetByUsername(username)
|
||
if err == nil && existingUser != nil {
|
||
return nil, ErrUsernameExists
|
||
}
|
||
|
||
// 检查邮箱是否已存在（如果提供）
|
||
if email != "" {
|
||
existingUser, err = s.userRepo.GetByEmail(email)
|
||
if err == nil && existingUser != nil {
|
||
return nil, ErrEmailExists
|
||
}
|
||
}
|
||
|
||
// 检查手机号是否已存在（如果提供）
|
||
if phone != "" {
|
||
existingUser, err = s.userRepo.GetByPhone(phone)
|
||
if err == nil && existingUser != nil {
|
||
return nil, ErrPhoneExists
|
||
}
|
||
}
|
||
|
||
// 密码哈希
|
||
hashedPassword, err := utils.HashPassword(password)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
// 创建用户
|
||
user := &model.User{
|
||
Username: username,
|
||
Nickname: nickname,
|
||
EmailVerified: true,
|
||
PasswordHash: hashedPassword,
|
||
Status: model.UserStatusActive,
|
||
}
|
||
|
||
// 如果提供了邮箱，设置指针值
|
||
if email != "" {
|
||
user.Email = &email
|
||
}
|
||
|
||
// 如果提供了手机号，设置指针值
|
||
if phone != "" {
|
||
user.Phone = &phone
|
||
}
|
||
|
||
err = s.userRepo.Create(user)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
return user, nil
|
||
}
|
||
|
||
// Login 用户登录
|
||
func (s *UserService) Login(ctx context.Context, account, password string) (*model.User, error) {
|
||
account = strings.TrimSpace(account)
|
||
var (
|
||
user *model.User
|
||
err error
|
||
)
|
||
if utils.ValidateEmail(account) {
|
||
user, err = s.userRepo.GetByEmail(account)
|
||
} else if utils.ValidatePhone(account) {
|
||
user, err = s.userRepo.GetByPhone(account)
|
||
} else {
|
||
user, err = s.userRepo.GetByUsername(account)
|
||
}
|
||
if err != nil || user == nil {
|
||
return nil, ErrInvalidCredentials
|
||
}
|
||
|
||
if !utils.CheckPasswordHash(password, user.PasswordHash) {
|
||
return nil, ErrInvalidCredentials
|
||
}
|
||
|
||
if user.Status != model.UserStatusActive {
|
||
return nil, ErrUserBanned
|
||
}
|
||
|
||
return user, nil
|
||
}
|
||
|
||
// GetUserByID 根据ID获取用户
|
||
func (s *UserService) GetUserByID(ctx context.Context, id string) (*model.User, error) {
|
||
return s.userRepo.GetByID(id)
|
||
}
|
||
|
||
// GetUserPostCount 获取用户帖子数（实时计算）
|
||
func (s *UserService) GetUserPostCount(ctx context.Context, userID string) (int64, error) {
|
||
return s.userRepo.GetPostsCount(userID)
|
||
}
|
||
|
||
// GetUserPostCountBatch 批量获取用户帖子数（实时计算）
|
||
func (s *UserService) GetUserPostCountBatch(ctx context.Context, userIDs []string) (map[string]int64, error) {
|
||
return s.userRepo.GetPostsCountBatch(userIDs)
|
||
}
|
||
|
||
// GetUserByIDWithFollowingStatus 根据ID获取用户（包含当前用户是否关注的状态）
|
||
func (s *UserService) GetUserByIDWithFollowingStatus(ctx context.Context, userID, currentUserID string) (*model.User, bool, error) {
|
||
user, err := s.userRepo.GetByID(userID)
|
||
if err != nil {
|
||
return nil, false, err
|
||
}
|
||
|
||
// 如果查询的是当前用户自己，不需要检查关注状态
|
||
if userID == currentUserID {
|
||
return user, false, nil
|
||
}
|
||
|
||
isFollowing, err := s.userRepo.IsFollowing(currentUserID, userID)
|
||
if err != nil {
|
||
return user, false, err
|
||
}
|
||
|
||
return user, isFollowing, nil
|
||
}
|
||
|
||
// GetUserByIDWithMutualFollowStatus 根据ID获取用户（包含双向关注状态）
|
||
func (s *UserService) GetUserByIDWithMutualFollowStatus(ctx context.Context, userID, currentUserID string) (*model.User, bool, bool, error) {
|
||
user, err := s.userRepo.GetByID(userID)
|
||
if err != nil {
|
||
return nil, false, false, err
|
||
}
|
||
|
||
// 如果查询的是当前用户自己，不需要检查关注状态
|
||
if userID == currentUserID {
|
||
return user, false, false, nil
|
||
}
|
||
|
||
// 当前用户是否关注了该用户
|
||
isFollowing, err := s.userRepo.IsFollowing(currentUserID, userID)
|
||
if err != nil {
|
||
return user, false, false, err
|
||
}
|
||
|
||
// 该用户是否关注了当前用户
|
||
isFollowingMe, err := s.userRepo.IsFollowing(userID, currentUserID)
|
||
if err != nil {
|
||
return user, isFollowing, false, err
|
||
}
|
||
|
||
return user, isFollowing, isFollowingMe, nil
|
||
}
|
||
|
||
// UpdateUser 更新用户
|
||
func (s *UserService) UpdateUser(ctx context.Context, user *model.User) error {
|
||
return s.userRepo.Update(user)
|
||
}
|
||
|
||
// GetFollowers 获取粉丝
|
||
func (s *UserService) GetFollowers(ctx context.Context, userID string, page, pageSize int) ([]*model.User, int64, error) {
|
||
return s.userRepo.GetFollowers(userID, page, pageSize)
|
||
}
|
||
|
||
// GetFollowing 获取关注
|
||
func (s *UserService) GetFollowing(ctx context.Context, userID string, page, pageSize int) ([]*model.User, int64, error) {
|
||
return s.userRepo.GetFollowing(userID, page, pageSize)
|
||
}
|
||
|
||
// FollowUser 关注用户
|
||
func (s *UserService) FollowUser(ctx context.Context, followerID, followeeID string) error {
|
||
blocked, err := s.userRepo.IsBlockedEitherDirection(followerID, followeeID)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if blocked {
|
||
return ErrUserBlocked
|
||
}
|
||
|
||
// 检查是否已经关注
|
||
isFollowing, err := s.userRepo.IsFollowing(followerID, followeeID)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if isFollowing {
|
||
return nil // 已经关注，直接返回成功
|
||
}
|
||
|
||
// 创建关注关系
|
||
follow := &model.Follow{
|
||
FollowerID: followerID,
|
||
FollowingID: followeeID,
|
||
}
|
||
|
||
err = s.userRepo.CreateFollow(follow)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
// 刷新关注者的关注数（通过实际计数，更可靠）
|
||
err = s.userRepo.RefreshFollowingCount(followerID)
|
||
if err != nil {
|
||
// 不回滚，计数可以通过其他方式修复
|
||
}
|
||
|
||
// 刷新被关注者的粉丝数（通过实际计数，更可靠）
|
||
err = s.userRepo.RefreshFollowersCount(followeeID)
|
||
if err != nil {
|
||
// 不回滚，计数可以通过其他方式修复
|
||
}
|
||
|
||
// 发送关注通知给被关注者
|
||
if s.systemMessageService != nil {
|
||
// 异步发送通知，不阻塞主流程
|
||
go func() {
|
||
_ = s.systemMessageService.SendFollowNotification(context.Background(), followeeID, followerID)
|
||
}()
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// UnfollowUser 取消关注用户
|
||
func (s *UserService) UnfollowUser(ctx context.Context, followerID, followeeID string) error {
|
||
// 检查是否已经关注
|
||
isFollowing, err := s.userRepo.IsFollowing(followerID, followeeID)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if !isFollowing {
|
||
return nil // 没有关注，直接返回成功
|
||
}
|
||
|
||
// 删除关注关系
|
||
err = s.userRepo.DeleteFollow(followerID, followeeID)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
// 刷新关注者的关注数（通过实际计数，更可靠）
|
||
err = s.userRepo.RefreshFollowingCount(followerID)
|
||
if err != nil {
|
||
}
|
||
|
||
// 刷新被关注者的粉丝数（通过实际计数，更可靠）
|
||
err = s.userRepo.RefreshFollowersCount(followeeID)
|
||
if err != nil {
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// BlockUser 拉黑用户，并自动清理双向关注/粉丝关系
|
||
func (s *UserService) BlockUser(ctx context.Context, blockerID, blockedID string) error {
|
||
if blockerID == blockedID {
|
||
return ErrInvalidOperation
|
||
}
|
||
return s.userRepo.BlockUserAndCleanupRelations(blockerID, blockedID)
|
||
}
|
||
|
||
// UnblockUser 取消拉黑
|
||
func (s *UserService) UnblockUser(ctx context.Context, blockerID, blockedID string) error {
|
||
if blockerID == blockedID {
|
||
return ErrInvalidOperation
|
||
}
|
||
return s.userRepo.UnblockUser(blockerID, blockedID)
|
||
}
|
||
|
||
// GetBlockedUsers 获取黑名单列表
|
||
func (s *UserService) GetBlockedUsers(ctx context.Context, blockerID string, page, pageSize int) ([]*model.User, int64, error) {
|
||
return s.userRepo.GetBlockedUsers(blockerID, page, pageSize)
|
||
}
|
||
|
||
// IsBlocked 检查当前用户是否已拉黑目标用户
|
||
func (s *UserService) IsBlocked(ctx context.Context, blockerID, blockedID string) (bool, error) {
|
||
return s.userRepo.IsBlocked(blockerID, blockedID)
|
||
}
|
||
|
||
// GetFollowingList 获取关注列表（字符串参数版本）
|
||
func (s *UserService) GetFollowingList(ctx context.Context, userID, page, pageSize string) ([]*model.User, error) {
|
||
// 转换字符串参数为整数
|
||
pageInt := 1
|
||
pageSizeInt := 20
|
||
if page != "" {
|
||
_, err := fmt.Sscanf(page, "%d", &pageInt)
|
||
if err != nil {
|
||
pageInt = 1
|
||
}
|
||
}
|
||
if pageSize != "" {
|
||
_, err := fmt.Sscanf(pageSize, "%d", &pageSizeInt)
|
||
if err != nil {
|
||
pageSizeInt = 20
|
||
}
|
||
}
|
||
|
||
users, _, err := s.userRepo.GetFollowing(userID, pageInt, pageSizeInt)
|
||
return users, err
|
||
}
|
||
|
||
// GetFollowersList 获取粉丝列表（字符串参数版本）
|
||
func (s *UserService) GetFollowersList(ctx context.Context, userID, page, pageSize string) ([]*model.User, error) {
|
||
// 转换字符串参数为整数
|
||
pageInt := 1
|
||
pageSizeInt := 20
|
||
if page != "" {
|
||
_, err := fmt.Sscanf(page, "%d", &pageInt)
|
||
if err != nil {
|
||
pageInt = 1
|
||
}
|
||
}
|
||
if pageSize != "" {
|
||
_, err := fmt.Sscanf(pageSize, "%d", &pageSizeInt)
|
||
if err != nil {
|
||
pageSizeInt = 20
|
||
}
|
||
}
|
||
|
||
users, _, err := s.userRepo.GetFollowers(userID, pageInt, pageSizeInt)
|
||
return users, err
|
||
}
|
||
|
||
// GetMutualFollowStatus 批量获取双向关注状态
|
||
func (s *UserService) GetMutualFollowStatus(ctx context.Context, currentUserID string, targetUserIDs []string) (map[string][2]bool, error) {
|
||
return s.userRepo.GetMutualFollowStatus(currentUserID, targetUserIDs)
|
||
}
|
||
|
||
// CheckUsernameAvailable 检查用户名是否可用
|
||
func (s *UserService) CheckUsernameAvailable(ctx context.Context, username string) (bool, error) {
|
||
user, err := s.userRepo.GetByUsername(username)
|
||
if err != nil {
|
||
return true, nil // 用户不存在，可用
|
||
}
|
||
return user == nil, nil
|
||
}
|
||
|
||
// ChangePassword 修改密码
|
||
func (s *UserService) ChangePassword(ctx context.Context, userID, oldPassword, newPassword, verificationCode string) error {
|
||
// 获取用户
|
||
user, err := s.userRepo.GetByID(userID)
|
||
if err != nil {
|
||
return ErrUserNotFound
|
||
}
|
||
if user.Email == nil || strings.TrimSpace(*user.Email) == "" {
|
||
return ErrEmailNotBound
|
||
}
|
||
if err := s.emailCodeService.VerifyCode(CodePurposeChangePassword, *user.Email, verificationCode); err != nil {
|
||
return err
|
||
}
|
||
|
||
// 验证旧密码
|
||
if !utils.CheckPasswordHash(oldPassword, user.PasswordHash) {
|
||
return ErrInvalidCredentials
|
||
}
|
||
|
||
// 哈希新密码
|
||
hashedPassword, err := utils.HashPassword(newPassword)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
// 更新密码
|
||
user.PasswordHash = hashedPassword
|
||
return s.userRepo.Update(user)
|
||
}
|
||
|
||
// ResetPasswordByEmail 通过邮箱重置密码
|
||
func (s *UserService) ResetPasswordByEmail(ctx context.Context, email, verificationCode, newPassword string) error {
|
||
email = strings.TrimSpace(email)
|
||
if !utils.ValidateEmail(email) {
|
||
return ErrInvalidEmail
|
||
}
|
||
if !utils.ValidatePassword(newPassword) {
|
||
return ErrWeakPassword
|
||
}
|
||
if err := s.emailCodeService.VerifyCode(CodePurposePasswordReset, email, verificationCode); err != nil {
|
||
return err
|
||
}
|
||
|
||
user, err := s.userRepo.GetByEmail(email)
|
||
if err != nil || user == nil {
|
||
return ErrUserNotFound
|
||
}
|
||
|
||
hashedPassword, err := utils.HashPassword(newPassword)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
user.PasswordHash = hashedPassword
|
||
return s.userRepo.Update(user)
|
||
}
|
||
|
||
// Search 搜索用户
|
||
func (s *UserService) Search(ctx context.Context, keyword string, page, pageSize int) ([]*model.User, int64, error) {
|
||
return s.userRepo.Search(keyword, page, pageSize)
|
||
}
|
||
|
||
// 错误定义
|
||
var (
|
||
ErrInvalidUsername = &ServiceError{Code: 400, Message: "invalid username"}
|
||
ErrInvalidEmail = &ServiceError{Code: 400, Message: "invalid email"}
|
||
ErrInvalidPhone = &ServiceError{Code: 400, Message: "invalid phone number"}
|
||
ErrWeakPassword = &ServiceError{Code: 400, Message: "password too weak"}
|
||
ErrUsernameExists = &ServiceError{Code: 400, Message: "username already exists"}
|
||
ErrEmailExists = &ServiceError{Code: 400, Message: "email already exists"}
|
||
ErrPhoneExists = &ServiceError{Code: 400, Message: "phone number already exists"}
|
||
ErrUserNotFound = &ServiceError{Code: 404, Message: "user not found"}
|
||
ErrUserBanned = &ServiceError{Code: 403, Message: "user is banned"}
|
||
ErrUserBlocked = &ServiceError{Code: 403, Message: "blocked relationship exists"}
|
||
ErrInvalidOperation = &ServiceError{Code: 400, Message: "invalid operation"}
|
||
ErrEmailServiceUnavailable = &ServiceError{Code: 503, Message: "email service unavailable"}
|
||
ErrVerificationCodeTooFrequent = &ServiceError{Code: 429, Message: "verification code sent too frequently"}
|
||
ErrVerificationCodeInvalid = &ServiceError{Code: 400, Message: "invalid verification code"}
|
||
ErrVerificationCodeExpired = &ServiceError{Code: 400, Message: "verification code expired"}
|
||
ErrVerificationCodeUnavailable = &ServiceError{Code: 500, Message: "verification code storage unavailable"}
|
||
ErrEmailAlreadyVerified = &ServiceError{Code: 400, Message: "email already verified"}
|
||
ErrEmailNotBound = &ServiceError{Code: 400, Message: "email not bound"}
|
||
)
|
||
|
||
// ServiceError 服务错误
|
||
type ServiceError struct {
|
||
Code int
|
||
Message string
|
||
}
|
||
|
||
func (e *ServiceError) Error() string {
|
||
return e.Message
|
||
}
|
||
|
||
var ErrInvalidCredentials = &ServiceError{Code: 401, Message: "invalid username or password"}
|