根据List对象的特定属性合并另一属性

ゝ一世哀愁。 2024-02-05 19:09 48阅读 0赞

Java中使用stream流进行List对象的特定属性合并另一属性（包括单个属性以及两个属性的合并求和以及没有属性的合并求和）

首先建立一个测试类：

@Data
    @AllArgsConstructor
    @NoArgsConstructor
    public class Person {
        
        //id
        private String id;
        //姓名
        private String name;
        //年龄
        private int age;
    }

建立一个List并放入值：

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求和

int tTotal=list.stream().collect(Collectors.summingInt(Person::getAge()));

二、根据单个属性合并（id），合并后将age求和

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源码）

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求和

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()函数，这个函数的源码如下代码块

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中的结果是

[("1", "一号选手", 6),("2", "二号选手", 2),("3", "三号选手", 3),("1", "四号选手", 4)]