more wip

DataDog · Jul 13, 2023 · 672022a · 672022a
1 parent f4a8079
commit 672022a
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/datadog_checks_base/datadog_checks/base/data/agent_requirements.in b/datadog_checks_base/datadog_checks/base/data/agent_requirements.in
@@ -57,6 +57,7 @@ protobuf==3.17.3; python_version < '3.0'
 protobuf==3.20.2; python_version > '3.0'
 psutil==5.9.0
 psycopg[binary]==3.1.9; python_version > '3.0'
+psycopg[pool]==3.1.7; python_version > '3.0'
 pyasn1==0.4.6
 pycryptodomex==3.10.1
 pydantic==1.10.8; python_version > '3.0'

diff --git a/postgres/datadog_checks/postgres/postgres.py b/postgres/datadog_checks/postgres/postgres.py
@@ -9,6 +9,7 @@
 
 import psycopg
 from psycopg.rows import dict_row
+from psycopg_pool import ConnectionPool
 from six import iteritems
 
 from datadog_checks.base import AgentCheck

diff --git a/postgres/pyproject.toml b/postgres/pyproject.toml
@@ -42,6 +42,7 @@ deps = [
     "cachetools==5.3.1; python_version > '3.0'",
     "psycopg[binary]==3.1.9; python_version > '3.0'",
     "semver==3.0.1; python_version > '3.0'",
+    "psycopg[pool]==3.1.7; python_version > '3.0'",
 ]
 
 [project.urls]