Question

在一次采访中，我被要求检查以下代码是否按预期工作。

ConcurrentHashMap<Integer, Integer> chm = new ConcurrentHashMap<>();

if (chm.get(key) != null) {
    chm.get(key).doSomething();
    chm.remove(key);
}

根据JavaDocs，get返回上次完成的更新操作的值。因此，如果线程1已经调用chm.remove(key)并且如果线程2进入if语句并且即将调用get方法，那么我们可能会得到一个异常。这是对的吗？

如何使这个线程安全？

Answer 1

Map.remove(key)如果已删除，则返回该值。在许多情况下，这是一个非常好的技巧，包括你的：

Object value = chm.remove(key)
if(value != null)
{
     value.doSomething();
}

你不能安全地使用get然后删除，因为如果两个线程同时调用你的方法，那么他们总是有两次或多次调用doSomething的风险，之前密钥已被删除。

如果先将其删除，则无法进行此操作。上面的代码是Threadsafe，也更简单。

Answer 2

你是对的。如果多个线程可以修改此Map，则对chm.get(key)的第一次调用将返回非空值，第二次调用将返回null（由于删除了来自Map由另一个线程完成的密钥），因此chm.get(key).doSomething()将抛出NullPointerException。

您可以使用局部变量存储chm.get(key)的结果来使此代码线程安全：

ConcurrentHashMap<Integer, Integer> chm = new ConcurrentHashMap<Integer, Integer>();
Integer value = chm.get(key);

if(value != null) {
    value.doSomething(); // P.S. Integer class doesn't have a doSomething() method
                         // but I guess this is just an example of calling some arbitrary 
                         // instance method
    chm.remove(key);
}

顺便说一句，即使那个Map不是ConcurentHashMap并且只有一个线程可以访问它，我仍然会使用局部变量，因为它比调用{{1}更有效方法两次。

编辑：

如下所述，此修复程序不会阻止get()被不同的线程多次调用相同的键/值。这是否是所期望的行为并不清楚。

如果您希望防止多个线程为同一个键/值调用doSomething()的可能性，您可以使用doSomething()删除密钥并在同一步骤获取值。

然而，这会冒一些键/值根本不会执行chm.remove(key)的风险，因为如果第一次调用doSomething()导致异常，则不会再调用另一个帖子doSomething()，因为键/值对将不再位于doSomething()中。另一方面，如果仅在Map成功执行后从地图中删除键/值对，则保证doSomething()成功执行至少一次所有已重新修改的键/值对来自doSomething()。

如何执行线程安全获取然后使用ConcurrentHashMap删除？

2 个答案: