Use a single remote call when querying accelerators in stonith-lambda

Signed-off-by: Ryan Emerson <[email protected]>
keycloak · Aug 12, 2024 · 2f94ada · 2f94ada
1 parent dd78da6
commit 2f94ada
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/provision/opentofu/modules/aws/accelerator/src/stonith_lambda.py b/provision/opentofu/modules/aws/accelerator/src/stonith_lambda.py
@@ -11,9 +11,7 @@ def handle_site_offline(labels):
     a_client = boto3.client('globalaccelerator', region_name='us-west-2')
 
     acceleratorDNS = labels['accelerator']
-    accelerator = jmespath.search(f"Accelerators[?DnsName=='{acceleratorDNS}']", a_client.list_accelerators())
-    if not accelerator:
-        accelerator = jmespath.search(f"Accelerators[?DualStackDnsName=='{acceleratorDNS}']", a_client.list_accelerators())
+    accelerator = jmespath.search(f"Accelerators[?(DnsName=='{acceleratorDNS}'|| DualStackDnsName=='{acceleratorDNS}')]", a_client.list_accelerators())
     if not accelerator:
         print(f"Ignoring SiteOffline alert as accelerator with DnsName '{acceleratorDNS}' not found")
         return