288 lines
7.1 KiB
Go
288 lines
7.1 KiB
Go
package cms
|
||
|
||
import (
|
||
"fmt"
|
||
"time"
|
||
|
||
"git.rosy.net.cn/jx-callback/business/jxutils/jxcontext"
|
||
"git.rosy.net.cn/jx-callback/business/model"
|
||
"git.rosy.net.cn/jx-callback/business/model/dao"
|
||
)
|
||
|
||
func GetMenu(ctx *jxcontext.Context, userID string) (menus []*model.Menu, err error) {
|
||
if userID == "" {
|
||
return dao.GetMenu(dao.GetDB(), "", 0, userID)
|
||
} else {
|
||
return dao.GetMenuWithUser(dao.GetDB(), "", 0, userID)
|
||
}
|
||
}
|
||
|
||
func AddMenu(ctx *jxcontext.Context, menu *model.Menu) (err error) {
|
||
var (
|
||
db = dao.GetDB()
|
||
)
|
||
if menu == nil {
|
||
return fmt.Errorf("添加失败！menu nil")
|
||
}
|
||
if menu.Name == "" || menu.Level == 0 {
|
||
return fmt.Errorf("添加失败！menu 名称和等级必须有值！")
|
||
}
|
||
menus, err := dao.GetMenu(db, menu.Name, menu.Level, "")
|
||
if len(menus) > 0 {
|
||
return fmt.Errorf("添加失败！已存在相同名称的 menu name : %v", menu.Name)
|
||
}
|
||
dao.Begin(db)
|
||
defer func() {
|
||
if r := recover(); r != nil {
|
||
dao.Rollback(db)
|
||
panic(r)
|
||
}
|
||
}()
|
||
dao.WrapAddIDCULDEntity(menu, ctx.GetUserName())
|
||
err = dao.CreateEntity(db, menu)
|
||
dao.Commit(db)
|
||
return err
|
||
}
|
||
|
||
func UpdateMenu(ctx *jxcontext.Context, menuID int, payload map[string]interface{}, isDelete bool) (num int64, err error) {
|
||
var (
|
||
db = dao.GetDB()
|
||
)
|
||
menu := &model.Menu{}
|
||
menu.ID = menuID
|
||
err = dao.GetEntity(db, menu)
|
||
if err != nil {
|
||
return 0, err
|
||
}
|
||
dao.Begin(db)
|
||
defer func() {
|
||
if r := recover(); r != nil {
|
||
dao.Rollback(db)
|
||
panic(r)
|
||
}
|
||
}()
|
||
if !isDelete {
|
||
valid := dao.StrictMakeMapByStructObject(payload, menu, ctx.GetUserName())
|
||
if len(valid) > 0 {
|
||
if num, err = dao.UpdateEntityLogically(db, menu, valid, ctx.GetUserName(), nil); err != nil {
|
||
dao.Rollback(db)
|
||
return 0, err
|
||
}
|
||
}
|
||
} else {
|
||
menu.DeletedAt = time.Now()
|
||
num, err = dao.UpdateEntity(db, menu, "DeletedAt")
|
||
}
|
||
dao.Commit(db)
|
||
return num, err
|
||
}
|
||
|
||
func GetRole(ctx *jxcontext.Context) (roles []*model.Role, err error) {
|
||
return dao.GetRole(dao.GetDB(), "")
|
||
}
|
||
|
||
func AddRole(ctx *jxcontext.Context, name string) (err error) {
|
||
var (
|
||
db = dao.GetDB()
|
||
)
|
||
roles, err := dao.GetRole(db, name)
|
||
if len(roles) > 0 {
|
||
return fmt.Errorf("添加失败！已存在相同名称的 role name : %v", name)
|
||
}
|
||
role := &model.Role{
|
||
Name: name,
|
||
}
|
||
dao.Begin(db)
|
||
defer func() {
|
||
if r := recover(); r != nil {
|
||
dao.Rollback(db)
|
||
panic(r)
|
||
}
|
||
}()
|
||
dao.WrapAddIDCULDEntity(role, ctx.GetUserName())
|
||
err = dao.CreateEntity(db, role)
|
||
dao.Commit(db)
|
||
return err
|
||
}
|
||
|
||
func UpdateRole(ctx *jxcontext.Context, roleID int, name string, isDelete bool) (num int64, err error) {
|
||
var (
|
||
db = dao.GetDB()
|
||
)
|
||
if roleID == 1 {
|
||
return 0, fmt.Errorf("管理员不允许修改！")
|
||
}
|
||
role := &model.Role{}
|
||
role.ID = roleID
|
||
err = dao.GetEntity(db, role)
|
||
if err != nil {
|
||
return 0, err
|
||
}
|
||
dao.Begin(db)
|
||
defer func() {
|
||
if r := recover(); r != nil {
|
||
dao.Rollback(db)
|
||
panic(r)
|
||
}
|
||
}()
|
||
if !isDelete {
|
||
role.Name = name
|
||
role.UpdatedAt = time.Now()
|
||
role.LastOperator = ctx.GetUserName()
|
||
num, err = dao.UpdateEntity(db, role, "Name", "UpdatedAt", "LastOperator")
|
||
} else {
|
||
role.DeletedAt = time.Now()
|
||
num, err = dao.UpdateEntity(db, role, "DeletedAt")
|
||
}
|
||
dao.Commit(db)
|
||
return num, err
|
||
}
|
||
|
||
func GetUserRole(ctx *jxcontext.Context, userID string) (userRoles []*model.UserRole, err error) {
|
||
return dao.GetUserRole(dao.GetDB(), []string{userID}, nil)
|
||
}
|
||
|
||
func UpdateUserRole(ctx *jxcontext.Context, userIDs []string, roleIDs []int) (err error) {
|
||
var (
|
||
db = dao.GetDB()
|
||
roleIDMap = make(map[int]int)
|
||
nowRoleIDMap = make(map[int]int)
|
||
userRoleMap = make(map[string][]int)
|
||
addUserRoleMap = make(map[string][]int)
|
||
deleteUserRoleMap = make(map[string][]int)
|
||
)
|
||
for _, v := range roleIDs {
|
||
roleIDMap[v] = 1
|
||
}
|
||
userRoles, err := dao.GetUserRole(db, userIDs, nil)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if len(userRoles) > 0 {
|
||
for _, v := range userRoles {
|
||
userRoleMap[v.UserID] = append(userRoleMap[v.UserID], v.RoleID)
|
||
}
|
||
}
|
||
for _, userID := range userIDs {
|
||
nowRoleIDs := userRoleMap[userID]
|
||
for _, nowRoleID := range nowRoleIDs {
|
||
if roleIDMap[nowRoleID] == 0 {
|
||
deleteUserRoleMap[userID] = append(deleteUserRoleMap[userID], nowRoleID)
|
||
}
|
||
nowRoleIDMap[nowRoleID] = 1
|
||
}
|
||
for _, roleID := range roleIDMap {
|
||
if nowRoleIDMap[roleID] == 0 {
|
||
addUserRoleMap[userID] = append(addUserRoleMap[userID], roleID)
|
||
}
|
||
}
|
||
}
|
||
dao.Begin(db)
|
||
defer func() {
|
||
if r := recover(); r != nil {
|
||
dao.Rollback(db)
|
||
panic(r)
|
||
}
|
||
}()
|
||
if len(addUserRoleMap) > 0 {
|
||
for userID, roleIDs := range addUserRoleMap {
|
||
for _, roleID := range roleIDs {
|
||
userRole := &model.UserRole{
|
||
UserID: userID,
|
||
RoleID: roleID,
|
||
}
|
||
dao.WrapAddIDCULDEntity(userRole, ctx.GetUserName())
|
||
err = dao.CreateEntity(db, userRole)
|
||
}
|
||
}
|
||
}
|
||
if len(deleteUserRoleMap) > 0 {
|
||
for userID, roleIDs := range deleteUserRoleMap {
|
||
for _, roleID := range roleIDs {
|
||
userRoles, _ := dao.GetUserRole(db, []string{userID}, []int{roleID})
|
||
if len(userRoles) > 0 {
|
||
userRoles[0].DeletedAt = time.Now()
|
||
userRoles[0].LastOperator = ctx.GetUserName()
|
||
_, err = dao.UpdateEntity(db, userRoles[0], "DeletedAt", "LastOperator")
|
||
}
|
||
}
|
||
}
|
||
}
|
||
dao.Commit(db)
|
||
return err
|
||
}
|
||
|
||
func GetRoleMenu(ctx *jxcontext.Context, roleID int) (roleMenus []*model.RoleMenu, err error) {
|
||
return dao.GetRoleMenu(dao.GetDB(), []int{roleID}, nil)
|
||
}
|
||
|
||
func UpdateRoleMenu(ctx *jxcontext.Context, roleIDs, menuIDs []int) (err error) {
|
||
var (
|
||
db = dao.GetDB()
|
||
menuIDMap = make(map[int]int)
|
||
nowMenuIDMap = make(map[int]int)
|
||
roleMenuMap = make(map[int][]int)
|
||
addRoleMenuMap = make(map[int][]int)
|
||
deleteRoleMenuMap = make(map[int][]int)
|
||
)
|
||
for _, v := range menuIDs {
|
||
menuIDMap[v] = 1
|
||
}
|
||
roleMenus, err := dao.GetRoleMenu(db, roleIDs, nil)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if len(roleMenus) > 0 {
|
||
for _, v := range roleMenus {
|
||
roleMenuMap[v.RoleID] = append(roleMenuMap[v.RoleID], v.MenuID)
|
||
}
|
||
}
|
||
for _, roleID := range roleIDs {
|
||
nowMenuIDs := roleMenuMap[roleID]
|
||
for _, nowMenuID := range nowMenuIDs {
|
||
if menuIDMap[nowMenuID] == 0 {
|
||
deleteRoleMenuMap[roleID] = append(deleteRoleMenuMap[roleID], nowMenuID)
|
||
}
|
||
nowMenuIDMap[nowMenuID] = 1
|
||
}
|
||
for _, menuID := range menuIDMap {
|
||
if nowMenuIDMap[menuID] == 0 {
|
||
addRoleMenuMap[menuID] = append(addRoleMenuMap[menuID], menuID)
|
||
}
|
||
}
|
||
}
|
||
dao.Begin(db)
|
||
defer func() {
|
||
if r := recover(); r != nil {
|
||
dao.Rollback(db)
|
||
panic(r)
|
||
}
|
||
}()
|
||
if len(addRoleMenuMap) > 0 {
|
||
for roleID, menuIDs := range addRoleMenuMap {
|
||
for _, menuID := range menuIDs {
|
||
roleMenu := &model.RoleMenu{
|
||
RoleID: roleID,
|
||
MenuID: menuID,
|
||
}
|
||
dao.WrapAddIDCULDEntity(roleMenu, ctx.GetUserName())
|
||
err = dao.CreateEntity(db, roleMenu)
|
||
}
|
||
}
|
||
}
|
||
if len(deleteRoleMenuMap) > 0 {
|
||
for roleID, menuIDs := range deleteRoleMenuMap {
|
||
for _, menuID := range menuIDs {
|
||
roleMenus, _ := dao.GetRoleMenu(db, []int{roleID}, []int{menuID})
|
||
if len(roleMenus) > 0 {
|
||
roleMenus[0].DeletedAt = time.Now()
|
||
roleMenus[0].LastOperator = ctx.GetUserName()
|
||
_, err = dao.UpdateEntity(db, roleMenus[0], "DeletedAt", "LastOperator")
|
||
}
|
||
}
|
||
}
|
||
}
|
||
dao.Commit(db)
|
||
return err
|
||
}
|