//获取用户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); }优化代码

可以将以下部分代码进行优化：

if(userlist.contains(userId)){
    BeanUtils.copyProperties(schemaRowRuleEo,schemaRowRule);
    userRowRuleList.add(schemaRowRule);
}else if(orglist.contains(orgCode)){
    BeanUtils.copyProperties(schemaRowRuleEo,schemaRowRule);
    userRowRuleList.add(schemaRowRule);
}

可以改成：

if(userlist.contains(userId) || orglist.contains(orgCode)){
    BeanUtils.copyProperties(schemaRowRuleEo,schemaRowRule);
    userRowRuleList.add(schemaRowRule);
}

这样可以减少代码重复度，提高代码可读性。另外，需要注意的是，`DataPermissionSchema` 对象的 `schemaRowRuleList` 属性应该在循环结束后再进行设置。可以将 `build.setSchemaRowRuleList(userRowRuleList);` 移动到循环结束后。

相关问题

//获取用户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<>(); DataPermissionSchema build = DataPermissionSchema.builder().build(); //设置行权限 根据用户id和用户组织去获取 for (SchemaRowRuleEo schemaRowRuleEo : schemaRowRuleList) { SchemaRowRule schemaRowRule = SchemaRowRule.builder().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);代码优化及原因

可以将循环中的if语句提取出来，避免重复代码。同时，使用stream流简化代码，并提高代码可读性。代码优化后如下：

String userId = UserAgent.get().getUserId();
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 = schemaRowRuleList.stream()
    .filter(schemaRowRuleEo -> {
        List<String> userList = JSONArray.parseArray(schemaRowRuleEo.getRowPermissionUserList(), String.class);
        List<String> orgList = JSONArray.parseArray(schemaRowRuleEo.getRowPermissionOrgList(), String.class);
        return userList.contains(userId) || orgList.contains(orgCode);
    })
    .map(schemaRowRuleEo -> {
        SchemaRowRule schemaRowRule = SchemaRowRule.builder().build();
        BeanUtils.copyProperties(schemaRowRuleEo, schemaRowRule);
        return schemaRowRule;
    })
    .collect(Collectors.toList());

DataPermissionSchema build = DataPermissionSchema.builder().build();
build.setSchemaRowRuleList(userRowRuleList);

优化后的代码更加简洁易读，使用stream流可以避免重复代码，同时也可以提高代码的执行效率。

//获取用户权限 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(); }帮我优化这段代码

可以将代码优化为：

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. 将变量声明和初始化放在一起，提高代码的可读性。