Add support for pre active connections for pre pg 9.1

dalibo · Jul 18, 2020 · 3d1c7ea · 3d1c7ea
1 parent 1bbc7f1
commit 3d1c7ea
Showing 1 changed file with 16 additions and 6 deletions.
diff --git a/pgactivity/Data.py b/pgactivity/Data.py
@@ -317,12 +317,22 @@ def pg_get_active_connections(self,):
         """
         Get total of active connections.
         """
-        query = """
-        SELECT
-            COUNT(*) as active_connections
-        FROM pg_stat_activity
-        WHERE state = 'active'
-        """
+
+        if self.pg_num_version <= 90100:
+            # prior to PostgreSQL 9.1, there was no state column
+            query = """
+    SELECT
+        COUNT(*) as active_connections
+    FROM pg_stat_activity
+    WHERE query NOT LIKE '<IDLE>%'
+            """
+        else:
+            query = """
+    SELECT
+        COUNT(*) as active_connections
+    FROM pg_stat_activity
+    WHERE state = 'active'
+            """
 
         cur = self.pg_conn.cursor()
         cur.execute(query,)