关于 MongoDB 实现计数器时并发计数错误的问题

[DovaKeen]

我用 MongoDB 的 upsert 和$inc 实现了一个简单的自增计数器，但是却发现当对一个键并发自增计数时，会丢失数据，让人百思不得其解啊

我的 MongoDB 进程是在本机运行的，计数器主要代码如下： 计数器的 incr 实现：

public boolean incr(String key) {

    checkKey(key);
    
    UpdateResult result = mongoTemplate.upsert(query(Criteria.where("key").is(key)),
            new Update().inc("value", 1),
            MongoCounter.class, COLLECTION_NAME);
    return result.wasAcknowledged();

}

测试代码：

    int count = 200;
    ExecutorService executor = Executors.newCachedThreadPool();
    String key = "incrkey";
    for (int i = 0; i < count; i++) {
        executor.submit(() -> counter.incr(key));
    }
    assertTrue(counter.getCounter(key) == count);
    executor.shutdown();
    counter.delCounter(key);

在测试时，开了一个线程池，用多个线程同时向一个初始没有值的键连续调用 n 次 incr，但是结果发现总是会丢失几次更新，难道原子性更新的前提是并发量不能太大吗？

请懂 MongoDB 的人指点一二啊

[xmh51]

不应该立马 executor.shutdown();
你等等 java 执行完异步线程再关闭.........

[DovaKeen]

@xmh51 executor.shutdown();执行之后，已经提交过的任务还是会跑完才关闭的，不会直接关闭线程池呀，而且我注释掉了这一句后，还是会丢失更新，，

[xmh51]

@DovaKeen 用错了方法你 This method does not wait for previously submitted tasks to
* complete execution 该方法会立即返回, 但不一定代表之前提交的任务已经全部完成了. 如果需要一个阻塞的方法, 可以调用 awaitTermination 方法

[DovaKeen]

@xmh51 确实是这里的问题，并且还发现当多个线程同时对一个没有值的键调用 $inc 时，可能会产生多个值为 1 而键名相同的文档，这里也是会导致计数器结果错误的地方，现在给键名加上 unique 约束，并且在调用后捕获异常重设值，就可以得到正确的结果了。谢谢你呀