SERVER-63207 Speedup very slow move_primary_with_writes.js test

1 files changed, 16 insertions, 4 deletions

diff --git a/src/mongo/db/s/sharding_ddl_coordinator.cpp b/src/mongo/db/s/sharding_ddl_coordinator.cpp
index 15e875918ac..e6da6be570a 100644
--- a/src/mongo/db/s/sharding_ddl_coordinator.cpp
+++ b/src/mongo/db/s/sharding_ddl_coordinator.cpp
@@ -50,6 +50,7 @@
 namespace mongo {
 
 MONGO_FAIL_POINT_DEFINE(hangBeforeRunningCoordinatorInstance);
+MONGO_FAIL_POINT_DEFINE(overrideDDLLockTimeout);
 
 namespace {
 
@@ -143,12 +144,23 @@ ExecutorFuture<void> ShardingDDLCoordinator::_acquireLockAsync(
 
                const auto coorName = DDLCoordinatorType_serializer(_coordId.getOperationType());
 
-               auto distLock = distLockManager->lockDirectLocally(
-                   opCtx, resource, DistLockManager::kDefaultLockTimeout);
+               const auto lockTimeOut = [&]() -> Milliseconds {
+                   if (auto sfp = overrideDDLLockTimeout.scoped(); MONGO_unlikely(sfp.isActive())) {
+                       if (auto timeoutElem = sfp.getData()["timeoutMillisecs"]; timeoutElem.ok()) {
+                           const auto timeoutMillisecs = Milliseconds(timeoutElem.safeNumberLong());
+                           LOGV2(6320700,
+                                 "Overriding DDL lock timeout",
+                                 "timeout"_attr = timeoutMillisecs);
+                           return timeoutMillisecs;
+                       }
+                   }
+                   return DistLockManager::kDefaultLockTimeout;
+               }();
+
+               auto distLock = distLockManager->lockDirectLocally(opCtx, resource, lockTimeOut);
                _scopedLocks.emplace(std::move(distLock));
 
-               uassertStatusOK(distLockManager->lockDirect(
-                   opCtx, resource, coorName, DistLockManager::kDefaultLockTimeout));
+               uassertStatusOK(distLockManager->lockDirect(opCtx, resource, coorName, lockTimeOut));
            })
         .until([this](Status status) { return (!_recoveredFromDisk) || status.isOK(); })
         .withBackoffBetweenIterations(kExponentialBackoff)