MongoDB（80）如何在MongoDB中使用多文档事务？

在MongoDB中使用多文档事务可以确保多个操作要么全部成功，要么全部失败，从而确保数据的一致性。MongoDB的多文档事务类似于传统关系型数据库的事务，支持ACID（原子性、一致性、隔离性和持久性）特性。以下是详细的步骤和Java代码示例，展示如何在MongoDB中使用多文档事务。

前提条件

复制集：事务需要在MongoDB复制集（Replica Set）或分片集群（Sharded Cluster）中运行。
MongoDB版本：确保使用MongoDB 4.0或更高版本。

1. 引入Maven依赖

首先，在 pom.xml 中引入 MongoDB Java 驱动依赖：

xml 复制代码

<dependency>
  <groupId>org.mongodb</groupId>
  <artifactId>mongodb-driver-sync</artifactId>
  <version>4.4.0</version>
</dependency>

2. Java代码示例

以下是一个完整的Java代码示例，展示如何在MongoDB中使用多文档事务：

java 复制代码

import com.mongodb.MongoClientSettings;
import com.mongodb.MongoException;
import com.mongodb.ReadConcern;
import com.mongodb.ReadPreference;
import com.mongodb.TransactionOptions;
import com.mongodb.WriteConcern;
import com.mongodb.client.ClientSession;
import com.mongodb.client.MongoClient;
import com.mongodb.client.MongoClients;
import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoDatabase;
import org.bson.Document;

public class MongoDBTransactionExample {

    public static void main(String[] args) {

        // 创建 MongoClient
        MongoClient mongoClient = MongoClients.create("mongodb://localhost:27017");

        // 获取数据库和集合
        MongoDatabase database = mongoClient.getDatabase("testdb");
        MongoCollection<Document> collection1 = database.getCollection("collection1");
        MongoCollection<Document> collection2 = database.getCollection("collection2");

        // 插入一些初始数据
        collection1.insertOne(new Document("account", "Alice").append("balance", 100));
        collection2.insertOne(new Document("account", "Bob").append("balance", 0));

        // 开始会话
        ClientSession session = mongoClient.startSession();

        // 配置事务选项
        TransactionOptions txnOptions = TransactionOptions.builder()
                .readPreference(ReadPreference.primary())
                .readConcern(ReadConcern.LOCAL)
                .writeConcern(WriteConcern.MAJORITY)
                .build();

        try {
            // 开始事务
            session.startTransaction(txnOptions);

            // 执行跨集合的写操作
            Document aliceAccount = collection1.find(session, new Document("account", "Alice")).first();
            if (aliceAccount != null && aliceAccount.getInteger("balance") >= 50) {
                collection1.updateOne(session, new Document("account", "Alice"),
                        new Document("$inc", new Document("balance", -50)));
                collection2.updateOne(session, new Document("account", "Bob"),
                        new Document("$inc", new Document("balance", 50)));
            } else {
                throw new RuntimeException("Insufficient balance!");
            }

            // 提交事务
            session.commitTransaction();
            System.out.println("Transaction committed.");
        } catch (MongoException | RuntimeException e) {
            // 回滚事务
            session.abortTransaction();
            System.err.println("Transaction aborted due to: " + e.getMessage());
        } finally {
            session.close();
        }

        mongoClient.close();
    }
}

解释代码

创建 MongoClient：连接到 MongoDB 实例。

java 复制代码

MongoClient mongoClient = MongoClients.create("mongodb://localhost:27017");

获取数据库和集合：获取所需的数据库和集合引用。

java 复制代码

MongoDatabase database = mongoClient.getDatabase("testdb");
MongoCollection<Document> collection1 = database.getCollection("collection1");
MongoCollection<Document> collection2 = database.getCollection("collection2");

插入初始数据：向集合中插入一些示例数据。

java 复制代码

collection1.insertOne(new Document("account", "Alice").append("balance", 100));
collection2.insertOne(new Document("account", "Bob").append("balance", 0));

开始会话 ：通过 mongoClient.startSession() 开始一个会话。

java 复制代码

ClientSession session = mongoClient.startSession();

配置事务选项 ：使用 TransactionOptions 配置事务的选项，比如读偏好、读关注和写关注。

java 复制代码

TransactionOptions txnOptions = TransactionOptions.builder()
        .readPreference(ReadPreference.primary())
        .readConcern(ReadConcern.LOCAL)
        .writeConcern(WriteConcern.MAJORITY)
        .build();

开始事务 ：通过 session.startTransaction(txnOptions) 开始事务。

java 复制代码

session.startTransaction(txnOptions);

执行跨集合的写操作：在事务中执行多个集合的写操作。

java 复制代码

Document aliceAccount = collection1.find(session, new Document("account", "Alice")).first();
if (aliceAccount != null && aliceAccount.getInteger("balance") >= 50) {
    collection1.updateOne(session, new Document("account", "Alice"),
            new Document("$inc", new Document("balance", -50)));
    collection2.updateOne(session, new Document("account", "Bob"),
            new Document("$inc", new Document("balance", 50)));
} else {
    throw new RuntimeException("Insufficient balance!");
}

提交事务 ：如果所有操作成功，通过 session.commitTransaction() 提交事务。

java 复制代码

session.commitTransaction();
System.out.println("Transaction committed.");

回滚事务 ：如果发生异常，通过 session.abortTransaction() 回滚事务。

java 复制代码

catch (MongoException | RuntimeException e) {
    session.abortTransaction();
    System.err.println("Transaction aborted due to: " + e.getMessage());
}

关闭会话和客户端：最后关闭会话和 MongoDB 客户端连接。

java 复制代码

finally {
    session.close();
}

mongoClient.close();

总结

通过上述代码示例，展示了如何在MongoDB中使用多文档事务来实现ACID特性。合理利用这些特性，可以在复杂应用场景中确保数据的一致性和可靠性。通过事务，可以保证在多个集合中的操作具有原子性、一致性、隔离性和持久性，从而确保数据的可靠性和一致性。