mirror of
https://gitee.com/dromara/mayfly-go
synced 2025-11-02 23:40:24 +08:00
92 lines
2.5 KiB
Go
92 lines
2.5 KiB
Go
package application
|
||
|
||
import (
|
||
"mayfly-go/internal/sys/domain/entity"
|
||
"mayfly-go/internal/sys/domain/repository"
|
||
"mayfly-go/pkg/biz"
|
||
"mayfly-go/pkg/model"
|
||
"strings"
|
||
)
|
||
|
||
type Resource interface {
|
||
GetResourceList(condition *entity.Resource, toEntity any, orderBy ...string)
|
||
|
||
GetById(id uint64, cols ...string) *entity.Resource
|
||
|
||
GetByIdIn(ids []uint64, toEntity any, cols ...string)
|
||
|
||
Save(entity *entity.Resource)
|
||
|
||
Delete(id uint64)
|
||
|
||
GetAccountResources(accountId uint64, toEntity any)
|
||
}
|
||
|
||
func newResourceApp(resourceRepo repository.Resource) Resource {
|
||
return &resourceAppImpl{
|
||
resourceRepo: resourceRepo,
|
||
}
|
||
}
|
||
|
||
type resourceAppImpl struct {
|
||
resourceRepo repository.Resource
|
||
}
|
||
|
||
func (r *resourceAppImpl) GetResourceList(condition *entity.Resource, toEntity any, orderBy ...string) {
|
||
r.resourceRepo.GetResourceList(condition, toEntity, orderBy...)
|
||
}
|
||
|
||
func (r *resourceAppImpl) GetById(id uint64, cols ...string) *entity.Resource {
|
||
return r.resourceRepo.GetById(id, cols...)
|
||
}
|
||
|
||
func (r *resourceAppImpl) GetByIdIn(ids []uint64, toEntity any, orderBy ...string) {
|
||
r.resourceRepo.GetByIdIn(ids, toEntity, orderBy...)
|
||
}
|
||
|
||
func (r *resourceAppImpl) Save(resource *entity.Resource) {
|
||
if resource.Id != 0 {
|
||
if resource.Code != "" {
|
||
oldRes := r.GetById(resource.Id, "Code")
|
||
// 如果修改了code，则校验新code是否存在
|
||
if oldRes.Code != resource.Code {
|
||
r.checkCode(resource.Code)
|
||
}
|
||
}
|
||
model.UpdateById(resource)
|
||
} else {
|
||
if pid := resource.Pid; pid != 0 {
|
||
biz.IsTrue(r.GetById(uint64(pid)) != nil, "该父资源不存在")
|
||
}
|
||
// 默认启用状态
|
||
if resource.Status == 0 {
|
||
resource.Status = entity.ResourceStatusEnable
|
||
}
|
||
r.checkCode(resource.Code)
|
||
model.Insert(resource)
|
||
}
|
||
}
|
||
|
||
func (r *resourceAppImpl) checkCode(code string) {
|
||
biz.IsTrue(!strings.Contains(code, ","), "code不能包含','")
|
||
biz.IsEquals(model.CountBy(&entity.Resource{Code: code}), int64(0), "该code已存在")
|
||
}
|
||
|
||
func (r *resourceAppImpl) Delete(id uint64) {
|
||
// 查找pid == id的资源
|
||
condition := &entity.Resource{Pid: int(id)}
|
||
var resources resourceList
|
||
r.resourceRepo.GetResourceList(condition, &resources)
|
||
|
||
biz.IsTrue(len(resources) == 0, "请先删除该资源的所有子资源")
|
||
model.DeleteById(condition, id)
|
||
// 删除角色关联的资源信息
|
||
model.DeleteByCondition(&entity.RoleResource{ResourceId: id})
|
||
}
|
||
|
||
func (r *resourceAppImpl) GetAccountResources(accountId uint64, toEntity any) {
|
||
r.resourceRepo.GetAccountResources(accountId, toEntity)
|
||
}
|
||
|
||
type resourceList []entity.Resource
|