从嵌套的 HashMap 中收集

Question

考虑到 NullPointerException，我的以下代码有一点问题：

@Override
public List<FlowEdge<V>> getEdges() {
    List<FlowEdge<V>> edges = new ArrayList<FlowEdge<V>>();
    Collection<FlowEdge<V>> collection = new ArrayList<>();
    collection = graph.get(getNodes()).values();
    edges.addAll(collection);
    return edges;
}

我知道这可以写得更短一些，但我想稍微拆分一下代码以找出我在哪里得到 NullPointerException。所以问题出在“collection =...”这一行。

Graph 是一个嵌套的 HashMap：

Map<V, Map<V, FlowEdge<V>> graph = new HashMap<>();

和 getNodes() 是在同一类中返回集合的方法。我已经测试过 getNodes() 是否提供了一个空集，但事实并非如此。

@Override
public Set<V> getNodes() {
    Set<V> set = graph.keySet();
    return set; 
}

提前谢谢你。

Answer 1

graph 映射的键是V 类型，因此将Set<V> 传递给Map 的get() 方法将始终返回null。

如果要获取内部Maps的所有FlowEdge<V>值，可以这样写：

List<FlowEdge<V>> collection =
    graph.values() // a Collection<Map<V, FlowEdge<V>>
         .stream() // a Stream<Map<V, FlowEdge<V>>
         .flatMap(m -> m.values().stream()) // a Stream<FlowEdge<V>>
         .collect(Collectors.toList()); // a List<FlowEdge<V>>

编辑：

请注意，上面的 sn-p 是一条语句。为了可读性，它只分为多行。

同样可以写在一行代码中：

List<FlowEdge<V>> collection = graph.values().stream().flatMap(m -> m.values().stream()).collect(Collectors.toList());