无源码时如何基于Employee的id对TreeMap进行排序？

嘿，这个场景其实挺常见的——当依赖类封装在JAR里没法修改源码时，TreeMap的设计早就考虑到这种情况了，根本不需要让Employee实现Comparable接口，直接用**自定义排序器（Comparator）**就能搞定！

核心思路 #

TreeMap的排序逻辑有两种来源：

• 键本身实现了Comparable接口（自然排序）
• 初始化TreeMap时传入一个自定义的Comparator（定制排序）

我们要用到第二种方式，因为没法修改Employee类，所以只需要写一个Comparator，让它根据Employee的id字段来比较两个实例就行。

具体实现 #

题目里明确说Employee有对应的getter方法（getId()），所以我们直接调用这个方法就行——哪怕没有源码，只要JAR包在类路径里，编译器和JVM都能识别并调用这个public方法。

1. 编写自定义Comparator

如果是Java 8之前的版本，用匿名内部类实现：

Comparator<Employee> employeeIdComparator = new Comparator<Employee>() {
    @Override
    public int compare(Employee e1, Employee e2) {
        // 根据id类型调整比较逻辑，这里假设id是Integer
        return Integer.compare(e1.getId(), e2.getId());
        // 如果id是字符串，用 return e1.getId().compareTo(e2.getId());
        // 如果是Long，用 Long.compare(e1.getId(), e2.getId());
    }
};

Java 8及以上可以用Lambda或者方法引用简化，代码更简洁：

// 假设id是int/Integer类型
Comparator<Employee> employeeIdComparator = Comparator.comparingInt(Employee::getId);

// 如果id是Long类型
// Comparator<Employee> employeeIdComparator = Comparator.comparingLong(Employee::getId);

// 如果id是字符串
// Comparator<Employee> employeeIdComparator = Comparator.comparing(Employee::getId);

2. 初始化TreeMap时传入Comparator

创建TreeMap的时候，把上面写的排序器传进去就行：

TreeMap<Employee, String> employeeMap = new TreeMap<>(employeeIdComparator);

这样之后，往这个TreeMap里放Employee实例，它就会自动按照id字段的大小来排序键了！

额外注意点：处理null值 #

如果你的业务场景中可能出现Employee实例为null，或者id字段为null的情况，建议给Comparator加上null处理逻辑，避免抛出NullPointerException：

Comparator<Employee> safeComparator = Comparator.nullsFirst(
    Comparator.comparing(Employee::getId, Comparator.nullsFirst(Integer::compare))
);
TreeMap<Employee, String> safeEmployeeMap = new TreeMap<>(safeComparator);

这个排序器会把null的Employee放在最前面，id为null的Employee也会排在id非null的前面，非常安全。

内容的提问来源于stack exchange，提问作者Manish Patil