//获取用户id String userId = UserAgent.get().getUserId(); //根据id拿到用户组织 Map<String, UserEo> userMapByUid = comDataComponent.getUserMapByUid(); String orgCode = userMapByUid.get(userId).getOrgCode(); //所有的行权限 列权限 List<SchemaRowRuleEo> schemaRowRuleList = comDataComponent.getSchemaRowRuleList(); List<SchemaColumnRuleEo> schemaColumnRuleList = comDataComponent.getSchemaColumnRuleList(); List<SchemaRowRule> userRowRuleList = new ArrayList<>(); //设置行权限 根据用户id和用户组织去获取 for (SchemaRowRuleEo schemaRowRuleEo : schemaRowRuleList) { SchemaRowRule schemaRowRule = SchemaRowRule.builder().build(); DataPermissionSchema build = DataPermissionSchema.builder().schemaCode(schemaRowRuleEo.getSchemaCode()).databaseCode(schemaRowRuleEo.getDatabaseCode()).build(); if(StringUtils.isNotEmpty(schemaRowRuleEo.getRowPermissionUserList())){ List<String> userlist = JSONArray.parseArray(schemaRowRuleEo.getRowPermissionUserList(), String.class); if(userlist.contains(userId)){ BeanUtils.copyProperties(schemaRowRuleEo,schemaRowRule); userRowRuleList.add(schemaRowRule); } }else if(StringUtils.isNotEmpty(schemaRowRuleEo.getRowPermissionOrgList())){ List<String> orglist = JSONArray.parseArray(schemaRowRuleEo.getRowPermissionOrgList(), String.class); if(orglist.contains(orgCode)){ BeanUtils.copyProperties(schemaRowRuleEo,schemaRowRule); userRowRuleList.add(schemaRowRule); } } build.setSchemaRowRuleList(userRowRuleList); }优化代码
时间: 2023-12-10 16:38:39 浏览: 138
可以使用流式编程来优化代码,代码如下:
```
//获取用户id
String userId = UserAgent.get().getUserId();
//根据id拿到用户组织
Map<String, UserEo> userMapByUid = comDataComponent.getUserMapByUid();
String orgCode = userMapByUid.get(userId).getOrgCode();
//所有的行权限 列权限
List<SchemaRowRuleEo> schemaRowRuleList = comDataComponent.getSchemaRowRuleList();
List<SchemaColumnRuleEo> schemaColumnRuleList = comDataComponent.getSchemaColumnRuleList();
//设置行权限 根据用户id和用户组织去获取
List<SchemaRowRule> userRowRuleList = schemaRowRuleList.stream()
.filter(schemaRowRuleEo -> StringUtils.isNotEmpty(schemaRowRuleEo.getRowPermissionUserList())
&& JSONArray.parseArray(schemaRowRuleEo.getRowPermissionUserList(), String.class).contains(userId)
|| StringUtils.isNotEmpty(schemaRowRuleEo.getRowPermissionOrgList())
&& JSONArray.parseArray(schemaRowRuleEo.getRowPermissionOrgList(), String.class).contains(orgCode))
.map(schemaRowRuleEo -> {
SchemaRowRule schemaRowRule = SchemaRowRule.builder().build();
BeanUtils.copyProperties(schemaRowRuleEo, schemaRowRule);
return schemaRowRule;
})
.collect(Collectors.toList());
DataPermissionSchema build = DataPermissionSchema.builder()
.schemaCode(schemaRowRuleList.get(0).getSchemaCode())
.databaseCode(schemaRowRuleList.get(0).getDatabaseCode())
.schemaRowRuleList(userRowRuleList)
.build();
```
阅读全文