我有以下兩類:的Java 8流映射分組操作
Person
:
public class Person {
private final Long id;
private final String address;
private final String phone;
public Person(Long id, String address, String phone) {
this.id = id;
this.address = address;
this.phone = phone;
}
public Long getId() {
return id;
}
public String getAddress() {
return address;
}
public String getPhone() {
return phone;
}
@Override
public String toString() {
return "Person [id=" + id + ", address=" + address + ", phone=" + phone + "]";
}
}
CollectivePerson
:
import java.util.HashSet;
import java.util.Set;
public class CollectivePerson {
private final Long id;
private final Set<String> addresses;
private final Set<String> phones;
public CollectivePerson(Long id) {
this.id = id;
this.addresses = new HashSet<>();
this.phones = new HashSet<>();
}
public Long getId() {
return id;
}
public Set<String> getAddresses() {
return addresses;
}
public Set<String> getPhones() {
return phones;
}
@Override
public String toString() {
return "CollectivePerson [id=" + id + ", addresses=" + addresses + ", phones=" + phones + "]";
}
}
我想有流操作,以便:
- The
Person
映射到CollectivePerson
- 的
address
和Person
phone
在CollectivePerson
分別合併成addresses
和phones
對於具有相同id
所有Person
的I寫了下面的代碼段用於此目的:
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.stream.Collectors;
public class Main {
public static void main(String[] args) {
Person person1 = new Person(1L, "Address 1", "Phone 1");
Person person2 = new Person(2L, "Address 2", "Phone 2");
Person person3 = new Person(3L, "Address 3", "Phone 3");
Person person11 = new Person(1L, "Address 4", "Phone 4");
Person person21 = new Person(2L, "Address 5", "Phone 5");
Person person22 = new Person(2L, "Address 6", "Phone 6");
List<Person> persons = new ArrayList<>();
persons.add(person1);
persons.add(person11);
persons.add(person2);
persons.add(person21);
persons.add(person22);
persons.add(person3);
Map<Long, CollectivePerson> map = new HashMap<>();
List<CollectivePerson> collectivePersons = persons.stream()
.map((Person person) -> {
CollectivePerson collectivePerson = map.get(person.getId());
if (Objects.isNull(collectivePerson)) {
collectivePerson = new CollectivePerson(person.getId());
map.put(person.getId(), collectivePerson);
collectivePerson.getAddresses().add(person.getAddress());
collectivePerson.getPhones().add(person.getPhone());
return collectivePerson;
} else {
collectivePerson.getAddresses().add(person.getAddress());
collectivePerson.getPhones().add(person.getPhone());
return null;
}
})
.filter(Objects::nonNull)
.collect(Collectors.<CollectivePerson>toList());
collectivePersons.forEach(System.out::println);
}
}
它做的工作和輸出爲:
CollectivePerson [id=1, addresses=[Address 1, Address 4], phones=[Phone 1, Phone 4]]
CollectivePerson [id=2, addresses=[Address 2, Address 6, Address 5], phones=[Phone 5, Phone 2, Phone 6]]
CollectivePerson [id=3, addresses=[Address 3], phones=[Phone 3]]
但我相信有可能是一個更好的辦法,分組來實現相同的流路。任何指針都會很棒。
謝謝你,我採取了你的第二個變種,它非常快。在約10秒鐘內操作~1000000個'人員'。 –