def send_start_flow_msg(dag_name, dag_run_id): step_finish_event = { "dag_name": dag_name, "dag_run_id": dag_run_id, "operation": EventOperation.Start_Flow, "time": time.time(), } print(step_finish_event) mq_broker = get_MQ_Broker_Object() mq_broker.send_msg(step_finish_event)
def send_finish_step_msg(dag_name, dag_run_id, step_name, status=None, message=None, result=None): if not status: status = StepStatus.Succeeded step_finish_event = { "dag_name": dag_name, "dag_run_id": dag_run_id, "operation": EventOperation.Finish_Step, "time": time.time(), "step_name": step_name, "status": status, "message": message, "result": result, } print(step_finish_event) mq_broker = get_MQ_Broker_Object() mq_broker.send_msg(step_finish_event)
__author__ = 'godq' import unittest import time from dagflow.loader import get_MQ_Broker_Object, get_DagRepo_Object from dagflow.event import EventOperation from dagflow.flow_operation import send_finish_step_msg, send_start_flow_msg mq_broker = get_MQ_Broker_Object() dag_repo = get_DagRepo_Object() class StartFlowAsyncEventTest(unittest.TestCase): def test_start_async_flow(self): dag_name = "dag_def_test_{}".format(time.time()) dag = { "name": dag_name, "steps": [ { "name": "step1", "task_func": "hello_plugin", "args": { "name": "hello world1" }, "upstreams": [], "downstreams": ["step2"] }, { "name": "step2",
def send_event_message(event): if isinstance(event, dict): event = json.dumps(event) # message_queue.put_message(event) mq_broker = get_MQ_Broker_Object() mq_broker.send_msg(event)