public <T, K> void assembleProjectInfo(List<T> list, Map<Function<T, String>, Map<Function<K, String>, BiConsumer<T, String>>> idFunc_nameBic) throws CustomException {
if (CollectionUtils.isEmpty(list) || MapUtils.isEmpty(idFunc_nameBic))
return;
// 获取所有id
List<String> idList = getIdList(list, idFunc_nameBic.keySet());
if (CollectionUtils.isEmpty(idList)) return;
ProjectQueryModel queryModel = new ProjectQueryModel();
queryModel.setIdList(idList);
// 根据ID调用其它服务接口获取数据
List<ProjectResp> respList = getProjectPageInfo(queryModel);
// list 转换为map
Map<String, Object> projectRespMap = respList.stream().collect(Collectors.toMap(ProjectResp::getId, resp -> resp));
// 处理数据方法
assemble(
list,
idFunc_nameBic,
projectRespMap
);
}
private <T, K> void assemble(List<T> list, Map<Function<T, String>, Map<Function<K, String>, BiConsumer<T, String>>> idFunc_nameBic, Map<String, Object> id_obj) {
if (MapUtils.isNotEmpty(id_obj)) {
list.forEach(dto -> idFunc_nameBic.forEach(
(idFuncItem, nameBicItem) -> {
String ids = idFuncItem.apply(dto);
if (StringUtils.isBlank(ids)) return;
Object obj = id_obj.get(ids);
if (obj == null) return;
nameBicItem.forEach(
(get, set) -> {
String val = get.apply((K) obj);
if (val == null) {
return;
}
set.accept(dto, val);
}
);
}
)
);
}
}
方法调用:
assembleProjectInfo(
result.getContent(), // 原数据,需要填充的数据
new HashMap<Function<HappyRiverResp, String>, Map<Function<ProjectResp, String>, BiConsumer<HappyRiverResp, String>>>() {{
// key:需要用来查询数据的属性
put(HappyRiverResp::getRiverId, new HashMap<Function<ProjectResp, String>, BiConsumer<HappyRiverResp, String>>() {{
put(ProjectResp::getName, HappyRiverResp::setRiverName); // 需要填充的属性,根据获取到的对象的某属性,填充到原对象的某属性
put(ProjectResp::getAreaName, HappyRiverResp::setAreaName); // 需要填充的属性
}});
}}
);