项目统计指标开发

jwtech-system/src/main/java/com/kms/common/api/CommonApi.java

@@ -74,6 +74,7 @@ public class CommonApi {
     @Value("${jianwei.profile}")
     private String filePath;
 
+
     @ApiOperation("字典数据")
     @GetMapping("dict/{dictType}")
     public AjaxResult dict(@PathVariable("dictType") String dictType) {
@@ -174,10 +175,11 @@ public class CommonApi {
         SysUser user = UserUtils.getUser();
         List<SysRole> sysRoles = sysRoleService.getRoleByUserId(user.getId());
         List<SysRole> legalPerson = sysRoles.stream().filter(x -> x.getRoleKey().equals("legalPerson")).collect(Collectors.toList());
+        List<SysRole> enterprise = sysRoles.stream().filter(x -> x.getRoleKey().equals("enterprise")).collect(Collectors.toList());
 
-        if (CollectionUtil.isEmpty(legalPerson)) {
+        if (CollectionUtil.isEmpty(legalPerson)||CollectionUtil.isEmpty(enterprise)) {
             return AjaxResult.success(true);
-        } else {
+        } else if (CollectionUtil.isNotEmpty(legalPerson)){
             LambdaQueryWrapper<SysUserPro> queryWrapper = Wrappers.lambdaQuery(SysUserPro.class)
                     .eq(SysUserPro::getUserId, user.getId());
 
@@ -189,6 +191,14 @@ public class CommonApi {
             }else {
                 return AjaxResult.success(true);
             }
+        }else {
+            List<SysUserEnterprise> list = sysUserEnterpriseService.list(Wrappers.lambdaQuery(SysUserEnterprise.class)
+                    .eq(SysUserEnterprise::getUserId, user.getId()));
+            if (CollectionUtil.isEmpty(list)) {
+                return AjaxResult.success(false);
+            } else {
+                return AjaxResult.success(true);
+            }
         }
 
     }