使用 Java 8 toMap 函数将 HashMap 排序到桶中

Question

我想通过附加到每个键的特定 rank 对 HashMap 进行分组。键可以是 rank 1 或 2。所以最后，像 <String, Integer> 这样的给定输入 hashmap 应该以这种形式映射和输出 <Integer, Set<String>> - 其中输出映射的键是 rank，值是原始输入映射中属于该等级的所有字符串的集合。如果输出映射中的集合按原始映射中的值降序排列，那也很好。

我原来是这样实现的，效果很好，但是我想用toMap函数，感觉这个方法有点大材小用。

private Set<String> sortByValue(Map<String, Long> iMap) {
    //This is where we are going to bucket our stuff
    Map<Integer, Set<String>> rankSort = new HashMap<Integer, Set<String>>() {{
        put(1, new LinkedHashSet<>());
        put(2, new LinkedHashSet<>());
    }};

    //Just populating rankSort with elements from the map
    iMap.entrySet().stream()
            .sorted(Collections.reverseOrder(Map.Entry.comparingByValue()))
            .forEachOrdered(e -> rankSort.get(isRank1(e.getKey()) ? 1 : 2).add(e.getKey()));

    return new LinkedHashSet<String>() {{
        addAll(rankSort.get(2));
        addAll(rankSort.get(1));
    }};
}

private boolean isRank1(String val) {
    try {
        return extService.getIt(val).getParent() == null;
    } catch (Exception ex) {
        return false;
    }
}

Answer 1

你要找的不是toMap而是

iMap.entrySet().stream()
    .collect(groupingBy(
        Entry::getValue,
        mapping(Entry::getKey, toSet())));

使用 Java 8 toMap 函数将 HashMap 排序到桶中

Sort HashMap into buckets using Java 8 toMap function

java

mapping

lambda

java-8