Потоки Java: группируйте список в карту карт

Как я могу сделать следующее с потоками Java?

Скажем, у меня есть следующие классы:

class Foo {
    Bar b;
}

class Bar {
    String id;
    String date;
}

У меня есть List<Foo>, и я хочу преобразовать его в Map <Foo.b.id, Map<Foo.b.date, Foo>. I.e: сначала сначала Foo.b.id, а затем Foo.b.date.

Я борюсь со следующим двухэтапным подходом, но второй даже не компилируется:

Map<String, List<Foo>> groupById =
        myList
                .stream()
                .collect(
                        Collectors.groupingBy(
                                foo -> foo.getBar().getId()
                        )
                );

Map<String, Map<String, Foo>> output = groupById.entrySet()
        .stream()
        .map(
                entry -> entry.getKey(),
                entry -> entry.getValue()
                        .stream()
                        .collect(
                                Collectors.groupingBy(
                                        bar -> bar.getDate()
                                )
                        )
        );

Спасибо заранее.

Ответы

Ответ 1

Вы можете группировать свои данные за один раз, предполагая, что существуют только разные Foo:

Map<String, Map<String, Foo>> map = list.stream()
        .collect(Collectors.groupingBy(f -> f.b.id, 
                 Collectors.toMap(f -> f.b.date, Function.identity())));

Сохранение некоторых символов с помощью статического импорта:

Map<String, Map<String, Foo>> map = list.stream()
        .collect(groupingBy(f -> f.b.id, toMap(f -> f.b.date, identity())));

Ответ 2

Предположим, что пары (b.id, b.date) различны. Если так, на втором этапе вам не нужна группировка, просто собирая Map, где ключ foo.b.date, а значение foo:

Map<String, Map<String, Foo>> map = 
       myList.stream()
             .collect(Collectors.groupingBy(f -> f.b.id))    // map {Foo.b.id -> List<Foo>}
             .entrySet().stream()
             .collect(Collectors.toMap(e -> e.getKey(),                 // id
                                       e -> e.getValue().stream()       // stream of foos
                                             .collect(Collectors.toMap(f -> f.b.date, 
                                                                       f -> f))));

Или еще проще:

Map<String, Map<String, Foo>> map = 
       myList.stream()
             .collect(Collectors.groupingBy(f -> f.b.id, 
                                            Collectors.toMap(f -> f.b.date, 
                                                             f -> f)));

Ответ 3

Альтернативой является поддержка договора равенства на вашем ключе, Bar:

class Bar {
    String id;
    String date;

    public boolean equals(Object o){
       if (o == null) return false;
       if (!o.getClass().equals(getClass())) return false;
       Bar other = (Bar)o;
       return Objects.equals(o.id, id) && Objects.equals(o.date, date);
    }

    public int hashCode(){
       return id.hashCode*31 + date.hashCode;
    }    
}

Теперь вы можете просто иметь Map<Bar, Foo>.