我有一个 List<Person>
对象,想将其转换为 Map<Integer, List<Person>>
,其中映射的键表示人员属性等级。源列表中可能有多个具有相同等级的 Person 对象,在这种情况下,我希望将它们全部分组到相应等级的 List 中,并在生成的 Map 中与之对应。
到目前为止,我有以下代码:
public class PersonMain
{
public static void main(String[] args)
{
Person person1 = new Person();
person1.setName("person1");
person1.setGrade(1);
Person person2 = new Person();
person2.setName("person2");
person2.setGrade(2);
Person person3 = new Person();
person3.setName("person3");
person3.setGrade(1);
List<Person> persons = Arrays.asList(person1, person2, person3);
Map<Integer, List<Person>> personsMap = persons.stream()
.collect(Collectors.toMap(Person::getGrade, PersonMain::getPerson, PersonMain::combinePerson));
System.out.println(personsMap);
}
private static List<Person> getPerson(Person p)
{
List<Person> persons = new ArrayList<>();
persons.add(p);
return persons;
}
private static List<Person> combinePerson(List<Person> oldVal, List<Person> newVal)
{
oldVal.addAll(newVal);
return oldVal;
}
}
有没有更好的实现方式?