Enable Data Jobs Monitoring for Apache Airflow
Data Jobs Monitoring for Apache Airflow is in Preview
To try the preview for Airflow monitoring, follow the setup instructions below.
Data Jobs Monitoring provides visibility into the performance and reliability of workflows run by Apache Airflow DAGs.
Requirements
Setup
Data Jobs Monitoring supports Apache Airflow deployments with apache-airflow-providers-openlineage installed.
To get started, follow the instructions below.
Install openlineage provider by adding the following into your requirements.txt file or wherever your Airflow depedencies are managed:
apache-airflow-providers-openlineage>=1.11.0
Configure openlineage provider. The simplest option is to set the following environment variables and make them available to pods where you run Airflow schedulers and Airflow workers:
OPENLINEAGE_URL=<DD_DATA_OBSERVABILITY_INTAKE>
OPENLINEAGE_API_KEY=<DD_API_KEY>
- Install and configure
openlineage provider for both Airflow schedulers and Airflow workers. - Replace
<DD_DATA_OBSERVABILITY_INTAKE> with https://data-obs-intake. - Replace
<DD_API_KEY> with your valid Datadog API key.
Optional:
- Set
AIRFLOW__OPENLINEAGE__NAMESPACE with a unique name for your Airflow deployment. This allows Datadog to logically separate this deployment’s jobs from those of other Airflow deployments. - Set
OPENLINEAGE_CLIENT_LOGGING to DEBUG for OpenLineage client and its child modules. This can be useful in troubleshooting during the configuration of openlineage provider.
Check official documentation configuration-openlineage for other supported configurations of the openlineage provider.
Trigger an update to your Airflow pods and wait for them to finish.
Validation
In Datadog, view the Data Jobs Monitoring page to see a list of your Airflow job runs after the setup.
Requirements
Setup
Data Jobs Monitoring is supported for Apache Airflow deployment with apache-airflow-providers-openlineage installed.
To get started, follow the instructions below.
Install openlineage provider by adding the following into your requirements.txt file:
apache-airflow-providers-openlineage>=1.11.0
Ensure the openlineage provider version is compatible with your constraints file. If no constraints file is specified in requirements.txt, ensure compatibility with the default Apache Airflow constraints for your Airflow version. Refer to the Amazon MWAA User Guide for guidance on specifying Python dependencies in requirements.txt.
Configure openlineage provider. The simplest option is to set the following environment variables in your Amazon MWAA start script:
#!/bin/sh
export OPENLINEAGE_URL=<DD_DATA_OBSERVABILITY_INTAKE>
export OPENLINEAGE_API_KEY=<DD_API_KEY>
- Replace
<DD_DATA_OBSERVABILITY_INTAKE> fully with https://data-obs-intake. - Replace
<DD_API_KEY> fully with your valid Datadog API key.
Optional:
- Set
AIRFLOW__OPENLINEAGE__NAMESPACE with a unique name for your Airflow deployment. This allows Datadog to logically separate this deployment’s jobs from those of other Airflow deployments. - Set
OPENLINEAGE_CLIENT_LOGGING to DEBUG for OpenLineage client and its child modules. This can be useful in troubleshooting during the configuration of openlineage provider.
Check official documentation configuration-openlineage for other supported configurations of openlineage provider.
Deploy your updated requirements.txt and Amazon MWAA start script to your Amazon S3 folder configured for your Amazon MWAA Environment.
Ensure your Execution role configured for your Amazon MWAA Environment has the right permissions to the requirements.txt and Amazon MWAA start script. This is required if you are managing your own Execution role and it’s the first time you are adding those supporting files. See official guide Amazon MWAA execution role for details if needed.
Validation
In Datadog, view the Data Jobs Monitoring page to see a list of your Airflow job runs after the setup.
Requirements
Setup
Install the OpenLineage provider (apache-airflow-providers-openlineage) 1.11.0+ and openlineage-python 1.23.0+. Add the following to your requirements.txt file inside your Astro project:
apache-airflow-providers-openlineage>=1.11.0
openlineage-python>=1.23.0
To set up the OpenLineage provider, define the following environment variables. You can configure these variables in your Astronomer deployment using either of the following methods:
- From the Astro UI: Navigate to your deployment settings and add the environment variables directly.
- In the Dockerfile: Define the environment variables in your
Dockerfile to ensure they are included during the build process.
OPENLINEAGE__TRANSPORT__TYPE=composite
OPENLINEAGE__TRANSPORT__TRANSPORTS__DATADOG__TYPE=http
OPENLINEAGE__TRANSPORT__TRANSPORTS__DATADOG__URL=<DD_DATA_OBSERVABILITY_INTAKE>
OPENLINEAGE__TRANSPORT__TRANSPORTS__DATADOG__AUTH__TYPE=api_key
OPENLINEAGE__TRANSPORT__TRANSPORTS__DATADOG__AUTH__API_KEY=<DD_API_KEY>
OPENLINEAGE__TRANSPORT__TRANSPORTS__DATADOG__COMPRESSION=gzip
- replace
<DD_DATA_OBSERVABILITY_INTAKE> with https://data-obs-intake. - replace
<DD_API_KEY> with your valid Datadog API key.
Optional:
- Set
AIRFLOW__OPENLINEAGE__NAMESPACE with a unique name for your Airflow deployment. This allows Datadog to logically separate this deployment’s jobs from those of other Airflow deployments. - Set
OPENLINEAGE_CLIENT_LOGGING to DEBUG for the OpenLineage client and its child modules to log at a DEBUG logging level. This can be useful for troubleshooting during the configuration of an OpenLineage provider.
See the Astronomer official guide for managing environment variables for a deployment. See Apache Airflow’s OpenLineage Configuration Reference for other supported configurations of the OpenLineage provider.
Trigger a update to your deployment and wait for it to finish.
Validation
In Datadog, view the Data Jobs Monitoring page to see a list of your Airflow job runs after the setup.
Troubleshooting
Check that the OpenLineage environment variables are correctly set on the Astronomer deployment.
Note: Using the .env file to add the environment variables does not work because the variables are only applied to the local Airflow environment.
Advanced Configuration
Link your Spark jobs with Airflow task
You can troubleshoot Airflow tasks that run Spark jobs more efficiently by connecting the Spark job run info and telemetry with the respective Airflow task.
Prerequisites: your Spark jobs are currently monitored through Data Jobs Monitoring and are submitted through SparkSubmitOperators from your Airflow jobs.
To see the link between Airflow task and the the Spark application it submitted, follow these steps:
Configure Airflow to turn off lazy loading of Airflow plugins by setting lazy_load_plugins config to False in your airflow.cfg or exporting the following environment variable where your Airflow schedulers and Airflow workers run:
export AIRFLOW__CORE__LAZY_LOAD_PLUGINS='False'
Update your Airflow job’s DAG file by adding the following Spark configurations to your SparkSubmitOperator where you submit your Spark Application:
SparkSubmitOperator(
conf={
"spark.openlineage.parentJobNamespace": "{{ macros.OpenLineageProviderPlugin.lineage_job_namespace() }}",
"spark.openlineage.parentJobName": "{{ macros.OpenLineageProviderPlugin.lineage_job_name(task_instance) }}",
"spark.openlineage.parentRunId": "{{ macros.OpenLineageProviderPlugin.lineage_run_id(task_instance) }}",
},
)
See Lineage job & run macros for the definitions of referenced macros.
Once you have re-deployed your Airflow environment with the updated lazy_load_plugins config and the updated DAG file, and your Airflow DAG as been re-run, go to Data Jobs Monitoring page. You can then find your latest Airflow job run and see a SpanLink in the Airflow Job Run trace to the trace of the launched Spark Application. This makes it possible to debug issues in Airflow or Spark all in one place.
Further Reading
Additional helpful documentation, links, and articles:
