void UpdateStage::doRestoreState() {
const UpdateRequest& request = *_params.request;
const NamespaceString& nsString(request.getNamespaceString());
// We may have stepped down during the yield.
bool userInitiatedWritesAndNotPrimary = getOpCtx()->writesAreReplicated() &&
!repl::ReplicationCoordinator::get(getOpCtx())->canAcceptWritesFor(getOpCtx(), nsString);
if (userInitiatedWritesAndNotPrimary) {
uasserted(ErrorCodes::PrimarySteppedDown,
str::stream() << "Demoted from primary while performing update on "
<< nsString.ns());
}
if (request.getLifecycle()) {
UpdateLifecycle* lifecycle = request.getLifecycle();
lifecycle->setCollection(_collection);
if (!lifecycle->canContinue()) {
uasserted(17270, "Update aborted due to invalid state transitions after yield.");
}
_params.driver->refreshIndexKeys(lifecycle->getIndexKeys(getOpCtx()));
}
}
Status UpdateStage::restoreUpdateState(OperationContext* opCtx) {
const UpdateRequest& request = *_params.request;
const NamespaceString& nsString(request.getNamespaceString());
// We may have stepped down during the yield.
if (request.shouldCallLogOp() &&
!repl::getGlobalReplicationCoordinator()->canAcceptWritesForDatabase(nsString.db())) {
return Status(ErrorCodes::NotMaster,
str::stream() << "Demoted from primary while performing update on "
<< nsString.ns());
}
if (request.getLifecycle()) {
UpdateLifecycle* lifecycle = request.getLifecycle();
lifecycle->setCollection(_collection);
if (!lifecycle->canContinue()) {
return Status(ErrorCodes::IllegalOperation,
"Update aborted due to invalid state transitions after yield.",
17270);
}
_params.driver->refreshIndexKeys(lifecycle->getIndexKeys(opCtx));
}
return Status::OK();
}
