Java中使用stream流进行List对象的特定属性合并另一属性(包括单个属性以及两个属性的合并求和以及没有属性的合并求和)
首先建立一个测试类:
java
@Data
@AllArgsConstructor
@NoArgsConstructor
public class Person {
//id
private String id;
//姓名
private String name;
//年龄
private int age;
}
建立一个List并放入值:
java
List<Person> list = new ArrayList<>();
list.add(new Person("1", "一号选手", 1));
list.add(new Person("2", "二号选手", 2));
list.add(new Person("3", "三号选手", 3));
list.add(new Person("1", "四号选手", 4));
一、没有属性的合并求和,代码如下
//对list中的age求和
java
int tTotal=list.stream().collect(Collectors.summingInt(Person::getAge()));
二、根据单个属性合并(id),合并后将age求和
java
list = list.stream().collect(Collectors.toMap(Person::getId(), a -> a, (o1, o2) -> {
o1.setAge(o1.getAge().add(o2.getAge()));
return o1;
})).values().stream().collect(Collectors.toList());
这里主要讲解下这个stream流的含义,主要在Collectors.toMap()这个函数,后面带有的三个参数,我们查看源码可以知道(下面代码块为toMap源码)
java
public static <T, K, U>
Collector<T, ?, Map<K,U>> toMap(Function<? super T, ? extends K> keyMapper,
Function<? super T, ? extends U> valueMapper,
BinaryOperator<U> mergeFunction) {
return toMap(keyMapper, valueMapper, mergeFunction, HashMap::new);
}
三个参数的含义分别是:
1.keyMapper:Map的键
2.valueMapper:Map的值
3.mergeFunction:如果键重复,值的合并方式(这里可以写你自己的逻辑)
结合我们的例子,就是将Person类的id作为键,值还是Person对象,如果id重复,那么处理方式是将两个对象(o1, o2)的age相加。最后list中的结果将是
[("1", "一号选手", 10),("2", "二号选手", 2),("3", "三号选手", 3)]
三、根据两个属性合并(id, name),合并后将age求和
java
List<Person> filterList = new ArrayList<>();
list.stream().collect(Collectors.groupingBy(item -> (item.getId() + item.getName()), Collectors.toList())).forEach(
(id, transfer) -> {
transfer.stream().reduce((a,b) -> new Person(a.getId(), a.getName(),a.getAge()+b.getAge())).ifPresent(filterList::add);
}
);
上述代码可以分为两部分看,第一部分使用了Collectors.groupingBy()函数,这个函数的源码如下代码块
java
public static <T, K, A, D>
Collector<T, ?, Map<K, D>> groupingBy(Function<? super T, ? extends K> classifier,
Collector<? super T, A, D> downstream) {
return groupingBy(classifier, HashMap::new, downstream);
}
可以看出实际上这个函数是返回了一个HashMap,那么第一个参数classifier实际是HashMap的key是什么,第二个参数downstream表示你的值的类型,结合我们的例子可以看出,我们的key是id+name,我们值的类型是List。
第二部分就是将生成的map进行foreach,那么tranfer即为值的list,里面对象的id和name不变,将年龄相加合并,然后放入新的filterlist中。
最后filterlist中的结果是
java
[("1", "一号选手", 6),("2", "二号选手", 2),("3", "三号选手", 3),("1", "四号选手", 4)]