[client] Adapt client to use session (performance)

Author: richard-julien

pycti/api/opencti_api_client.py

@@ -101,6 +101,7 @@ def __init__(self, url, token, log_level="info", ssl_verify=False, proxies={}):
         self.api_token = token
         self.api_url = url + "/graphql"
         self.request_headers = {"Authorization": "Bearer " + token}
+        self.session = requests.session()
 
         # Define the dependencies
         self.work = OpenCTIApiWork(self)
@@ -241,7 +242,7 @@ def query(self, query, variables={}):
                     multipart_files.append(file_multi)
                     file_index += 1
             # Send the multipart request
-            r = requests.post(
+            r = self.session.post(
                 self.api_url,
                 data=multipart_data,
                 files=multipart_files,
@@ -251,7 +252,7 @@ def query(self, query, variables={}):
             )
         # If no
         else:
-            r = requests.post(
+            r = self.session.post(
                 self.api_url,
                 json={"query": query, "variables": variables},
                 headers=self.request_headers,
@@ -295,7 +296,7 @@ def fetch_opencti_file(self, fetch_uri, binary=False):
         :rtype: str or bytes
         """
 
-        r = requests.get(fetch_uri, headers=self.request_headers)
+        r = self.session.get(fetch_uri, headers=self.request_headers)
         if binary:
             return r.content
         return r.text

pycti/connector/opencti_connector_helper.py

@@ -206,6 +206,7 @@ def __init__(
         self.connector_last_event_id = connector_last_event_id
         self.last_event_id = last_event_id
         self.stream_connection_id = stream_connection_id
+        self.session = requests.session()
 
     def get_range(self, from_id):
         payload = {
@@ -214,7 +215,7 @@ def get_range(self, from_id):
             "connectionId": self.stream_connection_id,
         }
         headers = {"Authorization": "Bearer " + self.opencti_token}
-        r = requests.post(
+        r = self.session.post(
             self.opencti_url + "/stream/history", json=payload, headers=headers
         )
         result = r.json()