2016-04-25 10 views
2

Ich versuche, nach einer Sammlung nach einem Wert zu gruppieren, der in meinem Objekt als Liste angezeigt wird.Java Stream - Gruppierung nach Schlüssel in einer Liste

Dies ist das Modell, das ich

public class Student { 
    String stud_id; 
    String stud_name; 
    List<String> stud_location = new ArrayList<>(); 

    public Student(String stud_id, String stud_name, String... stud_location) { 
     this.stud_id = stud_id; 
     this.stud_name = stud_name; 
     this.stud_location.addAll(Arrays.asList(stud_location)); 
    } 
} 

haben, wenn ich sie mit folgendem initialisieren:

California -> Student(1726),Student(4321),Student(2234),Student(7765) 
New York -> Student(1726),Student(5223),Student(3442) 
Los Angeles => Student(2234) 

Ich versuche:

List<Student> studlist = new ArrayList<Student>(); 
    studlist.add(new Student("1726", "John", "New York","California")); 
    studlist.add(new Student("4321", "Max", "California")); 
    studlist.add(new Student("2234", "Andrew", "Los Angeles","California")); 
    studlist.add(new Student("5223", "Michael", "New York")); 
    studlist.add(new Student("7765", "Sam", "California")); 
    studlist.add(new Student("3442", "Mark", "New York")); 

ich folgendes erhalten möchten schreibe folgendes

Map<Student, List<String>> x = studlist.stream() 
      .flatMap(student -> student.getStud_location().stream().map(loc -> new Tuple(loc, student))) 
      .collect(Collectors.groupingBy(y->y.getLocation(), mapping(Entry::getValue, toList()))); 

Aber ich habe Schwierigkeiten, es zu vervollständigen - wie behalte ich den ursprünglichen Studenten nach dem Mapping?

+4

Es sollte 'Map >' sein. Außerdem haben Sie Ihre 'Tuple'-Klasse nicht angezeigt, aber ich vermute, dass' Entry :: getValue' nicht das ist, was Sie wollen. Ich würde für y -> y.getStudent() 'gehen. Und es funktioniert :). – Tunaki

Antwort

1

die Kommentare oben zusammenfassend, würde die gesammelte Weisheit vorschlagen:

Map<String, List<Student>> x = studlist.stream() 
      .flatMap(student -> student.getStud_location().stream().map(loc -> new AbstractMap.SimpleEntry<>(loc, student))) 
      .collect(Collectors.groupingBy(Map.Entry::getKey, Collectors.mapping(Map.Entry::getValue, toList()))); 

Als alternativen Ansatz, wenn Sie diesen Ort enthalten, nicht die Studenten in jeder Liste kümmern, könnten Sie die Studenten Abflachung betrachten Liste Studierende mit nur einem Ort:

Map<String, List<Student>> x = studlist.stream() 
     .flatMap(student -> 
       student.stud_location.stream().map(loc -> 
         new Student(student.stud_id, student.stud_name, loc)) 
     ).collect(Collectors.groupingBy(student -> student.stud_location.get(0)));