mindol1004/spring-data-mongodb
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

DATAMONGO-2360 - Apply query hint to count queries.

Browse Source 
Original pull request: #788.
This commit is contained in:
Christoph Strobl
2019-09-12 10:01:17 +02:00
committed by Mark Paluch
parent dc859953f4
commit 313ffb5426
2 changed files with 16 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
spring-data-mongodb/src/main/java/org/springframework/data/mongodb/core/MongoTemplate.java
View File

@@ -1130,6 +1130,10 @@ public class MongoTemplate implements MongoOperations, ApplicationContextAware,
CountOptions options = new CountOptions(); CountOptions options = new CountOptions();
query.getCollation().map(Collation::toMongoCollation).ifPresent(options::collation); query.getCollation().map(Collation::toMongoCollation).ifPresent(options::collation);
if (StringUtils.hasText(query.getHint())) {
options.hint(Document.parse(query.getHint()));
}
Document document = queryMapper.getMappedObject(query.getQueryObject(), Document document = queryMapper.getMappedObject(query.getQueryObject(),
Optional.ofNullable(entityClass).map(it -> mappingContext.getPersistentEntity(entityClass))); Optional.ofNullable(entityClass).map(it -> mappingContext.getPersistentEntity(entityClass)));

12
spring-data-mongodb/src/test/java/org/springframework/data/mongodb/core/MongoTemplateUnitTests.java
View File

@@ -910,6 +910,18 @@ public class MongoTemplateUnitTests extends MongoOperationsUnitTests {
is(equalTo(com.mongodb.client.model.Collation.builder().locale("fr").build()))); is(equalTo(com.mongodb.client.model.Collation.builder().locale("fr").build())));
} }
@Test // DATAMONGO-2360
public void countShouldApplyQueryHintIfPresent() {
Document queryHint = new Document("age", 1);
template.count(new BasicQuery("{}").withHint(queryHint), AutogenerateableId.class);
ArgumentCaptor<CountOptions> options = ArgumentCaptor.forClass(CountOptions.class);
verify(collection).count(any(), options.capture());
assertThat(options.getValue().getHint()).isEqualTo(queryHint);
}
@Test // DATAMONGO-1733 @Test // DATAMONGO-1733
public void appliesFieldsWhenInterfaceProjectionIsClosedAndQueryDoesNotDefineFields() { public void appliesFieldsWhenInterfaceProjectionIsClosedAndQueryDoesNotDefineFields() {