Update batch_test.py

batch_test.py

@@ -8,6 +8,13 @@ from data.gasan_data import gasanData   # import cj-gasan data
 # set timezone
 local_tz = pendulum.timezone("Asia/Seoul")
 
+def fail_alert(context):
+    alert = BashOperator(
+        task_id='fail_alert',
+        bash_command="curl -d \'{\"@context\":\"https://schema.org/extensions\",\"@type\":\"MessageCard\",\"themeColor\":\"0072C6\",\"title\":\"Batch Job Error\",\"text\":\"[health_check] Error Task\"}\' -H \"Content-Type: Application/JSON\" -X POST https://lottegroup.webhook.office.com/webhookb2/2ed9f7fc-4c60-4d2d-a61c-aa50c0075564@dc742f86-8941-4de1-8d2c-d2dfef93cfe8/IncomingWebhook/1047eeaf7bde45a08e5ccb4d6c80f08d/d7352368-8126-4827-aab7-4a62b0b5abc2"
+    )
+    return alert.execute(context)
+
 with DAG(
     'batch_test',
     default_args={
@@ -23,7 +30,8 @@ with DAG(
     # (Task1) 헬스체크
     health_check = BashOperator(
         task_id='health_check',
-        bash_command="curl -X GET -v http://10.231.238.224:30999/api/v1/core/health \'-H accept: application/json\'"
+        bash_command="curl -X GET -v http://10.231.238.224:30999/api/v1/core/health \'-H accept: application/json\'",
+        on_failure_callback: fail_alert
     )
 
     # (Task2) gasan 작업 병렬처리