如何在MongoDB Java驱动中覆盖默认ObjectId生成器实现自动自定义ID？

当然可以做到！在MongoDB Java驱动（4.x及以上版本）中，你完全可以替换默认的ObjectId生成逻辑，让插入操作自动使用你自定义的_id，不用每次手动显式设置。下面是具体的实现步骤：

首先，你需要实现org.bson.types.ObjectIdGenerator接口，重写generate()方法来返回符合你需求的自定义ObjectId。比如下面这个示例，我们生成带有自定义时间戳的ObjectId（你可以根据自己的需求修改生成逻辑）：

import org.bson.types.ObjectId;
import org.bson.types.ObjectIdGenerator;

public class CustomObjectIdGenerator implements ObjectIdGenerator {
    @Override
    public ObjectId generate() {
        // 这里编写你的自定义生成逻辑
        // 示例：使用提前1小时的时间戳 + 默认的机器/进程/计数器部分生成ObjectId
        long customTimestamp = System.currentTimeMillis() - 3600000;
        
        byte[] objectIdBytes = new byte[12];
        // 写入自定义时间戳（前4字节）
        objectIdBytes[0] = (byte) (customTimestamp >> 24);
        objectIdBytes[1] = (byte) (customTimestamp >> 16);
        objectIdBytes[2] = (byte) (customTimestamp >> 8);
        objectIdBytes[3] = (byte) customTimestamp;
        
        // 复用默认的机器标识、进程ID和递增计数器（也可以自定义这部分）
        System.arraycopy(ObjectId.getMachineIdentifier(), 0, objectIdBytes, 4, 3);
        System.arraycopy(ObjectId.getProcessIdentifier(), 0, objectIdBytes, 7, 2);
        byte[] counter = ObjectId.getCounter();
        System.arraycopy(counter, 0, objectIdBytes, 9, 3);
        
        return new ObjectId(objectIdBytes);
    }
}

接下来，在创建MongoClient的时候，通过MongoClientSettings把自定义生成器配置进去：

import com.mongodb.MongoClientSettings;
import com.mongodb.client.MongoClients;
import com.mongodb.client.MongoClient;
import com.mongodb.ConnectionString;

public class CustomMongoClientFactory {
    public static MongoClient createClient() {
        MongoClientSettings settings = MongoClientSettings.builder()
                .objectIdGenerator(new CustomObjectIdGenerator()) // 绑定自定义生成器
                .applyConnectionString(new ConnectionString("mongodb://localhost:27017"))
                .build();
        return MongoClients.create(settings);
    }
}

现在当你插入文档时，不需要手动设置_id字段，驱动会自动调用你的自定义生成器来生成_id：

import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoDatabase;
import org.bson.Document;

public class InsertDemo {
    public static void main(String[] args) {
        try (MongoClient client = CustomMongoClientFactory.createClient()) {
            MongoDatabase db = client.getDatabase("my_db");
            MongoCollection<Document> collection = db.getCollection("my_collection");
            
            // 创建一个不包含_id的文档
            Document doc = new Document("product_name", "MongoDB Guide")
                    .append("price", 59.9);
            
            // 插入文档，驱动会自动生成自定义的_id
            collection.insertOne(doc);
            
            // 打印生成的_id，验证是否是自定义的
            System.out.println("自动生成的_id: " + doc.get("_id"));
        }
    }
}

• 如果你使用的是3.x版本的MongoDB Java驱动，配置方式略有不同：需要通过MongoClientOptions.Builder的objectIdGenerator()方法来设置，核心的生成器实现逻辑是一样的。
• 自定义生成器一定要保证生成的ObjectId全局唯一，否则会导致插入文档时抛出重复键错误。建议参考默认ObjectId的实现逻辑，确保机器标识、进程ID和递增计数器的唯一性。

内容的提问来源于stack exchange，提问作者Rasekaran