mirror of
https://gitee.com/dromara/mayfly-go
synced 2025-11-03 07:50:25 +08:00
85 lines
2.7 KiB
Go
85 lines
2.7 KiB
Go
package persistence
|
||
|
||
import (
|
||
"mayfly-go/internal/sys/domain/entity"
|
||
"mayfly-go/internal/sys/domain/repository"
|
||
"mayfly-go/pkg/biz"
|
||
"mayfly-go/pkg/gormx"
|
||
"mayfly-go/pkg/model"
|
||
)
|
||
|
||
type roleRepoImpl struct{}
|
||
|
||
func newRoleRepo() repository.Role {
|
||
return new(roleRepoImpl)
|
||
}
|
||
|
||
func (m *roleRepoImpl) GetPageList(condition *entity.Role, pageParam *model.PageParam, toEntity any, orderBy ...string) *model.PageResult[any] {
|
||
qd := gormx.NewQuery(condition).WithCondModel(condition).WithOrderBy(orderBy...)
|
||
return gormx.PageQuery(qd, pageParam, toEntity)
|
||
}
|
||
|
||
func (m *roleRepoImpl) Delete(id uint64) {
|
||
biz.ErrIsNil(gormx.DeleteById(new(entity.Role), id), "删除角色失败")
|
||
}
|
||
|
||
// 获取角色拥有的资源id数组，从role_resource表获取
|
||
func (m *roleRepoImpl) GetRoleResourceIds(roleId uint64) []uint64 {
|
||
var rrs []entity.RoleResource
|
||
|
||
condtion := &entity.RoleResource{RoleId: roleId}
|
||
gormx.ListBy(condtion, &rrs, "ResourceId")
|
||
|
||
var rids []uint64
|
||
for _, v := range rrs {
|
||
rids = append(rids, v.ResourceId)
|
||
}
|
||
return rids
|
||
}
|
||
|
||
func (m *roleRepoImpl) GetRoleResources(roleId uint64, toEntity any) {
|
||
sql := "select rr.creator AS creator, rr.create_time AS CreateTime, rr.resource_id AS id, r.pid AS pid, " +
|
||
"r.name AS name, r.type AS type, r.status AS status " +
|
||
"FROM t_sys_role_resource rr JOIN t_sys_resource r ON rr.resource_id = r.id " +
|
||
"WHERE rr.role_id = ? AND rr.is_deleted = 0 AND r.is_deleted = 0 " +
|
||
"ORDER BY r.pid ASC, r.weight ASC"
|
||
gormx.GetListBySql2Model(sql, toEntity, roleId)
|
||
}
|
||
|
||
func (m *roleRepoImpl) SaveRoleResource(rr []*entity.RoleResource) {
|
||
gormx.BatchInsert(rr)
|
||
}
|
||
|
||
func (m *roleRepoImpl) DeleteRoleResource(roleId uint64, resourceId uint64) {
|
||
gormx.DeleteByCondition(&entity.RoleResource{RoleId: roleId, ResourceId: resourceId})
|
||
}
|
||
|
||
func (m *roleRepoImpl) GetAccountRoleIds(accountId uint64) []uint64 {
|
||
var rrs []entity.AccountRole
|
||
|
||
condtion := &entity.AccountRole{AccountId: accountId}
|
||
gormx.ListBy(condtion, &rrs, "RoleId")
|
||
|
||
var rids []uint64
|
||
for _, v := range rrs {
|
||
rids = append(rids, v.RoleId)
|
||
}
|
||
return rids
|
||
}
|
||
|
||
func (m *roleRepoImpl) SaveAccountRole(ar *entity.AccountRole) {
|
||
gormx.Insert(ar)
|
||
}
|
||
|
||
func (m *roleRepoImpl) DeleteAccountRole(accountId, roleId uint64) {
|
||
gormx.DeleteByCondition(&entity.AccountRole{RoleId: roleId, AccountId: accountId})
|
||
}
|
||
|
||
// 获取账号角色信息列表
|
||
func (m *roleRepoImpl) GetAccountRoles(accountId uint64, toEntity any) {
|
||
sql := "SELECT r.status, r.name, ar.create_time AS CreateTime, ar.creator AS creator " +
|
||
"FROM t_sys_role r JOIN t_sys_account_role ar ON r.id = ar.role_id AND ar.account_id = ? AND r.is_deleted = 0 AND ar.is_deleted = 0 " +
|
||
"ORDER BY ar.create_time DESC"
|
||
gormx.GetListBySql2Model(sql, toEntity, accountId)
|
||
}
|