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

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

ConcurrentHashMap chm = new ConcurrentHashMap<>();

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

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

我怎样才能使这个线程安全？

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

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

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

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

你是对的。如果Map可以由多个线程修改此方法，则对的第一个调用可能chm.get(key)会返回非null值，而第二个调用可能会返回null（由于Map另一个线程从完成的键中删除了键），因此chm.get(key).doSomething()将抛出一个NullPointerException。

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

ConcurrentHashMap chm = new ConcurrentHashMap();
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不是a ConcurentHashMap，只有一个线程可以访问它，我仍将使用局部变量，因为它比get()两次调用方法更有效。

编辑：

如下所述，此修补程序不会阻止doSomething()不同线程针对同一键/值多次调用此修复程序。这是否是所需的行为尚不清楚。

如果希望防止doSomething()多个线程为同一键/值调用该方法，则可以chm.remove(key)在同一步骤中用于删除键和获取值。

但是，这冒着doSomething()对于某些键/值根本不会执行的风险，因为如果第一次调用doSomething()导致异常，则不会再有doSomething()另一个线程再次调用，因为键/值对将不再在Map。另一方面，如果仅在doSomething()成功执行后从映射中删除键/值对，则可以保证对于从中删除的doSomething()所有键/值对至少成功执行一次Map。