Java替换ArrayList为TreeSet存储Report对象抛出ClassCastException如何解决

TreeSet 是基于红黑树实现的有序去重集合，默认要求存入的元素必须实现 java.lang.Comparable 接口，用于插入时的排序比较。你直接替换ArrayList为TreeSet后，Report 类没有实现Comparable接口，TreeSet在插入第二个元素做排序比较时，会尝试将Report强制转换为Comparable类型，就触发了ClassCastException。仅展示1条报告是因为第一条元素插入时不需要做比较，后续插入触发异常直接中断，所以集合中仅剩第一条元素。

你可以从下面两种方案中任选一种适配： #

方案1：修改Report类实现Comparable接口 #

适合可以修改Report源码的场景，按照业务排序需求重写compareTo方法即可：

// 原Report类声明修改为实现Comparable泛型接口
public class Report implements Comparable<Report> {
    // 原有类属性、方法保持不变

    @Override
    public int compareTo(Report other) {
        // 这里按你的业务排序需求写逻辑，示例为按报告ID升序
        return Integer.compare(this.getId(), other.getId());
        // 如果要按报告时间升序，可写为 return this.getCreateTime().compareTo(other.getCreateTime());
        // *注意：compareTo返回0时TreeSet会判定两个元素相等，不会存入新元素，所以请确保比较逻辑和业务去重逻辑一致*
    }
}

方案2：构造TreeSet时传入自定义Comparator #

适合不能修改Report源码的场景，不需要改动Report类，初始化集合时指定比较规则：

// 初始化TreeSet时直接传入Comparator，用lambda写法更简洁
TreeSet<Report> reports = new TreeSet<>(Comparator.comparingInt(Report::getId));
// 也可以写多条件排序，比如先按时间升序，时间相同按ID升序
// TreeSet<Report> reports = new TreeSet<>(Comparator.comparing(Report::getCreateTime).thenComparingInt(Report::getId));

如果你的需求仅为排序不需要去重，不需要替换为TreeSet，直接沿用原有ArrayList，在需要展示排序结果时调用Collections.sort(reports, 自定义Comparator)或者reports.stream().sorted(自定义Comparator)做排序即可，避免TreeSet的去重规则误删业务数据。

内容的提问来源于stack exchange，提问作者ilie alexandru