diff options
Diffstat (limited to 'src/mongo/db/s/scoped_operation_completion_sharding_actions.cpp')
-rw-r--r-- | src/mongo/db/s/scoped_operation_completion_sharding_actions.cpp | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/src/mongo/db/s/scoped_operation_completion_sharding_actions.cpp b/src/mongo/db/s/scoped_operation_completion_sharding_actions.cpp index 625f62a51f3..4d544948696 100644 --- a/src/mongo/db/s/scoped_operation_completion_sharding_actions.cpp +++ b/src/mongo/db/s/scoped_operation_completion_sharding_actions.cpp @@ -73,6 +73,13 @@ ScopedOperationCompletionShardingActions::~ScopedOperationCompletionShardingActi } if (auto staleInfo = status->extraInfo<StaleConfigInfo>()) { + if (staleInfo->getCriticalSectionSignal()) { + // Set migration critical section on operation sharding state: operation will wait for + // the migration to finish before returning. + auto& oss = OperationShardingState::get(_opCtx); + oss.setMigrationCriticalSectionSignal(staleInfo->getCriticalSectionSignal()); + } + auto handleMismatchStatus = onShardVersionMismatchNoExcept( _opCtx, staleInfo->getNss(), staleInfo->getVersionReceived()); if (!handleMismatchStatus.isOK()) |