在HashMap中添加到列表的快捷方式

Question

我经常需要获取对象列表，然后根据对象中包含的值将它们分组为Map。例如。列出用户和国家（地区）分组。

我的代码通常如下所示：

Map<String, List<User>> usersByCountry = new HashMap<String, List<User>>();
for(User user : listOfUsers) {
    if(usersByCountry.containsKey(user.getCountry())) {
        //Add to existing list
        usersByCountry.get(user.getCountry()).add(user);

    } else {
        //Create new list
        List<User> users = new ArrayList<User>(1);
        users.add(user);
        usersByCountry.put(user.getCountry(), users);
    }
}

但是，我不禁会认为这很尴尬，某些专家有更好的方法。到目前为止，我能看到的最接近的是Google Collections的MultiMap。

有没有标准的方法？

谢谢！

Answer 1

在Java
8中，您可以使用Map#computeIfAbsent()。

Map<String, List<User>> usersByCountry = new HashMap<>();

for (User user : listOfUsers) {
    usersByCountry.computeIfAbsent(user.getCountry(), k -> new ArrayList<>()).add(user);
}

或者，请使用流API的是Collectors#groupingBy()从去List到Map直接：

Map<String, List<User>> usersByCountry = listOfUsers.stream().collect(Collectors.groupingBy(User::getCountry));

在Java 7或更低版​​本中，最好的获得如下：

Map<String, List<User>> usersByCountry = new HashMap<>();

for (User user : listOfUsers) {
    List<User> users = usersByCountry.get(user.getCountry());
    if (users == null) {
        users = new ArrayList<>();
        usersByCountry.put(user.getCountry(), users);
    }
    users.add(user);
}

Commons Collections有一个LazyMap，但没有参数化。番石榴没有LazyMap或的种类LazyList，但是您可以使用Multimap它，如下面的的答案