Я пытаюсь проверить, что mongodb ведет себя так, как я ожидаю, и заблокировать документ, когда он изменяется в транзакции. я использую

<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-parent</artifactId>
<version>3.3.1</version>

и Spring-данные с набором реплик

mongo 7.0.5 community

Моя цель состоит в том, чтобы сделать следующее:

• Я создаю транзакцию
• Я «блокирую» документ, записывая в него запись
• Другой процесс пытается записать в тот же документ
• Этот второй процесс «записи» будет либо зависать до тех пор, пока транзакция не будет совершена, либо ему будет отказано из-за WriteConflict (я бы предпочел, чтобы он был отклонен)

Я использую этот код:

@Getter
@Setter
class DummyDocument {
    @MongoId
    private String uuid;
    private String lock;
    private List<String> simpleCollection = new ArrayList<>();

    public DummyDocument(String uuid) {
        this.uuid = uuid;
    }
}

public class ConcurrentDocumentAccess {

    @Autowired
    private MongoClient client;

    @Autowired
    private MongoTemplate mongoTemplate;

    @After
    public void cleanup() {
        DummyDocument doc = mongoTemplate.findById("test", DummyDocument.class);
        mongoTemplate.remove(doc);
    }

    @Test
    public void documentLockingTest() {

        // Create a document
        DummyDocument doc = new DummyDocument("test");
        mongoTemplate.save(doc);

        // A query to find the doc
        Query findDoc = new Query().addCriteria(Criteria.where("uuid").is("test"));

        // An update to change the lock value in the doc
        Update lock = new Update().set("lock", "locked");

        // An update of the doc
        Update updateCollection = new Update().addToSet("simpleCollection", "something");

        try (ClientSession session = client.startSession()) {

            session.startTransaction();

            // Acquire lock on doc by writing on it
            UpdateResult lockResult = mongoTemplate.withSession(session).updateFirst(findDoc, lock,
                    DummyDocument.class);
            assertThat(lockResult.getModifiedCount() == 1L).isTrue();

            // Try to update the collection out of the transaction
            UpdateResult changeResult = mongoTemplate.updateFirst(findDoc, updateCollection, DummyDocument.class);
            // assertThat(changeResult.getModifiedCount() == 0L).isTrue();
            
            session.commitTransaction();
        } catch (MongoCommandException e) {
            e.printStackTrace();
        }
        DummyDocument updatedDoc = mongoTemplate.findOne(findDoc, DummyDocument.class);
        assertThat(updatedDoc.getLock()).isEqualTo("locked");
        assertThat(updatedDoc.getSimpleCollection()).doesNotContain("something");
    }
}

Что я наблюдаю, так это то, что запрос вне транзакции принудительно выполняется в документе, затем транзакция зависает до тех пор, пока не произойдет сбой с ответом: {"errorLabels": ["TransientTransactionError"], "ok": 0.0, "errmsg": "Transaction with { txnNumber: 2 } has been aborted.", "code": 251, "codeName": "NoSuchTransaction"

Я что-то делаю неправильно или это ожидаемое поведение?

Обновлять

Что ж, я провел еще один тест, и поведение все еще остается загадочным. Когда я использую сеанс для обновления документа, поведение меняется. Это код, который используется:

    // Create a document
    DummyDocument doc = new DummyDocument("test");
    mongoTemplate.save(doc);

    // A query to find the doc
    Query findDoc = new Query().addCriteria(Criteria.where("uuid").is("test"));
    // findDoc.getQueryObject().toBsonDocument();

    // An update to change the lock value in the doc
    Update lock1 = new Update().set("lock", "lock1");
    Update lock2 = new Update().set("lock", "lock2");

    // An update of the doc
    Update update1 = new Update().addToSet("simpleCollection", "update1");
    Update update2 = new Update().addToSet("simpleCollection", "update2");

    // Create a first session
    ClientSession session1 = client.startSession();
    session1.startTransaction();

    // Create a second session
    ClientSession session2 = client.startSession();
    session2.startTransaction();

    // Acquire lock on doc by writing on it
    try {
        UpdateResult lock1Result = mongoTemplate.withSession(session1).updateFirst(findDoc, lock1,
                DummyDocument.class);
        assertThat(lock1Result.getMatchedCount()).isOne();
        assertThat(lock1Result.getModifiedCount()).isOne();
    } catch (MongoCommandException e) {
        e.printStackTrace();
    } catch (UncategorizedMongoDbException e) {
        e.printStackTrace();
    } catch (Exception e) {
        e.printStackTrace();
    }

    // Try to update the collection out of the transaction
    UpdateResult changeResult = null;
    try {
        changeResult = mongoTemplate.withSession(session2).updateFirst(findDoc, update1, DummyDocument.class);
    } catch (MongoCommandException e) {
        e.printStackTrace();
    } catch (UncategorizedMongoDbException e) {
        // This catch a write error
        e.printStackTrace();
    } catch (Exception e) {
        e.printStackTrace();
    }
    assertThat(changeResult).isNull();

    // Try to commit the change on the collection before 1st session commit
    try {
        session2.commitTransaction();
    } catch (MongoCommandException e) {
        // This catch a "NoSuchTransaction" error
        e.printStackTrace();
    } catch (UncategorizedMongoDbException e) {
        e.printStackTrace();
    } catch (Exception e) {
        e.printStackTrace();
    }

    // Uncomment this to timeout session 1
    // Thread.sleep(120005L);

    // Try to commit the change oof session 1, this should work
    try {
        session1.commitTransaction();
    } catch (MongoCommandException e) {
        e.printStackTrace();
    } catch (UncategorizedMongoDbException e) {
        e.printStackTrace();
    } catch (Exception e) {
        e.printStackTrace();
    }

    session1.close();
    session2.close();

    DummyDocument updatedDoc = mongoTemplate.findOne(findDoc, DummyDocument.class);
    assertThat(updatedDoc.getLock()).isEqualTo("lock1");
    assertThat(updatedDoc.getSimpleCollection()).doesNotContain("update1");

Здесь обновление документа session2 отклонено из-за конфликта записи, чего я и ожидал. Так почему же эти два метода ведут себя по-разному? Могу ли я настроить клиент так, чтобы первая реализация вела себя как вторая?

Вот логи теста:

{"t":{"$date":"2024-07-03T19:36:01.127+00:00"},"s":"I",  "c":"WRITE",    "id":51803,   "ctx":"conn218","msg":"Slow query","attr":{"type":"update","ns":"test.dummyDocument","command":{"q":{"_id":"test"},"u":{"_id":"test","simpleCollection":[],"_class":"com.test.concurrency.DummyDocument"},"multi":false,"upsert":true},"planSummary":"IDHACK","totalOplogSlotDurationMicros":142,"keysExamined":0,"docsExamined":0,"nMatched":0,"nModified":0,"nUpserted":1,"keysInserted":1,"numYields":0,"locks":{"ParallelBatchWriterMode":{"acquireCount":{"r":2}},"FeatureCompatibilityVersion":{"acquireCount":{"w":2}},"ReplicationStateTransition":{"acquireCount":{"w":2}},"Global":{"acquireCount":{"w":2}},"Database":{"acquireCount":{"w":2}},"Collection":{"acquireCount":{"w":2}}},"flowControl":{"acquireCount":1},"readConcern":{"provenance":"implicitDefault"},"storage":{},"cpuNanos":362193,"remote":"172.18.0.1:34942","durationMillis":0}}
{"t":{"$date":"2024-07-03T19:36:01.220+00:00"},"s":"I",  "c":"WRITE",    "id":51803,   "ctx":"conn218","msg":"Slow query","attr":{"type":"update","ns":"test.dummyDocument","command":{"q":{"_id":"test"},"u":{"$set":{"lock":"lock1"}},"multi":false,"upsert":false},"planSummary":"IDHACK","keysExamined":1,"docsExamined":1,"nMatched":1,"nModified":1,"nUpserted":0,"keysInserted":0,"keysDeleted":0,"numYields":0,"locks":{"Database":{"acquireCount":{"w":1}},"Collection":{"acquireCount":{"w":1}}},"flowControl":{"acquireCount":1},"readConcern":{"level":"local","provenance":"implicitDefault"},"storage":{},"cpuNanos":316767,"remote":"172.18.0.1:34942","durationMillis":0}}
{"t":{"$date":"2024-07-03T19:36:01.281+00:00"},"s":"I",  "c":"WRITE",    "id":51803,   "ctx":"conn218","msg":"Slow query","attr":{"type":"update","ns":"test.dummyDocument","command":{"q":{"_id":"test"},"u":{"$addToSet":{"simpleCollection":"update1"}},"multi":false,"upsert":false},"planSummary":"IDHACK","numYields":0,"ok":0,"errMsg":"Caused by :: Write conflict during plan execution and yielding is disabled. :: Please retry your operation or multi-document transaction.","errName":"WriteConflict","errCode":112,"locks":{"Database":{"acquireCount":{"w":1}},"Collection":{"acquireCount":{"w":1}}},"flowControl":{"acquireCount":2},"readConcern":{"level":"local","provenance":"implicitDefault"},"storage":{},"cpuNanos":391742,"remote":"172.18.0.1:34942","durationMillis":0}}