Improve identifier for zookeeper based lockfiles.
authorScott Gasch <[email protected]>
Fri, 9 Dec 2022 21:10:20 +0000 (13:10 -0800)
committerScott Gasch <[email protected]>
Fri, 9 Dec 2022 21:10:20 +0000 (13:10 -0800)
src/pyutils/files/lockfile.py

index 26adfb3c31ff6700ca9c826119fd8e3977633622..158a636653eefd7b66ee31c698a7dc909dc0e770 100644 (file)
@@ -26,6 +26,7 @@ import datetime
 import json
 import logging
 import os
+import platform
 import signal
 import sys
 import warnings
@@ -149,11 +150,14 @@ class LockFile(contextlib.AbstractContextManager):
 
     def _try_acquire_zk_lock(self) -> bool:
         assert self.expiration_timestamp
+        identifier = f"Lockfile for pid={os.getpid()} on machine {platform.node()}"
+        if self.override_command:
+            identifier += f" running {self.override_command}"
         self.zk_lease = zookeeper.RenewableReleasableLease(
             self.zk_client,
             self.lockfile,
             datetime.timedelta(seconds=self.expiration_timestamp),
-            f"Pyutils lockfile pid={os.getpid()}",
+            identifier,
         )
         return self.zk_lease