//获取用户权限 List<DataPermissionSchemaEo> dataPermissionsRedisByUserUid = DataPermissionUtils.getDataPermissionsRedisByUserUid(UserAgent.get().getUserId()); if (CollectionUtils.isEmpty(dataPermissionsRedisByUserUid)) { return joinPoint.proceed(); } Map<String, DataPermissionSchemaEo> dataPermissionSchemaEoMap = dataPermissionsRedisByUserUid.stream().collect(Collectors.toMap(key -> key.getDatabaseCode() + key.getSchemaCode(), value -> value)); //获取用户对于这个表的行权限 DataPermissionSchemaEo dataPermissionSchemaEo = dataPermissionSchemaEoMap.get(databaseName + tableName); if(dataPermissionSchemaEo == null){ return joinPoint.proceed(); } List<SchemaRowRuleEo> schemaRowRuleList = dataPermissionSchemaEo.getSchemaRowRuleList(); if(CollectionUtils.isEmpty(schemaRowRuleList)){ return joinPoint.proceed(); }帮我优化这段代码
时间: 2024-02-20 19:00:27 浏览: 18
可以将代码优化为:
```java
List<DataPermissionSchemaEo> dataPermissionsRedisByUserUid = DataPermissionUtils.getDataPermissionsRedisByUserUid(UserAgent.get().getUserId());
if (CollectionUtils.isEmpty(dataPermissionsRedisByUserUid)) {
return joinPoint.proceed();
}
DataPermissionSchemaEo dataPermissionSchemaEo = dataPermissionsRedisByUserUid.stream()
.filter(d -> (d.getDatabaseCode() + d.getSchemaCode()).equals(databaseName + tableName))
.findFirst()
.orElse(null);
if (dataPermissionSchemaEo == null) {
return joinPoint.proceed();
}
List<SchemaRowRuleEo> schemaRowRuleList = dataPermissionSchemaEo.getSchemaRowRuleList();
if (CollectionUtils.isEmpty(schemaRowRuleList)) {
return joinPoint.proceed();
}
```
这里采用了:
1. 使用 Stream API 的 filter() 和 findFirst() 方法来优雅地查找用户对于这个表的行权限。
2. 使用 orElse(null) 方法来取代 if 判断,使得代码更加简洁。
3. 将变量声明和初始化放在一起,提高代码的可读性。