hoelee/tubesync
1
0
Fork 0
mirror of https://github.com/meeb/tubesync.git synced 2025-06-22 13:06:34 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add nodename to Task.locked_by

Browse Source
This commit is contained in:
tcely 2025-03-24 13:46:59 -04:00 committed by GitHub
parent e5e29885f1
commit c10b1aa255
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 15 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
patches/background_task/models.py
View File

@ -195,9 +195,14 @@ class Task(models.Model):
Check if the locked_by process is still running.
"""
if self.locked_by:
pid, node = self.locked_by.split('/', 1)
# locked by a process on this node?
if os.uname().nodename[:(64-10)] != node:
return False
# is the process still running?
try:
# won't kill the process. kill is a bad named system call
os.kill(int(self.locked_by), 0)
# Signal number zero won't kill the process.
os.kill(int(pid), 0)
return True
except:
return False
@ -220,8 +225,9 @@ class Task(models.Model):
def lock(self, locked_by):
now = timezone.now()
owner = f'{locked_by[:8]}/{os.uname().nodename[:(64-10)}'
unlocked = Task.objects.unlocked(now).filter(pk=self.pk)
updated = unlocked.update(locked_by=locked_by, locked_at=now)
updated = unlocked.update(locked_by=owner, locked_at=now)
if updated:
return Task.objects.get(pk=self.pk)
return None
@ -423,9 +429,14 @@ class CompletedTask(models.Model):
Check if the locked_by process is still running.
"""
if self.locked_by:
pid, node = self.locked_by.split('/', 1)
# locked by a process on this node?
if os.uname().nodename[:(64-10)] != node:
return False
# is the process still running?
try:
# won't kill the process. kill is a bad named system call
os.kill(int(self.locked_by), 0)
os.kill(int(pid), 0)
return True
except:
return False