feat(auth): 完整认证系统 — 注册/登录/JWT/路由(Task 4 + Task 5)
Task 4 — 认证服务层: - pkg/utils/password.go: bcrypt 密码加密与校验 - pkg/utils/jwt.go: Access/Refresh Token 生成与解析(jwt/v5@v5.2.1,兼容 Go 1.23) - app/dto/auth_dto.go: 认证相关请求/响应 DTO(含参数校验 binding tag) - app/auth/service/auth_service.go: 核心业务逻辑(注册、登录、管理员登录、Token 刷新、个人信息、改密码) - pkg/middleware/auth.go: JWT 认证中间件 + 角色权限检查中间件 Task 5 — Controller 与路由注册: - app/auth/controller/auth_controller.go: 前台认证接口(7 个 API) - app/auth/controller/admin_auth_controller.go: 后台管理认证接口 - app/auth/router.go: auth 模块路由定义(模块自包含) - router/router.go: 主路由汇总入口(仅做调度,不含具体路由定义) - cmd/server/main.go: 路由注册迁移到 router.Setup() 已验证全部接口:注册 ✓ 登录 ✓ 获取 profile ✓ 错误密码 ✓ 未认证拒绝 ✓ 非管理员拒绝 ✓ Made-with: Cursor
This commit is contained in:
@@ -0,0 +1,48 @@
|
||||
package controller
|
||||
|
||||
import (
|
||||
"github.com/echochat/backend/app/auth/service"
|
||||
"github.com/echochat/backend/app/dto"
|
||||
"github.com/echochat/backend/pkg/logs"
|
||||
"github.com/echochat/backend/pkg/utils"
|
||||
"github.com/gin-gonic/gin"
|
||||
"go.uber.org/zap"
|
||||
)
|
||||
|
||||
// AdminAuthController 后台管理认证控制器
|
||||
// 与前台 AuthController 的区别:登录后额外检查管理员角色权限
|
||||
type AdminAuthController struct {
|
||||
authService *service.AuthService
|
||||
}
|
||||
|
||||
// NewAdminAuthController 创建后台认证控制器实例
|
||||
func NewAdminAuthController(authService *service.AuthService) *AdminAuthController {
|
||||
return &AdminAuthController{authService: authService}
|
||||
}
|
||||
|
||||
// AdminLogin 管理员登录
|
||||
// POST /api/v1/admin/auth/login
|
||||
func (ctrl *AdminAuthController) AdminLogin(c *gin.Context) {
|
||||
funcName := "controller.admin_auth_controller.AdminLogin"
|
||||
ctx := c.Request.Context()
|
||||
|
||||
var req dto.LoginRequest
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
logs.Warn(ctx, funcName, "参数校验失败", zap.Error(err))
|
||||
utils.ResponseBadRequest(c, "参数校验失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
logs.Info(ctx, funcName, "管理员登录请求",
|
||||
zap.String("account", req.Account),
|
||||
zap.String("ip", c.ClientIP()),
|
||||
)
|
||||
|
||||
resp, err := ctrl.authService.AdminLogin(ctx, &req, c.ClientIP())
|
||||
if err != nil {
|
||||
handleAuthError(c, err)
|
||||
return
|
||||
}
|
||||
|
||||
utils.ResponseOK(c, resp)
|
||||
}
|
||||
217
backend/go-service/app/auth/controller/auth_controller.go
Normal file
217
backend/go-service/app/auth/controller/auth_controller.go
Normal file
@@ -0,0 +1,217 @@
|
||||
// Package controller 提供 auth 模块的 HTTP 接口处理
|
||||
package controller
|
||||
|
||||
import (
|
||||
"github.com/echochat/backend/app/auth/service"
|
||||
"github.com/echochat/backend/app/dto"
|
||||
"github.com/echochat/backend/pkg/logs"
|
||||
"github.com/echochat/backend/pkg/middleware"
|
||||
"github.com/echochat/backend/pkg/utils"
|
||||
"github.com/gin-gonic/gin"
|
||||
"go.uber.org/zap"
|
||||
)
|
||||
|
||||
// AuthController 前台认证控制器
|
||||
// 处理用户注册、登录、Token 刷新、个人信息管理等接口
|
||||
type AuthController struct {
|
||||
authService *service.AuthService
|
||||
}
|
||||
|
||||
// NewAuthController 创建前台认证控制器实例
|
||||
func NewAuthController(authService *service.AuthService) *AuthController {
|
||||
return &AuthController{authService: authService}
|
||||
}
|
||||
|
||||
// Register 用户注册
|
||||
// POST /api/v1/auth/register
|
||||
func (ctrl *AuthController) Register(c *gin.Context) {
|
||||
funcName := "controller.auth_controller.Register"
|
||||
ctx := c.Request.Context()
|
||||
|
||||
var req dto.RegisterRequest
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
logs.Warn(ctx, funcName, "参数校验失败", zap.Error(err))
|
||||
utils.ResponseBadRequest(c, "参数校验失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
logs.Info(ctx, funcName, "注册请求",
|
||||
zap.String("username", req.Username),
|
||||
zap.String("email", logs.MaskEmail(req.Email)),
|
||||
)
|
||||
|
||||
resp, err := ctrl.authService.Register(ctx, &req)
|
||||
if err != nil {
|
||||
handleAuthError(c, err)
|
||||
return
|
||||
}
|
||||
|
||||
utils.ResponseCreated(c, resp)
|
||||
}
|
||||
|
||||
// Login 用户登录
|
||||
// POST /api/v1/auth/login
|
||||
func (ctrl *AuthController) Login(c *gin.Context) {
|
||||
funcName := "controller.auth_controller.Login"
|
||||
ctx := c.Request.Context()
|
||||
|
||||
var req dto.LoginRequest
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
logs.Warn(ctx, funcName, "参数校验失败", zap.Error(err))
|
||||
utils.ResponseBadRequest(c, "参数校验失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
logs.Info(ctx, funcName, "登录请求",
|
||||
zap.String("account", req.Account),
|
||||
zap.String("ip", c.ClientIP()),
|
||||
)
|
||||
|
||||
resp, err := ctrl.authService.Login(ctx, &req, c.ClientIP())
|
||||
if err != nil {
|
||||
handleAuthError(c, err)
|
||||
return
|
||||
}
|
||||
|
||||
utils.ResponseOK(c, resp)
|
||||
}
|
||||
|
||||
// Logout 用户登出
|
||||
// POST /api/v1/auth/logout(需认证)
|
||||
func (ctrl *AuthController) Logout(c *gin.Context) {
|
||||
funcName := "controller.auth_controller.Logout"
|
||||
ctx := c.Request.Context()
|
||||
|
||||
userID, _ := middleware.GetCurrentUserID(c)
|
||||
logs.Info(ctx, funcName, "用户登出", zap.Int64("user_id", userID))
|
||||
|
||||
// 当前为无状态 JWT,登出由客户端删除 Token 实现
|
||||
// 后续可扩展:将 Token 加入 Redis 黑名单
|
||||
utils.ResponseOK(c, nil)
|
||||
}
|
||||
|
||||
// RefreshToken 刷新 Access Token
|
||||
// POST /api/v1/auth/refresh-token
|
||||
func (ctrl *AuthController) RefreshToken(c *gin.Context) {
|
||||
funcName := "controller.auth_controller.RefreshToken"
|
||||
ctx := c.Request.Context()
|
||||
|
||||
var req dto.RefreshTokenRequest
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
logs.Warn(ctx, funcName, "参数校验失败", zap.Error(err))
|
||||
utils.ResponseBadRequest(c, "参数校验失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
resp, err := ctrl.authService.RefreshToken(ctx, req.RefreshToken)
|
||||
if err != nil {
|
||||
handleAuthError(c, err)
|
||||
return
|
||||
}
|
||||
|
||||
utils.ResponseOK(c, resp)
|
||||
}
|
||||
|
||||
// GetProfile 获取当前用户信息
|
||||
// GET /api/v1/auth/profile(需认证)
|
||||
func (ctrl *AuthController) GetProfile(c *gin.Context) {
|
||||
funcName := "controller.auth_controller.GetProfile"
|
||||
ctx := c.Request.Context()
|
||||
|
||||
userID, ok := middleware.GetCurrentUserID(c)
|
||||
if !ok {
|
||||
utils.ResponseUnauthorized(c, "无法获取用户信息")
|
||||
return
|
||||
}
|
||||
|
||||
logs.Debug(ctx, funcName, "获取个人信息", zap.Int64("user_id", userID))
|
||||
|
||||
userInfo, err := ctrl.authService.GetProfile(ctx, userID)
|
||||
if err != nil {
|
||||
handleAuthError(c, err)
|
||||
return
|
||||
}
|
||||
|
||||
utils.ResponseOK(c, userInfo)
|
||||
}
|
||||
|
||||
// UpdateProfile 更新个人资料
|
||||
// PUT /api/v1/auth/profile(需认证)
|
||||
func (ctrl *AuthController) UpdateProfile(c *gin.Context) {
|
||||
funcName := "controller.auth_controller.UpdateProfile"
|
||||
ctx := c.Request.Context()
|
||||
|
||||
userID, ok := middleware.GetCurrentUserID(c)
|
||||
if !ok {
|
||||
utils.ResponseUnauthorized(c, "无法获取用户信息")
|
||||
return
|
||||
}
|
||||
|
||||
var req dto.UpdateProfileRequest
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
logs.Warn(ctx, funcName, "参数校验失败", zap.Error(err))
|
||||
utils.ResponseBadRequest(c, "参数校验失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
logs.Info(ctx, funcName, "更新个人资料", zap.Int64("user_id", userID))
|
||||
|
||||
userInfo, err := ctrl.authService.UpdateProfile(ctx, userID, &req)
|
||||
if err != nil {
|
||||
handleAuthError(c, err)
|
||||
return
|
||||
}
|
||||
|
||||
utils.ResponseOK(c, userInfo)
|
||||
}
|
||||
|
||||
// ChangePassword 修改密码
|
||||
// PUT /api/v1/auth/password(需认证)
|
||||
func (ctrl *AuthController) ChangePassword(c *gin.Context) {
|
||||
funcName := "controller.auth_controller.ChangePassword"
|
||||
ctx := c.Request.Context()
|
||||
|
||||
userID, ok := middleware.GetCurrentUserID(c)
|
||||
if !ok {
|
||||
utils.ResponseUnauthorized(c, "无法获取用户信息")
|
||||
return
|
||||
}
|
||||
|
||||
var req dto.ChangePasswordRequest
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
logs.Warn(ctx, funcName, "参数校验失败", zap.Error(err))
|
||||
utils.ResponseBadRequest(c, "参数校验失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
logs.Info(ctx, funcName, "修改密码", zap.Int64("user_id", userID))
|
||||
|
||||
if err := ctrl.authService.ChangePassword(ctx, userID, &req); err != nil {
|
||||
handleAuthError(c, err)
|
||||
return
|
||||
}
|
||||
|
||||
utils.ResponseOK(c, nil)
|
||||
}
|
||||
|
||||
// handleAuthError 统一处理认证相关业务错误,映射到对应的 HTTP 状态码
|
||||
func handleAuthError(c *gin.Context, err error) {
|
||||
switch err {
|
||||
case service.ErrUserAlreadyExists:
|
||||
utils.ResponseBadRequest(c, err.Error())
|
||||
case service.ErrUserNotFound:
|
||||
utils.ResponseNotFound(c, err.Error())
|
||||
case service.ErrPasswordWrong:
|
||||
utils.ResponseUnauthorized(c, "账号或密码错误")
|
||||
case service.ErrUserDisabled:
|
||||
utils.ResponseForbidden(c, err.Error())
|
||||
case service.ErrUserDeleted:
|
||||
utils.ResponseForbidden(c, err.Error())
|
||||
case service.ErrNotAdmin:
|
||||
utils.ResponseForbidden(c, err.Error())
|
||||
case service.ErrRefreshTokenType:
|
||||
utils.ResponseBadRequest(c, err.Error())
|
||||
default:
|
||||
utils.ResponseError(c, "服务器内部错误")
|
||||
}
|
||||
}
|
||||
@@ -3,12 +3,13 @@ package model
|
||||
import "time"
|
||||
|
||||
// Role 角色表模型,对应 auth_roles 表
|
||||
// 系统预置三种角色:user(普通用户)、admin(管理员)、super_admin(超级管理员)
|
||||
type Role struct {
|
||||
ID int `json:"id" gorm:"primaryKey;autoIncrement"`
|
||||
Code string `json:"code" gorm:"uniqueIndex;size:50;not null"` // 角色代码: user/admin/super_admin
|
||||
Name string `json:"name" gorm:"size:50;not null"` // 角色显示名称
|
||||
Description string `json:"description" gorm:"size:200;default:''"`
|
||||
CreatedAt time.Time `json:"created_at" gorm:"not null;autoCreateTime"`
|
||||
ID int `json:"id" gorm:"primaryKey;autoIncrement"` // 角色唯一标识,自增主键
|
||||
Code string `json:"code" gorm:"uniqueIndex;size:50;not null"` // 角色代码,全局唯一,如 user/admin/super_admin(对应 constants.RoleCode*)
|
||||
Name string `json:"name" gorm:"size:50;not null"` // 角色中文显示名称,如「普通用户」「管理员」
|
||||
Description string `json:"description" gorm:"size:200;default:''"` // 角色描述说明,用于后台管理界面展示
|
||||
CreatedAt time.Time `json:"created_at" gorm:"not null;autoCreateTime"` // 创建时间,由 GORM 自动填充
|
||||
}
|
||||
|
||||
// TableName 指定数据库表名
|
||||
@@ -17,10 +18,11 @@ func (Role) TableName() string {
|
||||
}
|
||||
|
||||
// UserRole 用户角色关联表模型,对应 auth_user_roles 表
|
||||
// 采用联合主键(user_id + role_id),一个用户可拥有多个角色
|
||||
type UserRole struct {
|
||||
UserID int64 `json:"user_id" gorm:"primaryKey"`
|
||||
RoleID int `json:"role_id" gorm:"primaryKey"`
|
||||
CreatedAt time.Time `json:"created_at" gorm:"not null;autoCreateTime"`
|
||||
UserID int64 `json:"user_id" gorm:"primaryKey"` // 关联的用户 ID,外键指向 auth_users.id
|
||||
RoleID int `json:"role_id" gorm:"primaryKey"` // 关联的角色 ID,外键指向 auth_roles.id
|
||||
CreatedAt time.Time `json:"created_at" gorm:"not null;autoCreateTime"` // 角色分配时间,由 GORM 自动填充
|
||||
}
|
||||
|
||||
// TableName 指定数据库表名
|
||||
|
||||
@@ -5,19 +5,19 @@ import "time"
|
||||
|
||||
// User 用户主表模型,对应 auth_users 表
|
||||
type User struct {
|
||||
ID int64 `json:"id" gorm:"primaryKey;autoIncrement"`
|
||||
Username string `json:"username" gorm:"uniqueIndex;size:50;not null"`
|
||||
Email string `json:"email" gorm:"uniqueIndex;size:100;not null"`
|
||||
PasswordHash string `json:"-" gorm:"column:password_hash;size:255;not null"` // JSON 序列化时隐藏密码
|
||||
Nickname string `json:"nickname" gorm:"size:50;not null;default:''"`
|
||||
Avatar string `json:"avatar" gorm:"size:500;not null;default:''"`
|
||||
Gender int `json:"gender" gorm:"not null;default:0"` // 0=未知, 1=男, 2=女
|
||||
Phone *string `json:"phone" gorm:"size:20"` // 可选字段
|
||||
Status int `json:"status" gorm:"not null;default:1"` // 1=正常, 2=禁用, 3=注销
|
||||
LastLoginAt *time.Time `json:"last_login_at"`
|
||||
LastLoginIP *string `json:"last_login_ip" gorm:"column:last_login_ip;size:50"`
|
||||
CreatedAt time.Time `json:"created_at" gorm:"not null;autoCreateTime"`
|
||||
UpdatedAt time.Time `json:"updated_at" gorm:"not null;autoUpdateTime"`
|
||||
ID int64 `json:"id" gorm:"primaryKey;autoIncrement"` // 用户唯一标识,自增主键
|
||||
Username string `json:"username" gorm:"uniqueIndex;size:50;not null"` // 登录用户名,全局唯一
|
||||
Email string `json:"email" gorm:"uniqueIndex;size:100;not null"` // 邮箱地址,全局唯一,用于登录和找回密码
|
||||
PasswordHash string `json:"-" gorm:"column:password_hash;size:255;not null"` // bcrypt 加密后的密码哈希,JSON 序列化时隐藏
|
||||
Nickname string `json:"nickname" gorm:"size:50;not null;default:''"` // 用户昵称,用于页面展示,允许重复
|
||||
Avatar string `json:"avatar" gorm:"size:500;not null;default:''"` // 头像 URL 地址,为空则使用默认头像
|
||||
Gender int `json:"gender" gorm:"not null;default:0"` // 性别:0=未知, 1=男, 2=女(对应 constants.Gender*)
|
||||
Phone *string `json:"phone" gorm:"size:20"` // 手机号,可选字段,指针类型允许 NULL
|
||||
Status int `json:"status" gorm:"not null;default:1"` // 账号状态:1=正常, 2=禁用, 3=注销(对应 constants.UserStatus*)
|
||||
LastLoginAt *time.Time `json:"last_login_at"` // 最后登录时间,首次注册时为 NULL
|
||||
LastLoginIP *string `json:"last_login_ip" gorm:"column:last_login_ip;size:50"` // 最后登录 IP 地址,用于安全审计
|
||||
CreatedAt time.Time `json:"created_at" gorm:"not null;autoCreateTime"` // 账号创建时间,由 GORM 自动填充
|
||||
UpdatedAt time.Time `json:"updated_at" gorm:"not null;autoUpdateTime"` // 最后更新时间,由 GORM 自动更新
|
||||
}
|
||||
|
||||
// TableName 指定数据库表名
|
||||
|
||||
@@ -2,12 +2,18 @@
|
||||
package auth
|
||||
|
||||
import (
|
||||
"github.com/echochat/backend/app/auth/controller"
|
||||
"github.com/echochat/backend/app/auth/dao"
|
||||
"github.com/echochat/backend/app/auth/service"
|
||||
"github.com/google/wire"
|
||||
)
|
||||
|
||||
// AuthSet Auth 模块依赖注入 Provider Set
|
||||
// 提供 DAO 层、Service 层和 Controller 层的所有组件
|
||||
var AuthSet = wire.NewSet(
|
||||
dao.NewUserDAO,
|
||||
dao.NewRoleDAO,
|
||||
service.NewAuthService,
|
||||
controller.NewAuthController,
|
||||
controller.NewAdminAuthController,
|
||||
)
|
||||
|
||||
40
backend/go-service/app/auth/router.go
Normal file
40
backend/go-service/app/auth/router.go
Normal file
@@ -0,0 +1,40 @@
|
||||
package auth
|
||||
|
||||
import (
|
||||
"github.com/echochat/backend/app/auth/controller"
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
// RegisterRoutes 注册 auth 模块的所有路由
|
||||
// 包含前台公开路由、前台需认证路由、后台管理认证路由
|
||||
// authMiddleware 为 JWT 认证中间件,由外部传入以解耦
|
||||
func RegisterRoutes(
|
||||
r *gin.Engine,
|
||||
ctrl *controller.AuthController,
|
||||
adminCtrl *controller.AdminAuthController,
|
||||
authMiddleware gin.HandlerFunc,
|
||||
) {
|
||||
// 前台公开路由(无需认证)
|
||||
public := r.Group("/api/v1/auth")
|
||||
{
|
||||
public.POST("/register", ctrl.Register)
|
||||
public.POST("/login", ctrl.Login)
|
||||
public.POST("/refresh-token", ctrl.RefreshToken)
|
||||
}
|
||||
|
||||
// 前台需认证路由
|
||||
authed := r.Group("/api/v1/auth")
|
||||
authed.Use(authMiddleware)
|
||||
{
|
||||
authed.POST("/logout", ctrl.Logout)
|
||||
authed.GET("/profile", ctrl.GetProfile)
|
||||
authed.PUT("/profile", ctrl.UpdateProfile)
|
||||
authed.PUT("/password", ctrl.ChangePassword)
|
||||
}
|
||||
|
||||
// 后台管理认证路由(无需认证,登录后在 Service 层检查管理员角色)
|
||||
admin := r.Group("/api/v1/admin/auth")
|
||||
{
|
||||
admin.POST("/login", adminCtrl.AdminLogin)
|
||||
}
|
||||
}
|
||||
390
backend/go-service/app/auth/service/auth_service.go
Normal file
390
backend/go-service/app/auth/service/auth_service.go
Normal file
@@ -0,0 +1,390 @@
|
||||
// Package service 提供 auth 模块的核心业务逻辑
|
||||
package service
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
|
||||
"github.com/echochat/backend/app/auth/dao"
|
||||
"github.com/echochat/backend/app/auth/model"
|
||||
"github.com/echochat/backend/app/constants"
|
||||
"github.com/echochat/backend/app/dto"
|
||||
"github.com/echochat/backend/config"
|
||||
"github.com/echochat/backend/pkg/logs"
|
||||
"github.com/echochat/backend/pkg/utils"
|
||||
"go.uber.org/zap"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
var (
|
||||
ErrUserAlreadyExists = errors.New("用户名或邮箱已被注册")
|
||||
ErrUserNotFound = errors.New("用户不存在")
|
||||
ErrPasswordWrong = errors.New("密码错误")
|
||||
ErrUserDisabled = errors.New("账号已被禁用")
|
||||
ErrUserDeleted = errors.New("账号已注销")
|
||||
ErrNotAdmin = errors.New("该账号无管理员权限")
|
||||
ErrRefreshTokenType = errors.New("无效的 Refresh Token 类型")
|
||||
)
|
||||
|
||||
// AuthService 认证服务,处理注册、登录、Token 管理、个人信息等业务逻辑
|
||||
type AuthService struct {
|
||||
userDAO *dao.UserDAO
|
||||
roleDAO *dao.RoleDAO
|
||||
jwtCfg *config.JWTConfig
|
||||
}
|
||||
|
||||
// NewAuthService 创建认证服务实例
|
||||
func NewAuthService(userDAO *dao.UserDAO, roleDAO *dao.RoleDAO, jwtCfg *config.JWTConfig) *AuthService {
|
||||
return &AuthService{
|
||||
userDAO: userDAO,
|
||||
roleDAO: roleDAO,
|
||||
jwtCfg: jwtCfg,
|
||||
}
|
||||
}
|
||||
|
||||
// Register 用户注册
|
||||
// 流程:检查用户名/邮箱是否重复 → 加密密码 → 创建用户 → 分配默认角色 → 生成 Token
|
||||
func (s *AuthService) Register(ctx context.Context, req *dto.RegisterRequest) (*dto.LoginResponse, error) {
|
||||
funcName := "service.auth_service.Register"
|
||||
logs.Info(ctx, funcName, "开始处理注册",
|
||||
zap.String("username", req.Username),
|
||||
zap.String("email", logs.MaskEmail(req.Email)),
|
||||
)
|
||||
|
||||
var err error
|
||||
defer func() {
|
||||
if err != nil {
|
||||
logs.Error(ctx, funcName, "注册处理失败",
|
||||
zap.String("username", req.Username),
|
||||
zap.Error(err),
|
||||
)
|
||||
}
|
||||
}()
|
||||
|
||||
// 检查用户名是否已存在
|
||||
existing, _ := s.userDAO.FindByUsername(ctx, req.Username)
|
||||
if existing != nil {
|
||||
err = ErrUserAlreadyExists
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// 检查邮箱是否已存在
|
||||
existing, _ = s.userDAO.FindByEmail(ctx, req.Email)
|
||||
if existing != nil {
|
||||
err = ErrUserAlreadyExists
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// 加密密码
|
||||
hashedPassword, hashErr := utils.HashPassword(req.Password)
|
||||
if hashErr != nil {
|
||||
err = hashErr
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// 创建用户
|
||||
nickname := req.Nickname
|
||||
if nickname == "" {
|
||||
nickname = req.Username
|
||||
}
|
||||
user := &model.User{
|
||||
Username: req.Username,
|
||||
Email: req.Email,
|
||||
PasswordHash: hashedPassword,
|
||||
Nickname: nickname,
|
||||
Status: constants.UserStatusActive,
|
||||
}
|
||||
if err = s.userDAO.Create(ctx, user); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// 分配默认角色(普通用户)
|
||||
defaultRole, roleErr := s.roleDAO.FindByCode(ctx, constants.RoleUser)
|
||||
if roleErr != nil {
|
||||
logs.Warn(ctx, funcName, "查找默认角色失败,跳过角色分配", zap.Error(roleErr))
|
||||
} else {
|
||||
if assignErr := s.roleDAO.AssignRole(ctx, user.ID, defaultRole.ID); assignErr != nil {
|
||||
logs.Warn(ctx, funcName, "分配默认角色失败", zap.Error(assignErr))
|
||||
}
|
||||
}
|
||||
|
||||
// 获取角色列表并生成 Token
|
||||
roles, _ := s.roleDAO.GetUserRoleCodes(ctx, user.ID)
|
||||
resp, err := s.buildLoginResponse(user, roles)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
logs.Info(ctx, funcName, "注册成功", zap.Int64("user_id", user.ID))
|
||||
return resp, nil
|
||||
}
|
||||
|
||||
// Login 用户登录
|
||||
// 流程:查找用户 → 校验密码 → 检查账号状态 → 更新登录信息 → 生成 Token
|
||||
func (s *AuthService) Login(ctx context.Context, req *dto.LoginRequest, clientIP string) (*dto.LoginResponse, error) {
|
||||
funcName := "service.auth_service.Login"
|
||||
logs.Info(ctx, funcName, "开始处理登录",
|
||||
zap.String("account", req.Account),
|
||||
)
|
||||
|
||||
var err error
|
||||
defer func() {
|
||||
if err != nil {
|
||||
logs.Error(ctx, funcName, "登录处理失败",
|
||||
zap.String("account", req.Account),
|
||||
zap.Error(err),
|
||||
)
|
||||
}
|
||||
}()
|
||||
|
||||
// 按用户名或邮箱查找用户
|
||||
user, findErr := s.userDAO.FindByAccount(ctx, req.Account)
|
||||
if findErr != nil {
|
||||
if errors.Is(findErr, gorm.ErrRecordNotFound) {
|
||||
err = ErrUserNotFound
|
||||
} else {
|
||||
err = findErr
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// 校验密码
|
||||
if !utils.CheckPassword(req.Password, user.PasswordHash) {
|
||||
err = ErrPasswordWrong
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// 检查账号状态
|
||||
if err = s.checkUserStatus(user.Status); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// 更新最后登录信息
|
||||
_ = s.userDAO.UpdateLastLogin(ctx, user.ID, clientIP)
|
||||
|
||||
// 获取角色列表并生成 Token
|
||||
roles, _ := s.roleDAO.GetUserRoleCodes(ctx, user.ID)
|
||||
resp, err := s.buildLoginResponse(user, roles)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
logs.Info(ctx, funcName, "登录成功",
|
||||
zap.Int64("user_id", user.ID),
|
||||
zap.String("username", user.Username),
|
||||
)
|
||||
return resp, nil
|
||||
}
|
||||
|
||||
// AdminLogin 管理后台登录
|
||||
// 与普通登录相同,但额外检查用户是否拥有 admin 或 super_admin 角色
|
||||
func (s *AuthService) AdminLogin(ctx context.Context, req *dto.LoginRequest, clientIP string) (*dto.LoginResponse, error) {
|
||||
funcName := "service.auth_service.AdminLogin"
|
||||
logs.Info(ctx, funcName, "开始处理管理员登录",
|
||||
zap.String("account", req.Account),
|
||||
)
|
||||
|
||||
resp, err := s.Login(ctx, req, clientIP)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// 检查是否拥有管理员角色
|
||||
hasAdmin := false
|
||||
for _, role := range resp.User.Roles {
|
||||
if role == constants.RoleAdmin || role == constants.RoleSuperAdmin {
|
||||
hasAdmin = true
|
||||
break
|
||||
}
|
||||
}
|
||||
if !hasAdmin {
|
||||
logs.Warn(ctx, funcName, "非管理员尝试后台登录",
|
||||
zap.String("account", req.Account),
|
||||
)
|
||||
return nil, ErrNotAdmin
|
||||
}
|
||||
|
||||
logs.Info(ctx, funcName, "管理员登录成功",
|
||||
zap.String("account", req.Account),
|
||||
)
|
||||
return resp, nil
|
||||
}
|
||||
|
||||
// RefreshToken 刷新 Access Token
|
||||
// 验证 Refresh Token 有效性后,重新生成一对新的 Token
|
||||
func (s *AuthService) RefreshToken(ctx context.Context, refreshToken string) (*dto.LoginResponse, error) {
|
||||
funcName := "service.auth_service.RefreshToken"
|
||||
|
||||
// 解析 Refresh Token
|
||||
claims, err := utils.ParseToken(s.jwtCfg, refreshToken)
|
||||
if err != nil {
|
||||
logs.Warn(ctx, funcName, "Refresh Token 解析失败", zap.Error(err))
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// 验证 Token 类型
|
||||
if claims.Subject != "refresh" {
|
||||
return nil, ErrRefreshTokenType
|
||||
}
|
||||
|
||||
// 查找用户(确保用户仍然有效)
|
||||
user, err := s.userDAO.FindByID(ctx, claims.UserID)
|
||||
if err != nil {
|
||||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
return nil, ErrUserNotFound
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if err = s.checkUserStatus(user.Status); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// 获取角色并生成新 Token
|
||||
roles, _ := s.roleDAO.GetUserRoleCodes(ctx, user.ID)
|
||||
resp, err := s.buildLoginResponse(user, roles)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
logs.Info(ctx, funcName, "Token 刷新成功", zap.Int64("user_id", user.ID))
|
||||
return resp, nil
|
||||
}
|
||||
|
||||
// GetProfile 获取用户个人信息
|
||||
func (s *AuthService) GetProfile(ctx context.Context, userID int64) (*dto.UserInfo, error) {
|
||||
funcName := "service.auth_service.GetProfile"
|
||||
logs.Debug(ctx, funcName, "获取用户信息", zap.Int64("user_id", userID))
|
||||
|
||||
user, err := s.userDAO.FindByID(ctx, userID)
|
||||
if err != nil {
|
||||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
return nil, ErrUserNotFound
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
|
||||
roles, _ := s.roleDAO.GetUserRoleCodes(ctx, user.ID)
|
||||
|
||||
return s.buildUserInfo(user, roles), nil
|
||||
}
|
||||
|
||||
// UpdateProfile 更新用户个人资料
|
||||
func (s *AuthService) UpdateProfile(ctx context.Context, userID int64, req *dto.UpdateProfileRequest) (*dto.UserInfo, error) {
|
||||
funcName := "service.auth_service.UpdateProfile"
|
||||
logs.Info(ctx, funcName, "更新用户资料", zap.Int64("user_id", userID))
|
||||
|
||||
user, err := s.userDAO.FindByID(ctx, userID)
|
||||
if err != nil {
|
||||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
return nil, ErrUserNotFound
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if req.Nickname != "" {
|
||||
user.Nickname = req.Nickname
|
||||
}
|
||||
if req.Avatar != "" {
|
||||
user.Avatar = req.Avatar
|
||||
}
|
||||
if req.Gender != nil {
|
||||
user.Gender = *req.Gender
|
||||
}
|
||||
if req.Phone != "" {
|
||||
user.Phone = &req.Phone
|
||||
}
|
||||
|
||||
if err = s.userDAO.Update(ctx, user); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
roles, _ := s.roleDAO.GetUserRoleCodes(ctx, user.ID)
|
||||
logs.Info(ctx, funcName, "用户资料更新成功", zap.Int64("user_id", userID))
|
||||
return s.buildUserInfo(user, roles), nil
|
||||
}
|
||||
|
||||
// ChangePassword 修改密码
|
||||
func (s *AuthService) ChangePassword(ctx context.Context, userID int64, req *dto.ChangePasswordRequest) error {
|
||||
funcName := "service.auth_service.ChangePassword"
|
||||
logs.Info(ctx, funcName, "修改密码", zap.Int64("user_id", userID))
|
||||
|
||||
user, err := s.userDAO.FindByID(ctx, userID)
|
||||
if err != nil {
|
||||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
return ErrUserNotFound
|
||||
}
|
||||
return err
|
||||
}
|
||||
|
||||
// 校验旧密码
|
||||
if !utils.CheckPassword(req.OldPassword, user.PasswordHash) {
|
||||
return ErrPasswordWrong
|
||||
}
|
||||
|
||||
// 加密新密码
|
||||
hashedPassword, err := utils.HashPassword(req.NewPassword)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
user.PasswordHash = hashedPassword
|
||||
if err = s.userDAO.Update(ctx, user); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
logs.Info(ctx, funcName, "密码修改成功", zap.Int64("user_id", userID))
|
||||
return nil
|
||||
}
|
||||
|
||||
// checkUserStatus 检查用户账号状态
|
||||
func (s *AuthService) checkUserStatus(status int) error {
|
||||
switch status {
|
||||
case constants.UserStatusDisabled:
|
||||
return ErrUserDisabled
|
||||
case constants.UserStatusDeleted:
|
||||
return ErrUserDeleted
|
||||
default:
|
||||
return nil
|
||||
}
|
||||
}
|
||||
|
||||
// buildLoginResponse 构建登录响应(生成 Token + 用户信息)
|
||||
func (s *AuthService) buildLoginResponse(user *model.User, roles []string) (*dto.LoginResponse, error) {
|
||||
if roles == nil {
|
||||
roles = []string{}
|
||||
}
|
||||
|
||||
token, err := utils.GenerateToken(s.jwtCfg, user.ID, user.Username, roles)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
refreshToken, err := utils.GenerateRefreshToken(s.jwtCfg, user.ID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &dto.LoginResponse{
|
||||
Token: token,
|
||||
RefreshToken: refreshToken,
|
||||
ExpiresIn: int64(s.jwtCfg.AccessExpireMin) * 60,
|
||||
User: *s.buildUserInfo(user, roles),
|
||||
}, nil
|
||||
}
|
||||
|
||||
// buildUserInfo 从 model.User 构建 dto.UserInfo
|
||||
func (s *AuthService) buildUserInfo(user *model.User, roles []string) *dto.UserInfo {
|
||||
if roles == nil {
|
||||
roles = []string{}
|
||||
}
|
||||
return &dto.UserInfo{
|
||||
ID: user.ID,
|
||||
Username: user.Username,
|
||||
Email: user.Email,
|
||||
Nickname: user.Nickname,
|
||||
Avatar: user.Avatar,
|
||||
Gender: user.Gender,
|
||||
Roles: roles,
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user