Spring Boot không thể cập nhật bộ sưu tập đã phân đoạn trên azure cosmos db (MongoDb)

Tôi đã gặp vấn đề tương tự, được giải quyết bằng cách hack sau:

@Configuration
public class ReactiveMongoConfig {

    @Bean
    public ReactiveMongoTemplate reactiveMongoTemplate(ReactiveMongoDatabaseFactory reactiveMongoDatabaseFactory,
            MongoConverter converter, MyService service) {
        return new ReactiveMongoTemplate(reactiveMongoDatabaseFactory, converter) {
            @Override
            protected Mono<UpdateResult> doUpdate(String collectionName, Query query, UpdateDefinition update,
                    Class<?> entityClass, boolean upsert, boolean multi) {
                query.addCriteria(new Criteria("shardKey").is(service.getShardKey()));
                return super.doUpdate(collectionName, query, update, entityClass, upsert, multi);
            }
        };
    }
}

Rất vui nếu có chú thích @ShardKey để đánh dấu trường tài liệu là phân đoạn và thêm nó vào truy vấn tự động.