使用流来添加多个哈希映射的内容

Question

我有一个HashMaps矩阵,我试图根据给定的输入添加那些HashMaps的内容,这将匹配"外部"HashMap的键.拿下面的SSCCE(有1个错误说明我的进度),有4个HashMaps的HashMap行,我想在第1行和第3行中添加HashMaps的内容.

当然,我可以用旧的Java 7方式来做这件事.但我想用Java 8流做这个,而不必迭代.

public class HashMapCollector {
    public static void main(String[] args) {
        HashMap<Integer, HashMap<Integer, BigDecimal>> map = new HashMap<>();
        map.put(1, new LinkedHashMap<>()); // row 1
        map.put(2, new LinkedHashMap<>()); // row 2
        map.put(3, new LinkedHashMap<>()); // row 3
        map.put(4, new LinkedHashMap<>()); // row 4

        // row 1
        map.get(1).merge(1, BigDecimal.TEN, BigDecimal::add);
        map.get(1).merge(2, BigDecimal.TEN, BigDecimal::add);
        map.get(1).merge(3, BigDecimal.TEN, BigDecimal::add);
        map.get(1).merge(4, BigDecimal.TEN, BigDecimal::add);

        // row 2
        map.get(2).merge(1, BigDecimal.TEN, BigDecimal::add);
        map.get(2).merge(2, BigDecimal.TEN, BigDecimal::add);
        map.get(2).merge(3, BigDecimal.TEN, BigDecimal::add);
        map.get(2).merge(4, BigDecimal.TEN, BigDecimal::add);

        // row 3
        map.get(3).merge(1, BigDecimal.TEN, BigDecimal::add);
        map.get(3).merge(2, BigDecimal.TEN, BigDecimal::add);
        map.get(3).merge(3, BigDecimal.TEN, BigDecimal::add);
        map.get(3).merge(4, BigDecimal.TEN, BigDecimal::add);

        // row 4
        map.get(4).merge(1, BigDecimal.TEN, BigDecimal::add);
        map.get(4).merge(2, BigDecimal.TEN, BigDecimal::add);
        map.get(4).merge(3, BigDecimal.TEN, BigDecimal::add);
        map.get(4).merge(4, BigDecimal.TEN, BigDecimal::add);

        // Using a HashSet to contain the keys of the rows I want
        HashSet<Integer> keys_to_add = new HashSet<>();
        keys_to_add.add(1);
        keys_to_add.add(3);

        // Java 7 boring vanilla iteration
        BigDecimal total = BigDecimal.ZERO;
        for(Entry<Integer, HashMap<Integer, BigDecimal>> entry : map.entrySet()) {
            if(keys_to_add.contains(entry.getKey())) {
                for(BigDecimal bd : entry.getValue().values()) {
                    total = total.add(bd);
                }
            }
        }

        System.out.println(total);

        // Java 8ish method -- still some boring iteration
        total = BigDecimal.ZERO;
        for(Entry<Integer, HashMap<Integer, BigDecimal>> entry : map.entrySet()) {
            if(keys_to_add.contains(entry.getKey())) {
                total = total.add(entry.getValue().values().stream().reduce(BigDecimal.ZERO, BigDecimal::add));
            }
        }

        System.out.println(total);


        // Java 8 method -- no iteration
        total = BigDecimal.ZERO;
        map.entrySet().stream().filter(e -> keys_to_add.contains(e.getKey())).map(Map.Entry::getValue). // This is as far as I got

        System.out.println(total);
    }
}

Answer 1

我认为开始时更有意义,keys_to_add而不是遍历所有条目并检查密钥是否在keys_to_add.

BigDecimal total = keys_to_add.stream()
        .filter(map::containsKey)
        .map(map::get)
        .map(Map::values)
        .flatMap(Collection::stream)
        .reduce(BigDecimal::add)
        .orElse(BigDecimal.ZERO);

或者,如果您更喜欢保留垂直空间,

BigDecimal total = keys_to_add.stream()
        .filter(map::containsKey)
        .flatMap(k -> map.get(k).values().stream())
        .reduce(BigDecimal.ZERO, BigDecimal::add);