Skip to content

Commit a3301b9

Browse files
joykim1005joykim1005
committed
add index hint support for distinct command
1 parent 6bb4005 commit a3301b9

File tree

17 files changed

+327
-9
lines changed

17 files changed

+327
-9
lines changed

‎driver-core/src/main/com/mongodb/internal/operation/AsyncOperations.java

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -181,8 +181,9 @@ public <TResult> AsyncReadOperation<AsyncBatchCursor<TResult>> find(final MongoN
181181
}
182182

183183
public <TResult> AsyncReadOperation<AsyncBatchCursor<TResult>> distinct(final String fieldName, final Bson filter,
184-
final Class<TResult> resultClass, final Collation collation, final BsonValue comment) {
185-
return operations.distinct(fieldName, filter, resultClass, collation, comment);
184+
final Class<TResult> resultClass, final Collation collation, final BsonValue comment, final Bson hint,
185+
final String hintString) {
186+
return operations.distinct(fieldName, filter, resultClass, collation, comment, hint, hintString);
186187
}
187188

188189
public <TResult> AsyncExplainableReadOperation<AsyncBatchCursor<TResult>> aggregate(

‎driver-core/src/main/com/mongodb/internal/operation/DistinctOperation.java

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,8 @@ public class DistinctOperation<T> implements AsyncReadOperation<AsyncBatchCursor
5454
private BsonDocument filter;
5555
private Collation collation;
5656
private BsonValue comment;
57+
private BsonValue hint;
58+
private String hintString;
5759

5860
public DistinctOperation(final MongoNamespace namespace, final String fieldName, final Decoder<T> decoder) {
5961
this.namespace = notNull("namespace", namespace);
@@ -97,6 +99,24 @@ public DistinctOperation<T> comment(final BsonValue comment) {
9799
return this;
98100
}
99101

102+
public BsonValue getHint() {
103+
return hint;
104+
}
105+
106+
public DistinctOperation<T> hint(@Nullable final BsonValue hint) {
107+
this.hint = hint;
108+
return this;
109+
}
110+
111+
public String getHintString() {
112+
return hintString;
113+
}
114+
115+
public DistinctOperation<T> hintString(@Nullable final String hint) {
116+
this.hintString = hint;
117+
return this;
118+
}
119+
100120
@Override
101121
public BatchCursor<T> execute(final ReadBinding binding) {
102122
return executeRetryableRead(binding, namespace.getDatabaseName(), getCommandCreator(), createCommandDecoder(),
@@ -124,6 +144,7 @@ private CommandCreator getCommandCreator() {
124144
commandDocument.put("collation", collation.asDocument());
125145
}
126146
putIfNotNull(commandDocument, "comment", comment);
147+
putIfNotNull(commandDocument, "hint", hint);
127148
return commandDocument;
128149
};
129150
}

‎driver-core/src/main/com/mongodb/internal/operation/Operations.java

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -217,13 +217,14 @@ private <TResult> FindOperation<TResult> createFindOperation(final MongoNamespac
217217
}
218218

219219
<TResult> DistinctOperation<TResult> distinct(final String fieldName, @Nullable final Bson filter, final Class<TResult> resultClass,
220-
final Collation collation, final BsonValue comment) {
220+
final Collation collation, final BsonValue comment, @Nullable final Bson hint, @Nullable final String hintString) {
221221
return new DistinctOperation<>(assertNotNull(namespace),
222222
fieldName, codecRegistry.get(resultClass))
223223
.retryReads(retryReads)
224224
.filter(filter == null ? null : filter.toBsonDocument(documentClass, codecRegistry))
225225
.collation(collation)
226-
.comment(comment);
226+
.comment(comment)
227+
.hint(hint != null ? toBsonDocument(hint) : (hintString != null ? new BsonString(hintString) : null));
227228
}
228229

229230
<TResult> AggregateOperation<TResult> aggregate(final List<? extends Bson> pipeline, final Class<TResult> resultClass,

‎driver-core/src/main/com/mongodb/internal/operation/SyncOperations.java

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -165,8 +165,9 @@ public <TResult> ReadOperation<BatchCursor<TResult>> find(final MongoNamespace f
165165

166166
public <TResult> ReadOperation<BatchCursor<TResult>> distinct(final String fieldName, final Bson filter,
167167
final Class<TResult> resultClass,
168-
final Collation collation, final BsonValue comment) {
169-
return operations.distinct(fieldName, filter, resultClass, collation, comment);
168+
final Collation collation, final BsonValue comment,
169+
final Bson hint, final String hintString) {
170+
return operations.distinct(fieldName, filter, resultClass, collation, comment, hint, hintString);
170171
}
171172

172173
public <TResult> ExplainableReadOperation<BatchCursor<TResult>> aggregate(final List<? extends Bson> pipeline,

‎driver-core/src/test/resources/unified-test-format/crud/distinct-hint.json

Lines changed: 139 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,139 @@
1+
{
2+
"description": "distinct-hint",
3+
"schemaVersion": "1.0",
4+
"runOnRequirements": [
5+
{
6+
"minServerVersion": "7.1.0"
7+
}
8+
],
9+
"createEntities": [
10+
{
11+
"client": {
12+
"id": "client0",
13+
"observeEvents": [
14+
"commandStartedEvent"
15+
]
16+
}
17+
},
18+
{
19+
"database": {
20+
"id": "database0",
21+
"client": "client0",
22+
"databaseName": "distinct-hint-tests"
23+
}
24+
},
25+
{
26+
"collection": {
27+
"id": "collection0",
28+
"database": "database0",
29+
"collectionName": "coll0"
30+
}
31+
}
32+
],
33+
"initialData": [
34+
{
35+
"collectionName": "coll0",
36+
"databaseName": "distinct-hint-tests",
37+
"documents": [
38+
{
39+
"_id": 1,
40+
"x": 11
41+
},
42+
{
43+
"_id": 2,
44+
"x": 22
45+
},
46+
{
47+
"_id": 3,
48+
"x": 33
49+
}
50+
]
51+
}
52+
],
53+
"tests": [
54+
{
55+
"description": "distinct with hint string",
56+
"operations": [
57+
{
58+
"name": "distinct",
59+
"object": "collection0",
60+
"arguments": {
61+
"fieldName": "x",
62+
"filter": {
63+
"_id": 1
64+
},
65+
"hint": "_id_"
66+
},
67+
"expectResult": [
68+
11
69+
]
70+
}
71+
],
72+
"expectEvents": [
73+
{
74+
"client": "client0",
75+
"events": [
76+
{
77+
"commandStartedEvent": {
78+
"command": {
79+
"distinct": "coll0",
80+
"key": "x",
81+
"query": {
82+
"_id": 1
83+
},
84+
"hint": "_id_"
85+
},
86+
"commandName": "distinct",
87+
"databaseName": "distinct-hint-tests"
88+
}
89+
}
90+
]
91+
}
92+
]
93+
},
94+
{
95+
"description": "distinct with hint document",
96+
"operations": [
97+
{
98+
"name": "distinct",
99+
"object": "collection0",
100+
"arguments": {
101+
"fieldName": "x",
102+
"filter": {
103+
"_id": 1
104+
},
105+
"hint": {
106+
"_id": 1
107+
}
108+
},
109+
"expectResult": [
110+
11
111+
]
112+
}
113+
],
114+
"expectEvents": [
115+
{
116+
"client": "client0",
117+
"events": [
118+
{
119+
"commandStartedEvent": {
120+
"command": {
121+
"distinct": "coll0",
122+
"key": "x",
123+
"query": {
124+
"_id": 1
125+
},
126+
"hint": {
127+
"_id": 1
128+
}
129+
},
130+
"commandName": "distinct",
131+
"databaseName": "distinct-hint-tests"
132+
}
133+
}
134+
]
135+
}
136+
]
137+
}
138+
]
139+
}

‎driver-kotlin-coroutine/src/integration/kotlin/com/mongodb/kotlin/client/coroutine/syncadapter/SyncDistinctIterable.kt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,8 @@ data class SyncDistinctIterable<T : Any>(val wrapped: DistinctFlow<T>) :
3333
override fun collation(collation: Collation?): SyncDistinctIterable<T> = apply { wrapped.collation(collation) }
3434
override fun comment(comment: String?): SyncDistinctIterable<T> = apply { wrapped.comment(comment) }
3535
override fun comment(comment: BsonValue?): SyncDistinctIterable<T> = apply { wrapped.comment(comment) }
36+
override fun hint(hint: Bson?): SyncDistinctIterable<T> = apply { wrapped.hint(hint) }
37+
override fun hintString(hint: String?): SyncDistinctIterable<T> = apply { wrapped.hintString(hint) }
3638
override fun timeoutMode(timeoutMode: TimeoutMode): SyncDistinctIterable<T> = apply {
3739
wrapped.timeoutMode(timeoutMode)
3840
}

‎driver-kotlin-coroutine/src/main/kotlin/com/mongodb/kotlin/client/coroutine/DistinctFlow.kt

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@ import com.mongodb.annotations.Alpha
1919
import com.mongodb.annotations.Reason
2020
import com.mongodb.client.cursor.TimeoutMode
2121
import com.mongodb.client.model.Collation
22+
import com.mongodb.lang.Nullable
2223
import com.mongodb.reactivestreams.client.DistinctPublisher
2324
import java.util.concurrent.TimeUnit
2425
import kotlinx.coroutines.flow.Flow
@@ -103,5 +104,21 @@ public class DistinctFlow<T : Any>(private val wrapped: DistinctPublisher<T>) :
103104
*/
104105
public fun comment(comment: BsonValue?): DistinctFlow<T> = apply { wrapped.comment(comment) }
105106

107+
/**
108+
* Sets the hint for which index to use. A null value means no hint is set.
109+
*
110+
* @param hint the hint
111+
* @return this
112+
*/
113+
public fun hint(@Nullable hint: Bson?): DistinctFlow<T> = apply { wrapped.hint(hint) }
114+
115+
/**
116+
* Sets the hint for which index to use. A null value means no hint is set.
117+
*
118+
* @param hint the name of the index which should be used for the operation
119+
* @return this
120+
*/
121+
public fun hintString(@Nullable hint: String?): DistinctFlow<T> = apply { wrapped.hintString(hint) }
122+
106123
public override suspend fun collect(collector: FlowCollector<T>): Unit = wrapped.asFlow().collect(collector)
107124
}

‎driver-kotlin-sync/src/integration/kotlin/com/mongodb/kotlin/client/syncadapter/SyncDistinctIterable.kt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,4 +36,6 @@ internal class SyncDistinctIterable<T : Any>(val wrapped: DistinctIterable<T>) :
3636
override fun collation(collation: Collation?): SyncDistinctIterable<T> = apply { wrapped.collation(collation) }
3737
override fun comment(comment: String?): SyncDistinctIterable<T> = apply { wrapped.comment(comment) }
3838
override fun comment(comment: BsonValue?): SyncDistinctIterable<T> = apply { wrapped.comment(comment) }
39+
override fun hint(hint: Bson?): SyncDistinctIterable<T> = apply { wrapped.hint(hint) }
40+
override fun hintString(hint: String?): SyncDistinctIterable<T> = apply { wrapped.hintString(hint) }
3941
}

‎driver-kotlin-sync/src/main/kotlin/com/mongodb/kotlin/client/DistinctIterable.kt

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ import com.mongodb.annotations.Reason
2020
import com.mongodb.client.DistinctIterable as JDistinctIterable
2121
import com.mongodb.client.cursor.TimeoutMode
2222
import com.mongodb.client.model.Collation
23+
import com.mongodb.lang.Nullable
2324
import java.util.concurrent.TimeUnit
2425
import org.bson.BsonValue
2526
import org.bson.conversions.Bson
@@ -99,4 +100,20 @@ public class DistinctIterable<T : Any?>(private val wrapped: JDistinctIterable<T
99100
* @return this
100101
*/
101102
public fun comment(comment: BsonValue?): DistinctIterable<T> = apply { wrapped.comment(comment) }
103+
104+
/**
105+
* Sets the hint for which index to use. A null value means no hint is set.
106+
*
107+
* @param hint the hint
108+
* @return this
109+
*/
110+
public fun hint(@Nullable hint: Bson?): DistinctIterable<T> = apply { wrapped.hint(hint) }
111+
112+
/**
113+
* Sets the hint for which index to use. A null value means no hint is set.
114+
*
115+
* @param hint the name of the index which should be used for the operation
116+
* @return this
117+
*/
118+
public fun hintString(@Nullable hint: String?): DistinctIterable<T> = apply { wrapped.hintString(hint) }
102119
}

‎driver-reactive-streams/src/main/com/mongodb/reactivestreams/client/DistinctPublisher.java

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -97,6 +97,24 @@ public interface DistinctPublisher<TResult> extends Publisher<TResult> {
9797
*/
9898
DistinctPublisher<TResult> comment(@Nullable BsonValue comment);
9999

100+
/**
101+
* Sets the hint for which index to use. A null value means no hint is set.
102+
*
103+
* @param hint the hint
104+
* @return this
105+
* @since 5.3
106+
*/
107+
DistinctPublisher<TResult> hint(@Nullable Bson hint);
108+
109+
/**
110+
* Sets the hint for which index to use. A null value means no hint is set.
111+
*
112+
* @param hint the name of the index which should be used for the operation
113+
* @return this
114+
* @since 5.3
115+
*/
116+
DistinctPublisher<TResult> hintString(@Nullable String hint);
117+
100118
/**
101119
* Sets the timeoutMode for the cursor.
102120
*

0 commit comments

Comments
 (0)