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

女爷i 2023-09-30 17:06 20阅读 0赞

## 前言 ##

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

## 一、list转map ##

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

> 代码如下（示例）：

运行结果:  
![在这里插入图片描述][90e1619d57dc4c5ab6e30ac455c1d579.png]

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

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

代码如下（示例）：

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;
    }

运行结果：  
![在这里插入图片描述][ba2f36bbd2f04261a5afdbd330fdefee.png]  
从运行结果可以看出，根据address分组后取每组年龄最大的对象信息仅需要  
小小的几行代码就可以搞定

## 总结 ##

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

[90e1619d57dc4c5ab6e30ac455c1d579.png]: https://img-blog.csdnimg.cn/90e1619d57dc4c5ab6e30ac455c1d579.png
[ba2f36bbd2f04261a5afdbd330fdefee.png]: https://img-blog.csdnimg.cn/ba2f36bbd2f04261a5afdbd330fdefee.png