在Java中利用MongoDB唯一索引高效处理重复文档插入

// ...
DBObject duplicate = match.findOne(filter); // 注意：DBObject 已被弃用
try {
    if (duplicate != null) { // 逻辑错误，应该在 duplicate 为 null 时才插入
        InsertOneResult result = match.insertOne(zeroCmd);
    }
    throw new Exception("[Error] duplicate insertion");
} catch (Exception me) {
    System.out.println(me.getMessage());
}
// ...

import com.mongodb.MongoWriteException;
import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoDatabase;
import com.mongodb.client.result.InsertOneResult;
import org.bson.Document;
import com.mongodb.client.MongoClient;
import com.mongodb.client.MongoClients;
import com.mongodb.client.model.Indexes;
import com.mongodb.client.model.IndexOptions;
import com.mongodb.MongoCommandException;

public class ProductInserter {

    private MongoCollection<Document> productCollection;

    /**
     * 构造函数，初始化集合并确保唯一复合索引的存在。
     * @param database MongoDB数据库实例
     */
    public ProductInserter(MongoDatabase database) {
        this.productCollection = database.getCollection("products"); // 假设集合名为 "products"

        // 确保集合上存在复合唯一索引
        // 这一步通常在应用启动或首次使用集合时执行一次
        try {
            productCollection.createIndex(
                    Indexes.compoundIndex(
                            Indexes.ascending("name"),
                            Indexes.ascending("supplier"),
                            Indexes.ascending("food"),
                            Indexes.ascending("countryOfOrigin") // 注意：字段名应与实际文档一致
                    ),
                    new IndexOptions().unique(true)
            );
            System.out.println("Unique compound index created or already exists on 'products' collection.");
        } catch (MongoCommandException e) {
            // 错误码 85 表示索引已存在
            if (e.getErrorCode() == 85) {
                System.out.println("Unique compound index already exists.");
            } else {
                System.err.println("Error creating index: " + e.getMessage());
                throw e; // 抛出其他关键错误
            }
        } catch (Exception e) {
            System.err.println("An unexpected error occurred during index creation: " + e.getMessage());
            throw e;
        }
    }

    /**
     * 尝试插入一个新产品文档。如果存在重复，则捕获异常并处理。
     * @param name 产品名称
     * @param supplier 供应商
     * @param food 食品类型
     * @param countryOfOrigin 原产国
     */
    public void insertProduct(String name, String supplier, String food, String countryOfOrigin) {
        Document productToInsert = new Document()
                .append("name", name)
                .append("supplier", supplier)
                .append("food", food)
                .append("countryOfOrigin", countryOfOrigin); // 确保字段名与索引一致

        try {
            InsertOneResult result = productCollection.insertOne(productToInsert);
            System.out.println("Document inserted successfully with _id: " + result.getInsertedId());
        } catch (MongoWriteException e) {
            // 错误码 11000 表示唯一索引冲突（DuplicateKey error）
            if (e.getError().getCode() == 11000) {
                System.err.println("[Error] Duplicate insertion attempted for product: " + productToInsert.toJson());
                // 在这里可以添加更详细的业务逻辑，例如记录日志、返回特定错误码给调用方等
            } else {
                System.err.println("An unexpected MongoDB write error occurred: " + e.getMessage());
                throw e; // 重新抛出其他类型的写入异常
            }
        } catch (Exception e) {
            System.err.println("An unexpected error occurred: " + e.getMessage());
            e.printStackTrace();
        }
    }

    public static void main(String[] args) {
        // 1. 建立MongoDB连接
        // 请根据您的MongoDB实例地址和端口进行修改
        String connectionString = "mongodb://localhost:27017";
        String databaseName = "mydatabase"; //