前言
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流的时候在其中写太复杂的逻辑。