Properly cache thread-local database connection object. #1466

perfsonar · Aug 30, 2024 · 099f451 · 099f451
1 parent 3ac8ca1
commit 099f451
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pscheduler-server/pscheduler-server/api-server/pschedulerapiserver/dbcursor.py b/pscheduler-server/pscheduler-server/api-server/pschedulerapiserver/dbcursor.py
@@ -74,7 +74,12 @@ def cursor(self):
 
 def dbcursor():
     """Get this thread's database cursor"""
-    return getattr(module.threadlocal, "cursor", DBCursor()).cursor()
+    try:
+        cursor = getattr(module.threadlocal, 'cursor')
+    except AttributeError:
+        cursor = DBCursor()
+        setattr(module.threadlocal, 'cursor', cursor)
+    return cursor.cursor()
 
 
 def dbcursor_query(query,