mirror of https://github.com/jumpserver/jumpserver
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
26 lines
774 B
26 lines
774 B
2 years ago
|
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)
|