8a6576a2f7
- constants: 用户状态、性别、角色代码常量定义 - model: User/Role/UserRole GORM 模型 - dao: UserDAO (CRUD + 账号查询) + RoleDAO (角色分配/查询/检查) - provider: Auth 模块 Wire Provider Set Made-with: Cursor
120 lines
3.2 KiB
Go
120 lines
3.2 KiB
Go
package dao
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/echochat/backend/app/auth/model"
|
|
"github.com/echochat/backend/pkg/logs"
|
|
"go.uber.org/zap"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// RoleDAO 角色数据访问对象
|
|
type RoleDAO struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
// NewRoleDAO 创建 RoleDAO 实例
|
|
func NewRoleDAO(db *gorm.DB) *RoleDAO {
|
|
return &RoleDAO{db: db}
|
|
}
|
|
|
|
// FindByCode 按角色代码查询角色
|
|
// 利用 code 唯一索引，精确匹配
|
|
func (d *RoleDAO) FindByCode(ctx context.Context, code string) (*model.Role, error) {
|
|
funcName := "dao.role_dao.FindByCode"
|
|
logs.Debug(ctx, funcName, "按代码查询角色", zap.String("code", code))
|
|
|
|
var role model.Role
|
|
err := d.db.WithContext(ctx).Where("code = ?", code).First(&role).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &role, nil
|
|
}
|
|
|
|
// AssignRole 为用户分配角色
|
|
// 插入 user_id + role_id 关联记录
|
|
func (d *RoleDAO) AssignRole(ctx context.Context, userID int64, roleID int) error {
|
|
funcName := "dao.role_dao.AssignRole"
|
|
logs.Info(ctx, funcName, "分配角色",
|
|
zap.Int64("user_id", userID),
|
|
zap.Int("role_id", roleID),
|
|
)
|
|
|
|
var err error
|
|
defer func() {
|
|
if err != nil {
|
|
logs.Error(ctx, funcName, "分配角色失败", zap.Error(err))
|
|
}
|
|
}()
|
|
|
|
userRole := model.UserRole{
|
|
UserID: userID,
|
|
RoleID: roleID,
|
|
}
|
|
err = d.db.WithContext(ctx).Create(&userRole).Error
|
|
return err
|
|
}
|
|
|
|
// GetUserRoles 获取用户的所有角色列表
|
|
// 通过 JOIN auth_roles 表获取角色完整信息
|
|
func (d *RoleDAO) GetUserRoles(ctx context.Context, userID int64) ([]model.Role, error) {
|
|
funcName := "dao.role_dao.GetUserRoles"
|
|
logs.Debug(ctx, funcName, "获取用户角色", zap.Int64("user_id", userID))
|
|
|
|
var roles []model.Role
|
|
err := d.db.WithContext(ctx).
|
|
Joins("JOIN auth_user_roles ON auth_user_roles.role_id = auth_roles.id").
|
|
Where("auth_user_roles.user_id = ?", userID).
|
|
Find(&roles).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return roles, nil
|
|
}
|
|
|
|
// GetUserRoleCodes 获取用户的角色代码列表（便于权限判断）
|
|
func (d *RoleDAO) GetUserRoleCodes(ctx context.Context, userID int64) ([]string, error) {
|
|
roles, err := d.GetUserRoles(ctx, userID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
codes := make([]string, 0, len(roles))
|
|
for _, r := range roles {
|
|
codes = append(codes, r.Code)
|
|
}
|
|
return codes, nil
|
|
}
|
|
|
|
// HasRole 检查用户是否拥有指定角色
|
|
func (d *RoleDAO) HasRole(ctx context.Context, userID int64, roleCode string) (bool, error) {
|
|
funcName := "dao.role_dao.HasRole"
|
|
logs.Debug(ctx, funcName, "检查用户角色",
|
|
zap.Int64("user_id", userID),
|
|
zap.String("role_code", roleCode),
|
|
)
|
|
|
|
var count int64
|
|
err := d.db.WithContext(ctx).
|
|
Model(&model.UserRole{}).
|
|
Joins("JOIN auth_roles ON auth_roles.id = auth_user_roles.role_id").
|
|
Where("auth_user_roles.user_id = ? AND auth_roles.code = ?", userID, roleCode).
|
|
Count(&count).Error
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
return count > 0, nil
|
|
}
|
|
|
|
// RemoveUserRoles 移除用户的所有角色（用于重新分配）
|
|
func (d *RoleDAO) RemoveUserRoles(ctx context.Context, userID int64) error {
|
|
funcName := "dao.role_dao.RemoveUserRoles"
|
|
logs.Info(ctx, funcName, "移除用户所有角色", zap.Int64("user_id", userID))
|
|
|
|
return d.db.WithContext(ctx).
|
|
Where("user_id = ?", userID).
|
|
Delete(&model.UserRole{}).Error
|
|
}
|