java8 新特性 stream，set 转 map<String, Set<>> 问题

Question

class User{
    String type;
    String userId;
    ...
}

// 现有 Set<User> users; 需要将相同 type 转成 Map<Stirng, Set<User>>
// 需要怎样用 stream
// 使用的 for 循环

Map<String, Set<User>> map = new HashMap<>();
for (User u : users) {
    if (!map.containsKey(u.getType()))
        map.put(u.getType(), new HashSet<User>());
    map.get(u.getType()).add(u);
}

// 使用 stream 怎样实现相同同能

Answer 1

一楼的回答基本正确，纠正一下，type是String类型，正确的返回值应该是这样

Map<String, List<User>> collect = users.stream().collect(Collectors.groupingBy(User::getType));