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 进行收集，然后传入根据哪个属性进行比较的比较器，这样就可以了。

以上关于JDK 8 Stream 实现根据对象的属性去重的文章就介绍到这了，更多相关内容请搜索码云笔记以前的文章或继续浏览下面的相关文章，希望大家以后多多支持码云笔记。