Why monitoring prevents silent pipeline failures in Apache Airflow - See It in Action

Why monitoring prevents silent pipeline failures

📖 Scenario: You are managing a data pipeline using Apache Airflow. Sometimes, tasks fail silently without you noticing, causing data issues downstream. To avoid this, you want to add simple monitoring that alerts you when a task fails.

🎯 Goal: Build a basic Airflow DAG with a task that can fail. Add a monitoring step that checks the task status and prints an alert if the task failed. This will help you understand how monitoring prevents silent failures in pipelines.

📋 What You'll Learn

Create an Airflow DAG with one task that can fail

Add a variable to simulate failure condition

Write a Python function to check task status and print alert if failed

Print the monitoring alert message as output

💡 Why This Matters

🌍 Real World

In real data pipelines, tasks can fail without obvious errors. Monitoring helps detect these failures early to prevent bad data or downtime.

💼 Career

DevOps engineers and data engineers use monitoring to maintain reliable pipelines and quickly respond to issues.

1

Create a failing task in Airflow DAG

Create a Python Airflow DAG named monitoring_dag with one task called task_to_monitor that runs a Python function possibly_failing_task. The function should raise an exception if a variable should_fail is True.

Apache Airflow

from airflow import DAG
from airflow.operators.python import PythonOperator
from datetime import datetime

# Your code here

Hint

Define a Python function that raises an exception if should_fail is True. Then create a DAG and a PythonOperator task that calls this function.

2

Add a monitoring function to check task status

Add a Python function called monitor_task_status that takes a parameter task_instance. It should check if task_instance.state is 'failed' and print 'ALERT: Task failed!' if so.

Apache Airflow

from airflow import DAG
from airflow.operators.python import PythonOperator
from datetime import datetime

should_fail = True

def possibly_failing_task():
    if should_fail:
        raise Exception("Task failed due to simulated error")

def monitor_task_status(task_instance):
    # Your code here

with DAG(dag_id='monitoring_dag', start_date=datetime(2024, 1, 1), schedule_interval='@daily', catchup=False) as dag:
    task_to_monitor = PythonOperator(
        task_id='task_to_monitor',
        python_callable=possibly_failing_task
    )

Hint

Check the state attribute of task_instance. If it equals 'failed', print the alert message.

3

Add a task to run the monitoring function

Add a PythonOperator task called monitoring_task to the DAG that calls monitor_task_status. Pass the task_to_monitor instance to it using op_kwargs={'task_instance': task_to_monitor}. Set task_to_monitor to run before monitoring_task.

Apache Airflow

from airflow import DAG
from airflow.operators.python import PythonOperator
from datetime import datetime

should_fail = True

def possibly_failing_task():
    if should_fail:
        raise Exception("Task failed due to simulated error")

def monitor_task_status(task_instance):
    if task_instance.state == 'failed':
        print('ALERT: Task failed!')

with DAG(dag_id='monitoring_dag', start_date=datetime(2024, 1, 1), schedule_interval='@daily', catchup=False) as dag:
    task_to_monitor = PythonOperator(
        task_id='task_to_monitor',
        python_callable=possibly_failing_task
    )
    # Your code here

Hint

Create a new PythonOperator that calls monitor_task_status and pass task_to_monitor as task_instance using op_kwargs. Set the task order so task_to_monitor runs before monitoring_task.

4

Print the monitoring alert output

Add a print statement inside monitor_task_status that outputs 'ALERT: Task failed!' when the task state is 'failed'. Run the DAG and show the printed alert message as output.

Apache Airflow

from airflow import DAG
from airflow.operators.python import PythonOperator
from datetime import datetime

should_fail = True

def possibly_failing_task():
    if should_fail:
        raise Exception("Task failed due to simulated error")

def monitor_task_status(task_instance):
    if task_instance.state == 'failed':
        # Your code here

with DAG(dag_id='monitoring_dag', start_date=datetime(2024, 1, 1), schedule_interval='@daily', catchup=False) as dag:
    task_to_monitor = PythonOperator(
        task_id='task_to_monitor',
        python_callable=possibly_failing_task
    )
    monitoring_task = PythonOperator(
        task_id='monitoring_task',
        python_callable=monitor_task_status,
        op_kwargs={'task_instance': task_to_monitor}
    )
    task_to_monitor >> monitoring_task

Hint

Make sure the print statement is inside the if block checking for failure. When the DAG runs, this message should appear if the task failed.