128 lines
3.5 KiB
Go
128 lines
3.5 KiB
Go
package rbac
|
|
|
|
import (
|
|
"context"
|
|
|
|
"portal-mini-go/internal/model/admin"
|
|
"portal-mini-go/internal/pkg/errors"
|
|
adminrepo "portal-mini-go/internal/repository/admin"
|
|
)
|
|
|
|
// RoleService 角色业务：CRUD 与权限分配。
|
|
type RoleService struct {
|
|
repo RoleRepository
|
|
invalidator CacheInvalidator
|
|
}
|
|
|
|
func NewRoleService(repo RoleRepository, invalidator CacheInvalidator) *RoleService {
|
|
return &RoleService{repo: repo, invalidator: invalidator}
|
|
}
|
|
|
|
// CreateRoleReq 新建角色请求。
|
|
type CreateRoleReq struct {
|
|
Code string `json:"code" binding:"required"`
|
|
Name string `json:"name" binding:"required"`
|
|
Description string `json:"description"`
|
|
Status int8 `json:"status"`
|
|
}
|
|
|
|
// Create 新建角色，code 唯一。
|
|
func (s *RoleService) Create(ctx context.Context, req CreateRoleReq) (*admin.Role, error) {
|
|
if req.Status == 0 {
|
|
req.Status = admin.UserStatusEnabled
|
|
}
|
|
if req.Status != admin.UserStatusEnabled && req.Status != admin.UserStatusDisabled {
|
|
return nil, errors.New(40000, "非法的角色状态")
|
|
}
|
|
if _, err := s.repo.GetByCode(ctx, req.Code); err == nil {
|
|
return nil, errors.New(40900, "角色 code 已存在")
|
|
} else if !errors.Is(err, errors.ErrNotFound) {
|
|
return nil, err
|
|
}
|
|
r := &admin.Role{Code: req.Code, Name: req.Name, Description: req.Description, Status: req.Status}
|
|
if err := s.repo.Create(ctx, r); err != nil {
|
|
return nil, err
|
|
}
|
|
return r, nil
|
|
}
|
|
|
|
// RoleListReq 角色列表过滤条件。
|
|
type RoleListReq struct {
|
|
Keyword string
|
|
Status int8
|
|
Page int
|
|
Size int
|
|
}
|
|
|
|
// List 分页查询角色。
|
|
func (s *RoleService) List(ctx context.Context, req RoleListReq) ([]admin.Role, int64, error) {
|
|
page, size := req.Page, req.Size
|
|
if page <= 0 {
|
|
page = 1
|
|
}
|
|
if size <= 0 {
|
|
size = 20
|
|
}
|
|
return s.repo.List(ctx, adminrepo.RoleListFilter{
|
|
Keyword: req.Keyword,
|
|
Status: req.Status,
|
|
Page: page,
|
|
Size: size,
|
|
})
|
|
}
|
|
|
|
// GetByID 按 ID 查询角色。
|
|
func (s *RoleService) GetByID(ctx context.Context, id uint64) (*admin.Role, error) {
|
|
return s.repo.GetByID(ctx, id)
|
|
}
|
|
|
|
// UpdateRoleReq 编辑角色请求；指针区分“未提供”与“置空”。
|
|
type UpdateRoleReq struct {
|
|
Name *string `json:"name"`
|
|
Description *string `json:"description"`
|
|
Status *int8 `json:"status"`
|
|
}
|
|
|
|
// Update 局部更新角色。
|
|
func (s *RoleService) Update(ctx context.Context, id uint64, req UpdateRoleReq) (*admin.Role, error) {
|
|
cols := map[string]interface{}{}
|
|
if req.Name != nil {
|
|
cols["name"] = *req.Name
|
|
}
|
|
if req.Description != nil {
|
|
cols["description"] = *req.Description
|
|
}
|
|
if req.Status != nil {
|
|
if *req.Status != admin.UserStatusEnabled && *req.Status != admin.UserStatusDisabled {
|
|
return nil, errors.New(40000, "非法的角色状态")
|
|
}
|
|
cols["status"] = *req.Status
|
|
}
|
|
if len(cols) == 0 {
|
|
return s.repo.GetByID(ctx, id)
|
|
}
|
|
if err := s.repo.Update(ctx, id, cols); err != nil {
|
|
return nil, err
|
|
}
|
|
s.invalidator.InvalidateAll(ctx)
|
|
return s.repo.GetByID(ctx, id)
|
|
}
|
|
|
|
// Delete 删除角色（清理关联并失效全部缓存）。
|
|
func (s *RoleService) Delete(ctx context.Context, id uint64) error {
|
|
if err := s.repo.Delete(ctx, id); err != nil {
|
|
return err
|
|
}
|
|
s.invalidator.InvalidateAll(ctx)
|
|
return nil
|
|
}
|
|
|
|
// SetPermissions 为角色分配权限（全量替换）。
|
|
func (s *RoleService) SetPermissions(ctx context.Context, id uint64, permIDs []uint64) error {
|
|
if err := s.repo.SetPermissions(ctx, id, permIDs); err != nil {
|
|
return err
|
|
}
|
|
s.invalidator.InvalidateAll(ctx)
|
|
return nil
|
|
}
|