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"}))