Java 8映射计算,设置为值

时间:2017-04-02 00:26:06

标签: java-8

我的地图看起来像这样

private Map<String, LinkedHashSet<String>> map = new HashMap<>();

在传统方法中,我可以使用键检查为地图添加值,如下所示

public void addEdge(String node1, String node2) {
        LinkedHashSet<String> adjacent = map.get(node1);
        if (adjacent == null) {
            adjacent = new LinkedHashSet();
            map.put(node1, adjacent);
        }
        adjacent.add(node2);
    }

使用java 8,我可以做这样的事情,我也可以得到相同的输出。

map.compute(node1, (k,v)-> {
            if(v==null) {
                v=new LinkedHashSet<>();
            }
            v.add(node2);
            return v;
            });

有没有更好的方法来处理java 8?

1 个答案:

答案 0 :(得分:5)

使用

map.computeIfAbsent(node1, k -> new LinkedHashSet<>()).add(node2);

如果地图中已找到node1,则相当于:

map.get(node1).add(node2);

如果地图中尚未显示node1,则相当于:

map.put(node1, new LinkedHashSet<>()).add(node2);

这正是您正在寻找的内容,甚至被描述为documentation中的用例。