java - 像切换条件一样对 Java Stream 进行排序

Question

我正在尝试按其中一个字段的特定顺序对流进行排序。

现在我通过将流转换为列表并使用开关然后将它们重新加入到所需顺序的列表中来实现这一点。

    fruits.forEach(fruit -> {                switch (fruit.getName()) {                    case "Orange":                        orangesList.add(fruit);                        break;                    case "Apple":                        applesList.add(fruit);                        break;                    case "WaterMelon":                        watermelonList.add(fruit);                        break;                    default:                        otherFruits.add(fruit);                        break;                }    });    genericFruitList.addAll(0, orangeList);    genericFruitList.addAll(1, applesList);    genericFruitList.addAll(2, watermelonList);    genericFruitList.addAll(3, otherFruits);

我想知道是否有任何更改可以使用流排序方法实现此目的并使用自定义比较器或类似的东西。

提前致谢。

Answer 1

您可以使用像这样的显式顺序创建比较器

List<String> order = Arrays.asList("Orange", "Apple", "WaterMelon");
Comparator<String> comp
    = Comparator.comparingInt(name -> order.indexOf(name)-Integer.MIN_VALUE);

可以像这样使用

List<Fruit> genericFruitList = fruits
    .sorted(Comparator.comparing(fruit -> fruit.getName(), comp))
    .collect(Collectors.toList());

但是，对整个列表进行排序，尤其是使用基于 List.indexOf 的比较器时，效率可能非常低。另一种选择是

List<Fruit> genericFruitList = fruits
    .collect(Collectors.groupingBy(fruit -> fruit.getName()))
    .entrySet().stream()
    .sorted(Map.Entry.comparingByKey(comp))
    .flatMap(e -> e.getValue().stream())
    .collect(Collectors.toList());

它只是对每个 Fruit 执行哈希查找，并且只对不同的映射进行排序。

这可以看作是 Bucket Sort 的变体.