如何在java中使用带有条件的比较器

Question

我正在尝试根据从 API 收到的排序键和排序顺序对列表进行排序。 例如， 我有一个带有 sortkey 和 sortorder 的列表，我需要根据它进行排序。

List<SortList> sortlist;

我有一个对象列表：

List<Employee> employee;

我可以使用

进行排序

Collections.sort(sourceList, Comparator
                .comparing(Employee::getAge).reversed()
                .thenComparing(Employee::getCount));

但我需要在一个条件下检查 sortfeild，并基于该条件仅考虑对该字段进行排序。

例如：

如果（排序键 = “名称”） 按排序顺序从 sortlist 中 sortbythatkey

if (sortkey = "place") sortbythat key 从 sortlist 按排序顺序

所以这里如果 sortlist 有名称和位置，那么它应该按键和顺序排序

知道我怎么能做到这一点吗？

排序列表包含：

{
    "sortKey":"name",
    "sortOrder":"ASC"

},
{
    "sortKey":"place",
    "sortOrder":"DESC"

}

要求是像 SQL 中的 ORDER BY 一样将它们链接在一起

Answer 1

假设sortlist是SortCriteria的一个列表，就是这样一个类：

class SortCritera {
    private String key;
    private String order;

    public String getKey() {
        return key;
    }

    public String getOrder() {
        return order;
    }

    // constructors, setters...
}

您首先需要一个HashMap<String, Comparator<Employee>> 来存储每个可能键的所有对应比较器：

HashMap<String, Comparator<Employee>> comparators = new HashMap<>();
comparators.put("name", Comparator.comparing(Employee::getName));
comparators.put("age", Comparator.comparing(Employee::getAge));
// ...

然后你可以遍历sortlist并继续调用thenComparing：

Comparator<Employee> comparator = comparators.get(sortlist.get(0).getKey());
if (sortlist.get(0).getOrder().equals("DESC")) {
    comparator = comparator.reversed();
}
for(int i = 1 ; i < sortlist.size() ; i++) {
    if (sortlist.get(i).getOrder().equals("DESC")) {
        comparator = comparator.thenComparing(comparators.get(sortlist.get(i).getKey()).reversed());
    } else {
        comparator = comparator.thenComparing(comparators.get(sortlist.get(i).getKey()));
    }
}
// now you can sort with "comparator".

正如 Holger 所建议的，您也可以使用 Stream API 来执行此操作：

sortlist.stream().map(sc -> {
    Comparator<Employee> c = comparators.get(sc.getKey()); 
    return sc.getOrder().equals("DESC")? c.reversed(): c; 
}).reduce(Comparator::thenComparing)
.ifPresent(x -> Collections.sort(originalList, x));

Answer 2

您可以创建一个方法，在传递排序键时提供适当的比较器：

public Comparator<Employee> getComparator(String sortKey) {
    if("name".equals(sortKey)) {
        return Comparator.comparing(Employee::getName);
    } else if ("place".equals(sortKey) {
        return Comparator.comparing(Employee::getPlace);
    } else {
        throw new IllegalArgumentException();
    }
}

简单来说就是：

Collections.sort(sourceList, getComparator(sortKey).reversed()
                .thenComparing(Employee::getCount));

虽然您也可以自己编写，但我发现最好委派“标准”部分并简单地编写与此不同的部分。

如果您发现自己有很多这样的排序键，那么更合适的方法是使用地图：

private static final Map<String, Comparator<Employee>> COMPARE_MAP = new HashMap<>() {{
    put.("name", Comparator.comparing(Employee::getName));
    put.("place", Comparator.comparing(Employee::getPlace));
}});

public Comparator<Employee> getComparator(String sortKey) {
    if(COMPARE_MAP.containsKey(sortKey)) {
        return COMPARE_MAP.get(sortKey);
    } else {
        throw new IllegalArgumentException();
    }
}

反射也是一种选择，但我会谨慎使用反射，除非它变得不切实际。在这种情况下，您可以创建自己的注释来确定 Employee 类的哪些字段可用于排序。