feat():learning后台管理项目初始化

2025-12-04 16:23:46 +08:00
parent 39886d50d2
commit 88e048f4d1
154 changed files with 28966 additions and 6 deletions
--- a/internal/services/goalfymax_user_service.go
+++ b/internal/services/goalfymax_user_service.go
@@ -0,0 +1,512 @@
+package services
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"goalfymax-admin/internal/models"
+	"goalfymax-admin/internal/storage"
+	"goalfymax-admin/pkg/redis"
+	"goalfymax-admin/pkg/utils"
+	"strconv"
+
+	goredis "github.com/redis/go-redis/v9"
+	"go.uber.org/zap"
+)
+
+type GoalfyMaxUserService interface {
+	List(req *models.GoalfyMaxUserListRequest) ([]models.GoalfyMaxUser, int64, error)
+	GetByID(id uint) (*models.GoalfyMaxUser, error)
+	Create(req *models.GoalfyMaxUserCreateRequest) (*models.GoalfyMaxUser, error)
+	Update(id uint, req *models.GoalfyMaxUserUpdateRequest) (*models.GoalfyMaxUser, error)
+	Delete(id uint) error
+	Ban(id uint, req *models.GoalfyMaxUserBanRequest, adminID int) error
+	Unban(id uint) error
+	AddBalance(id uint, req *models.GoalfyMaxUserAddBalanceRequest, operatorID int, operatorEmail string, ipAddress string, userAgent string) error
+	DeductBalance(id uint, req *models.GoalfyMaxUserAddBalanceRequest, operatorID int, operatorEmail string, ipAddress string, userAgent string) error
+}
+
+type goalfyMaxUserService struct {
+	storage                    storage.GoalfyMaxUserStorage
+	messagePushService         MessagePushService
+	ssoAdminService            SSOAdminService
+	redisClient                *redis.Client
+	balanceOperationLogStorage storage.BalanceOperationLogStorage
+	auditLogService            AuditLogService
+	logger                     *utils.Logger
+}
+
+func NewGoalfyMaxUserService(s storage.GoalfyMaxUserStorage, messagePushService MessagePushService, ssoAdminService SSOAdminService, redisClient *redis.Client, balanceOperationLogStorage storage.BalanceOperationLogStorage, auditLogService AuditLogService, logger *utils.Logger) GoalfyMaxUserService {
+	return &goalfyMaxUserService{
+		storage:                    s,
+		messagePushService:         messagePushService,
+		ssoAdminService:            ssoAdminService,
+		redisClient:                redisClient,
+		balanceOperationLogStorage: balanceOperationLogStorage,
+		auditLogService:            auditLogService,
+		logger:                     logger,
+	}
+}
+
+func (s *goalfyMaxUserService) List(req *models.GoalfyMaxUserListRequest) ([]models.GoalfyMaxUser, int64, error) {
+	users, total, err := s.storage.List(req)
+	if err != nil {
+		return nil, 0, err
+	}
+
+	// 如果Redis客户端可用，查询每个用户的余额
+	if s.redisClient != nil {
+		ctx := context.Background()
+		// 余额换算比例：1美元 = 100,000000 (即100000000)
+		const balanceMultiplier = 100000000.0
+
+		for i := range users {
+			// 默认余额为0
+			balanceUSD := 0.0
+
+			// Redis key格式: GW:QU_{用户ID}
+			redisKey := fmt.Sprintf("GW:QU_%d", users[i].UserID)
+
+			// 从Redis获取余额
+			balanceStr, err := s.redisClient.Rdb.Get(ctx, redisKey).Result()
+			if err == nil {
+				// 解析余额值
+				balanceValue, err := strconv.ParseInt(balanceStr, 10, 64)
+				if err == nil {
+					// 转换为美元：除以100000000
+					balanceUSD = float64(balanceValue) / balanceMultiplier
+				}
+			}
+			// 无论是否查询成功，都设置余额（查询失败则为0）
+			users[i].Balance = &balanceUSD
+		}
+	} else {
+		// 如果Redis客户端不可用，设置所有用户余额为0
+		zeroBalance := 0.0
+		for i := range users {
+			users[i].Balance = &zeroBalance
+		}
+	}
+
+	return users, total, nil
+}
+
+func (s *goalfyMaxUserService) GetByID(id uint) (*models.GoalfyMaxUser, error) {
+	return s.storage.GetByID(id)
+}
+
+func (s *goalfyMaxUserService) Create(req *models.GoalfyMaxUserCreateRequest) (*models.GoalfyMaxUser, error) {
+	// 检查用户名是否已存在
+	_, err := s.storage.GetByUsername(req.Username)
+	if err == nil {
+		return nil, errors.New("用户名已存在")
+	}
+
+	// 检查邮箱是否已存在
+	_, err = s.storage.GetByEmail(req.Email)
+	if err == nil {
+		return nil, errors.New("邮箱已存在")
+	}
+
+	// 1. 先调用SSO创建用户
+	ctx := context.Background()
+	ssoReq := &SSOAdminUserCreateRequest{
+		Username: req.Username,
+		Email:    req.Email,
+		Phone:    "",           // 默认空手机号，可以根据需要修改
+		Password: req.Password, // 使用用户输入的密码
+	}
+
+	ssoUser, err := s.ssoAdminService.CreateUser(ctx, ssoReq)
+	if err != nil {
+		s.logger.Error("create goalfymax user failed", zap.Error(err))
+		return nil, fmt.Errorf("SSO创建用户失败: %w", err)
+	}
+
+	// 2. 设置系统角色：默认 custom（而非固定角色ID）
+	if err := s.ssoAdminService.SetSystemRole(ctx, ssoUser.ID, "custom"); err != nil {
+		// 如果设置失败，记录错误但不阻止用户创建
+		if s.logger != nil {
+			s.logger.Error("设置系统角色失败", zap.Int("user_id", ssoUser.ID), zap.Error(err))
+		}
+	}
+
+	// 3. 在本地数据库创建用户记录，使用SSO返回的用户ID
+	// 如果未指定用户等级，默认为 normal
+	userLevelCode := req.UserLevelCode
+	if userLevelCode == "" {
+		userLevelCode = "normal"
+	}
+
+	// 如果未指定版本，默认为 1（用户版）
+	version := req.Version
+	if version == 0 {
+		version = 1
+	}
+
+	user := &models.GoalfyMaxUser{
+		UserID:        ssoUser.ID, // 使用SSO返回的用户ID
+		Username:      req.Username,
+		Email:         req.Email,
+		Nickname:      req.Nickname,
+		Avatar:        req.Avatar,
+		UserLevelCode: userLevelCode,
+		Version:       version,
+		IsBanned:      false,
+	}
+
+	if err := s.storage.Create(user); err != nil {
+		// 如果本地创建失败，需要清理SSO用户（这里简化处理，实际应该调用SSO删除接口）
+		if s.logger != nil {
+			s.logger.Warn("本地用户创建失败，但SSO用户已创建", zap.Int("sso_user_id", ssoUser.ID), zap.Error(err))
+		}
+		return nil, fmt.Errorf("创建用户失败: %w", err)
+	}
+
+	return user, nil
+}
+
+func (s *goalfyMaxUserService) Update(id uint, req *models.GoalfyMaxUserUpdateRequest) (*models.GoalfyMaxUser, error) {
+	user, err := s.storage.GetByID(id)
+	if err != nil {
+		return nil, errors.New("用户不存在")
+	}
+	if req.Nickname != "" {
+		user.Nickname = req.Nickname
+	}
+	if req.Email != "" {
+		user.Email = req.Email
+	}
+	if req.Avatar != "" {
+		user.Avatar = req.Avatar
+	}
+	if req.UserLevelCode != "" {
+		user.UserLevelCode = req.UserLevelCode
+	}
+	if req.Version != nil {
+		user.Version = *req.Version
+	}
+	if req.GoalfyHubPermission != nil {
+		user.GoalfyHubPermission = *req.GoalfyHubPermission
+	}
+	if err := s.storage.Update(user); err != nil {
+		return nil, err
+	}
+	return user, nil
+}
+
+func (s *goalfyMaxUserService) Delete(id uint) error {
+	_, err := s.storage.GetByID(id)
+	if err != nil {
+		return errors.New("用户不存在")
+	}
+	return s.storage.Delete(id)
+}
+
+func (s *goalfyMaxUserService) Ban(id uint, req *models.GoalfyMaxUserBanRequest, adminID int) error {
+	user, err := s.storage.GetByID(id)
+	if err != nil {
+		return errors.New("用户不存在")
+	}
+
+	// 执行封禁操作
+	if err := s.storage.SetBanned(id, req.Reason, adminID); err != nil {
+		return err
+	}
+
+	// 发送封禁通知
+	ctx := context.Background()
+	banMessage := &models.MessagePushRequest{
+		Title:   "封禁通知",
+		Content: fmt.Sprintf("您的账户已被封禁。封禁原因：%s。如有疑问，请联系客服。", req.Reason),
+		UserIDs: []int{user.UserID},
+	}
+
+	// 异步发送通知，不阻塞封禁操作
+	go func() {
+		if _, err := s.messagePushService.SendMessage(ctx, banMessage, adminID, "系统管理员"); err != nil {
+			// 记录错误日志，但不影响封禁操作
+			if s.logger != nil {
+				s.logger.Error("发送封禁通知失败", zap.Int("user_id", user.UserID), zap.Int("admin_id", adminID), zap.Error(err))
+			}
+		}
+	}()
+
+	return nil
+}
+
+func (s *goalfyMaxUserService) Unban(id uint) error {
+	user, err := s.storage.GetByID(id)
+	if err != nil {
+		return errors.New("用户不存在")
+	}
+
+	// 执行解封操作
+	if err := s.storage.Unban(id); err != nil {
+		return err
+	}
+
+	// 发送解封通知
+	ctx := context.Background()
+	unbanMessage := &models.MessagePushRequest{
+		Title:   "解封通知",
+		Content: "您的账户已被解封，现在可以正常使用所有功能。感谢您的理解与配合。",
+		UserIDs: []int{user.UserID},
+	}
+
+	// 异步发送通知，不阻塞解封操作
+	go func() {
+		if _, err := s.messagePushService.SendMessage(ctx, unbanMessage, 0, "系统管理员"); err != nil {
+			// 记录错误日志，但不影响解封操作
+			if s.logger != nil {
+				s.logger.Error("发送解封通知失败", zap.Int("user_id", user.UserID), zap.Error(err))
+			}
+		}
+	}()
+
+	return nil
+}
+
+func (s *goalfyMaxUserService) AddBalance(id uint, req *models.GoalfyMaxUserAddBalanceRequest, operatorID int, operatorEmail string, ipAddress string, userAgent string) error {
+	// 检查用户是否存在
+	user, err := s.storage.GetByID(id)
+	if err != nil {
+		return errors.New("用户不存在")
+	}
+
+	// 检查Redis客户端是否可用
+	if s.redisClient == nil {
+		return errors.New("Redis客户端不可用，无法增加余额")
+	}
+
+	ctx := context.Background()
+	// 余额换算比例：1美元 = 100,000000 (即100000000)
+	const balanceMultiplier = 100000000.0
+
+	// Redis key格式: GW:QU_{用户ID}
+	redisKey := fmt.Sprintf("GW:QU_%d", user.UserID)
+
+	// 获取操作前余额
+	balanceBeforeUSD := 0.0
+	balanceBeforeStr, err := s.redisClient.Rdb.Get(ctx, redisKey).Result()
+	if err == nil {
+		balanceBeforeValue, err := strconv.ParseInt(balanceBeforeStr, 10, 64)
+		if err == nil {
+			balanceBeforeUSD = float64(balanceBeforeValue) / balanceMultiplier
+		}
+	}
+
+	// 将美元金额转换为Redis存储的数值
+	amountToAdd := int64(req.Amount * balanceMultiplier)
+
+	// 使用Redis的INCRBY命令增加余额（如果key不存在，会自动创建并设置为0，然后加上amountToAdd）
+	newBalance, err := s.redisClient.Rdb.IncrBy(ctx, redisKey, amountToAdd).Result()
+	if err != nil {
+		return fmt.Errorf("增加余额失败: %w", err)
+	}
+
+	// 计算操作后余额
+	balanceAfterUSD := float64(newBalance) / balanceMultiplier
+
+	// 异步记录操作日志（保留原有的余额操作日志）
+	go func() {
+		logEntry := &models.BalanceOperationLog{
+			UserID:        user.UserID,
+			OperationType: models.OperationTypeAdd,
+			Amount:        req.Amount,
+			BalanceBefore: balanceBeforeUSD,
+			BalanceAfter:  balanceAfterUSD,
+			OperatorID:    operatorID,
+			OperatorName:  operatorEmail,
+			Remark:        "", // 可以后续扩展
+		}
+		if err := s.balanceOperationLogStorage.Create(logEntry); err != nil {
+			// 记录错误但不影响主流程
+			if s.logger != nil {
+				s.logger.Error("记录余额操作日志失败",
+					zap.Int("user_id", user.UserID),
+					zap.String("operation_type", models.OperationTypeAdd),
+					zap.Float64("amount", req.Amount),
+					zap.Error(err))
+			}
+		}
+	}()
+
+	// 异步记录审计日志
+	if s.auditLogService != nil {
+		go func() {
+			operationDetails := models.OperationDetails{
+				"action":         "recharge",
+				"amount":         req.Amount,
+				"currency":       "USD",
+				"before_balance": balanceBeforeUSD,
+				"after_balance":  balanceAfterUSD,
+				"remark":         "",
+			}
+
+			auditLogReq := &models.CreateAuditLogRequest{
+				OperationType:    models.OperationTypeBalanceAdjustment,
+				OperatorID:       operatorID,
+				OperatorEmail:    operatorEmail,
+				TargetType:       models.TargetTypeUser,
+				TargetID:         &user.UserID,
+				TargetEmail:      user.Email,
+				OperationDetails: operationDetails,
+				IPAddress:        ipAddress,
+				UserAgent:        userAgent,
+				Status:           models.AuditLogStatusSuccess,
+			}
+
+			if err := s.auditLogService.Create(auditLogReq); err != nil {
+				// 记录错误但不影响主流程
+				if s.logger != nil {
+					s.logger.Error("记录审计日志失败",
+						zap.Int("user_id", user.UserID),
+						zap.String("operation_type", models.OperationTypeBalanceAdjustment),
+						zap.Error(err))
+				}
+			}
+		}()
+	}
+
+	// 记录操作日志
+	if s.logger != nil {
+		s.logger.Info("用户余额增加成功",
+			zap.Int("user_id", user.UserID),
+			zap.String("username", user.Username),
+			zap.Float64("amount", req.Amount),
+			zap.Float64("balance_before", balanceBeforeUSD),
+			zap.Float64("balance_after", balanceAfterUSD),
+			zap.Int64("redis_balance_raw", newBalance),
+			zap.Int("operator_id", operatorID),
+			zap.String("operator_email", operatorEmail))
+	}
+
+	return nil
+}
+
+func (s *goalfyMaxUserService) DeductBalance(id uint, req *models.GoalfyMaxUserAddBalanceRequest, operatorID int, operatorEmail string, ipAddress string, userAgent string) error {
+	// 检查用户是否存在
+	user, err := s.storage.GetByID(id)
+	if err != nil {
+		return errors.New("用户不存在")
+	}
+
+	// 检查Redis客户端是否可用
+	if s.redisClient == nil {
+		return errors.New("Redis客户端不可用，无法减少余额")
+	}
+
+	ctx := context.Background()
+	// 余额换算比例：1美元 = 100,000000 (即100000000)
+	const balanceMultiplier = 100000000.0
+
+	// Redis key格式: GW:QU_{用户ID}
+	redisKey := fmt.Sprintf("GW:QU_%d", user.UserID)
+
+	// 将美元金额转换为Redis存储的数值
+	amountToDeduct := int64(req.Amount * balanceMultiplier)
+
+	// 先获取当前余额，检查是否足够
+	currentBalance, err := s.redisClient.Rdb.Get(ctx, redisKey).Int64()
+	if err != nil {
+		// 如果key不存在，表示余额为0，不能减少
+		if err == goredis.Nil {
+			return errors.New("余额不足，无法减少")
+		}
+		return fmt.Errorf("查询余额失败: %w", err)
+	}
+
+	// 计算操作前余额
+	balanceBeforeUSD := float64(currentBalance) / balanceMultiplier
+
+	// 检查余额是否足够
+	if currentBalance < amountToDeduct {
+		return errors.New("余额不足，无法减少")
+	}
+
+	// 使用Redis的DECRBY命令减少余额
+	newBalance, err := s.redisClient.Rdb.DecrBy(ctx, redisKey, amountToDeduct).Result()
+	if err != nil {
+		return fmt.Errorf("减少余额失败: %w", err)
+	}
+
+	// 计算操作后余额
+	balanceAfterUSD := float64(newBalance) / balanceMultiplier
+
+	// 异步记录操作日志（保留原有的余额操作日志）
+	go func() {
+		logEntry := &models.BalanceOperationLog{
+			UserID:        user.UserID,
+			OperationType: models.OperationTypeDeduct,
+			Amount:        req.Amount,
+			BalanceBefore: balanceBeforeUSD,
+			BalanceAfter:  balanceAfterUSD,
+			OperatorID:    operatorID,
+			OperatorName:  operatorEmail,
+			Remark:        "", // 可以后续扩展
+		}
+		if err := s.balanceOperationLogStorage.Create(logEntry); err != nil {
+			// 记录错误但不影响主流程
+			if s.logger != nil {
+				s.logger.Error("记录余额操作日志失败",
+					zap.Int("user_id", user.UserID),
+					zap.String("operation_type", models.OperationTypeDeduct),
+					zap.Float64("amount", req.Amount),
+					zap.Error(err))
+			}
+		}
+	}()
+
+	// 异步记录审计日志
+	if s.auditLogService != nil {
+		go func() {
+			operationDetails := models.OperationDetails{
+				"action":         "deduct",
+				"amount":         req.Amount,
+				"currency":       "USD",
+				"before_balance": balanceBeforeUSD,
+				"after_balance":  balanceAfterUSD,
+				"remark":         "",
+			}
+
+			auditLogReq := &models.CreateAuditLogRequest{
+				OperationType:    models.OperationTypeBalanceAdjustment,
+				OperatorID:       operatorID,
+				OperatorEmail:    operatorEmail,
+				TargetType:       models.TargetTypeUser,
+				TargetID:         &user.UserID,
+				TargetEmail:      user.Email,
+				OperationDetails: operationDetails,
+				IPAddress:        ipAddress,
+				UserAgent:        userAgent,
+				Status:           models.AuditLogStatusSuccess,
+			}
+
+			if err := s.auditLogService.Create(auditLogReq); err != nil {
+				// 记录错误但不影响主流程
+				if s.logger != nil {
+					s.logger.Error("记录审计日志失败",
+						zap.Int("user_id", user.UserID),
+						zap.String("operation_type", models.OperationTypeBalanceAdjustment),
+						zap.Error(err))
+				}
+			}
+		}()
+	}
+
+	// 记录操作日志
+	if s.logger != nil {
+		s.logger.Info("用户余额减少成功",
+			zap.Int("user_id", user.UserID),
+			zap.String("username", user.Username),
+			zap.Float64("amount", req.Amount),
+			zap.Float64("balance_before", balanceBeforeUSD),
+			zap.Float64("balance_after", balanceAfterUSD),
+			zap.Int64("redis_balance_raw", newBalance),
+			zap.Int("operator_id", operatorID),
+			zap.String("operator_email", operatorEmail))
+	}
+
+	return nil
+}