mirror of https://github.com/jumpserver/jumpserver
26 lines
774 B
Python
26 lines
774 B
Python
|
from pathlib import Path
|
||
|
|
||
|
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.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.touch()
|
||
|
|
||
|
|
||
|
@worker_shutdown.connect
|
||
|
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.unlink(missing_ok=True)
|