Java 8 groupingby mit benutzerdefinierten Schlüssel

Habe ich eine Reihe von Eingabe-Strings im folgenden format:

typeA:code1,
typeA:code2,
typeA:code3,
typeB:code4,
typeB:code5,
typeB:code6,
typeC:code7,
...

und ich brauche, um eine Map<String, List<String>> mit der folgenden Struktur:

typeA, [code1, code2, code3]
typeB, [code4, code5, code6]
typeC, [code7, code8, ...]

Der Haken ist, dass zum generieren der einzelnen geben, die ich brauche, um eine Funktion aufzurufen, wie in diesem Fall auf jeden input String:

public static String getType(String code)
{
  return code.split(":")[0];  //yes this is horrible code, it's just for the example, honestly
}

Ich bin ziemlich zuversichtlich, dass Bäche und Sammler kann dies tun, aber ich bin kämpfen, um die richtige Beschwörungsformel zu zaubern, um Sie geschehen.

InformationsquelleAutor endian | 2015-08-28
  1. 8

    Hier ist ein Weg, es zu tun (vorausgesetzt die Klasse ist benannt A):

    Map<String, List<String>> result = Stream.of(input)
                          .collect(groupingBy(A::getType, mapping(A::getValue, toList())));

    Wenn Sie möchten, dass die Ausgabe sortiert, Sie können eine TreeMap anstelle der Standard-HashMap:

    .collect(groupingBy(A::getType, TreeMap::new, mapping(A::getValue, toList())));

    Vollständigen Beispiel:

    public static void main(String[] args) {
  String input[] = ("typeA:code1," +
                "typeA:code2," +
                "typeA:code3," +
                "typeB:code4," +
                "typeB:code5," +
                "typeB:code6," +
                "typeC:code7").split(",");

  Map<String, List<String>> result = Stream.of(input)
                    .collect(groupingBy(A::getType, mapping(A::getValue, toList())));
  System.out.println(result);
}

public static String getType(String code) {
  return code.split(":")[0];
}
public static String getValue(String code) {
  return code.split(":")[1];
}
    • Danke, das ist ziemlich nahe an dem, was ich kam mit. Ich erkannte, dass der Typ des ersten Parameters in groupingBy ist eine func, und ich könnte meine getType-Funktion gibt.
    InformationsquelleAutor assylias
  2. 6

    Der code wird einfach, wenn man bedenkt, was Sie ausgelassen haben, dass müssen Sie den zweiten Teil der split-string-sowie:

    Map<String, List<String>> result = Stream.of(input).map(s->s.split(":", 2))
    .collect(groupingBy(a->a[0], mapping(a->a[1], toList())));

    (vorausgesetzt, Sie haben eine import static java.util.stream.Collectors.*;)

    Es ist nichts falsch mit der Aufteilung eines String in ein array, die Umsetzung hat sogar ein "fast-path" für den häufigen Fall, Sie Spalten mit einem einfachen Charakter, anstatt einer komplizierten regulären Ausdruck.

    • Danke, das ist auch ein guter Ansatz.
    InformationsquelleAutor Holger
  3. 2

    Obwohl ich war zu langsam, hier ist ein MCVE zeigen, wie dies gelöst werden kann mit Collectors#groupingBy.

    Gibt es natürlich verschiedene Optionen für die Definition der "classifier" und "mapper". Hier bin ich einfach mit String#substring den Teil vor und nach der ":".

    import static java.util.stream.Collectors.groupingBy;
import static java.util.stream.Collectors.mapping;
import static java.util.stream.Collectors.toList;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;

public class GroupingBySubstringsTest
{
    public static void main(String[] args)
    {
        List<String> strings = new ArrayList<String>();
        strings.add("typeA:code1");
        strings.add("typeA:code2");
        strings.add("typeA:code3");
        strings.add("typeB:code4");
        strings.add("typeB:code5");
        strings.add("typeB:code6");
        strings.add("typeC:code7");

        Map<String, List<String>> result = strings.stream().collect(
            groupingBy(s -> s.substring(0, s.indexOf(":")), 
                mapping(s -> s.substring(s.indexOf(":")+1), toList())));

        for (Entry<String, List<String>> entry : result.entrySet())
        {
            System.out.println(entry);
        }
    }
}
    InformationsquelleAutor Marco13
Schreibe einen Kommentar