87 lines
2.4 KiB
Go
87 lines
2.4 KiB
Go
package sysrolemenu
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"be.ems/lib/core/datasource"
|
|
"be.ems/lib/core/utils/parse"
|
|
"be.ems/lib/log"
|
|
)
|
|
|
|
// 实例化数据层 RepoSysRoleMenu 结构体
|
|
var NewRepoSysRoleMenu = &RepoSysRoleMenu{}
|
|
|
|
// RepoSysRoleMenu 角色与菜单关联表 数据层处理
|
|
type RepoSysRoleMenu struct{}
|
|
|
|
// CheckMenuExistRole 查询菜单分配给角色使用数量
|
|
func (r *RepoSysRoleMenu) CheckMenuExistRole(menuId string) int64 {
|
|
querySql := "select count(1) as 'total' from sys_role_menu where menu_id = ?"
|
|
results, err := datasource.RawDB("", querySql, []any{menuId})
|
|
if err != nil {
|
|
log.Errorf("query err => %v", err)
|
|
return 0
|
|
}
|
|
if len(results) > 0 {
|
|
return parse.Number(results[0]["total"])
|
|
}
|
|
return 0
|
|
}
|
|
|
|
// DeleteRoleMenu 批量删除角色和菜单关联
|
|
func (r *RepoSysRoleMenu) DeleteRoleMenu(roleIds []string) int64 {
|
|
placeholder := datasource.KeyPlaceholderByQuery(len(roleIds))
|
|
sql := "delete from sys_role_menu where role_id in (" + placeholder + ")"
|
|
parameters := datasource.ConvertIdsSlice(roleIds)
|
|
results, err := datasource.ExecDB("", sql, parameters)
|
|
if err != nil {
|
|
log.Errorf("delete err => %v", err)
|
|
return 0
|
|
}
|
|
affected, err := results.RowsAffected()
|
|
if err != nil {
|
|
log.Errorf("delete err => %v", err)
|
|
return 0
|
|
}
|
|
return affected
|
|
}
|
|
|
|
// DeleteMenuRole 批量删除菜单和角色关联
|
|
func (r *RepoSysRoleMenu) DeleteMenuRole(menuIds []string) int64 {
|
|
placeholder := datasource.KeyPlaceholderByQuery(len(menuIds))
|
|
sql := "delete from sys_role_menu where menu_id in (" + placeholder + ")"
|
|
parameters := datasource.ConvertIdsSlice(menuIds)
|
|
results, err := datasource.ExecDB("", sql, parameters)
|
|
if err != nil {
|
|
log.Errorf("delete err => %v", err)
|
|
return 0
|
|
}
|
|
affected, err := results.RowsAffected()
|
|
if err != nil {
|
|
log.Errorf("delete err => %v", err)
|
|
return 0
|
|
}
|
|
return affected
|
|
}
|
|
|
|
// BatchRoleMenu 批量新增角色菜单信息
|
|
func (r *RepoSysRoleMenu) BatchRoleMenu(sysRoleMenus []SysRoleMenu) int64 {
|
|
keyValues := make([]string, 0)
|
|
for _, item := range sysRoleMenus {
|
|
keyValues = append(keyValues, fmt.Sprintf("(%s,%s)", item.RoleID, item.MenuID))
|
|
}
|
|
sql := "insert into sys_role_menu(role_id, menu_id) values " + strings.Join(keyValues, ",")
|
|
results, err := datasource.ExecDB("", sql, nil)
|
|
if err != nil {
|
|
log.Errorf("insert err => %v", err)
|
|
return 0
|
|
}
|
|
insertId, err := results.LastInsertId()
|
|
if err != nil {
|
|
log.Errorf("insert row : %v", err.Error())
|
|
return 0
|
|
}
|
|
return insertId
|
|
}
|