88 lines
2.5 KiB
Go
88 lines
2.5 KiB
Go
package sysuserrole
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"ems.agt/lib/core/datasource"
|
|
"ems.agt/lib/core/utils/parse"
|
|
"ems.agt/lib/log"
|
|
)
|
|
|
|
// 实例化数据层 RepoSysUserRole 结构体
|
|
var NewRepoSysUserRole = &RepoSysUserRole{}
|
|
|
|
// RepoSysUserRole 用户与角色关联表 数据层处理
|
|
type RepoSysUserRole struct{}
|
|
|
|
// CountUserRoleByRoleId 通过角色ID查询角色使用数量
|
|
func (r *RepoSysUserRole) CountUserRoleByRoleId(roleId string) int64 {
|
|
querySql := "select count(1) as total from sys_user_role where role_id = ?"
|
|
results, err := datasource.RawDB("", querySql, []any{roleId})
|
|
if err != nil {
|
|
log.Errorf("query err => %v", err)
|
|
return 0
|
|
}
|
|
if len(results) > 0 {
|
|
return parse.Number(results[0]["total"])
|
|
}
|
|
return 0
|
|
}
|
|
|
|
// BatchUserRole 批量新增用户角色信息
|
|
func (r *RepoSysUserRole) BatchUserRole(sysUserRoles []SysUserRole) int64 {
|
|
keyValues := make([]string, 0)
|
|
for _, item := range sysUserRoles {
|
|
keyValues = append(keyValues, fmt.Sprintf("(%s,%s)", item.UserID, item.RoleID))
|
|
}
|
|
sql := "insert into sys_user_role(user_id, role_id) values " + strings.Join(keyValues, ",")
|
|
results, err := datasource.ExecDB("", sql, nil)
|
|
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
|
|
}
|
|
|
|
// DeleteUserRole 批量删除用户和角色关联
|
|
func (r *RepoSysUserRole) DeleteUserRole(userIds []string) int64 {
|
|
placeholder := datasource.KeyPlaceholderByQuery(len(userIds))
|
|
sql := "delete from sys_user_role where user_id in (" + placeholder + ")"
|
|
parameters := datasource.ConvertIdsSlice(userIds)
|
|
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
|
|
}
|
|
|
|
// DeleteUserRoleByRoleId 批量取消授权用户角色
|
|
func (r *RepoSysUserRole) DeleteUserRoleByRoleId(roleId string, userIds []string) int64 {
|
|
placeholder := datasource.KeyPlaceholderByQuery(len(userIds))
|
|
sql := "delete from sys_user_role where role_id= ? and user_id in (" + placeholder + ")"
|
|
parameters := datasource.ConvertIdsSlice(userIds)
|
|
parameters = append([]any{roleId}, parameters...)
|
|
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
|
|
}
|