浏览代码

#MOD 角色数据权限

谭斌 2 年之前
父节点
当前提交
3e9c7b45b3
共有 1 个文件被更改,包括 6 次插入2 次删除
  1. 6 2
      authorize-api-service/src/main/java/com/xy/service/impl/SysRoleMenuServiceImpl.java

+ 6 - 2
authorize-api-service/src/main/java/com/xy/service/impl/SysRoleMenuServiceImpl.java

@@ -76,6 +76,7 @@ public class SysRoleMenuServiceImpl extends ServiceImpl<SysRoleMenuMapper, SysRo
             return Boolean.TRUE;
         }
         List<SysRoleMenu> updateList = new ArrayList<>(16);
+        List<SysRoleMenu> addList = new ArrayList<>(16);
         sysRoleMenuList.forEach(sysRoleMenu -> {
             SysRoleMenu roleMenu = this.getOne(Wrappers.<SysRoleMenu>lambdaQuery()
                     .eq(SysRoleMenu::getSysId, sysRoleMenu.getSysId())
@@ -86,11 +87,14 @@ public class SysRoleMenuServiceImpl extends ServiceImpl<SysRoleMenuMapper, SysRo
                 roleMenu.setSelectAdmin(sysRoleMenu.getSelectAdmin());
                 updateList.add(roleMenu);
             } else {
-                updateList.add(sysRoleMenu);
+                addList.add(sysRoleMenu);
             }
         });
         if (CollUtil.isNotEmpty(updateList)) {
-            saveOrUpdateBatch(updateList);
+            updateBatchById(updateList);
+        }
+        if (CollUtil.isNotEmpty(addList)) {
+            saveBatch(addList);
         }
         return Boolean.TRUE;
     }