背景:
从数据库多表查询数据,需要分多次关联多个表格,每次查出最终返回数据的一部分,最终根据相同属性进行合并。今天来实现合并的部分。
比如入下图类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
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再去获取其他数据,可以在对象返回前去操作。
以上,就是根据属性进行合并的操作方法。