MongoDB Java开发：如何高效处理和防止重复文档插入

import com.mongodb.client.MongoCollection;
import com.mongodb.client.model.Indexes;
import com.mongodb.MongoWriteException;
import org.bson.Document;

public class DocumentService {

    private final MongoCollection<Document> collection;

    public DocumentService(MongoCollection<Document> collection) {
        this.collection = collection;
        // 确保集合上存在唯一索引
        createUniqueIndex();
    }

    private void createUniqueIndex() {
        try {
            // 为 name, supplier, food, country of origin 字段创建复合唯一索引
            // 确保这些字段的组合是唯一的
            collection.createIndex(
                Indexes.compoundIndex(
                    Indexes.ascending("name"),
                    Indexes.ascending("supplier"),
                    Indexes.ascending("food"),
                    Indexes.ascending("country of origin")
                ),
                new com.mongodb.client.model.IndexOptions().unique(true)
            );
            System.out.println("复合唯一索引创建成功或已存在。");
        } catch (MongoWriteException e) {
            // 如果索引已存在，MongoDB会抛出异常，但通常可以忽略
            // 除非是索引定义冲突等更严重的问题
            if (e.getError().getCode() == 85) { // 85 is the code for IndexAlreadyExists
                System.out.println("复合唯一索引已存在，无需重复创建。");
            } else {
                System.err.println("创建复合唯一索引时发生错误: " + e.getMessage());
                throw new RuntimeException("索引创建失败", e);
            }
        }
    }

    // ... 其他方法
}

import com.mongodb.client.MongoCollection;
import com.mongodb.client.result.InsertOneResult;
import com.mongodb.MongoWriteException;
import com.mongodb.ErrorCategory;
import org.bson.Document;

public class DocumentService {

    private final MongoCollection<Document> collection;

    // 构造函数和 createUniqueIndex 方法如前所示

    /**
     * 尝试插入一个新文档。如果文档的唯一键组合已存在，则抛出自定义异常。
     * @param document 要插入的文档
     * @throws DuplicateDocumentException 如果文档的唯一键组合已存在
     */
    public void insertNewDocument(Document document) throws DuplicateDocumentException {
        try {
            InsertOneResult result = collection.insertOne(document);
            if (result.wasAcknowledged()) {
                System.out.println("文档插入成功，_id: " + result.getInsertedId());
            }
        } catch (MongoWriteException e) {
            // 检查错误类别是否为 DUPLICATE_KEY
            if (e.getError().getCategory() == ErrorCategory.DUPLICATE_KEY) {
                System.err.println("[Error] 尝试插入重复文档: " + document);
                throw new DuplicateDocumentException("文档已存在，无法插入重复记录。", e);
            } else {
                // 处理其他写入错误
                System.err.println("文档插入失败，发生MongoDB写入错误: " + e.getMessage());
                throw new RuntimeException("文档插入失败", e);
            }
        } catch (Exception e) {
            // 捕获其他潜在异常
            System.err.println("文档插入过程中发生未知错误: " + e.getMessage());
            throw new RuntimeException("文档插入失败", e);
        }
    }

    // 自定义异常类
    public static class DuplicateDocumentException extends Exception {
        public DuplicateDocumentException(String message) {
            super(message);
        }
        public DuplicateDocumentException(String message, Throwable cause) {
            super(message, cause);
        }
    }

    public static void main(String[] args) {
        // 假设已经初始化了 MongoClient 和 MongoDatabase
        // MongoClient mongoClient = MongoClients.create("mongodb://localhost:27017");
        // MongoDatabase database = mongoClient.getDatabase("your_database_name");
        // MongoCollection<Document> myCollection = database.getCollection("your_collection_name");

        // 示例：初始化 DocumentService
        // DocumentService service = new DocumentService(myCollection);

        // 示例文档
        Document doc1 = new Document()
                .append("name", "Apple")
                .append("supplier", "FruitCorp")
                .append("food", "Fruit")
                .append("country of origin", "USA");

        Document doc2 = new Document() // 这是一个重复的文档
                .append("name", "Apple")
                .append("supplier", "FruitCorp")
                .append("food", "Fruit")
                .append("country of origin", "USA");

        Document doc3 = new Document() // 这是一个新的文档
                .append("name", "Banana")
                .append("supplier", "TropicalFruits")
                .append("food", "Fruit")
                .append("country of origin", "Ecuador");

        // 模拟插入操作
        try {
            // service.insertNewDocument(doc1); // 第一次插入成功
        } catch (DuplicateDocumentException e) {
            System.out.println(e.getMessage());
        }

        try {
            // service.insertNewDocument(doc2); // 第二次插入，会抛出 DuplicateDocumentException
        } catch (DuplicateDocumentException e) {
            System.out.println(e.getMessage()); // 输出：文档已存在，无法插入重复记录。
        }

        try {
            // service.insertNewDocument(doc3); // 再次插入，成功
        } catch (DuplicateDocumentException e) {
            System.out.println(e.getMessage());
        }
    }
}