Acceder a la interfaz de línea de comandos de Airflow

Cloud Composer 1 | Cloud Composer 2

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 1 proporciona dos sintaxis diferentes para los comandos de la CLI:

Antes de comenzar

Debes tener permisos suficientes para usar la herramienta de línea de comandos de gcloud con Cloud Composer. Para obtener más información, consulta Control de acceso.

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 de Compute Engine donde 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 del 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 los argumentos para los comandos compuestos como argumentos de subcomandos, después de un separador --.
gcloud composer environments run example-environment \
    dags list -- --output=json

Airflow 1

Para la sintaxis de la CLI de Airflow 1.10, haz lo siguiente:

  • Especifica los comandos de la CLI como subcomando.
  • Especifica los argumentos después de un separador --.
gcloud composer environments run example-environment \
    list_dags -- --report

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

gcloud composer environments run example-environment \
    --location us-central1 trigger_dag -- sample_quickstart \
    --run_id=5077

Ejecuta comandos en un entorno de IP privada

Para ejecutar los comandos de la CLI de Airflow en un entorno de IP privada, debes usar una máquina que pueda acceder al extremo de la instancia principal del clúster de GKE. Las opciones pueden variar según la configuración del clúster privado.

Si el acceso de extremo público está inhabilitado en tu clúster, debes ejecutar comandos de Airflow desde una VM en la misma red de VPC. Crea una VM en tu red de VPC para habilitar esta ruta.

Si el acceso al extremo público y las redes autorizadas del plano de control están habilitados, también puedes ejecutar comandos de Airflow desde una máquina con una dirección IP pública que se agregue a las redes autorizadas de la instancia principal. Para habilitar el acceso desde tu máquina, ajusta la configuración del clúster de GKE:

  1. Busca el nombre del clúster de GKE que ejecuta tu entorno de Cloud Composer:

    gcloud composer environments describe \
        ENVIRONMENT_NAME \
        --location LOCATION \
        --format="value(config.gkeCluster)"
    

    Reemplaza lo siguiente:

    • ENVIRONMENT_NAME por el nombre del entorno.
    • LOCATION por la región de Compute Engine donde se encuentra el entorno.
  2. Busca la IP pública de la máquina en la que deseas ejecutar los comandos de la CLI de Airflow en tu entorno. Si usas Cloud Shell, usa dig para buscar la dirección IP externa de tu Cloud Shell:

    dig +short myip.opendns.com @resolver1.opendns.com
    
  3. Agrega la dirección externa de tu máquina a la lista de redes autorizadas de la instancia principal del clúster:

    gcloud container clusters update GKE_CLUSTER \
        --enable-master-authorized-networks \
        --master-authorized-networks EXISTING_AUTH_NETS,EXTERNAL_IP/32
    

    Reemplaza lo siguiente:

    • EXISTING_AUTH_NETS es la lista existente de redes autorizadas de instancia principal. Para buscar las redes autorizadas de la instancia principal, ejecuta el siguiente comando:

      gcloud container clusters describe GKE_CLUSTER \
          --format "flattened(masterAuthorizedNetworksConfig.cidrBlocks[])"
      
    • EXTERNAL_IP con la dirección IP externa de tu Cloud Shell o la IP pública de la máquina en la que deseas ejecutar los comandos de la CLI de Airflow

  4. Ejecuta los comandos de la CLI de Airflow en tu entorno de IP privada.

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.