Blizzard
90 lines
2.5 KiB
Go
90 lines
2.5 KiB
Go
package system
|
|
|
|
import (
|
|
"errors"
|
|
"sundynix-go/global"
|
|
common "sundynix-go/model/commom/request"
|
|
"sundynix-go/model/system"
|
|
systemReq "sundynix-go/model/system/request"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type RoleService struct {
|
|
}
|
|
|
|
var RoleServiceApp = new(RoleService)
|
|
|
|
func (s *RoleService) SaveRole(role system.Role) error {
|
|
if !errors.Is(global.DB.Where("code = ?", role.Code).First(&system.Role{}).Error, gorm.ErrRecordNotFound) {
|
|
return errors.New("存在重复角色")
|
|
}
|
|
return global.DB.Create(&role).Error
|
|
}
|
|
|
|
func (s *RoleService) UpdateRole(role system.Role) error {
|
|
return global.DB.Model(&role).Where("id = ?", role.Id).Updates(&role).Error
|
|
}
|
|
|
|
func (s *RoleService) GetRoleList(info systemReq.GetRoleList) (list interface{}, total int64, err error) {
|
|
limit := info.PageSize
|
|
offset := info.PageSize * (info.Current - 1)
|
|
db := global.DB.Model(&system.Role{})
|
|
var roleList []system.Role
|
|
if info.Code != "" {
|
|
db = db.Where("code = ?", info.Code)
|
|
}
|
|
if info.Name != "" {
|
|
db = db.Where("name LIKE ?", "%"+info.Name+"%")
|
|
}
|
|
err = db.Count(&total).Error
|
|
if err != nil {
|
|
return
|
|
}
|
|
err = db.Limit(limit).Offset(offset).Find(&roleList).Error
|
|
return roleList, total, err
|
|
}
|
|
|
|
func (s *RoleService) DeleteRoleByIds(ids common.IdsReq) error {
|
|
return global.DB.Where("id in ?", ids.Ids).Delete(&system.Role{}).Error
|
|
}
|
|
|
|
func (s *RoleService) GetRoleById(id string) (role *system.Role, err error) {
|
|
var r system.Role
|
|
err = global.DB.Where("id = ?", id).First(&r).Error
|
|
return &r, err
|
|
}
|
|
|
|
func (s *RoleService) GrantRole(userId string, roleIds []string) error {
|
|
//1. 检查是否存在userid的授权记录 存在就删除 不存在就插入
|
|
//2. 插入新的数据
|
|
return global.DB.Transaction(func(tx *gorm.DB) error {
|
|
if err := tx.Where("user_id = ?", userId).Delete(&system.UserRole{}).Error; err != nil {
|
|
return err
|
|
}
|
|
for _, roleId := range roleIds {
|
|
if err := tx.Create(&system.UserRole{UserId: userId, RoleId: roleId}).Error; err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
})
|
|
|
|
}
|
|
|
|
func (s *RoleService) GrantMenu(roleId string, menuIds []string) error {
|
|
//1. 检查是否存在userid的授权记录 存在就删除 不存在就插入
|
|
//2. 插入新的数据
|
|
return global.DB.Transaction(func(tx *gorm.DB) error {
|
|
if err := tx.Where("role_id = ?", roleId).Delete(&system.RoleMenu{}).Error; err != nil {
|
|
return err
|
|
}
|
|
for _, menuId := range menuIds {
|
|
if err := tx.Create(&system.RoleMenu{RoleId: roleId, MenuId: menuId}).Error; err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
})
|
|
}
|