主页 / 问题 / 1502823
Accepted
SergeyMih
Asked:2023-03-07 15:00:18 +0000 UTC2023-03-07 15:00:18 +0000 UTC

不使用 Stream API 对集合项进行分组

  • 772

有一种方法可以对来自多个字段的集合的元素进行分组。

public void test() {
    List<Item> itemsList = ...//Чтение из XML;

    Map<List<Object>, List<Item>> groupBy =
            itemsList.stream().collect(Collectors.groupingBy(
                    i -> Arrays.asList(i.id, i.city, i.order_status)));

    for (Map.Entry<List<Object>, List<Item>> t : groupBy.entrySet()) {
        System.out.println(t.getKey());
        for (Item item : t.getValue()) {
            System.out.println("    " + item);
        }
    }
}

以及如何在没有 Stream API 的情况下做同样的事情？

java

1 个回答

  1. Best Answer

    简单地创建一个映射，并明确指示键和值的类型，在一个循环中，您遍历列表并使用此方法将元素添加到映射 - 如果需要，此方法创建一个新的或返回现有的列表Map::computeIfAbsent值，您可以立即调用该方法List:add。
    在这种情况下LinkedHashMap，最好使用使其中元素的顺序稳定（HashMap不保证顺序）。

    public static Map<List<Object>, List<Item>> groupBy(List<Item> list) {
    Map<List<Object>, List<Item>> map = new LinkedHashMap<>();
    for (Item item : list) {
        List<Object> key = Arrays.asList(item.getId(), item.getCity(), item.getOrderStatus());
        map.computeIfAbsent(key, k -> new ArrayList<>()).add(item);
    }
    return map;
}

    hashCode应该注意的是，使用一些你需要正确重新定义方法和的中间类而不是对象列表会更正确equals。record这对于自 Java 16 以来正式支持的元组或 Lombok 注释@Data/来说很容易做到@AllArgsConstructor。

    enum OrderStatus {NEW, PROCESSING, COMPLETED, CANCELED};

// определены конструктор, геттеры без префикса `get`, hashCode / equals, toString
record MyKey(long id, String city, OrderStatus status) {
    public MyKey(Item item) {
        this(item.getId(), item.getCity(), item.getOrderStatus());
    }
}

public static Map<MyKey, List<Item>> groupBy(List<Item> list) {
    Map<MyKey, List<Item>> map = new LinkedHashMap<>();
    list.forEach(item -> map
        .computeIfAbsent(new MyKey(item), k -> new ArrayList<>())
        .add(item)
    );

    // отладочный вывод
    map.forEach((key, val) -> {
        System.out.println(key);
        val.forEach(item -> System.out.println("\t" + item));
    });

    return map;
}

    如果需要，可以为任何键参数化创建这样的分组，如果您传递某个转换器函数Function<Item, Key>，在上一个示例中，这样的函数是元组构造函数MyKey(Item item)。例如，要生成列表键，可以定义一个单独的工厂方法：

    // MyClass
public static List<Object> keyIdCityStatus(Item item) {
    return Arrays.asList(item.getId(), item.getCity(), item.getOrderStatus());
}

    参数化方法如下所示：

    public static<K> Map<K, List<Item>> groupBy(Function<Item, K> keyBuilder, List<Item> list) {
    Map<K, List<Item>> map = new LinkedHashMap<>();

    list.forEach(item -> map
        .computeIfAbsent(keyBuilder.apply(item), k -> new ArrayList<>())
        .add(item)
    );

    return map;
}

    因此，可以为任何键调用它：

    // ссылка на фабричный метод MyClass::keyIdCityStatus
Map<List<Object>, List<Item>> keyListMap = groupBy(MyClass::keyIdCityStatus, list);

// группировка по двум полям с помощью лямбды
Map<List<Object>, List<Item>> byIdAndStatus = groupBy(
    item -> List.of(item.getId(), item.getOrderStatus()), 
    list
);

// ссылка на конструктор MyKey::new
Map<MyKey, List<Item>> keyRecordMap = groupBy(MyKey::new, list);
    • 2

相关问题