如何解决气流:在自定义的 SubDagOperator 中放大 Sub Dag 按钮
我设计了一个定制的 SubDagOperator
。一切正常,只是没有出现“放大 Sub Dag”按钮。似乎如果气流 UI 无法将任务识别为子操作符本身,则不会显示按钮。我试图覆盖 task_type
属性,正如它在旧问题中提到的那样,但它对我不起作用。您知道是否可以看到带有自定义 SubDagOperators 的按钮?
气流版本:1.10.12
这是我的尝试:
class EmrSubdagOperator(SubDagOperator):
template_fields = ()
template_ext = ()
@apply_defaults
def __init__(self,*args,**kwargs):
dag = kwargs.get('dag')
task_id = kwargs.get('task_id')
spark_steps = kwargs.get('spark_steps')
job_flow_overrides = kwargs.get('job_flow_overrides')
subdag = DAG(
'{}.{}'.format(dag.dag_id,task_id),schedule_interval=dag.schedule_interval,start_date=dag.start_date
)
cluster_creator = EmrCreateJobFlowOperator(
dag=subdag,task_id='create_job_flow',job_flow_overrides=job_flow_overrides
)
step_adder = EmrAddStepsOperator(
dag=subdag,task_id='add_steps',job_flow_id="{{ task_instance.xcom_pull(task_ids='create_job_flow',key='return_value') }}",aws_conn_id='aws_default',steps=spark_steps,)
#Check if the last step is completed,skipped or terminated
last_step = len(spark_steps) - 1
step_checker = EmrStepSensor(
dag=subdag,task_id='watch_step',job_flow_id="{{ task_instance.xcom_pull('create_job_flow',step_id="{{ task_instance.xcom_pull(task_ids='add_steps',key='return_value')[" + str(last_step) +"]}}",)
cluster_checker = EmrJobFlowSensor(
dag=subdag,task_id='watch_cluster',)
super(EmrSubdagOperator,self).__init__(subdag=subdag,**kwargs)
self.spark_steps = spark_steps
self.job_flow_overrides = job_flow_overrides
@property
def task_type(self):
return 'SubDagOperator'
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。