Acceder a la base de datos de Airflow

Cloud Composer 3 | Cloud Composer 2 | Cloud Composer 1

En esta página se explica cómo conectarse a una instancia de Cloud SQL que ejecuta la base de datos de Airflow de tu entorno de Cloud Composer y cómo ejecutar consultas SQL.

Por ejemplo, puede que quieras ejecutar consultas directamente en la base de datos de Airflow, hacer copias de seguridad de la base de datos, recoger estadísticas basadas en el contenido de la base de datos o recuperar cualquier otra información personalizada de la base de datos.

Antes de empezar

Ejecutar una consulta de SQL en la base de datos de Airflow

Para conectarte a la base de datos de Airflow, sigue estos pasos:

  1. Crea un DAG con uno o varios operadores SQLExecuteQueryOperator. Para empezar, puedes usar el DAG de ejemplo.

    .

  2. En el parámetro sql del operador, especifica tu consulta de SQL.

  3. Sube este DAG a tu entorno.

  4. Activa el DAG. Por ejemplo, puedes hacerlo manualmente o esperar a que se ejecute según una programación.

DAG de ejemplo:

import datetime
import os

import airflow
from airflow.providers.common.sql.operators.sql import SQLExecuteQueryOperator

SQL_DATABASE = os.environ["SQL_DATABASE"]

with airflow.DAG(
    "airflow_db_connection_example",
    start_date=datetime.datetime(2025, 1, 1),
    schedule_interval=None,
    catchup=False) as dag:

    SQLExecuteQueryOperator(
        task_id="run_airflow_db_query",
        dag=dag,
        conn_id="airflow_db",
        database=SQL_DATABASE,
        sql="SELECT * FROM dag LIMIT 10;",
    )

Para obtener más información sobre cómo usar SQLExecuteQueryOperator, consulta la guía práctica de Postgres con SQLExecuteQueryOperator en la documentación de Airflow.

Volcar el contenido de una base de datos y transferirlo a un segmento

Siguientes pasos