From 3b8960957e2b8a85f6e89df7cc97f77e8f35bd9a Mon Sep 17 00:00:00 2001
From: Jun Aishima <jaishima@bnl.gov>
Date: Tue, 1 Oct 2024 17:12:56 -0400
Subject: [PATCH] use a session instead of requests directly

     * enable persistent sessions to be used instead of an always-new
       session for each request
---
 analysisstore/client/asutils.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/analysisstore/client/asutils.py b/analysisstore/client/asutils.py
index 052e4e0..ceebd1e 100644
--- a/analysisstore/client/asutils.py
+++ b/analysisstore/client/asutils.py
@@ -5,8 +5,11 @@
 import six
 
 
+session = requests.Session()
+
+
 def get_document(url, doc_type, as_json, contents):
-    r = requests.get(url, params=ujson.dumps(contents))
+    r = session.get(url, params=ujson.dumps(contents))
     r.raise_for_status()
     content = ujson.loads(r.text)
     if as_json:
@@ -18,7 +21,7 @@ def get_document(url, doc_type, as_json, contents):
 
 def post_document(url, contents):
     try:
-        r = requests.post(url, data=ujson.dumps(contents))
+        r = session.post(url, data=ujson.dumps(contents))
     except ConnectionError:
         raise ConnectionError('No AnalysisStore server found')
     r.raise_for_status()  # this is for catching server side issue.