Migrer des environnements vers Cloud Composer 2 (à partir d'Airflow 1) à l'aide d'instantanés

Cloud Composer 3 | Cloud Composer 2 | Cloud Composer 1

Cette page explique comment transférer des DAG, des données et une configuration de vos environnements Cloud Composer 1, Airflow 1 existants vers Cloud Composer 2, Airflow 2.

Ce guide de migration utilise la fonctionnalité Snapshots (Instantanés).

Autres guides de migration

De À Méthode Guide
Cloud Composer 1, Airflow 2 Cloud Composer 2, Airflow 2 Côte à côte, à l'aide d'instantanés Guide de migration (instantanés)
Cloud Composer 1, Airflow 1 Cloud Composer 2, Airflow 2 Côte à côte, à l'aide d'instantanés Ce guide (instantanés)
Cloud Composer 1, Airflow 2 Cloud Composer 2, Airflow 2 Transfert manuel côte à côte Guide de migration manuelle
Cloud Composer 1, Airflow 1 Cloud Composer 2, Airflow 2 Transfert manuel côte à côte Guide de migration manuelle
Airflow 1 Airflow 2 Transfert manuel côte à côte Guide de migration manuelle

Avant de commencer

  • Les instantanés sont compatibles avec Cloud Composer 2 version 2.0.9 et ultérieure. Cloud Composer 1 est compatible avec l'enregistrement d'instantanés d'environnement dans la version 1.18.5.

  • Cloud Composer prend en charge la migration côte à côte de Cloud Composer 1 vers Cloud Composer 2. Il n'est pas possible de passer de Cloud Composer 1 à Cloud Composer 2 en place.

  • Consultez la liste des différences entre Cloud Composer 1 et Cloud Composer 2.

  • La taille maximale de la base de données Airflow compatible avec les instantanés est de 20 Go. Si la base de données de votre environnement occupe plus de 20 Go, réduisez sa taille.

  • Pour créer des instantanés, le nombre total d'objets dans les dossiers /dags,/plugins et /data du bucket de l'environnement doit être inférieur à 100 000.

  • Si vous utilisez le mécanisme XCom pour transférer des fichiers, assurez-vous de l'utiliser conformément aux consignes d'Airflow. Le transfert de fichiers volumineux ou d'un grand nombre de fichiers à l'aide de XCom a un impact sur les performances de la base de données Airflow et peut entraîner des échecs lors du chargement d'instantanés ou de la mise à niveau de votre environnement. Pensez à utiliser des solutions alternatives telles que Cloud Storage pour transférer de grands volumes de données.

  • Comme Cloud Composer 2 utilise Airflow 2, la migration inclut le passage de vos DAG et de la configuration de votre environnement à Airflow 2. Consultez le guide de migration d'Airflow 1 vers Airflow 2 pour en savoir plus sur les modifications non rétrocompatibles entre Airflow 1 et Airflow 2 dans Cloud Composer.

  • Dans ce guide, vous combinez la migration vers Airflow 2 et la migration vers Cloud Composer 2 dans une seule procédure de migration. De cette manière, vous n'avez pas besoin de migrer vers un environnement Cloud Composer 1 avec Airflow 2 avant de migrer vers Cloud Composer 2.

Étape 1 : Passer à Airflow 1.10.15

Si votre environnement utilise une version d'Airflow antérieure à 1.10.15, mettez à niveau votre environnement vers une version Cloud Composer qui utilise Airflow 1.10.15 et est compatible avec les instantanés.

Étape 2 : Vérifier la compatibilité avec Airflow 2

Pour vérifier les conflits potentiels avec Airflow 2, consultez la section sur la mise à jour des DAG du guide de mise à niveau vers Airflow 2.0 et versions ultérieures.

Un problème courant que vous pouvez rencontrer est lié aux chemins d'importation incompatibles. Pour plus d'informations sur la résolution de ce problème de compatibilité, dans le guide de mise à niveau vers Airflow 2.0 et versions ultérieures, consultez la section sur les fournisseurs de rétroportage.

Étape 3: Assurez-vous que vos DAG sont prêts pour Airflow 2

Avant de transférer des DAG vers votre environnement Cloud Composer 2, assurez-vous que les conditions suivantes sont remplies:

  1. Vos DAG s'exécutent correctement et il n'y a plus de problèmes de compatibilité.

  2. Vos DAG utilisent des instructions d'importation correctes.

    Par exemple, la nouvelle instruction d'importation pour BigQueryCreateDataTransferOperator peut se présenter comme suit :

    from airflow.providers.google.cloud.operators.bigquery_dts \
    import BigQueryCreateDataTransferOperator

  3. Vos DAG sont mis à jour pour Airflow 2. Cette modification est compatible avec Airflow 1.10.14 et versions ultérieures.

Étape 4: Suspendre les DAG dans votre environnement Cloud Composer 1

Pour éviter les exécutions de DAG en double, suspendez tous les DAG de votre environnement Cloud Composer 1 avant d'enregistrer son instantané.

Vous pouvez utiliser l'une des options suivantes:

  • Dans l'interface Web Airflow, accédez aux DAG et suspendez manuellement tous les DAG.

  • Utilisez le script composer_dags pour suspendre tous les DAG:

    python3 composer_dags.py --environment COMPOSER_1_ENV \
  --project PROJECT_ID \
  --location COMPOSER_1_LOCATION \
  --operation pause

    Remplacez :

    • COMPOSER_1_ENV par le nom de votre environnement Cloud Composer 1.
    • PROJECT_ID par l'ID du projet.
    • COMPOSER_1_LOCATION par la région où se trouve l'environnement.

Étape 5: Enregistrer l'instantané de votre environnement Cloud Composer 1

Console

Créez un instantané de votre environnement:

  1. Dans la console Google Cloud, accédez à la page Environnements.

    Accéder à la page Environnements

  2. Dans la liste des environnements, cliquez sur le nom de votre environnement Cloud Composer 1. La page Détails de l'environnement s'ouvre.

  3. Cliquez sur Create snapshot (Créer un instantané).

  4. Dans la boîte de dialogue Créer un instantané, cliquez sur Envoyer. Dans ce guide, vous enregistrez l'instantané dans le bucket de l'environnement Cloud Composer 1, mais vous pouvez sélectionner un autre emplacement si vous le souhaitez.

  5. Attendez que Cloud Composer crée l'instantané.

gcloud

  1. Obtenez l'URI du bucket de votre environnement Cloud Composer 1:

    1. Exécutez la commande suivante :

      gcloud composer environments describe COMPOSER_1_ENV \
    --location COMPOSER_1_LOCATION \
     --format="value(config.dagGcsPrefix)"

      Remplacez :

      • COMPOSER_1_ENV par le nom de votre environnement Cloud Composer 1.
      • COMPOSER_1_LOCATION par la région dans laquelle se trouve l'environnement.

    2. Dans le résultat, supprimez le dossier /dags. Le résultat correspond à l'URI du bucket de votre environnement Cloud Composer 1.

      Par exemple, remplacez gs://us-central1-example-916807e1-bucket/dags par gs://us-central1-example-916807e1-bucket.

  2. Créez un instantané de votre environnement Cloud Composer 1:

    gcloud composer environments snapshots save \
  COMPOSER_1_ENV \
  --location COMPOSER_1_LOCATION \
  --snapshot-location "COMPOSER_1_SNAPSHOTS_FOLDER"

    Remplacez :

    • COMPOSER_1_ENV par le nom de votre environnement Cloud Composer 1.
    • COMPOSER_1_LOCATION par la région où se trouve l'environnement Cloud Composer 1.
    • COMPOSER_1_SNAPSHOTS_FOLDER par l'URI du bucket de votre environnement Cloud Composer 1. Dans ce guide, vous enregistrez l'instantané dans le bucket de l'environnement Cloud Composer 1, mais vous pouvez sélectionner un autre emplacement si vous le souhaitez. Si vous spécifiez un emplacement personnalisé, les comptes de service des deux environnements doivent disposer d'autorisations de lecture et d'écriture pour l'emplacement spécifié.

Étape 6: Créer un environnement Cloud Composer 2

Créez un environnement Cloud Composer 2. Vous pouvez commencer par un environnement prédéfini correspondant aux besoins de ressources prévus, puis procéder au scaling et à l'optimisation supplémentaires de votre environnement.

Vous n'avez pas besoin de spécifier des remplacements de configuration et des variables d'environnement, car vous les remplacerez plus tard lorsque vous chargerez l'instantané de votre environnement Cloud Composer 1.

Certaines options de configuration d'Airflow 1 utilisent un nom et une section différents dans Airflow 2. Pour en savoir plus, consultez la section Modifications de configuration.

Étape 7: Charger l'instantané dans votre environnement Cloud Composer 2

Console

Pour charger l'instantané dans votre environnement Cloud Composer 2:

  1. Dans la console Google Cloud, accédez à la page Environnements.

    Accéder à la page Environnements

  2. Dans la liste des environnements, cliquez sur le nom de votre environnement Cloud Composer 2. La page Détails de l'environnement s'ouvre.

  3. Cliquez sur Charger un instantané.

  4. Dans la boîte de dialogue Charger un instantané, cliquez sur Parcourir.

  5. Sélectionnez le dossier contenant l'instantané. Si vous utilisez l'emplacement par défaut de ce guide, ce dossier se trouve dans le bucket de votre environnement Cloud Composer 1, dans le dossier /snapshots. Son nom correspond au code temporel de l'opération d'enregistrement de l'instantané. Exemple : us-central1-example-916807e1-bucket/snapshots_example-project_us-central1_example-environment/2022-01-05T18-59-00.

  6. Cliquez sur Load (Charger) et attendez que Cloud Composer charge l'instantané.

gcloud

Chargez l'instantané de votre environnement Cloud Composer 1 dans votre environnement Cloud Composer 2:

gcloud composer environments snapshots load \
  COMPOSER_2_ENV \
  --location COMPOSER_2_LOCATION \
  --snapshot-path "SNAPSHOT_PATH"

Remplacez :

  • COMPOSER_2_ENV par le nom de votre environnement Cloud Composer 2.
  • COMPOSER_2_LOCATION par la région où se trouve l'environnement Cloud Composer 2.
  • SNAPSHOT_PATH par l'URI du bucket de votre environnement Cloud Composer 1, suivi du chemin d'accès à l'instantané. Par exemple, gs://us-central1-example-916807e1-bucket/snapshots/example-project_us-central1_example-environment_2022-01-05T18-59-00.

Étape 8: Réactiver les DAG dans l'environnement Cloud Composer 2

Vous pouvez utiliser l'une des options suivantes:

  • Dans l'interface Web d'Airflow, accédez à DAG et réactivez manuellement tous les DAG un par un.

  • Utilisez le script composer_dags pour réactiver tous les DAG:

    python3 composer_dags.py --environment COMPOSER_2_ENV \
  --project PROJECT_ID \
  --location COMPOSER_2_LOCATION \
  --operation unpause

    Remplacez :

    • COMPOSER_2_ENV par le nom de votre environnement Cloud Composer 2.
    • PROJECT_ID par l'ID du projet.
    • COMPOSER_2_LOCATION par la région où se trouve l'environnement.

  • (versions Airflow 2.9.1 et ultérieures) En cas d'erreurs de quota lors de la réactivation d'un grand nombre de DAG, vous pouvez utiliser les commandes CLI Airflow suivantes pour réactiver tous les DAG en même temps:

    gcloud composer environments run COMPOSER_2_ENV dags unpause \
  --project PROJECT_ID \
  --location COMPOSER_2_LOCATION \
  -- -y --treat-dag-id-as-regex ".*"

  • (Versions d'Airflow antérieures à 2.9.1) En cas d'erreurs de quota lors de la réactivation d'un grand nombre de DAG, vous pouvez les réactiver à l'aide de l'API REST Airflow. Consultez également la section Essayer l'API dans la documentation Airflow.

Étape 9: Recherchez des erreurs dans les DAG

  1. Dans l'interface Web Airflow, accédez aux DAG et recherchez les erreurs de syntaxe des DAG signalés.

  2. Vérifiez que les exécutions du DAG sont planifiées au bon moment.

  3. Attendez que les exécutions du DAG se produisent dans l'environnement Cloud Composer 2 et vérifiez si elles ont réussi. Si l'exécution du DAG a réussi, ne la réactivez pas dans l'environnement Cloud Composer 1. Si vous le faites, un DAG s'exécutera à la même heure et à la même date dans votre environnement Cloud Composer 1.

  4. Si une exécution de DAG spécifique échoue, essayez de résoudre les problèmes liés au DAG jusqu'à ce que le DAG s'exécute avec succès dans Cloud Composer 2.

Étape 10: Surveiller votre environnement Cloud Composer 2

Une fois que vous avez transféré tous les DAG et la configuration vers l'environnement Cloud Composer 2, surveillez-les pour détecter les problèmes potentiels, les exécutions de DAG ayant échoué et l'état général de l'environnement.

Si l'environnement Cloud Composer 2 s'exécute sans problème pendant une période suffisante, envisagez de supprimer l'environnement Cloud Composer 1.

Étape suivante