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

Add a TaskManager.boot_time property

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

28
patches/background_task/models.py
View File

@ -39,6 +39,23 @@ class TaskQuerySet(models.QuerySet):
class TaskManager(models.Manager): class TaskManager(models.Manager):
_boot_time = posix_epoch = datetime(1970, 1, 1, tzinfo=tz.utc)
@property
def boot_time(self):
if self._boot_time > self.posix_epoch:
return self._boot_time
stats = None
boot_time = self.posix_epoch
kcore_path = Path('/proc/kcore')
if kcore_path.exists():
stats = kcore_path.stats()
if stats:
boot_time += timedelta(seconds=stats.st_mtime)
if boot_time > self._boot_time:
self._boot_time = boot_time
return self._boot_time
def get_queryset(self): def get_queryset(self):
return TaskQuerySet(self.model, using=self._db) return TaskQuerySet(self.model, using=self._db)
@ -70,21 +87,14 @@ class TaskManager(models.Manager):
max_run_time = app_settings.BACKGROUND_TASK_MAX_RUN_TIME max_run_time = app_settings.BACKGROUND_TASK_MAX_RUN_TIME
qs = self.get_queryset() qs = self.get_queryset()
expires_at = now - timedelta(seconds=max_run_time) expires_at = now - timedelta(seconds=max_run_time)
unlocked = Q(locked_by=None) | Q(locked_at__lt=expires_at) unlocked = Q(locked_by=None) | Q(locked_at__lt=expires_at) | Q(locked_at__lt=self.boot_time)
return qs.filter(unlocked) return qs.filter(unlocked)
def locked(self, now): def locked(self, now):
stats = None
kcore_path = Path('/proc/kcore')
boot_time = posix_epoch = datetime(1970, 1, 1, tzinfo=tz.utc)
if kcore_path.exists():
stats = kcore_path.stats()
if stats:
boot_time += timedelta(seconds=stats.st_mtime)
max_run_time = app_settings.BACKGROUND_TASK_MAX_RUN_TIME max_run_time = app_settings.BACKGROUND_TASK_MAX_RUN_TIME
qs = self.get_queryset() qs = self.get_queryset()
expires_at = now - timedelta(seconds=max_run_time) expires_at = now - timedelta(seconds=max_run_time)
locked = Q(locked_by__isnull=False) & Q(locked_at__gt=expires_at) & Q(locked_at__gt=boot_time) locked = Q(locked_by__isnull=False) & Q(locked_at__gt=expires_at) & Q(locked_at__gt=self.boot_time)
return qs.filter(locked) return qs.filter(locked)
def failed(self): def failed(self):