Cloud Composer 1 | Cloud Composer 2 | Cloud Composer 3
Panduan ini menunjukkan cara menulis directed acyclic graph (DAG) Apache Airflow yang berjalan di lingkungan Cloud Composer.
Karena Apache Airflow tidak menyediakan isolasi tugas dan DAG yang kuat, sebaiknya Anda menggunakan lingkungan produksi dan pengujian terpisah untuk mencegah gangguan DAG. Untuk informasi selengkapnya, lihat Menguji DAG.
Membuat struktur DAG Airflow
DAG Airflow ditentukan dalam file Python dan terdiri dari komponen berikut:
- Definisi DAG
- Operator Airflow
- Hubungan operator
Cuplikan kode berikut menunjukkan contoh setiap komponen di luar konteks.
Definisi DAG
Contoh berikut menunjukkan definisi DAG Alur Kerja:
Airflow 2
Aliran udara 1
Operator dan tugas
Operator Airflow menjelaskan pekerjaan yang akan dilakukan. Tugas tugas adalah instance operator tertentu.
Airflow 2
Aliran udara 1
Hubungan tugas
Hubungan tugas menjelaskan urutan pekerjaan yang harus diselesaikan.
Airflow 2
Aliran udara 1
Contoh alur kerja DAG lengkap di Python
Alur kerja berikut adalah template DAG lengkap yang berfungsi dan terdiri dari
dua tugas: tugas hello_python
dan tugas goodbye_bash
:
Airflow 2
Aliran udara 1
Untuk informasi selengkapnya tentang menentukan DAG Airflow, lihat tutorial Airflow dan konsep Airflow.
Operator Airflow
Contoh berikut menunjukkan beberapa operator Airflow populer. Untuk referensi resmi operator Airflow, lihat Referensi Operator dan Hook dan Indeks penyedia.
BashOperator
Gunakan BashOperator untuk menjalankan program command line.
Airflow 2
Aliran udara 1
Cloud Composer menjalankan perintah yang disediakan dalam skrip Bash di pekerja Airflow. Pekerja adalah penampung Docker berbasis Debian dan menyertakan beberapa paket.
- Perintah
gcloud
, termasuk sub-perintahgcloud storage
untuk menggunakan bucket Cloud Storage. - Perintah
bq
- Perintah
kubectl
PythonOperator
Gunakan PythonOperator untuk menjalankan kode Python arbitrer.
Cloud Composer menjalankan kode Python dalam penampung yang menyertakan paket untuk versi image Cloud Composer yang digunakan di lingkungan Anda.
Untuk menginstal paket Python tambahan, lihat Menginstal Dependensi Python.
Operator Google Cloud
Untuk menjalankan tugas yang menggunakan produk Google Cloud, gunakan operator Google Cloud Airflow. Misalnya, operator BigQuery membuat kueri dan memproses data di BigQuery.
Ada banyak lagi operator Airflow untuk Google Cloud dan layanan individual yang disediakan oleh Google Cloud. Lihat Operator Google Cloud untuk mengetahui daftar lengkapnya.
Airflow 2
Aliran udara 1
EmailOperator
Gunakan EmailOperator untuk mengirim email dari DAG. Untuk mengirim email dari lingkungan Cloud Composer, konfigurasi lingkungan Anda untuk menggunakan SendGrid.
Airflow 2
Aliran udara 1
Notifikasi tentang kegagalan operator
Tetapkan email_on_failure
ke True
untuk mengirim notifikasi email saat operator
dalam DAG gagal. Untuk mengirim notifikasi email dari lingkungan Cloud Composer, Anda harus mengonfigurasi lingkungan untuk menggunakan SendGrid.
Airflow 2
Aliran udara 1
Panduan alur kerja DAG
Tempatkan library Python kustom di arsip ZIP DAG dalam direktori bertingkat. Jangan menempatkan library di tingkat teratas direktori DAG.
Saat memindai folder
dags/
, Airflow hanya memeriksa DAG di modul Python yang berada di tingkat atas folder DAG dan di tingkat atas arsip ZIP yang juga terletak di folderdags/
tingkat atas. Jika Airflow menemukan modul Python dalam arsip ZIP yang tidak berisi substringairflow
danDAG
, Airflow akan berhenti memproses arsip ZIP. Airflow hanya menampilkan DAG yang ditemukan hingga titik tersebut.Gunakan Airflow 2, bukan Airflow 1.
Komunitas Airflow tidak lagi memublikasikan rilis minor atau patch baru untuk Airflow 1.
Untuk toleransi error, jangan tentukan beberapa objek DAG dalam modul Python yang sama.
Jangan gunakan SubDAG. Sebagai gantinya, kelompokkan tugas di dalam DAG.
Tempatkan file yang diperlukan pada waktu penguraian DAG ke folder
dags/
, bukan di folderdata/
.Uji DAG yang dikembangkan atau diubah seperti yang direkomendasikan dalam petunjuk untuk menguji DAG.
Pastikan DAG yang dikembangkan tidak terlalu meningkatkan waktu penguraian DAG.
Tugas Airflow dapat gagal karena beberapa alasan. Untuk menghindari kegagalan seluruh DAG yang berjalan, sebaiknya aktifkan percobaan ulang tugas. Menetapkan percobaan ulang maksimum ke
0
berarti tidak ada percobaan ulang yang dilakukan.Sebaiknya ganti opsi
default_task_retries
dengan nilai untuk percobaan ulang tugas selain0
. Selain itu, Anda dapat menetapkan parameterretries
di tingkat tugas.Jika Anda ingin menggunakan GPU dalam tugas Airflow, buat cluster GKE terpisah berdasarkan node yang menggunakan mesin dengan GPU. Gunakan GKEStartPodOperator untuk menjalankan tugas Anda.
Hindari menjalankan tugas yang menggunakan banyak CPU dan memori di node pool cluster tempat komponen Airflow lainnya (penjadwal, pekerja, server web) berjalan. Sebagai gantinya, gunakan KubernetesPodOperator atau GKEStartPodOperator.
Saat men-deploy DAG ke lingkungan, hanya upload file yang benar-benar diperlukan untuk menafsirkan dan mengeksekusi DAG ke folder
/dags
.Batasi jumlah file DAG di folder
/dags
.Airflow terus mengurai DAG di folder
/dags
. Mengurai adalah proses yang melakukan loop melalui folder DAG dan jumlah file yang perlu dimuat (dengan dependensinya) akan memengaruhi performa penguraian DAG dan penjadwalan tugas. Sebaiknya gunakan 100 file dengan masing-masing 100 DAG daripada 10.000 file dengan masing-masing 1 DAG karena pengoptimalan tersebut jauh lebih efisien. Pengoptimalan ini adalah keseimbangan antara waktu penguraian dan efisiensi pembuatan serta pengelolaan DAG.Anda juga dapat mempertimbangkan, misalnya, untuk men-deploy 10.000 file DAG, Anda dapat membuat 100 file ZIP yang masing-masing berisi 100 file DAG.
Selain petunjuk di atas, jika Anda memiliki lebih dari 10.000 file DAG, membuat DAG secara terprogram mungkin merupakan opsi yang baik. Misalnya, Anda dapat menerapkan satu file DAG Python yang menghasilkan sejumlah objek DAG (misalnya, 20, 100 objek DAG).
Hindari penggunaan operator Airflow yang tidak digunakan lagi. Sebagai gantinya, gunakan alternatif terbarunya.
FAQ untuk menulis DAG
Bagaimana cara meminimalkan pengulangan kode jika saya ingin menjalankan tugas yang sama atau serupa di beberapa DAG?
Sebaiknya tentukan library dan wrapper untuk meminimalkan pengulangan kode.
Bagaimana cara menggunakan kembali kode di antara file DAG?
Masukkan fungsi utilitas Anda di library Python lokal dan impor fungsi tersebut. Anda dapat mereferensikan fungsi dalam DAG apa pun yang terletak di folder dags/
di bucket lingkungan Anda.
Bagaimana cara meminimalkan risiko munculnya definisi yang berbeda?
Misalnya, Anda memiliki dua tim yang ingin menggabungkan data mentah menjadi metrik pendapatan. Tim menulis dua tugas yang sedikit berbeda yang mencapai hal yang sama. Menentukan library untuk menggunakan data pendapatan sehingga pengimplementasi DAG harus mengklarifikasi definisi pendapatan yang digabungkan.
Bagaimana cara menetapkan dependensi antar-DAG?
Hal ini bergantung pada cara Anda ingin menentukan dependensi.
Jika memiliki dua DAG (DAG A dan DAG B) dan ingin DAG B dipicu setelah DAG A, Anda dapat menempatkan TriggerDagRunOperator
di akhir DAG A.
Jika DAG B hanya bergantung pada artefak yang dihasilkan DAG A, seperti pesan Pub/Sub, sensor mungkin akan berfungsi lebih baik.
Jika DAG B terintegrasi erat dengan DAG A, Anda mungkin dapat menggabungkan kedua DAG menjadi satu DAG.
Bagaimana cara meneruskan ID run unik ke DAG dan tugasnya?
Misalnya, Anda ingin meneruskan nama cluster dan jalur file Dataproc.
Anda dapat membuat ID unik acak dengan menampilkan str(uuid.uuid4())
dalam
PythonOperator
. Tindakan ini akan menempatkan ID ke dalam
XComs
sehingga Anda dapat merujuk ke ID di operator lain
melalui kolom template.
Sebelum membuat uuid
, pertimbangkan apakah ID khusus DagRun akan lebih bernilai. Anda juga dapat mereferensikan ID ini dalam penggantian Jinja dengan menggunakan makro.
Bagaimana cara memisahkan tugas dalam DAG?
Setiap tugas harus berupa unit pekerjaan idempoten. Oleh karena itu, Anda harus menghindari
enkapsulasi alur kerja multi-langkah dalam satu tugas, seperti program
kompleks yang berjalan di PythonOperator
.
Haruskah saya menentukan beberapa tugas dalam satu DAG untuk menggabungkan data dari beberapa sumber?
Misalnya, Anda memiliki beberapa tabel dengan data mentah dan ingin membuat agregat harian untuk setiap tabel. Tugas-tugas tersebut tidak saling bergantung. Haruskah Anda membuat satu tugas dan DAG untuk setiap tabel atau membuat satu DAG umum?
Jika Anda tidak keberatan jika setiap tugas memiliki properti tingkat DAG yang sama, seperti
schedule_interval
, sebaiknya tentukan beberapa tugas dalam satu
DAG. Atau, untuk meminimalkan pengulangan kode, beberapa DAG dapat dihasilkan
dari satu modul Python dengan menempatkannya ke dalam globals()
modul.
Bagaimana cara membatasi jumlah tugas serentak yang berjalan di DAG?
Misalnya, Anda ingin menghindari melampaui batas/kuota penggunaan API atau menghindari menjalankan terlalu banyak proses serentak.
Anda dapat menentukan kumpulan Airflow di UI web Airflow dan mengaitkan tugas dengan kumpulan yang ada di DAG.
FAQ untuk menggunakan operator
Haruskah saya menggunakan DockerOperator
?
Sebaiknya jangan gunakan
DockerOperator
, kecuali jika digunakan untuk meluncurkan
penampung di penginstalan Docker jarak jauh (bukan dalam cluster
lingkungan). Di lingkungan Cloud Composer, operator tidak memiliki
akses ke daemon Docker.
Sebagai gantinya, gunakan KubernetesPodOperator
atau
GKEStartPodOperator
. Operator ini meluncurkan pod Kubernetes ke dalam
cluster Kubernetes atau GKE. Perhatikan bahwa kami tidak
menyarankan untuk meluncurkan pod ke cluster lingkungan, karena hal ini dapat menyebabkan
persaingan resource.
Haruskah saya menggunakan SubDagOperator
?
Sebaiknya jangan gunakan SubDagOperator
.
Gunakan alternatif seperti yang disarankan di Mengelompokkan tugas.
Haruskah saya menjalankan kode Python hanya di PythonOperators
untuk memisahkan operator Python sepenuhnya?
Bergantung pada sasaran Anda, Anda memiliki beberapa opsi.
Jika satu-satunya masalah Anda adalah mempertahankan dependensi Python terpisah, Anda
dapat menggunakan PythonVirtualenvOperator
.
Pertimbangkan untuk menggunakan KubernetesPodOperator
. Operator ini memungkinkan Anda
menentukan pod Kubernetes dan menjalankan pod di cluster lain.
Bagaimana cara menambahkan paket biner kustom atau non-PyPI?
Anda dapat menginstal paket yang dihosting di repositori paket pribadi.
Bagaimana cara meneruskan argumen secara seragam ke DAG dan tugasnya?
Anda dapat menggunakan dukungan bawaan Airflow untuk template Jinja guna meneruskan argumen yang dapat digunakan di kolom template.
Kapan penggantian template terjadi?
Penggantian template terjadi pada pekerja Airflow tepat sebelum fungsi pre_execute
operator dipanggil. Dalam praktiknya, ini berarti template
tidak diganti hingga tepat sebelum tugas berjalan.
Bagaimana cara mengetahui argumen operator yang mendukung penggantian template?
Argumen operator yang mendukung penggantian template Jinja2 ditandai secara eksplisit sebagai argumen tersebut.
Cari kolom template_fields
dalam definisi Operator,
yang berisi daftar nama argumen yang mengalami penggantian template.
Misalnya, lihat
BashOperator
, yang mendukung pembuatan template untuk
argumen bash_command
dan env
.
Operator Airflow yang tidak digunakan lagi dan dihapus
Operator Airflow yang tercantum dalam tabel berikut tidak digunakan lagi:
Hindari penggunaan operator ini di DAG Anda. Sebagai gantinya, gunakan operator pengganti terbaru yang disediakan.
Jika operator tercantum sebagai tersedia, artinya rilis pemeliharaan terbaru Cloud Composer (1.20.12) masih menyediakan operator ini.
Beberapa operator pengganti tidak didukung di versi Cloud Composer 1 apa pun. Untuk menggunakannya, pertimbangkan untuk mengupgrade ke Cloud Composer 3 atau Cloud Composer 2.
Operator yang tidak digunakan lagi | Status | Operator pengganti | Penggantian tersedia dari |
---|---|---|---|
CreateAutoMLTextTrainingJobOperator | Tersedia di 1.20.12 | SupervisedFineTuningTrainOperator | Operator pengganti tidak tersedia |
GKEDeploymentHook | Tersedia di 1.20.12 | GKEKubernetesHook | Operator pengganti tidak tersedia |
GKECustomResourceHook | Tersedia di 1.20.12 | GKEKubernetesHook | Operator pengganti tidak tersedia |
GKEPodHook | Tersedia di 1.20.12 | GKEKubernetesHook | Operator pengganti tidak tersedia |
GKEJobHook | Tersedia di 1.20.12 | GKEKubernetesHook | Operator pengganti tidak tersedia |
GKEPodAsyncHook | Tersedia di 1.20.12 | GKEKubernetesAsyncHook | Operator pengganti tidak tersedia |
SecretsManagerHook | Tersedia di 1.20.12 | GoogleCloudSecretManagerHook | Operator pengganti tidak tersedia |
BigQueryExecuteQueryOperator | Tersedia di 1.20.12 | BigQueryInsertJobOperator | Tersedia di 1.20.12 |
BigQueryPatchDatasetOperator | Tersedia di 1.20.12 | BigQueryUpdateDatasetOperator | Tersedia di 1.20.12 |
DataflowCreateJavaJobOperator | Tersedia di 1.20.12 | beam.BeamRunJavaPipelineOperator | Tersedia di 1.20.12 |
DataflowCreatePythonJobOperator | Tersedia di 1.20.12 | beam.BeamRunPythonPipelineOperator | Tersedia di 1.20.12 |
DataprocSubmitPigJobOperator | Tersedia di 1.20.12 | DataprocSubmitJobOperator | Tersedia di 1.20.12 |
DataprocSubmitHiveJobOperator | Tersedia di 1.20.12 | DataprocSubmitJobOperator | Tersedia di 1.20.12 |
DataprocSubmitSparkSqlJobOperator | Tersedia di 1.20.12 | DataprocSubmitJobOperator | Tersedia di 1.20.12 |
DataprocSubmitSparkJobOperator | Tersedia di 1.20.12 | DataprocSubmitJobOperator | Tersedia di 1.20.12 |
DataprocSubmitHadoopJobOperator | Tersedia di 1.20.12 | DataprocSubmitJobOperator | Tersedia di 1.20.12 |
DataprocSubmitPySparkJobOperator | Tersedia di 1.20.12 | DataprocSubmitJobOperator | Tersedia di 1.20.12 |
BigQueryTableExistenceAsyncSensor | Tersedia di 1.20.12 | BigQueryTableExistenceSensor | Operator pengganti tidak tersedia |
BigQueryTableExistencePartitionAsyncSensor | Tersedia di 1.20.12 | BigQueryTablePartitionExistenceSensor | Operator pengganti tidak tersedia |
CloudComposerEnvironmentSensor | Tersedia di 1.20.12 | CloudComposerCreateEnvironmentOperator, CloudComposerDeleteEnvironmentOperator, CloudComposerUpdateEnvironmentOperator | Operator pengganti tidak tersedia |
GCSObjectExistenceAsyncSensor | Tersedia di 1.20.12 | GCSObjectExistenceSensor | Operator pengganti tidak tersedia |
GoogleAnalyticsHook | Tersedia di 1.20.12 | GoogleAnalyticsAdminHook | Operator pengganti tidak tersedia |
GoogleAnalyticsListAccountsOperator | Tersedia di 1.20.12 | GoogleAnalyticsAdminListAccountsOperator | Operator pengganti tidak tersedia |
GoogleAnalyticsGetAdsLinkOperator | Tersedia di 1.20.12 | GoogleAnalyticsAdminGetGoogleAdsLinkOperator | Operator pengganti tidak tersedia |
GoogleAnalyticsRetrieveAdsLinksListOperator | Tersedia di 1.20.12 | GoogleAnalyticsAdminListGoogleAdsLinksOperator | Operator pengganti tidak tersedia |
GoogleAnalyticsDataImportUploadOperator | Tersedia di 1.20.12 | GoogleAnalyticsAdminCreateDataStreamOperator | Operator pengganti tidak tersedia |
GoogleAnalyticsDeletePreviousDataUploadsOperator | Tersedia di 1.20.12 | GoogleAnalyticsAdminDeleteDataStreamOperator | Operator pengganti tidak tersedia |
DataPipelineHook | Tersedia di 1.20.12 | DataflowHook | Operator pengganti tidak tersedia |
CreateDataPipelineOperator | Tersedia di 1.20.12 | DataflowCreatePipelineOperator | Operator pengganti tidak tersedia |
RunDataPipelineOperator | Tersedia di 1.20.12 | DataflowRunPipelineOperator | Operator pengganti tidak tersedia |
AutoMLDatasetLink | Tersedia di 1.20.12 | TranslationLegacyDatasetLink | Operator pengganti tidak tersedia |
AutoMLDatasetListLink | Tersedia di 1.20.12 | TranslationDatasetListLink | Operator pengganti tidak tersedia |
AutoMLModelLink | Tersedia di 1.20.12 | TranslationLegacyModelLink | Operator pengganti tidak tersedia |
AutoMLModelTrainLink | Tersedia di 1.20.12 | TranslationLegacyModelTrainLink | Operator pengganti tidak tersedia |
AutoMLModelPredictLink | Tersedia di 1.20.12 | TranslationLegacyModelPredictLink | Operator pengganti tidak tersedia |
AutoMLBatchPredictOperator | Tersedia di 1.20.12 | vertex_ai.batch_prediction_job | Operator pengganti tidak tersedia |
AutoMLPredictOperator | Tersedia di 1.20.12 | vertex_aigenerative_model. TextGenerationModelPredictOperator, translate.TranslateTextOperator | Operator pengganti tidak tersedia |
PromptLanguageModelOperator | Tersedia di 1.20.12 | TextGenerationModelPredictOperator | Operator pengganti tidak tersedia |
GenerateTextEmbeddingsOperator | Tersedia di 1.20.12 | TextEmbeddingModelGetEmbeddingsOperator | Operator pengganti tidak tersedia |
PromptMultimodalModelOperator | Tersedia di 1.20.12 | GenerativeModelGenerateContentOperator | Operator pengganti tidak tersedia |
PromptMultimodalModelWithMediaOperator | Tersedia di 1.20.12 | GenerativeModelGenerateContentOperator | Operator pengganti tidak tersedia |
DataflowStartSqlJobOperator | Tersedia di 1.20.12 | DataflowStartYamlJobOperator | Operator pengganti tidak tersedia |
LifeSciencesHook | Tersedia di 1.20.12 | Hook Operator Batch Google Cloud | Belum diumumkan |
DataprocScaleClusterOperator | Tersedia di 1.20.12 | DataprocUpdateClusterOperator | Belum diumumkan |
MLEngineStartBatchPredictionJobOperator | Tersedia di 1.20.12 | CreateBatchPredictionJobOperator | Belum diumumkan |
MLEngineManageModelOperator | Tersedia di 1.20.12 | MLEngineCreateModelOperator, MLEngineGetModelOperator | Belum diumumkan |
MLEngineGetModelOperator | Tersedia di 1.20.12 | GetModelOperator | Belum diumumkan |
MLEngineDeleteModelOperator | Tersedia di 1.20.12 | DeleteModelOperator | Belum diumumkan |
MLEngineManageVersionOperator | Tersedia di 1.20.12 | MLEngineCreateVersion, MLEngineSetDefaultVersion, MLEngineListVersions, MLEngineDeleteVersion | Belum diumumkan |
MLEngineCreateVersionOperator | Tersedia di 1.20.12 | Parameter parent_model untuk operator VertexAI | Belum diumumkan |
MLEngineSetDefaultVersionOperator | Tersedia di 1.20.12 | SetDefaultVersionOnModelOperator | Belum diumumkan |
MLEngineListVersionsOperator | Tersedia di 1.20.12 | ListModelVersionsOperator | Belum diumumkan |
MLEngineDeleteVersionOperator | Tersedia di 1.20.12 | DeleteModelVersionOperator | Belum diumumkan |
MLEngineStartTrainingJobOperator | Tersedia di 1.20.12 | CreateCustomPythonPackageTrainingJobOperator | Belum diumumkan |
MLEngineTrainingCancelJobOperator | Tersedia di 1.20.12 | CancelCustomTrainingJobOperator | Belum diumumkan |
LifeSciencesRunPipelineOperator | Tersedia di 1.20.12 | Operator Batch Google Cloud | Belum diumumkan |
MLEngineCreateModelOperator | Tersedia di 1.20.12 | operator VertexAI yang sesuai | Belum diumumkan |
Langkah selanjutnya
- Memecahkan masalah DAG
- Memecahkan masalah Penjadwal
- Operator Google
- Operator Google Cloud
- Tutorial Apache Airflow