refactor: change heartbeat file dir.

This commit is contained in:
liqiang-fit2cloud 2025-11-28 09:35:52 +08:00
parent f67aef11fd
commit 9f18d3ce07

View File

@ -6,14 +6,14 @@ from celery.signals import heartbeat_sent, worker_ready, worker_shutdown
@heartbeat_sent.connect
def heartbeat(sender, **kwargs):
worker_name = sender.eventer.hostname.split('@')[0]
heartbeat_path = Path('/tmp/worker_heartbeat_{}'.format(worker_name))
heartbeat_path = Path('/opt/maxkb-app/tmp/worker_heartbeat_{}'.format(worker_name))
heartbeat_path.touch()
@worker_ready.connect
def worker_ready(sender, **kwargs):
worker_name = sender.hostname.split('@')[0]
ready_path = Path('/tmp/worker_ready_{}'.format(worker_name))
ready_path = Path('/opt/maxkb-app/tmp/worker_ready_{}'.format(worker_name))
ready_path.touch()
@ -21,5 +21,5 @@ def worker_ready(sender, **kwargs):
def worker_shutdown(sender, **kwargs):
worker_name = sender.hostname.split('@')[0]
for signal in ['ready', 'heartbeat']:
path = Path('/tmp/worker_{}_{}'.format(signal, worker_name))
path = Path('/opt/maxkb-app/tmp/worker_{}_{}'.format(signal, worker_name))
path.unlink(missing_ok=True)