如何根据动态输入选择Comparator实现可扩展多条件排序？

核心思路 #

用策略模式将每个排序维度对应的Comparator封装为独立策略，通过映射表关联用户输入的排序标识（如byName、byAge）与对应策略。之后只需根据用户输入的排序优先级列表，依次链式组合对应的Comparator即可——这种方式完全遵循开闭原则，新增排序维度时无需修改核心逻辑。

具体实现（Java示例） #

1. 定义数据模型

class Person {
    private String name;
    private int age;

    public Person(String name, int age) {
        this.name = name;
        this.age = age;
    }

    // 必须提供getter用于Comparator取值
    public String getName() { return name; }
    public int getAge() { return age; }

    @Override
    public String toString() {
        return String.format("{%s, %d}", name, age);
    }
}

2. 构建排序策略映射表

将用户输入的标识与对应的Comparator绑定，后续新增排序维度只需在此添加条目：

import java.util.Comparator;
import java.util.Map;

// 静态映射表，全局复用
private static final Map<String, Comparator<Person>> SORT_STRATEGIES = Map.of(
    "byName", Comparator.comparing(Person::getName), // 按姓名自然排序
    "byAge", Comparator.comparingInt(Person::getAge) // 按年龄数值排序
);

注：Java 8及以下版本可改用new HashMap<>()手动put键值对，兼容旧环境。

3. 动态组合Comparator的工具方法

遍历用户输入的排序优先级列表，从映射表中取出对应策略并链式组合：

import java.util.List;
import java.util.stream.Collectors;

public static Comparator<Person> buildDynamicComparator(List<String> sortFields) {
    // 输入合法性校验
    if (sortFields == null || sortFields.isEmpty()) {
        throw new IllegalArgumentException("排序字段列表不能为空");
    }

    // 用Stream.reduce链式组合所有Comparator
    return sortFields.stream()
            .map(field -> {
                Comparator<Person> cmp = SORT_STRATEGIES.get(field);
                if (cmp == null) {
                    throw new IllegalArgumentException("不支持的排序字段：" + field);
                }
                return cmp;
            })
            .reduce(Comparator::thenComparing)
            .orElseThrow(() -> new IllegalArgumentException("无法构建有效比较器"));
}

4. 测试示例

import java.util.Arrays;
import java.util.List;

public static void main(String[] args) {
    List<Person> people = Arrays.asList(
            new Person("John", 81),
            new Person("Bob", 81),
            new Person("David", 47)
    );

    // 测试用户输入["byName", "byAge"]的场景
    List<String> sortOrder1 = Arrays.asList("byName", "byAge");
    people.sort(buildDynamicComparator(sortOrder1));
    System.out.println("按姓名优先排序结果：");
    people.forEach(System.out::println); 
    // 输出：[{Bob, 81}, {David, 47}, {John, 81}]

    // 测试用户输入["byAge", "byName"]的场景
    List<String> sortOrder2 = Arrays.asList("byAge", "byName");
    people.sort(buildDynamicComparator(sortOrder2));
    System.out.println("\n按年龄优先排序结果：");
    people.forEach(System.out::println);
    // 输出：[{David, 47}, {Bob, 81}, {John, 81}]
}

扩展性说明 #

如果后续需要新增排序维度（如按分数byScore），只需两步：

1. 在Person类中添加score字段及对应的getScore()方法；
2. 在SORT_STRATEGIES映射表中新增一条："byScore", Comparator.comparingInt(Person::getScore)。
   无需修改buildDynamicComparator的核心逻辑，完全满足可扩展需求。

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