Spring数据Mongodb批量保存错误继续

Question

我正在使用 spring-data-mongodb 并且有一个简单的存储库，它配置了以下配置：

@Configuration
@EnableMongoRepositories(basePackages = "com.my.package")
@Profile("default")
public class MongoConfig extends AbstractMongoConfiguration {

    @Value("${mongo.db.uri}")
    private String mongoDbUri;
    @Value("${mongo.db.database}")
    private String mongoDbDatabaseName;

    @Override
    protected String getDatabaseName() {
        return mongoDbDatabaseName;
    }

    @Override
    public MongoClient mongoClient() {
        return new MongoClient(new MongoClientURI(mongoDbUri));
    }
}

使用的存储库扩展了 CrudRepository，这使得我可以调用 saveAll() 方法。默认情况下，当一条记录失败时，在 mongodb 中执行 saveAll（批量操作）将停止，除非将选项传递给 insertMany/updateMany 命令以使“continueOnError”为 true 或具有“BulkMode.unordered”。有什么方法可以将 spring 数据配置为始终在出错时继续（或始终进行无序插入/更新），这样即使某些记录失败，执行 saveAll 也将始终尝试整个批量？

谢谢！

Answer 1

要进行 MongoDB 批量写入（能够选择无序操作，从而允许写入在批量组中的一项失败后继续），您需要使用 org.springframework.data.mongodb.core.BulkOperations（可从 Spring Data 获得>= 1.9.0.RELEASE)

例如：

BulkOperations bulkOperations = mongoTemplate.bulkOps(BulkMode.UNORDERED, YourDomainObjectClass.class);
for (YourDomainObject ydo : yourDomainObjects) {
    bulkOperations.insert(ydo);
}
BulkWriteResult result = bulkOperations.execute();
// inspect the result
if (result.getInsertCount() != yourDomainObjects.size()) {
    // oh oh
}

如果您使用的是 Spring Data 版本 DBCollection（可通过 MongoOperations.getCollection() 获得）并使用标准驱动程序调用，例如：

BulkWriteOperation bulkWriteOperation = collection.initializeUnorderedBulkOperation();
for (YourDomainObject ydo : yourDomainObjects) {
    bulkWriteOperation.insert(ydo);    
}
BulkWriteResult result = bulkWriteOperation.execute();
// inspect the result
if (result.getInsertCount() != yourDomainObjects.size()) {
    // oh oh
}