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

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
+18 -15
View File
@@ -1,6 +1,9 @@
package adminrole
import "errors"
import (
"context"
"errors"
)
var ErrDependencyUnavailable = errors.New("dependency unavailable")
@@ -12,54 +15,54 @@ func NewService(repo *Repository) *Service {
return &Service{repo: repo}
}
func (s *Service) List() ([]RoleDTO, error) {
func (s *Service) List(ctx context.Context) ([]RoleDTO, error) {
if s.repo == nil {
return nil, ErrDependencyUnavailable
}
return s.repo.List()
return s.repo.List(ctx)
}
func (s *Service) FindByID(id uint64) (*RoleDTO, error) {
func (s *Service) FindByID(ctx context.Context, id uint64) (*RoleDTO, error) {
if s.repo == nil {
return nil, ErrDependencyUnavailable
}
return s.repo.FindByID(id)
return s.repo.FindByID(ctx, id)
}
func (s *Service) Create(req CreateRoleRequest) (*RoleDTO, error) {
func (s *Service) Create(ctx context.Context, req CreateRoleRequest) (*RoleDTO, error) {
if s.repo == nil {
return nil, ErrDependencyUnavailable
}
if req.Code == "" || req.Name == "" {
return nil, errors.New("code and name are required")
}
return s.repo.Create(req)
return s.repo.Create(ctx, req)
}
func (s *Service) Update(id uint64, req UpdateRoleRequest) (*RoleDTO, error) {
func (s *Service) Update(ctx context.Context, id uint64, req UpdateRoleRequest) (*RoleDTO, error) {
if s.repo == nil {
return nil, ErrDependencyUnavailable
}
return s.repo.Update(id, req)
return s.repo.Update(ctx, id, req)
}
func (s *Service) Delete(id uint64) error {
func (s *Service) Delete(ctx context.Context, id uint64) error {
if s.repo == nil {
return ErrDependencyUnavailable
}
return s.repo.Delete(id)
return s.repo.Delete(ctx, id)
}
func (s *Service) AssignPermissions(roleID uint64, req AssignPermissionsRequest) error {
func (s *Service) AssignPermissions(ctx context.Context, roleID uint64, req AssignPermissionsRequest) error {
if s.repo == nil {
return ErrDependencyUnavailable
}
return s.repo.AssignPermissions(roleID, req.PermissionIDs)
return s.repo.AssignPermissions(ctx, roleID, req.PermissionIDs)
}
func (s *Service) ListPermissions() ([]PermissionDTO, error) {
func (s *Service) ListPermissions(ctx context.Context) ([]PermissionDTO, error) {
if s.repo == nil {
return nil, ErrDependencyUnavailable
}
return s.repo.ListPermissions()
return s.repo.ListPermissions(ctx)
}