ai.databand.schema

Class AirflowTaskContext

java.lang.Object

ai.databand.schema.AirflowTaskContext

public class AirflowTaskContext
extends java.lang.Object

Constructor Summary

Constructors

Constructor and Description
AirflowTaskContext(java.lang.String airflowInstanceUid, java.lang.String airflowName, java.lang.String dagId, java.lang.String executionDate, java.lang.String taskId, java.lang.String tryNumber)

Method Summary

Modifier and Type	Method and Description
protected java.lang.String	buildAirflowJobRunUid() def get_job_run_uid(dag_id, execution_date): if airflow_instance_uid is None: return uuid.uuid5(NAMESPACE_DBND_RUN, "{}:{}".format(dag_id, execution_date)) else: return uuid.uuid5(NAMESPACE_DBND_RUN, "{}:{}:{}".format(airflow_instance_uid, dag_id, execution_date))
protected java.lang.String	buildAirflowJobUid(AirflowTaskContext context) def get_job_uid(dag_id): if airflow_server_info_uid: return uuid.uuid5(NAMESPACE_DBND_JOB, "{}:{}".format(airflow_server_info_uid, dag_id)) else: return uuid.uuid5(NAMESPACE_DBND_JOB, dag_id)
protected java.lang.String	buildAirflowTaskDefUid(AirflowTaskContext context) def get_task_def_uid(dag_id, task_id): return uuid.uuid5(NAMESPACE_DBND_TASK_DEF, "{}.{}".format(dag_id, task_id))
protected java.lang.String	buildAirflowTaskRunAttemptUid(java.lang.String runUid, AirflowTaskContext context) def get_task_run_attempt_uid(run_uid, dag_id, task_id, try_number): return uuid.uuid5(run_uid, "{}.{}:{}".format(dag_id, task_id, try_number))
protected java.lang.String	buildAirflowTaskRunUid(java.lang.String runUid) def get_task_run_uid(run_uid, dag_id, task_id): return uuid.uuid5(run_uid, "{}.{}".format(dag_id, task_id))
java.lang.String	getAfOperatorUid()
java.lang.String	getAirflowInstanceUid()
java.lang.String	getAirflowName()
java.lang.String	getDagId()
java.lang.String	getExecutionDate()
java.lang.String	getTaskId()
java.lang.String	getTryNumber()
java.lang.String	jobName()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

AirflowTaskContext

public AirflowTaskContext(java.lang.String airflowInstanceUid,
                          java.lang.String airflowName,
                          java.lang.String dagId,
                          java.lang.String executionDate,
                          java.lang.String taskId,
                          java.lang.String tryNumber)

Method Detail

getAirflowInstanceUid

public java.lang.String getAirflowInstanceUid()

getAirflowName

public java.lang.String getAirflowName()

getDagId

public java.lang.String getDagId()

getExecutionDate

public java.lang.String getExecutionDate()

getTaskId

public java.lang.String getTaskId()

getTryNumber

public java.lang.String getTryNumber()

getAfOperatorUid

public java.lang.String getAfOperatorUid()

buildAirflowJobRunUid

protected java.lang.String buildAirflowJobRunUid()

def get_job_run_uid(dag_id, execution_date): if airflow_instance_uid is None: return uuid.uuid5(NAMESPACE_DBND_RUN, "{}:{}".format(dag_id, execution_date)) else: return uuid.uuid5(NAMESPACE_DBND_RUN, "{}:{}:{}".format(airflow_instance_uid, dag_id, execution_date))

buildAirflowTaskRunUid

protected java.lang.String buildAirflowTaskRunUid(java.lang.String runUid)

def get_task_run_uid(run_uid, dag_id, task_id): return uuid.uuid5(run_uid, "{}.{}".format(dag_id, task_id))

buildAirflowTaskDefUid

protected java.lang.String buildAirflowTaskDefUid(AirflowTaskContext context)

def get_task_def_uid(dag_id, task_id): return uuid.uuid5(NAMESPACE_DBND_TASK_DEF, "{}.{}".format(dag_id, task_id))

buildAirflowTaskRunAttemptUid

protected java.lang.String buildAirflowTaskRunAttemptUid(java.lang.String runUid,
                                                         AirflowTaskContext context)

def get_task_run_attempt_uid(run_uid, dag_id, task_id, try_number): return uuid.uuid5(run_uid, "{}.{}:{}".format(dag_id, task_id, try_number))

buildAirflowJobUid

protected java.lang.String buildAirflowJobUid(AirflowTaskContext context)

def get_job_uid(dag_id): if airflow_server_info_uid: return uuid.uuid5(NAMESPACE_DBND_JOB, "{}:{}".format(airflow_server_info_uid, dag_id)) else: return uuid.uuid5(NAMESPACE_DBND_JOB, dag_id)

jobName

public java.lang.String jobName()