def _handle_enqueue_action(message, run_queue, event_queue):
"""
Handle the ``'enqueue'`` action.
"""
run = Run('{0}{1}/'.format(
config.get('job_runner_worker', 'run_resource_uri'),
message['run_id']))
worker_list = Worker.get_list(
config.get('job_runner_worker', 'worker_resource_uri'))
if run.enqueue_dts:
logger.warning(
'Was expecting that run: {0} was not in queue yet'.format(run.id))
elif len(worker_list) != 1:
logger.warning('API returned multiple workers, expected one')
else:
run.patch({
'enqueue_dts': datetime.now(utc).isoformat(' '),
# set the worker so we know which worker of the pool claimed the
# run
'worker': worker_list[0].resource_uri,
})
run_queue.put(run)
event_queue.put(
json.dumps({
'event': 'enqueued',
'run_id': run.id,
'kind': 'run'
}))
def _handle_ping_action(message):
"""
Handle the ``'ping'`` action.
"""
worker_list = Worker.get_list(config.get("job_runner_worker", "worker_resource_uri"))
if len(worker_list) == 1:
worker_list[0].patch(
{
"ping_response_dts": datetime.now(utc).isoformat(" "),
"worker_version": job_runner_worker.__version__,
"concurrent_jobs": config.getint("job_runner_worker", "concurrent_jobs"),
}
)
else:
logger.warning("API returned multiple workers, expected one")
def _handle_ping_action(message):
"""
Handle the ``'ping'`` action.
"""
worker_list = Worker.get_list(
config.get('job_runner_worker', 'worker_resource_uri'),
params={
'api_key': config.get('job_runner_worker', 'api_key')
}
)
if len(worker_list) == 1:
worker_list[0].patch({
'ping_response_dts': datetime.now(utc).isoformat(' '),
})
else:
logger.warning('Workers by api_key query resulted in multiple results')
def _handle_ping_action(message):
"""
Handle the ``'ping'`` action.
"""
worker_list = Worker.get_list(
config.get('job_runner_worker', 'worker_resource_uri'))
if len(worker_list) == 1:
worker_list[0].patch({
'ping_response_dts':
datetime.now(utc).isoformat(' '),
'worker_version':
job_runner_worker.__version__,
'concurrent_jobs':
config.getint('job_runner_worker', 'concurrent_jobs')
})
else:
logger.warning('API returned multiple workers, expected one')
def _handle_enqueue_action(message, run_queue, event_queue):
"""
Handle the ``'enqueue'`` action.
"""
run = Run("{0}{1}/".format(config.get("job_runner_worker", "run_resource_uri"), message["run_id"]))
worker_list = Worker.get_list(config.get("job_runner_worker", "worker_resource_uri"))
if run.enqueue_dts:
logger.warning("Was expecting that run: {0} was not in queue yet".format(run.id))
elif len(worker_list) != 1:
logger.warning("API returned multiple workers, expected one")
else:
run.patch(
{
"enqueue_dts": datetime.now(utc).isoformat(" "),
# set the worker so we know which worker of the pool claimed the
# run
"worker": worker_list[0].resource_uri,
}
)
run_queue.put(run)
event_queue.put(json.dumps({"event": "enqueued", "run_id": run.id, "kind": "run"}))
