背景:
从数据库多表查询数据,需要分多次关联多个表格,每次查出最终返回数据的一部分,最终根据相同属性进行合并。今天来实现合并的部分。
比如入下图类UserData,第一次查询拿出id和num1,第二次查询拿出id和num2,要求,id相同,需要在同一个对象返回所有数据,这时候如何进行合并?
public class UserData(){
private Integer id;
private Integer num1;
private Integer num2;
}
两个list,都是UserData对象,只不过都是含有部分属性,但是共同属性是id,而且要求id相同的一起返回。
思路:利用stream,将两个list进行合并,然后再根据id分组,再合并不同属性的值
List<UserData> userDataList = Stream.of(list1, list2)
.flatMap(Collection::stream)
.collect(Collectors.groupingBy(UserData::getId))
.values()
.stream()
.map(dataList -> {
UserData userData= new UserData();
userDate.setId(dataList.get(0).getId());
dataList.forEach(UserData::merge);
return userData;
})
.collect(Collectors.toList());
如果有多于2个list,则Stream.of()一起放入,进行合并即可,后续相同
merge方法,是具体的合并方法,可以根据需要自行实现,比如属性为空的取另一个属性的值,merge方法如下:
public class UserData(){
private Integer id;
private Integer num1;
private Integer num2;
public void merge(UserData userData){
if(num1 == null){
this.num1 = userData.num1;
}
if(num2 == null){
this.num2 = userData.num2;
}
}
}
合并完成之后,如果还有其他操作,比如根据id再去获取其他数据,可以在对象返回前去操作。
以上,就是根据属性进行合并的操作方法。