use main thread db connection

DataDog · Aug 7, 2023 · 8664978 · 8664978
1 parent 375fe2e
commit 8664978
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/postgres/datadog_checks/postgres/postgres.py b/postgres/datadog_checks/postgres/postgres.py
@@ -168,7 +168,10 @@ def _new_query_executor(self, queries):
         )
 
     def execute_query_raw(self, query):
-        return self.db_pool.execute_main_db_safe(query)
+        with self.db.cursor() as cursor:
+            cursor.execute(query)
+            rows = cursor.fetchall()
+            return rows
 
     @property
     def dynamic_queries(self):