use both pid and cluster_id

This commit is contained in:
Jason McVetta
2020-02-12 18:23:47 +07:00
parent d22c916bfc
commit 14a602ee8b
4 changed files with 13 additions and 10 deletions
+2 -2
View File
@@ -76,8 +76,8 @@ class Cluster(object):
@property
def stat(self):
if self.sentinel:
return Stat.get(self.pid)
return Status(self.pid)
return Stat.get(pid=self.pid, cluster_id=self.cluster_id)
return Status(pid=self.pid, cluster_id=self.cluster_id)
@property
def is_starting(self):
+6 -5
View File
@@ -8,11 +8,12 @@ from django_q.signing import SignedPackage, BadSignature
class Status(object):
"""Cluster status base class."""
def __init__(self, pid):
def __init__(self, pid, cluster_id):
self.workers = []
self.tob = None
self.reincarnations = 0
self.cluster_id = pid
self.pid = pid
self.cluster_id = cluster_id
self.sentinel = 0
self.status = Conf.STOPPED
self.done_q_size = 0
@@ -27,7 +28,7 @@ class Stat(Status):
"""Status object for Cluster monitoring."""
def __init__(self, sentinel):
super(Stat, self).__init__(sentinel.parent_pid or sentinel.pid)
super(Stat, self).__init__(sentinel.parent_pid or sentinel.pid, cluster_id=sentinel.cluster_id)
self.broker = sentinel.broker or get_broker()
self.tob = sentinel.tob
self.reincarnations = sentinel.reincarnations
@@ -72,7 +73,7 @@ class Stat(Status):
return self.done_q_size + self.task_q_size == 0
@staticmethod
def get(cluster_id, broker=None):
def get(pid, cluster_id, broker=None):
"""
gets the current status for the cluster
:param cluster_id: id of the cluster
@@ -86,7 +87,7 @@ class Stat(Status):
return SignedPackage.loads(pack)
except BadSignature:
return None
return Status(cluster_id)
return Status(pid=pid, cluster_id=cluster_id)
@staticmethod
def get_all(broker=None):
+1 -1
View File
@@ -345,7 +345,7 @@ def test_bad_secret(broker, monkeypatch):
monkeypatch.setattr(Conf, "SECRET_KEY", "OOPS")
stat = Stat.get_all()
assert len(stat) == 0
assert Stat.get(s.parent_pid) is None
assert Stat.get(pid=s.parent_pid, cluster_id=cluster_id) is None
task_queue = Queue()
pusher(task_queue, stop_event, broker=broker)
result_queue = Queue()
+4 -2
View File
@@ -1,4 +1,5 @@
import pytest
import uuid
from django_q.tasks import async_task
from django_q.brokers import get_broker
@@ -10,7 +11,8 @@ from django_q.conf import Conf
@pytest.mark.django_db
def test_monitor(monkeypatch):
assert Stat.get(0).sentinel == 0
cluster_id = uuid.uuid4()
assert Stat.get(pid=0, cluster_id=4).sentinel == 0
c = Cluster()
c.start()
stats = monitor(run_once=True)
@@ -18,7 +20,7 @@ def test_monitor(monkeypatch):
assert len(stats) > 0
found_c = False
for stat in stats:
if stat.cluster_id == c.pid:
if stat.cluster_id == c.cluster_id:
found_c = True
assert stat.uptime() > 0
assert stat.empty_queues() is True