yml/hfb_sys
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

完成剩余模块 Context 超时控制改造

Browse Source
This commit is contained in:
yml2213
2026-06-10 12:58:59 +08:00
parent d2858c529d
commit 6ae8f0e830
24 changed files with 370 additions and 323 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/internal/modules/adminmgr/repository.go
+39 -35
View File
@@ -23,19 +23,20 @@ func NewRepository(db *gorm.DB, redis *redis.Client) *Repository {
return &Repository{db: db, redis: redis}
}
func (r *Repository) List(page, pageSize int) (*PaginatedResult, error) {
func (r *Repository) List(ctx context.Context, page, pageSize int) (*PaginatedResult, error) {
db := r.db.WithContext(ctx)
var total int64
if err := r.db.Model(&model.AdminUser{}).Count(&total).Error; err != nil {
if err := db.Model(&model.AdminUser{}).Count(&total).Error; err != nil {
return nil, err
}
offset := (page - 1) * pageSize
var admins []model.AdminUser
if err := r.db.Order("id DESC").Offset(offset).Limit(pageSize).Find(&admins).Error; err != nil {
if err := db.Order("id DESC").Offset(offset).Limit(pageSize).Find(&admins).Error; err != nil {
return nil, err
}
items := make([]AdminUserDTO, 0, len(admins))
for _, admin := range admins {
roles, _ := r.getAdminRoles(admin.ID)
roles, _ := r.getAdminRoles(ctx, admin.ID)
items = append(items, AdminUserDTO{
ID: admin.ID,
Username: admin.Username,
@@ -50,12 +51,12 @@ func (r *Repository) List(page, pageSize int) (*PaginatedResult, error) {
return &PaginatedResult{Items: items, Total: total, Page: page, PageSize: pageSize}, nil
}
func (r *Repository) FindByID(id uint64) (*AdminUserDTO, error) {
func (r *Repository) FindByID(ctx context.Context, id uint64) (*AdminUserDTO, error) {
var admin model.AdminUser
if err := r.db.First(&admin, id).Error; err != nil {
if err := r.db.WithContext(ctx).First(&admin, id).Error; err != nil {
return nil, err
}
roles, _ := r.getAdminRoles(admin.ID)
roles, _ := r.getAdminRoles(ctx, admin.ID)
dto := AdminUserDTO{
ID: admin.ID,
Username: admin.Username,
@@ -69,7 +70,7 @@ func (r *Repository) FindByID(id uint64) (*AdminUserDTO, error) {
return &dto, nil
}
func (r *Repository) Create(req CreateAdminRequest) (*AdminUserDTO, error) {
func (r *Repository) Create(ctx context.Context, req CreateAdminRequest) (*AdminUserDTO, error) {
hash, err := bcrypt.GenerateFromPassword([]byte(req.Password), bcrypt.DefaultCost)
if err != nil {
return nil, err
@@ -80,15 +81,16 @@ func (r *Repository) Create(req CreateAdminRequest) (*AdminUserDTO, error) {
Nickname: req.Nickname,
Status: "active",
}
if err := r.db.Create(&admin).Error; err != nil {
if err := r.db.WithContext(ctx).Create(&admin).Error; err != nil {
return nil, err
}
return r.FindByID(admin.ID)
return r.FindByID(ctx, admin.ID)
}
func (r *Repository) Update(id uint64, req UpdateAdminRequest) (*AdminUserDTO, error) {
func (r *Repository) Update(ctx context.Context, id uint64, req UpdateAdminRequest) (*AdminUserDTO, error) {
db := r.db.WithContext(ctx)
var admin model.AdminUser
if err := r.db.First(&admin, id).Error; err != nil {
if err := db.First(&admin, id).Error; err != nil {
return nil, err
}
if req.Nickname != "" {
@@ -97,46 +99,48 @@ func (r *Repository) Update(id uint64, req UpdateAdminRequest) (*AdminUserDTO, e
if req.Status != "" {
admin.Status = req.Status
}
if err := r.db.Save(&admin).Error; err != nil {
if err := db.Save(&admin).Error; err != nil {
return nil, err
}
return r.FindByID(id)
return r.FindByID(ctx, id)
}
func (r *Repository) Delete(id uint64, currentAdminID uint64) error {
func (r *Repository) Delete(ctx context.Context, id uint64, currentAdminID uint64) error {
if id == currentAdminID {
return ErrCannotDeleteSelf
}
db := r.db.WithContext(ctx)
var admin model.AdminUser
if err := r.db.First(&admin, id).Error; err != nil {
if err := db.First(&admin, id).Error; err != nil {
return err
}
// 检查是否是最后一个 super_admin
isLastSuper, err := r.isLastSuperAdmin(id)
isLastSuper, err := r.isLastSuperAdmin(ctx, id)
if err != nil {
return err
}
if isLastSuper {
return ErrLastSuperAdmin
}
return r.db.Transaction(func(tx *gorm.DB) error {
return db.Transaction(func(tx *gorm.DB) error {
if err := tx.Where("admin_user_id = ?", id).Delete(&model.AdminUserRole{}).Error; err != nil {
return err
}
if err := tx.Delete(&model.AdminUser{}, id).Error; err != nil {
return err
}
r.invalidatePermCache(id)
r.invalidatePermCache(ctx, id)
return nil
})
}
func (r *Repository) AssignRoles(adminID uint64, roleIDs []uint64) error {
func (r *Repository) AssignRoles(ctx context.Context, adminID uint64, roleIDs []uint64) error {
db := r.db.WithContext(ctx)
var admin model.AdminUser
if err := r.db.First(&admin, adminID).Error; err != nil {
if err := db.First(&admin, adminID).Error; err != nil {
return err
}
err := r.db.Transaction(func(tx *gorm.DB) error {
err := db.Transaction(func(tx *gorm.DB) error {
if err := tx.Where("admin_user_id = ?", adminID).Delete(&model.AdminUserRole{}).Error; err != nil {
return err
}
@@ -151,13 +155,13 @@ func (r *Repository) AssignRoles(adminID uint64, roleIDs []uint64) error {
if err != nil {
return err
}
r.invalidatePermCache(adminID)
r.invalidatePermCache(ctx, adminID)
return nil
}
func (r *Repository) ChangePassword(id uint64, oldPwd, newPwd string) error {
func (r *Repository) ChangePassword(ctx context.Context, id uint64, oldPwd, newPwd string) error {
var admin model.AdminUser
if err := r.db.First(&admin, id).Error; err != nil {
if err := r.db.WithContext(ctx).First(&admin, id).Error; err != nil {
return err
}
if err := bcrypt.CompareHashAndPassword([]byte(admin.PasswordHash), []byte(oldPwd)); err != nil {
@@ -167,12 +171,12 @@ func (r *Repository) ChangePassword(id uint64, oldPwd, newPwd string) error {
if err != nil {
return err
}
return r.db.Model(&admin).Update("password_hash", string(hash)).Error
return r.db.WithContext(ctx).Model(&admin).Update("password_hash", string(hash)).Error
}
func (r *Repository) GetPermissionCodes(adminID uint64) ([]string, error) {
func (r *Repository) GetPermissionCodes(ctx context.Context, adminID uint64) ([]string, error) {
var codes []string
err := r.db.Table("admin_user_roles aur").
err := r.db.WithContext(ctx).Table("admin_user_roles aur").
Select("DISTINCT p.code").
Joins("JOIN role_permissions rp ON rp.role_id = aur.role_id").
Joins("JOIN permissions p ON p.id = rp.permission_id").
@@ -181,9 +185,9 @@ func (r *Repository) GetPermissionCodes(adminID uint64) ([]string, error) {
return codes, err
}
func (r *Repository) getAdminRoles(adminID uint64) ([]adminrole.RoleDTO, error) {
func (r *Repository) getAdminRoles(ctx context.Context, adminID uint64) ([]adminrole.RoleDTO, error) {
var roles []model.Role
err := r.db.Table("roles").
err := r.db.WithContext(ctx).Table("roles").
Joins("JOIN admin_user_roles aur ON aur.role_id = roles.id").
Where("aur.admin_user_id = ?", adminID).
Find(&roles).Error
@@ -199,23 +203,23 @@ func (r *Repository) getAdminRoles(adminID uint64) ([]adminrole.RoleDTO, error)
return result, nil
}
func (r *Repository) isLastSuperAdmin(adminID uint64) (bool, error) {
func (r *Repository) isLastSuperAdmin(ctx context.Context, adminID uint64) (bool, error) {
var superRole model.Role
if err := r.db.Where("code = ?", "super_admin").First(&superRole).Error; err != nil {
db := r.db.WithContext(ctx)
if err := db.Where("code = ?", "super_admin").First(&superRole).Error; err != nil {
return false, nil // 没有 super_admin 角色，不限制
}
var count int64
err := r.db.Model(&model.AdminUserRole{}).
err := db.Model(&model.AdminUserRole{}).
Where("role_id = ? AND admin_user_id != ?", superRole.ID, adminID).
Count(&count).Error
return count == 0, err
}
func (r *Repository) invalidatePermCache(adminID uint64) {
func (r *Repository) invalidatePermCache(ctx context.Context, adminID uint64) {
if r.redis == nil {
return
}
ctx := context.Background()
r.redis.Del(ctx, permCacheKey(adminID))
}