mirror of
https://github.com/django-q2/django-q2.git
synced 2026-10-08 01:58:13 +08:00
AttributeError when start_event is None, and guard process faster stop (#305)
* AttributeError when start_event is None, and guard process faster stop This commit has two related fixes that come into play when frequently shutting down and restarting the cluster. If a SIGINT or SIGTERM signal is received while the main process is waiting for the sentinel process to set start_event, the signal handler sets start_event to None, and the loop that polls start_event will raise an unhandled AttributeError attempting to check start_event.is_set(). The second issue is that the guard process sleeps for the guard cycle interval in each iteration before checking the stop event, preventing the guard cycle from terminating until the sleep completes. The cluster can be made more responsive to a shutdown request by using the event's wait(cycle) method rather than time.sleep(cycle) so the process wakes immediately when the event is set. The commit also fixes the case where the guard process sleeps for an extra cycle when the cycle counter happens to have been reset to zero on the loop iterations when the scheduler is called. This fix helps in our environment where we are frequently stopping and starting the cluster and where we have increased the guard cycle setting to several seconds. * Add sentinel premature death check while waiting for sentinel to start. Suggested by github copilot [here](https://github.com/django-q2/django-q2/pull/305/changes#r3143964420) * Add test case test_cluster_early_stop Add a test case to validate stopping the cluster before the sentinel has set the cluster's start_event. The test deterministically triggers the AttributeError when start_event is None (without the PR's fix in cluster.py). As requested by copilot: https://github.com/django-q2/django-q2/pull/305#discussion_r3143964399 * Add test test_cluster_stop_responsive Ensure that stopping the cluster is responsive and does not wait for a full GUARD_CYCLE to stop the cluster. As requested by copilot: https://github.com/django-q2/django-q2/pull/305/changes#r3143964414 * Address copilot review comments. 1. Do not raise RuntimeError if sentinel early exit was caused by SIGINT/SIGTERM stopping the cluster while waiting for the sentinel to start. Addresses https://github.com/django-q2/django-q2/pull/305#discussion_r3298618046 2. Ensure that the monkeypatched broker instance remains pickleable so the test_cluster_early_stop test will also work on platforms that use the spawn Process start method. Addresses https://github.com/django-q2/django-q2/pull/305#discussion_r3298618086 3. Do not allow the test_cluster_early_stop test to hang if the sentinel process or main test process terminates unexpectedly without setting the sentinel_event or test_event. Addresses https://github.com/django-q2/django-q2/pull/305#discussion_r3298618102
This commit is contained in:
+27
-3
@@ -73,7 +73,30 @@ class Cluster:
|
||||
)
|
||||
self.sentinel.start()
|
||||
logger.info(_("Q Cluster %(name)s starting.") % {"name": self.name})
|
||||
while not self.start_event.is_set():
|
||||
# Wait for the sentinel to set start_event by polling the event's state. We
|
||||
# cannot call wait() as that will deadlock if a signal is received while blocked
|
||||
# in wait(), as the sentinel will block in start_event.set() while this process
|
||||
# is blocked in sentinel.join(). It is also necessary to check for start_event
|
||||
# being set to None by the signal handler.
|
||||
while self.start_event and not self.start_event.is_set():
|
||||
# While waiting for the sentinel to start, also check for sentinel premature
|
||||
# death
|
||||
if not self.sentinel.is_alive():
|
||||
if self.sentinel.exitcode == 0:
|
||||
# the cluster was stopped via SIGTERM/SIGINT and sentinel early
|
||||
# termination was intentional -- exit quietly
|
||||
break
|
||||
logger.error(
|
||||
_(
|
||||
"Q Cluster %(name)s failed to start. Sentinel exited "
|
||||
"with exit code %(exitcode)s."
|
||||
)
|
||||
% {"name": self.name, "exitcode": self.sentinel.exitcode}
|
||||
)
|
||||
raise RuntimeError(
|
||||
f"Q Cluster {self.name} failed to start: sentinel exited "
|
||||
f"with code {self.sentinel.exitcode}"
|
||||
)
|
||||
sleep(0.1)
|
||||
return self.pid
|
||||
|
||||
@@ -313,7 +336,7 @@ class Sentinel:
|
||||
counter = 0
|
||||
cycle = Conf.GUARD_CYCLE # guard loop sleep in seconds
|
||||
# Guard loop. Runs at least once
|
||||
while not self.stop_event.is_set() or not counter:
|
||||
while True:
|
||||
# Check Workers
|
||||
for p in self.pool:
|
||||
with p.timer.get_lock():
|
||||
@@ -337,7 +360,8 @@ class Sentinel:
|
||||
scheduler(broker=self.broker)
|
||||
# Save current status
|
||||
Stat(self).save()
|
||||
sleep(cycle)
|
||||
if self.stop_event.wait(cycle):
|
||||
break
|
||||
self.stop()
|
||||
|
||||
def stop(self):
|
||||
|
||||
Reference in New Issue
Block a user