Gruppieren von GroupBy Sammlern manuell

Question

Ich möchte eine Liste von Personen gruppieren. Eine Person hat einige Attribute wie Namen, Land, Stadt, Postleitzahl, etc. Ich den statischen Code geschrieben, die sehr gut funktioniert:

Object groupedData = data.stream().collect(groupingBy(Person::getName, Collectors.groupingBy(Person::getCountry, Collectors.groupingBy(Person::getTown)))); 

Aber das Problem ist, das ist es nicht dynamisch. Manchmal möchte ich nur nach Namen und Ort gruppieren, manchmal nach Attributen. Wie kann ich das machen? Non Java 8-Lösungen sind ebenfalls willkommen.

Answer 1

Sie könnten eine Funktion nimmt eine beliebige Anzahl von Attributen zu einer Gruppe von und konstruieren schaffen die groupingBy - Collector in einer Schleife, jedesmal, wenn die vorherige Version von sich selbst als Kollektor downstream vorbei.

public static <T> Map collectMany(List<T> data, Function<T, ?>... groupers) { 
    Iterator<Function<T, ?>> iter = Arrays.asList(groupers).iterator(); 
    Collector collector = Collectors.groupingBy(iter.next()); 
    while (iter.hasNext()) { 
     collector = Collectors.groupingBy(iter.next(), collector); 
    } 
    return (Map) data.stream().collect(collector); 
} 

Beachten Sie, dass die Reihenfolge der grouper Funktionen umgekehrt wird, so dass Sie sie in umgekehrter Reihenfolge durchlaufen müssen (oder sie innerhalb der Funktion umkehren, zum Beispiel Collections.reverse oder Guava der mit Lists.reverse, je nachdem, was Sie bevorzugen).

Object groupedData = collectMany(data, Person::getTown, Person::getCountry, Person::getName); 

Oder so, eine Old-School for Schleife mit dem Array in der Funktion rückgängig zu machen, das heißt, Sie müssen nicht die Barsche in umgekehrter Reihenfolge passieren (aber IMHO ist dies schwieriger zu verstehen):

public static <T> Map collectMany(List<T> data, Function<T, ?>... groupers) { 
    Collector collector = Collectors.groupingBy(groupers[groupers.length-1]); 
    for (int i = groupers.length - 2; i >= 0; i--) { 
     collector = Collectors.groupingBy(groupers[i], collector); 
    } 
    return (Map) data.stream().collect(collector); 
} 

Beide Ansätze geben eine Map<?,Map<?,Map<?,T>>> zurück, genau wie in Ihrem ursprünglichen Code. Abhängig davon, was Sie mit diesen Daten machen möchten, könnte es auch sinnvoll sein, eine Map<List<?>,T> zu verwenden, wie von Tunaki vorgeschlagen.

Answer 2

Sie können nach einer Liste gruppieren, die aus den Attributen besteht, die Sie gruppieren möchten.

Stellen Sie sich vor, Sie möchten nach dem Namen und dem Land gruppieren. Dann könnten Sie verwenden:

Map<List<Object>, List<Person>> groupedData = 
    data.stream().collect(groupingBy(p -> Arrays.asList(p.getName(), p.getCountry()))); 

Dies funktioniert, weil zwei Listen gleich betrachtet werden, wenn sie das gleiche Element in der gleichen Reihenfolge enthalten. Daher haben Sie in der resultierenden Karte einen Schlüssel für jedes unterschiedliche Name/Land-Paar und als Wert die Liste der Personen mit diesem spezifischen Namen und Land. Anders gesagt, anstatt "Gruppe nach Name, dann Gruppe nach Land" zu sagen, heißt es effektiv "Gruppe nach Name und Land". Der Vorteil ist, dass Sie nicht mit einer Karte von Karten von Karten etc. enden.