问题描述
因此,我有一个包含用户信息的用户列表,包括公司uid,该值可以在多个用户中找到,因为它定义了该用户所属的公司。我需要创建一个userList映射,其中映射键是公司uid。我已经使用以下代码做到了这一点:
HashMap<String,List<LDAPUser>> allUsers = new HashMap<>();
userService.findAllUsers().forEach(u -> {
Optional.ofNullable(allUsers.putIfAbsent(u.getCompanyUid(),new ArrayList<>(Collections.singletonList(u))))
.ifPresent(list -> list.add(u));
});
即使它工作正常,我认为必须使用一种 flatMap,映射或收集方法从流中使用更干净的方法,但是我基本上无法使它工作,因为我看不到如何创建包含所有用户的列表。
解决方法
有一个收集器来对事物进行分组,请考虑以下示例
Application.java
import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
public class Application {
private static class User {
private final String name;
private final Integer companyId;
public User(String name,Integer companyId) {
this.name = name;
this.companyId = companyId;
}
public String getName() {
return name;
}
public Integer getCompanyId() {
return companyId;
}
@Override
public String toString() {
return "User{" +
"name='" + name + '\'' +
",companyId=" + companyId +
'}';
}
}
public static void main(String[] args) {
final List<User> users = Arrays.asList(new User("A",1),new User("B",new User("C",2));
final Map<Integer,List<User>> byCompanyId = users.stream()
.collect(Collectors.groupingBy(User::getCompanyId));
System.out.println(byCompanyId);
}
}
将要打印
{1=[User{name='A',companyId=1},User{name='B',companyId=1}],2=[User{name='C',companyId=2}]}