修改超级管理员获取所有菜单

api/service/dept.go

@@ -48,3 +48,83 @@ func (r *DeptService) GetDeptList() ([]*deptResponse.GetDeptList, error) {
 
 	return deptTree, nil
 }
+
+// AddDept 新增部门
+// func (r *DeptService) AddDept(addDeptRequest requests.AddDept) (bool, error) {
+// 	adminDeptDao := dao.AdminDeptDao{}
+//
+// 	parentId, err := strconv.ParseInt(addDeptRequest.ParentId, 10, 64)
+// 	if err != nil {
+// 		return false, errors.New("新增失败")
+// 	}
+//
+// 	// 判断父级id
+// 	if parentId != 0 {
+// 		adminDept, _ := adminDeptDao.GetAdminDeptById(parentId)
+// 		if adminDept == nil {
+// 			return false, errors.New("上级部门错误")
+// 		}
+// 	}
+//
+// 	// 开始事务
+// 	tx := global.Db.Begin()
+// 	defer func() {
+// 		if r := recover(); r != nil {
+// 			tx.Rollback()
+// 		}
+// 	}()
+//
+// 	// 新增菜单
+// 	AdminMenuModel := &model.AdminMenu{
+// 		MenuName:   AddMenuRequest.MenuName,
+// 		MenuTitle:  AddMenuRequest.MenuTitle,
+// 		ParentId:   ParentId,
+// 		MenuStatus: AddMenuRequest.MenuStatus,
+// 		MenuType:   AddMenuRequest.MenuType,
+// 		Permission: AddMenuRequest.Permission,
+// 		OrderNum:   AddMenuRequest.OrderNum,
+// 		Icon:       AddMenuRequest.Icon,
+// 		Path:       AddMenuRequest.Path,
+// 		Component:  AddMenuRequest.Component,
+// 	}
+//
+// 	adminMenu, _ := AdminMenuDao.AddAdminMenu(tx, AdminMenuModel)
+// 	if adminMenu == nil {
+// 		tx.Rollback()
+// 		return false, errors.New("新增失败")
+// 	}
+//
+// 	// 新增菜单api
+// 	if len(AddMenuRequest.ApiIds) > 0 {
+// 		for _, v := range AddMenuRequest.ApiIds {
+// 			// 将字符串转换为int64类型
+// 			v, err := strconv.ParseInt(v, 10, 64)
+// 			if err != nil {
+// 				tx.Rollback()
+// 				return false, errors.New("新增失败")
+// 			}
+//
+// 			// 检测api
+// 			adminApi, _ := AdminApiDao.GetAdminApiById(v)
+// 			if adminApi == nil {
+// 				tx.Rollback()
+// 				return false, errors.New("接口不存在")
+// 			}
+//
+// 			AdminMenuApiModel := &model.AdminMenuApi{
+// 				MenuId: adminMenu.MenuId,
+// 				ApiId:  v,
+// 			}
+//
+// 			adminMenuApi, _ := AdminMenuApiDao.AddAdminMenuApi(tx, AdminMenuApiModel)
+// 			if adminMenuApi == nil {
+// 				tx.Rollback()
+// 				return false, errors.New("新增失败")
+// 			}
+// 		}
+// 	}
+//
+// 	tx.Commit()
+//
+// 	return true, nil
+// }

api/service/role.go

@@ -14,35 +14,39 @@ import (
 type RoleService struct{}
 
 // GetRoleMenuList 获取角色菜单
-func (r *RoleService) GetRoleMenuList(roleId int64) ([]*roleResponse.GetRoleMenuList, error) {
+func (r *RoleService) GetRoleMenuList(roleId int64, isAdmin bool) ([]*roleResponse.GetRoleMenuList, error) {
 	// 获取角色菜单
 	AdminRoleMenuDao := dao.AdminRoleMenuDao{}
-	adminRoleMenu, _ := AdminRoleMenuDao.GetAdminRoleMenuListByRoleId(roleId)
-	if adminRoleMenu == nil {
-		return []*roleResponse.GetRoleMenuList{}, nil
+	AdminMenuDao := dao.AdminMenuDao{}
+	var menuIDs []int64
+	var err error
+
+	if isAdmin {
+		// 超级管理员
+		menuIDs, err = AdminMenuDao.GetAdminMenuId()
+		if err != nil {
+			return []*roleResponse.GetRoleMenuList{}, nil
+		}
+	} else {
+		menuIDs, err = AdminRoleMenuDao.GetAdminRoleMenuIdByRoleId(roleId)
+		if err != nil {
+			return []*roleResponse.GetRoleMenuList{}, nil
+		}
 	}
 
 	// 获取全部正常菜单列表
-	AdminMenuDao := dao.AdminMenuDao{}
 	adminMenu, _ := AdminMenuDao.GetAdminMenuListNormalSortOrderNum()
 	if adminMenu == nil {
 		return []*roleResponse.GetRoleMenuList{}, nil
 	}
 
-	var menuIDs []int64
 	var menuTree []*roleResponse.GetRoleMenuList
 
-	if len(adminRoleMenu) > 0 {
-		for _, roleMenu := range adminRoleMenu {
-			menuIDs = append(menuIDs, roleMenu.MenuID)
-		}
-
-		if len(menuIDs) > 0 {
-			// 构建菜单树
-			menuTree = buildMenuTree(menuIDs, adminMenu)
-			if menuTree == nil {
-				return nil, nil
-			}
+	if len(menuIDs) > 0 {
+		// 构建菜单树
+		menuTree = buildMenuTree(menuIDs, adminMenu)
+		if menuTree == nil {
+			return nil, nil
 		}
 	}
 
@@ -388,3 +392,23 @@ func (r *RoleService) DeleteRole(c *gin.Context, DeleteRoleRequest requests.Dele
 
 	return true, nil
 }
+
+// GetRoleIden 获取角色身份
+func (r *RoleService) GetRoleIden(c *gin.Context) (bool, error) {
+	roleId := c.GetInt64("RoleId")
+	if roleId == 0 {
+		return false, errors.New("角色错误")
+	}
+
+	// 获取当前登陆接口角色数据
+	adminRoleDao := dao.AdminRoleDao{}
+	adminRole, err := adminRoleDao.GetAdminRoleFirstById(roleId)
+	if err != nil || adminRole == nil {
+		return false, errors.New("非法操作")
+	}
+
+	if adminRole.IsAdmin == 0 {
+		return false, nil
+	}
+	return true, nil
+}

api/service/user.go

@@ -73,7 +73,7 @@ func (r *UserService) AddUser(c *gin.Context, AddUserRequest requests.AddUser) (
 		return false, errors.New("部门错误")
 	}
 
-	adminDept, err := AdminDeptDao.GetAdminDeptFirstById(deptID)
+	adminDept, err := AdminDeptDao.GetAdminDeptById(deptID)
 	if err != nil || adminDept == nil {
 		return false, errors.New("部门错误")
 	}
@@ -284,7 +284,7 @@ func (r *UserService) PutUser(c *gin.Context, requestUserId int64, putUserReques
 
 	if deptID != adminUser.DeptID {
 		adminDeptDao := dao.AdminDeptDao{}
-		res, err := adminDeptDao.GetAdminDeptFirstById(deptID)
+		res, err := adminDeptDao.GetAdminDeptById(deptID)
 		if err != nil || res == nil {
 			return false, errors.New("部门错误")
 		}