Add docs

astronomer · Mar 19, 2024 · bcc7ffa · bcc7ffa
1 parent 2b6d4c8
commit bcc7ffa
Showing 1 changed file with 14 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
@@ -222,11 +222,25 @@ def revoke_inbound_rules(task_instance):
     response = client.revoke_security_group_ingress(
         CidrIp=ip_range,
         FromPort=22,
+        ToPort=22,
         GroupId=task_instance.xcom_pull(
             key="cluster_response_master_security_group", task_ids=["describe_created_cluster"]
         )[0],
         IpProtocol="tcp",
     )
+
+    logging.info("%s", response)
+
+    response = client.revoke_security_group_ingress(
+        CidrIp=ip_range,
+        FromPort=HIVE_OPERATOR_INGRESS_PORT,
+        ToPort=HIVE_OPERATOR_INGRESS_PORT,
+        GroupId=task_instance.xcom_pull(
+            key="cluster_response_master_security_group", task_ids=["describe_created_cluster"]
+        )[0],
+        IpProtocol="tcp",
+    )
+
     logging.info("%s", response)