Timescale listener: recreate dbconn if it is closed, should help if f…

…or some reason the connection is closed (for some reason) before logging end time.
SvenskaSpel · Apr 22, 2022 · 2bddc07 · 2bddc07
1 parent 07a9367
commit 2bddc07
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/locust_plugins/listeners.py b/locust_plugins/listeners.py
@@ -82,6 +82,8 @@ def __init__(
     def dbcursor(self):
         with self.dblock:
             try:
+                if self.dbconn.closed:
+                    self.dbconn = self._dbconn()
                 yield self.dbconn.cursor()
             except psycopg2.Error:
                 try: