TreeSet自定义比较器下的覆盖需求及重复元素问题排查

问题背景 #

需求是跟踪某一时刻以来涨跌幅最高的股票代码，当股票涨跌幅更新时，希望TreeSet能覆盖原有条目而非新增重复项。
实现方式：

• 定义SymbolAndPercentMove类，重写hashCode、equals（仅按symbol判断相等）和compareTo（按percentMove降序排序）
• 使用Collections.synchronizedSortedSet包装TreeSet保证线程安全

遇到的问题：测试环境中先remove再add能正常覆盖，但生产环境多次出现remove失效，同一symbol存在两个不同percentMove的条目，甚至单线程处理或使用同步集合时仍会出现。

核心原因 #

TreeSet（底层依赖TreeMap）的元素判断逻辑不依赖equals和hashCode，而是完全基于compareTo方法的返回值。根据SortedSet的规范：当a.compareTo(b) == 0时，会被判定为同一个元素，此时要求a.equals(b)也必须返回true（即compareTo与equals逻辑一致）。

你的实现中存在致命的逻辑冲突：

• equals仅按symbol判断相等
• compareTo按percentMove降序排序，同symbol但不同percentMove的元素compareTo返回值不为0

这就导致TreeSet会把同symbol但不同percentMove的元素判定为不同元素，即使equals返回true。比如：

1. 旧元素：SymbolAndPercentMove("AAPL", 5.0)已在集合中
2. 新元素：SymbolAndPercentMove("AAPL", 10.0)调用remove时，TreeSet通过compareTo查找元素，由于10.0>5.0（降序排序下compareTo返回-1），判定两者为不同元素，remove操作找不到旧元素，无法删除
3. 随后执行add操作，TreeSet认为这是新元素，直接插入，最终出现重复条目

解决方案 #

方案一：修正compareTo方法，对齐equals逻辑 #

调整compareTo的判断逻辑，保证同symbol的元素compareTo返回0（与equals逻辑一致），不同symbol的元素再按percentMove降序排序。这样TreeSet会正确识别同symbol的元素为重复项，remove操作也能准确定位元素。

代码示例：

import java.util.Objects;

public class SymbolAndPercentMove implements Comparable<SymbolAndPercentMove> {
    private String symbol;
    private double percentMove;

    public SymbolAndPercentMove(String symbol, double percentMove) {
        this.symbol = symbol;
        this.percentMove = percentMove;
    }

    // Getter、Setter省略

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;
        SymbolAndPercentMove that = (SymbolAndPercentMove) o;
        return Objects.equals(symbol, that.symbol);
    }

    @Override
    public int hashCode() {
        return Objects.hash(symbol);
    }

    @Override
    public int compareTo(SymbolAndPercentMove o) {
        // 优先按symbol比较，同symbol则视为相等
        int symbolCompare = this.symbol.compareTo(o.symbol);
        if (symbolCompare != 0) {
            // 不同symbol时，按percentMove降序排序
            return Double.compare(o.percentMove, this.percentMove);
        }
        return 0;
    }
}

此时更新元素的流程仍为remove -> add，但由于compareTo逻辑对齐，remove能正确找到同symbol的旧元素，不会出现重复。

方案二：分离去重与排序逻辑（更推荐） #

放弃用TreeSet同时承担去重和排序的职责，改用HashMap保证symbol的唯一性（天然支持覆盖），需要排序时再将集合元素取出排序。这种方式逻辑更清晰，避免了Comparable规范冲突的问题。

代码示例：

import java.util.*;

public class StockTracker {
    // 用HashMap存储，key为symbol，保证唯一
    private final Map<String, SymbolAndPercentMove> stockMap = Collections.synchronizedMap(new HashMap<>());

    // 更新股票涨跌幅，自动覆盖旧值
    public void updateStock(String symbol, double newPercentMove) {
        stockMap.put(symbol, new SymbolAndPercentMove(symbol, newPercentMove));
    }

    // 获取按涨跌幅降序排序的股票列表
    public List<SymbolAndPercentMove> getSortedStocks() {
        List<SymbolAndPercentMove> stockList = new ArrayList<>(stockMap.values());
        stockList.sort((a, b) -> Double.compare(b.getPercentMove(), a.getPercentMove()));
        return stockList;
    }
}

// SymbolAndPercentMove类仅需保证equals和hashCode按symbol实现
class SymbolAndPercentMove {
    private String symbol;
    private double percentMove;

    public SymbolAndPercentMove(String symbol, double percentMove) {
        this.symbol = symbol;
        this.percentMove = percentMove;
    }

    // Getter、Setter省略

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;
        SymbolAndPercentMove that = (SymbolAndPercentMove) o;
        return Objects.equals(symbol, that.symbol);
    }

    @Override
    public int hashCode() {
        return Objects.hash(symbol);
    }
}

总结 #

问题的根源是违反了Comparable接口的核心规范：compareTo与equals逻辑必须一致。选择方案一需严格对齐两者逻辑，方案二则通过分离职责彻底规避这类问题，更适合业务场景。

内容的提问来源于stack exchange，提问作者Kannan J