fix: Abort lease_keepalive_task when EtcdLockManager lock request tim…

…eout
lablup · May 13, 2024 · 7f2b1b3 · 7f2b1b3
1 parent a726848
commit 7f2b1b3
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/lock_manager.rs b/src/lock_manager.rs
@@ -112,6 +112,10 @@ impl EtcdLockManager {
                 None => ready(Ok(self.try_lock(&mut client).await)).await,
             };
 
+        if let Some(ref lease_keepalive_task) = self.lease_keepalive_task {
+            lease_keepalive_task.abort();
+        }
+
         match timeout_result {
             Ok(Ok(_)) => {}
             Ok(Err(e)) => return Err(e.into()),
@@ -129,10 +133,6 @@ impl EtcdLockManager {
             }
         }
 
-        if let Some(ref lease_keepalive_task) = self.lease_keepalive_task {
-            lease_keepalive_task.abort();
-        }
-
         Ok(PyCommunicator::new(client))
     }