rbac

internal/repository/admin/permission_repo.go

@@ -0,0 +1,116 @@
+package admin
+
+import (
+	"context"
+
+	"gorm.io/gorm"
+
+	"portal-mini-go/internal/model/admin"
+	"portal-mini-go/internal/pkg/errors"
+)
+
+// PermissionRepository 权限（接口 / 菜单）的数据访问。
+type PermissionRepository struct {
+	db *gorm.DB
+}
+
+func NewPermissionRepository(db *gorm.DB) *PermissionRepository {
+	return &PermissionRepository{db: db}
+}
+
+// GetByID 按 ID 查询权限，不存在返回 ErrNotFound。
+func (r *PermissionRepository) GetByID(ctx context.Context, id uint64) (*admin.Permission, error) {
+	var m admin.Permission
+	if err := r.db.WithContext(ctx).Where("id = ?", id).First(&m).Error; err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return nil, errors.ErrNotFound
+		}
+		return nil, errors.Wrap(50000, "query permission", err)
+	}
+	return &m, nil
+}
+
+// Create 新建权限。
+func (r *PermissionRepository) Create(ctx context.Context, m *admin.Permission) error {
+	if err := r.db.WithContext(ctx).Create(m).Error; err != nil {
+		return errors.Wrap(50000, "create permission", err)
+	}
+	return nil
+}
+
+// PermListFilter 权限列表过滤条件。Type 为空表示不过滤类型。
+type PermListFilter struct {
+	Type    string
+	Keyword string
+	Page    int
+	Size    int
+}
+
+// List 分页查询权限，支持类型与关键字（名称 / code / 路径）过滤。
+func (r *PermissionRepository) List(ctx context.Context, f PermListFilter) ([]admin.Permission, int64, error) {
+	q := r.db.WithContext(ctx).Model(&admin.Permission{})
+	if f.Type != "" {
+		q = q.Where("type = ?", f.Type)
+	}
+	if f.Keyword != "" {
+		kw := "%" + f.Keyword + "%"
+		q = q.Where("name LIKE ? OR code LIKE ? OR path LIKE ?", kw, kw, kw)
+	}
+	var total int64
+	if err := q.Count(&total).Error; err != nil {
+		return nil, 0, errors.Wrap(50000, "count permission", err)
+	}
+	var list []admin.Permission
+	if err := q.Order("sort ASC, id ASC").Offset((f.Page - 1) * f.Size).Limit(f.Size).Find(&list).Error; err != nil {
+		return nil, 0, errors.Wrap(50000, "list permission", err)
+	}
+	return list, total, nil
+}
+
+// Update 按 ID 局部更新权限非零值字段。
+func (r *PermissionRepository) Update(ctx context.Context, id uint64, cols map[string]interface{}) error {
+	res := r.db.WithContext(ctx).Model(&admin.Permission{}).Where("id = ?", id).Updates(cols)
+	if res.Error != nil {
+		return errors.Wrap(50000, "update permission", res.Error)
+	}
+	if res.RowsAffected == 0 {
+		return errors.ErrNotFound
+	}
+	return nil
+}
+
+// Delete 删除权限，并清理角色-权限关联。
+func (r *PermissionRepository) Delete(ctx context.Context, id uint64) error {
+	if err := r.db.WithContext(ctx).Where("permission_id = ?", id).Delete(&admin.RolePermission{}).Error; err != nil {
+		return errors.Wrap(50000, "delete role permissions", err)
+	}
+	res := r.db.WithContext(ctx).Where("id = ?", id).Delete(&admin.Permission{})
+	if res.Error != nil {
+		return errors.Wrap(50000, "delete permission", res.Error)
+	}
+	if res.RowsAffected == 0 {
+		return errors.ErrNotFound
+	}
+	return nil
+}
+
+// AllMenuPermissions 返回全部菜单类型权限（用于组装菜单树）。
+func (r *PermissionRepository) AllMenuPermissions(ctx context.Context) ([]admin.Permission, error) {
+	var list []admin.Permission
+	if err := r.db.WithContext(ctx).
+		Where("type = ?", admin.PermTypeMenu).
+		Order("sort ASC, id ASC").
+		Find(&list).Error; err != nil {
+		return nil, errors.Wrap(50000, "list menu permissions", err)
+	}
+	return list, nil
+}
+
+// AllCodes 返回全部权限 code（超级管理员短路放行时使用）。
+func (r *PermissionRepository) AllCodes(ctx context.Context) ([]string, error) {
+	var codes []string
+	if err := r.db.WithContext(ctx).Model(&admin.Permission{}).Pluck("code", &codes).Error; err != nil {
+		return nil, errors.Wrap(50000, "list permission codes", err)
+	}
+	return codes, nil
+}