SERVER-49821 Define utility function for getting recipient which would own document under new shard key

1 files changed, 9 insertions, 0 deletions

diff --git a/src/mongo/db/s/op_observer_sharding_impl.cpp b/src/mongo/db/s/op_observer_sharding_impl.cpp
index 60e93d7f0a2..60f809ba791 100644
--- a/src/mongo/db/s/op_observer_sharding_impl.cpp
+++ b/src/mongo/db/s/op_observer_sharding_impl.cpp
@@ -34,11 +34,13 @@
 #include "mongo/db/s/op_observer_sharding_impl.h"
 
 #include "mongo/db/catalog_raii.h"
+#include "mongo/db/repl/oplog_entry.h"
 #include "mongo/db/s/active_move_primaries_registry.h"
 #include "mongo/db/s/collection_sharding_runtime.h"
 #include "mongo/db/s/database_sharding_state.h"
 #include "mongo/db/s/migration_chunk_cloner_source_legacy.h"
 #include "mongo/db/s/migration_source_manager.h"
+#include "mongo/db/s/resharding_util.h"
 #include "mongo/logv2/log.h"
 
 namespace mongo {
@@ -214,4 +216,11 @@ void OpObserverShardingImpl::shardObserveTransactionPrepareOrUnpreparedCommit(
             opCtx->getServiceContext(), stmts, prepareOrCommitOptime));
 }
 
+void OpObserverShardingImpl::shardAnnotateOplogEntry(OperationContext* opCtx,
+                                                     const NamespaceString nss,
+                                                     const BSONObj& doc,
+                                                     repl::ReplOperation& op) {
+    op.setDestinedRecipient(getDestinedRecipient(opCtx, nss, doc));
+}
+
 }  // namespace mongo