Hubungkan ke VM GCE dengan SSHOperator

Cloud Composer 1 | Cloud Composer 2 | Cloud Composer 3

Halaman ini menjelaskan cara terhubung ke VM Compute Engine dari DAG.

Membuat DAG yang terhubung ke instance VM Compute Engine

Di parameter ssh_hook dari SSHOperator, gunakan ComputeEngineSSHHook dengan parameter yang mengarah ke ke VM Compute Engine.

Contoh berikut menunjukkan cara menggunakan SSHOperator untuk menjalankan perintah di di instance VM Compute Engine.

Ganti nilai:

  • GCE_INSTANCE dengan nama instance VM.
  • GCE_ZONE dengan zona Compute Engine tempat VM berada.
  • GCP_PROJECT_ID dengan Project ID project tempat VM dan lingkungan yang menjalankan DAG berada. 
import datetime

import airflow
from airflow.providers.ssh.operators.ssh import SSHOperator
from airflow.providers.google.cloud.hooks.compute_ssh import ComputeEngineSSHHook

GCE_INSTANCE = 'example-compute-instance'
GCE_ZONE = 'us-central1-a'
GCP_PROJECT_ID = 'example-project'

with airflow.DAG(
        'composer_compute_ssh_dag',
        start_date=datetime.datetime(2022, 1, 1)
        ) as dag:

  ssh_task = SSHOperator(
      task_id='composer_compute_ssh_task',
      ssh_hook=ComputeEngineSSHHook(
          instance_name=GCE_INSTANCE,
          zone=GCE_ZONE,
          project_id=GCP_PROJECT_ID,
          use_oslogin=True,
          use_iap_tunnel=False,
          use_internal_ip=True),
      command='echo This command is executed from a DAG',
      dag=dag)

Langkah selanjutnya