新增删除角色-批量

2023-06-27 12:00:15 +08:00
parent 4101d9560e
commit f414704688
10 changed files with 309 additions and 7 deletions
@@ -122,3 +122,73 @@ func (r *MenuService) AddMenu(AddMenuRequest requests.AddMenu) (bool, error) {

 	return true, nil
 }
+
+// PutMenu 修改菜单
+func (r *MenuService) PutMenu(PutMenuRequest requests.PutMenu) (bool, error) {
+	AdminMenuDao := dao.AdminMenuDao{}
+	AdminMenuApiDao := dao.AdminMenuApiDao{}
+	AdminApiDao := dao.AdminApiDao{}
+
+	// 判断父级id
+	if PutMenuRequest.ParentId != 0 {
+		adminMenu, _ := AdminMenuDao.GetAdminMenuById(PutMenuRequest.ParentId)
+		if adminMenu == nil {
+			return false, errors.New("上级菜单错误")
+		}
+	}
+
+	// 开始事务
+	tx := global.Db.Begin()
+	defer func() {
+		if r := recover(); r != nil {
+			tx.Rollback()
+		}
+	}()
+
+	// 新增菜单
+	AdminMenuModel := &model.AdminMenu{
+		MenuName:   PutMenuRequest.MenuName,
+		MenuTitle:  PutMenuRequest.MenuTitle,
+		ParentId:   PutMenuRequest.ParentId,
+		MenuStatus: PutMenuRequest.MenuStatus,
+		MenuType:   PutMenuRequest.MenuType,
+		Permission: PutMenuRequest.Permission,
+		OrderNum:   PutMenuRequest.OrderNum,
+		Icon:       PutMenuRequest.Icon,
+		Path:       PutMenuRequest.Path,
+		Component:  PutMenuRequest.Component,
+	}
+
+	adminMenu, _ := AdminMenuDao.AddAdminMenu(tx, AdminMenuModel)
+	if adminMenu == nil {
+		tx.Rollback()
+		return false, errors.New("新增失败")
+	}
+
+	// 新增菜单api
+	if len(PutMenuRequest.ApiIds) > 0 {
+		for _, v := range PutMenuRequest.ApiIds {
+			// 检测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
+}