first commit

app/main/api/internal/logic/user/gettokenlogic.go

@@ -0,0 +1,62 @@
+package user
+
+import (
+	"context"
+	"time"
+
+	"bdrp-server/app/main/api/internal/svc"
+	"bdrp-server/app/main/api/internal/types"
+	"bdrp-server/app/main/model"
+	"bdrp-server/common/ctxdata"
+	"bdrp-server/common/xerr"
+
+	"github.com/pkg/errors"
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type GetTokenLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewGetTokenLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetTokenLogic {
+	return &GetTokenLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx,
+	}
+}
+
+func (l *GetTokenLogic) GetToken() (resp *types.MobileCodeLoginResp, err error) {
+	claims, err := ctxdata.GetClaimsFromCtx(l.ctx)
+	if err != nil {
+		return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "用户信息, %v", err)
+	}
+	// 被封禁用户禁止刷新 token（临时用户 ID 在 user_temp 表）
+	if claims.UserType == model.UserTypeTemp {
+		_, err := l.svcCtx.UserTempModel.FindOne(l.ctx, claims.UserId)
+		if err != nil {
+			return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "用户信息, %v", err)
+		}
+	} else {
+		user, err := l.svcCtx.UserModel.FindOne(l.ctx, claims.UserId)
+		if err != nil {
+			return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "用户信息, %v", err)
+		}
+		if user.Disable == 1 {
+			return nil, errors.Wrapf(xerr.NewErrCode(xerr.USER_DISABLED), "账号已被封禁")
+		}
+	}
+	token, err := l.svcCtx.UserService.GeneralUserToken(l.ctx, claims.UserId, claims.UserType)
+	if err != nil {
+		return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "用户信息, %v", err)
+	}
+	// 获取当前时间戳
+	now := time.Now().Unix()
+	return &types.MobileCodeLoginResp{
+		AccessToken:  token,
+		AccessExpire: now + l.svcCtx.Config.JwtAuth.AccessExpire,
+		RefreshAfter: now + l.svcCtx.Config.JwtAuth.RefreshAfter,
+	}, nil
+}