【编程技巧】Stream流之list转map、分组取每组第一条

【编程技巧】Stream流之list转map、分组取每组第一条 2022-08-21 318

编程技巧—Stream流之list转map、分组取每组第一条

前言

JDK1.8推出的stream流能极大的简化对集合的操作，让代码更美观，老规矩，直接上代码。

一、list转map

取list中对象的某个属性作为唯一key，对象作为value形成一个map集合，能够便于在后续业务中取值而不用遍历list集合，使代码更优雅。

代码如下（示例）：

public class StreamSkill {
          
   

    public static void main(String[] args) {
          
   
        List<TestData> list = new ArrayList<>();
        TestData.TestDataBuilder builder = TestData.builder();
        builder.id(1).name("老王").age(50).address("隔壁");
        list.add(builder.build());
        builder.id(2).name("司马老贼").age(45).address("许昌");
        list.add(builder.build());

        System.out.println(list);

        Map<Integer, TestData> map = list.stream().collect(Collectors.toMap(TestData::getId, Function.identity()));

        System.out.println(map.get(1));

    }

}

@Data
@Builder
class TestData{
          
   
    private Integer id;
    private String name;
    private Integer age;
    private String address;
}

运行结果:

二、分组取每组第一条数据

通过分组取每组第一条数据的操作可以在很多场景适用，比如取多条数据中最新的一条数据等场景。

代码如下（示例）：

public class StreamSkill {
          
   

    public static void main(String[] args) {
          
   
        List<TestData> list = new ArrayList<>();
        TestData.TestDataBuilder builder = TestData.builder();
        builder.id(1).name("老王").age(50).address("隔壁");
        list.add(builder.build());
        builder.id(2).name("司马老贼").age(45).address("许昌");
        list.add(builder.build());
        builder.id(3).name("曹贼").age(43).address("许昌");
        list.add(builder.build());
        builder.id(4).name("大耳朵").age(52).address("益州");
        list.add(builder.build());
        builder.id(5).name("燕人").age(58).address("益州");
        list.add(builder.build());

        System.out.println(list);

        Map<String, TestData> map = list.stream().collect(
                Collectors.groupingBy(TestData::getAddress, Collectors.collectingAndThen(
                        Collectors.reducing((t1, t2) -> t1.getAge() > t2.getAge() ? t1 : t2),
                        Optional::get
                ))
        );

        System.out.println(map);
    }
}


@Data
@Builder
class TestData{
          
   
    private Integer id;
    private String name;
    private Integer age;
    private String address;
}

运行结果：从运行结果可以看出，根据address分组后取每组年龄最大的对象信息仅需要小小的几行代码就可以搞定