展平地图中的列表列表

Question

I have a stream of orders (the source being a list of orders). Each order has a Customer, and a list of OrderLine.

What I'm trying to achieve is to have a map with the customer as the key, and all order lines belonging to that customer, in a simple list, as value.

What I managed right now returns me a Map<Customer>, List<Set<OrderLine>>> , by doing the following:

orders
  .collect(
    Collectors.groupingBy(
      Order::getCustomer,
      Collectors.mapping(Order::getOrderLines, Collectors.toList())
    )
  );

I'm either looking to get a Map<Customer, List<OrderLine>> directly from the orders stream, or by somehow flattening the list from a stream of the Map<Customer>, List<Set<OrderLine>>> that I got above.

Answer 1

Another option would be to use a simple forEach call:

Map<Customer, List<OrderLine>> map = new HashMap<>();

orders.forEach(
    o -> map.computeIfAbsent(
            o.getCustomer(),
            c -> new ArrayList<OrderLine>()
        ).addAll(o.getOrderLines())
);

You can then continue to use streams on the result with map.entrySet().stream() .

For a groupingBy approach, try Flat-Mapping Collector for property of a Class using groupingBy

Answer 2

You can simply use Collectors.toMap .

Something like

    orders
        .stream()
        .collect(Collectors
                .toMap(Order::getCustomer
                        , Order::getOrderLines
                        , (v1, v2) -> { List<OrderLine> temp = new ArrayList<>(v1); 
                                        temp.addAll(v2); 
                                        return temp;});

The third argument to the toMap function is the merge function. If you don't explicitly provide that and it there is a duplicate key then it will throw the error while finishing the operation.