Accede a la CLI de Airflow

Cloud Composer 1 | Cloud Composer 2 | Cloud Composer 3

Apache Airflow tiene una interfaz de línea de comandos (CLI) que puedes usar para realizar tareas como activar y administrar DAG, obtener información sobre las ejecuciones y tareas de DAG, agregar y borrar conexiones y usuarios.

Acerca de las versiones de la sintaxis de la CLI

Airflow en Cloud Composer 2 usa la sintaxis de la CLI de Airflow 2.

Comandos compatibles con la CLI de Airflow

Para obtener una lista completa de los comandos de la CLI de Airflow admitidos, consulta la referencia de gcloud composer environments run.

Antes de comenzar

Ejecuta comandos de la CLI de Airflow

Para ejecutar comandos de la CLI de Airflow en tus entornos, usa gcloud:

gcloud composer environments run ENVIRONMENT_NAME \
    --location LOCATION \
    SUBCOMMAND \
    -- SUBCOMMAND_ARGUMENTS

Reemplaza lo siguiente:

  • ENVIRONMENT_NAME por el nombre del entorno.
  • LOCATION por la región en la que se encuentra el entorno
  • SUBCOMMAND por uno de los comandos de la CLI de Airflow admitidos
  • SUBCOMMAND_ARGUMENTS por argumentos para el comando de la CLI de Airflow

Separador de argumentos de subcomando

Separa los argumentos para el comando de la CLI de Airflow especificado con --:

Airflow 2

Para la sintaxis de la CLI de Airflow 2, sigue estos pasos:

  • Especifica los comandos compuestos de la CLI como subcomando.
  • Especifica argumentos para comandos compuestos como argumentos de subcomandos. después de un separador --.
gcloud composer environments run example-environment \
    dags list -- --output=json

Airflow 1

Cloud Composer 2 solo es compatible con Airflow 2.

Ubicación predeterminada

La mayoría de los comandos gcloud composer requiere una ubicación. Puedes especificar la ubicación con la marca --location o si configuras la ubicación predeterminada.

Ejemplo

Por ejemplo, para activar un DAG con el nombre sample_quickstart y el ID 5077 en tu entorno de Cloud Composer, ejecuta lo siguiente:

Airflow 2 

gcloud composer environments run example-environment \
    --location us-central1 dags trigger -- sample_quickstart \
    --run-id=5077

Airflow 1

Cloud Composer 2 solo es compatible con Airflow 2.

Ejecuta comandos en un entorno de IP privada

A partir de la versión 2.4.0 de Cloud Composer, puedes ejecutar la CLI de Airflow en un entorno de IP privada sin configuración adicional. Tu no necesita acceso al plano de control del clúster del entorno extremo para ejecutar estos comandos.

En las versiones de Cloud Composer anteriores a la 2.4.0, haz lo siguiente:

Para ejecutar comandos de la CLI de Airflow en un entorno de IP privada, sigue estos pasos: ejecutarlas en una máquina que pueda acceder a la clúster de GKE extremo del plano de control. Las opciones pueden variar según la configuración del clúster privado.

Si el acceso al extremo público está inhabilitado en el clúster de tu entorno, no es posible usar los comandos de gcloud composer para ejecutar la CLI de Airflow. Para poder ejecutar los comandos de la CLI de Airflow, sigue estos pasos:

  1. Crea una VM en tu red de VPC
  2. Adquiere credenciales de clúster. Ejecuta el siguiente comando: bash gcloud container clusters get-credentials CLUSTER_NAME \ --region REGION \ --project PROJECT \ --internal-ip
  • Usa kubectl para ejecutar tu comando de Airflow. Por ejemplo:
kubectl exec deployment/airflow-scheduler -n COMPOSER_NAMESPACE \
  --container airflow-scheduler -- airflow dags list

Reemplaza COMPOSER_NAMESPACE por un espacio de nombres similar a composer-2-0-28-airflow-2-3-394zxc12411. Puedes encontrar Cloud Composer en la lista de cargas de trabajo o con el comando kubectl get namespaces.

Si el acceso al extremo público está habilitado en el clúster de tu entorno, también puedes ejecutar comandos de la CLI de Airflow desde una máquina con una dirección IP externa que se agregue a las redes autorizadas. Para habilitar el acceso desde tu máquina, agrega el una dirección externa de la máquina a la de tu entorno lista de redes autorizadas.

Ejecuta comandos de la CLI de Airflow a través de la API de Cloud Composer

A partir de la versión 2.4.0 de Cloud Composer, puedes ejecutar comandos de la CLI de Airflow a través de la API de Cloud Composer.

Ejecutar un comando

Crea una solicitud a la API de environments.executeAirflowCommand:

{
  "environment": "projects/PROJECT_ID/locations/LOCATION/environments/ENVIRONMENT_NAME",
  "command": "AIRFLOW_COMMAND",
  "subcommand": "AIRFLOW_SUBCOMMAND",
  "parameters": [
    "SUBCOMMAND_PARAMETER"
  ]
}

Reemplaza lo siguiente:

  • PROJECT_ID: El ID del proyecto.
  • LOCATION: Es la región en la que se encuentra el entorno.
  • ENVIRONMENT_NAME: Es el nombre de tu entorno.
  • AIRFLOW_COMMAND: El comando de la CLI de Airflow que deseas ejecutar, como dags.
  • AIRFLOW_SUBCOMMAND: Subcomando del comando de la CLI de Airflow que que deseas ejecutar, como list.
  • SUBCOMMAND_PARAMETER: Son los parámetros del subcomando (opcional). Si Si desea usar más de un parámetro, agregue más elementos a la lista.

Ejemplo:

// POST https://composer.googleapis.com/v1/{environment=projects/*/locations/*/environments/*}:executeAirflowCommand
{
  "environment": "projects/example-project/locations/us-central1/environments/example-environment",
  "command": "dags",
  "subcommand": "list",
  "parameters": [
    "-o json",
    "--verbose"
  ]
}

Estado del comando de la encuesta

Después de ejecutar un comando de la CLI de Airflow a través de la API de Cloud Composer, verifica lo siguiente: el comando se completó correctamente creando un PollAirflowCommand y, luego, inspeccionar en exitInfo para ver los errores y códigos de estado. El campo output contiene las líneas de registro.

Para obtener el estado de ejecución del comando y recuperar registros, proporciona executionId. Valores de pod y podNamespace que muestra ExecuteAirflowCommandRequest:

Ejemplo:

// POST https://composer.googleapis.com/v1/{environment=projects/*/locations/*/environments/*}:pollAirflowCommand
{
  "executionId": "39b82312-3a19-4d21-abac-7f8f19855ce7",
  "pod": "airflow-scheduler-1327d8cd68-hblpd",
  "podNamespace": "composer-2-4-0-airflow-2-5-3-184dadaf",
  "nextLineNumber": 1
}

Soluciona problemas

Sin conectividad al plano de control del clúster

Cuando se ejecuta gcloud composer environments run o kubectl, es posible que encuentres el siguiente error:

Get "https://<IP Address>/api?timeout=32s": dial tcp <IP Address>:443: i/o timeout"

Síntoma: Este mensaje de error indica que no hay conectividad de red desde una computadora en la que ejecutas estos comandos.

Solución: Sigue los lineamientos que se presentan en la sección Ejecuta comandos en un entorno de IP privada o usa las instrucciones disponibles en la sección El comando kubectl se agota.