Add task to revoke ingress rule

astronomer · Mar 19, 2024 · 1d33b50 · 1d33b50
1 parent cf496ef
commit 1d33b50
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/astronomer/providers/apache/hive/example_dags/example_hive.py b/astronomer/providers/apache/hive/example_dags/example_hive.py
@@ -224,7 +224,9 @@ def revoke_inbound_rules(task_instance):
         GroupId=task_instance.xcom_pull(
                 key="cluster_response_master_security_group", task_ids=["describe_created_cluster"]
             )[0],
+        IpProtocol="tcp",
     )
+    logging.info("%s", response)
 
 
 def ssh_and_run_command(task_instance: Any, **kwargs: Any) -> None:

diff --git a/astronomer/providers/apache/livy/example_dags/example_livy.py b/astronomer/providers/apache/livy/example_dags/example_livy.py
@@ -188,7 +188,9 @@ def revoke_inbound_rules(task_instance):
         GroupId=task_instance.xcom_pull(
                 key="cluster_response_master_security_group", task_ids=["describe_created_cluster"]
             )[0],
+        IpProtocol="tcp",
     )
+    logging.info("%s", response)
 
 
 def ssh_and_run_command(task_instance: Any, **kwargs: Any) -> None:

diff --git a/astronomer/providers/sftp/example_dags/example_sftp.py b/astronomer/providers/sftp/example_dags/example_sftp.py
@@ -171,7 +171,9 @@ def revoke_inbound_rules(task_instance):
         GroupId=task_instance.xcom_pull(
                 key="cluster_response_master_security_group", task_ids=["describe_created_cluster"]
             )[0],
+        IpProtocol="tcp",
     )
+    logging.info("%s", response)
 
 
 def ssh_and_run_command(task_instance: Any, **kwargs: Any) -> None: