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

编程技巧—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分组后取每组年龄最大的对象信息仅需要
小小的几行代码就可以搞定

---

总结

实际业务中根据业务合理利用stream流能够很大程度的简化代码，但是这种写法在出现bug的时候排查较为复杂，建议不要在使用stream流的时候在其中写太复杂的逻辑。