From 9de2af872efc779fd1dd59e5565b9b3af2468e25 Mon Sep 17 00:00:00 2001 From: DarcyRaynerDD Date: Thu, 3 Jun 2021 23:18:34 -0400 Subject: [PATCH 1/2] Check extension exists before calling hello route --- datadog_lambda/extension.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/datadog_lambda/extension.py b/datadog_lambda/extension.py index f897eab6..da11d628 100644 --- a/datadog_lambda/extension.py +++ b/datadog_lambda/extension.py @@ -1,14 +1,17 @@ import logging import requests +from os import path AGENT_URL = "http://127.0.0.1:8124" HELLO_PATH = "/lambda/hello" FLUSH_PATH = "/lambda/flush" +EXTENSION_PATH = "/opt/extensions/datadog-agent" logger = logging.getLogger(__name__) - def is_extension_running(): + if not path.exists(EXTENSION_PATH): + return False try: requests.get(AGENT_URL + HELLO_PATH) except Exception as e: From 860a00770c3ef9c9e3a54c4f96b999f68a72ed61 Mon Sep 17 00:00:00 2001 From: DarcyRaynerDD Date: Thu, 3 Jun 2021 23:46:18 -0400 Subject: [PATCH 2/2] Fix formatting --- datadog_lambda/extension.py | 1 + 1 file changed, 1 insertion(+) diff --git a/datadog_lambda/extension.py b/datadog_lambda/extension.py index da11d628..c2087a5f 100644 --- a/datadog_lambda/extension.py +++ b/datadog_lambda/extension.py @@ -9,6 +9,7 @@ logger = logging.getLogger(__name__) + def is_extension_running(): if not path.exists(EXTENSION_PATH): return False