JDK 8 Stream 实现根据对象的属性去重

ExportTemperatureDto 实体对象:

@Getter
@Setter
@ToString
public class ExportTemperatureDto {
    private String name;
    private Double morningTemperature;
    private Double afternoonTemperature;
    private String classId;
    private String gradeId;
    private Integer personId;
}

在一个 ExportTemperatureDto 的集合中,根据 personId 属性去重,生成一个新的集合。

import static java.util.Comparator.comparing;
import static java.util.stream.Collectors.collectingAndThen;
import static java.util.stream.Collectors.toCollection;
 
public class StreamTest {
 
    public static void main(String[] args) {
        List<ExportTemperatureDto> temperatureList = Lists.newArrayList();
        temperatureList.add(new ExportTemperatureDto(1, "haha"));
        temperatureList.add(new ExportTemperatureDto(2, "haha"));
        temperatureList.add(new ExportTemperatureDto(3, "haha"));
        temperatureList.add(new ExportTemperatureDto(4, "haha"));
 
        temperatureList.add(new ExportTemperatureDto(1, "hahaasdas"));
        temperatureList.add(new ExportTemperatureDto(2, "hahaasdas"));
 
        List<ExportTemperatureDto> result = temperatureList.stream()
                .collect(
                        collectingAndThen(
                                toCollection(
                                        () -> new TreeSet<>(comparing(ExportTemperatureDto::getPersonId))
                                ),
                                ArrayList::new
                        )
                );
 
        result.forEach(System.out::println);
 
        /*
            输出结果为:
                personId 为 1 的,其名称为 haha
                personId 为 2 的,其名称为 haha
            因为 TreeSet 底层是使用 TreeMap 进行实现的,传入了根据 getPersonId 进行比较的比较器
            在判断 personId 相同时,其比较结果为 0,然后就会替换其 value 值,而 key 值是不会变化的,
            又因为 TreeSet 是将传入的元素作为 key 的,所以使用 TreeSet 时,当比较器比较的结果相同时,以不会将原来的值替换成比较后的值
         */
 
    }
}

知其然知其所以然,这个 stream 流的操作看起来还是有点难度的,这里记录一下。

使用到了 collectingAndThen 完成根据属性进行去重的操作,对于该去重操作的关键使用到了 collectingAndThen、toCollection、TreeSet,有点难以理解,当时我也是懵逼的,这里记录一下,以后肯定还会用的到。

理解根据对象的属性进行去重的核心是,将集合放到 TreeSet 中,然后再将 TreeSet 转为 List, 其中 TreeSet 要传入一个根据哪个属性进行比较的比较器,然后使用public ArrayList(Collection<? extends E> c)将 TreeSet 放入构造器中生成 List。

上面的 Stream 操作可以使用普通的集合:

TreeSet<ExportTemperatureDto> treeSet = new TreeSet<>(Comparator.comparing(ExportTemperatureDto::getPersonId));
for (ExportTemperatureDto temperatureDto : temperatureList){
    treeSet.add(temperatureDto);
}
List<ExportTemperatureDto> result2 =  new ArrayList<>(treeSet);

只要能够理解普通集合怎么操作的,那么使用 Stream 流操作时,就是要看对于 API 的使用是否熟悉,其实这个才是关键,只有理解了 collectingAndThen、toCollection、JDK8 的匿名函数这样内容,才能看懂这个式子。

下面就简单介绍一下:

首先说一下 collectingAndThen 方法的使用——-先进行结果集的收集,然后将收集到的结果集进行下一步的处理 ,红字的两句话是理解 collectingAndThen 的关键,首先看一下 collectingAndThen 需要传递的参数:

public static<T,A,R,RR> Collector<T,A,RR> 
         collectingAndThen(Collector<T,A,R> downstream,
         Function<R,RR> finisher)

可以看到第一个参数是 Collector 接口的子类,所以还是对于对于 Collector 的处理,Collectors 工具类里面的toList()toSet()joining()mapping()collectingAndThen()等几乎所有的方法都可以使用,这样感觉这个 collectingAndThen 就很强大了,可以嵌套的去使用。

第二个参数是一个 Function 函数,熟悉的同学都知道,Function 函数是这样的:R apply(T t),这个也是理解上面去重式子的关键,原来我想的是ArrayList::new调用的无参的构造方法,其实他调用的 ArrayList 的有参构造方法,

public ArrayList(Collection<? extends E> c)

调用的是上面那个构造方法,这样就很清晰,就是把第一个参数 downstream 的结果,交给第二个参数 Function 函数的参数里面,R apply(T t),也就是将结果设置成 t。

对于 toCollection 是一个通用的转为集合的操作,当然在 Collectors 类里面也有toList()toSet()方法,但是都不满足于使用 TreeSet 来收集集合的方法,所以使用 toCollection 是一个通用的方法,使用 TreeSet 进行收集,然后传入根据哪个属性进行比较的比较器,这样就可以了。

「点点赞赏,手留余香」

0

给作者打赏,鼓励TA抓紧创作!

微信微信 支付宝支付宝

还没有人赞赏,快来当第一个赞赏的人吧!

声明:本站所有文章,如无特殊说明或标注,均为本站原创发布。任何个人或组织,在未征得本站同意时,禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益,可联系我们进行处理。
码云笔记 » JDK 8 Stream 实现根据对象的属性去重

发表回复