Spring Boot nie może zaktualizować kolekcji podzielonej na fragmenty na azure cosmos db (MongoDb)

Miałem ten sam problem, rozwiązany następującym hackiem:

@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);
            }
        };
    }
}

Byłoby miło mieć adnotację @ShardKey, aby oznaczyć pole dokumentu jako fragment i automatycznie dodać je do zapytania.