|
1 | 1 | package org.hypertrace.core.documentstore.mongo.update; |
2 | 2 |
|
| 3 | +import static org.hypertrace.core.documentstore.model.options.DataFreshness.REALTIME_FRESHNESS; |
3 | 4 | import static org.hypertrace.core.documentstore.model.options.ReturnDocumentType.NONE; |
4 | 5 | import static org.hypertrace.core.documentstore.mongo.MongoUtils.getReturnDocument; |
5 | 6 | import static org.hypertrace.core.documentstore.mongo.query.parser.MongoFilterTypeExpressionParser.getFilter; |
|
19 | 20 | import org.hypertrace.core.documentstore.commons.CommonUpdateValidator; |
20 | 21 | import org.hypertrace.core.documentstore.commons.UpdateValidator; |
21 | 22 | import org.hypertrace.core.documentstore.model.config.ConnectionConfig; |
| 23 | +import org.hypertrace.core.documentstore.model.options.QueryOptions; |
22 | 24 | import org.hypertrace.core.documentstore.model.options.ReturnDocumentType; |
23 | 25 | import org.hypertrace.core.documentstore.model.options.UpdateOptions; |
24 | 26 | import org.hypertrace.core.documentstore.model.options.UpdateOptions.MissingDocumentStrategy; |
@@ -106,13 +108,17 @@ public Optional<MongoCursor<BasicDBObject>> bulkUpdate( |
106 | 108 |
|
107 | 109 | switch (returnDocumentType) { |
108 | 110 | case BEFORE_UPDATE: |
109 | | - cursor = queryExecutor.aggregate(query); |
| 111 | + cursor = |
| 112 | + queryExecutor.aggregate( |
| 113 | + query, QueryOptions.builder().dataFreshness(REALTIME_FRESHNESS).build()); |
110 | 114 | logAndUpdate(filter, updateObject, mongoUpdateOptions); |
111 | 115 | return Optional.of(cursor); |
112 | 116 |
|
113 | 117 | case AFTER_UPDATE: |
114 | 118 | logAndUpdate(filter, updateObject, mongoUpdateOptions); |
115 | | - cursor = queryExecutor.aggregate(query); |
| 119 | + cursor = |
| 120 | + queryExecutor.aggregate( |
| 121 | + query, QueryOptions.builder().dataFreshness(REALTIME_FRESHNESS).build()); |
116 | 122 | return Optional.of(cursor); |
117 | 123 |
|
118 | 124 | case NONE: |
|
0 commit comments