模块

fhKeeper/formulahousekeeper/management-platform/src/main/java/com/management/platform/controller/WeiXinCorpController.java

@@ -3291,71 +3291,62 @@ public class WeiXinCorpController {
 
     //为超级管理员角色赋予模块权限
     private void rootRole (Integer companyId){
-        System.err.println("##############################为超级管理员开通模块及权限");
         Company company = companyMapper.selectById(companyId);
-        SysRole sysRole = sysRoleMapper.selectOne(new QueryWrapper<SysRole>().eq("company_id", companyId).eq("rolename", "超级管理员"));
-        QueryWrapper<SysModule> sysModuleQueryWrapper = new QueryWrapper<>();
-//        //项目协作平台
-//        if (company.getPackageProject().equals(1)){
-//            sysModuleQueryWrapper.or().eq("package_project", 1);
-//        }
-        //OA平台;请假，出差等
-        if (company.getPackageOa().equals(1)){
-            sysModuleQueryWrapper.or().eq("package_oa", 1);
-        }
-//        //费用报销
-//        if (company.getPackageExpense().equals(1)){
-//            sysModuleQueryWrapper.or().eq("package_expense", 1);
-//        }
-        //客户管理
-        if (company.getPackageCustomer().equals(1)){
-            sysModuleQueryWrapper.or().eq("package_customer", 1);
-        }
-        //工程专业
-        if (company.getPackageEngineering().equals(1)){
-            sysModuleQueryWrapper.or().eq("package_engineering", 1);
-        }
-//        //供应商模块
-//        if (company.getPackageProvider().equals(1)){
-//            sysModuleQueryWrapper.or().eq("package_provider", 1);
-//        }
-        List<SysModule> sysModules = sysModuleMapper.selectList(sysModuleQueryWrapper);
-        //过滤掉已开通的模块
-        List<SysRoleModule> modular = sysRoleModuleMapper.selectList(new QueryWrapper<SysRoleModule>().eq("role_id", sysRole.getId()));
-        List<Integer> modularCollect = modular.stream().map((m) -> m.getModuleId()).distinct().collect(Collectors.toList());
-        List<SysModule> collect = sysModules.stream().filter((s) -> {
-            Boolean flag = true;
-            for (Integer integer : modularCollect) {
-                if (s.getId().equals(integer)){
-                    flag = false;
+        if (company.getPackageOa().equals(1) || company.getPackageCustomer().equals(1) || company.getPackageEngineering().equals(1)){
+            System.err.println("##############################为超级管理员开通模块及权限");
+            SysRole sysRole = sysRoleMapper.selectOne(new QueryWrapper<SysRole>().eq("company_id", companyId).eq("rolename", "超级管理员"));
+            QueryWrapper<SysModule> sysModuleQueryWrapper = new QueryWrapper<>();
+            if (company.getPackageOa().equals(1)){
+                sysModuleQueryWrapper.or().eq("package_oa", 1);
+            }
+            //客户管理
+            if (company.getPackageCustomer().equals(1)){
+                sysModuleQueryWrapper.or().eq("package_customer", 1);
+            }
+            //工程专业
+            if (company.getPackageEngineering().equals(1)){
+                sysModuleQueryWrapper.or().eq("package_engineering", 1);
+            }
+            List<SysModule> sysModules = sysModuleMapper.selectList(sysModuleQueryWrapper);
+            //过滤掉已开通的模块
+            List<SysRoleModule> modular = sysRoleModuleMapper.selectList(new QueryWrapper<SysRoleModule>().eq("role_id", sysRole.getId()));
+            List<Integer> modularCollect = modular.stream().map((m) -> m.getModuleId()).distinct().collect(Collectors.toList());
+            List<SysModule> collect = sysModules.stream().filter((s) -> {
+                Boolean flag = true;
+                for (Integer integer : modularCollect) {
+                    if (s.getId().equals(integer)){
+                        flag = false;
+                    }
+                }
+                return flag;
+            }).collect(Collectors.toList());
+            if (collect.size()>0){
+                //进行模块权限添加
+                ArrayList<SysRoleModule> sysRoleModules = new ArrayList<>();
+                for (SysModule sysModule : collect) {
+                    SysRoleModule sysRoleModule = new SysRoleModule();
+                    sysRoleModule.setRoleId(sysRole.getId());
+                    sysRoleModule.setModuleId(sysModule.getId());
+                    sysRoleModules.add(sysRoleModule);
                 }
+                sysRoleModuleService.saveBatch(sysRoleModules);
+                System.out.println("########################################");
+                System.err.println(company.getCompanyName() + "已开通模块；" + sysRoleModules.toString());
+                List<Integer> sysModulesId = collect.stream().map((s) -> s.getId()).collect(Collectors.toList());
+                List<SysFunction> sysFunctions = sysFunctionMapper.selectList(new QueryWrapper<SysFunction>().in("module_id", sysModulesId));
+                //进行权限添加
+                ArrayList<SysRoleFunction> sysRoleFunctions = new ArrayList<>();
+                for (SysFunction sysFunction : sysFunctions) {
+                    SysRoleFunction sysRoleFunction = new SysRoleFunction();
+                    sysRoleFunction.setRoleId(sysRole.getId());
+                    sysRoleFunction.setFunctionId(sysFunction.getId());
+                    sysRoleFunctions.add(sysRoleFunction);
+                }
+                sysRoleFunctionService.saveBatch(sysRoleFunctions);
+                System.out.println("########################################");
+                System.err.println(company.getCompanyName() + "已开通权限；" + sysRoleModules.toString());
             }
-            return flag;
-        }).collect(Collectors.toList());
-        //进行模块权限添加
-        ArrayList<SysRoleModule> sysRoleModules = new ArrayList<>();
-        for (SysModule sysModule : collect) {
-            SysRoleModule sysRoleModule = new SysRoleModule();
-            sysRoleModule.setRoleId(sysRole.getId());
-            sysRoleModule.setModuleId(sysModule.getId());
-            sysRoleModules.add(sysRoleModule);
-        }
-        sysRoleModuleService.saveBatch(sysRoleModules);
-        System.out.println("########################################");
-        System.err.println(company.getCompanyName() + "已开通模块；" + sysRoleModules.toString());
-        List<Integer> sysModulesId = collect.stream().map((s) -> s.getId()).collect(Collectors.toList());
-        List<SysFunction> sysFunctions = sysFunctionMapper.selectList(new QueryWrapper<SysFunction>().in("module_id", sysModulesId));
-        //进行权限添加
-        ArrayList<SysRoleFunction> sysRoleFunctions = new ArrayList<>();
-        for (SysFunction sysFunction : sysFunctions) {
-            SysRoleFunction sysRoleFunction = new SysRoleFunction();
-            sysRoleFunction.setRoleId(sysRole.getId());
-            sysRoleFunction.setFunctionId(sysFunction.getId());
-            sysRoleFunctions.add(sysRoleFunction);
-        }
-        sysRoleFunctionService.saveBatch(sysRoleFunctions);
-        System.out.println("########################################");
-        System.err.println(company.getCompanyName() + "已开通权限；" + sysRoleModules.toString());
+        }
     }
 
 }