[client] Put live stream information in cache (#838)

Author: SamuelHassine

pycti/connector/opencti_connector_helper.py

@@ -769,6 +769,9 @@ class TimeUnit(Enum):
     def __init__(self, config: Dict, playbook_compatible=False) -> None:
         sys.excepthook = killProgramHook
 
+        # Cache
+        self.stream_collections = {}
+
         # Load API config
         self.config = config
         self.opencti_url = get_config_variable(
@@ -1063,6 +1066,9 @@ def get_stream_collection(self):
                     "stream_live": True,
                     "stream_public": False,
                 }
+            # Get from cache
+            elif self.connect_live_stream_id in self.stream_collections:
+                return self.stream_collections[self.connect_live_stream_id]
             else:
                 query = """
                     query StreamCollection($id: String!) {
@@ -1076,6 +1082,10 @@ def get_stream_collection(self):
                     }
                 """
                 result = self.api.query(query, {"id": self.connect_live_stream_id})
+                # Put in cache
+                self.stream_collections[self.connect_live_stream_id] = result["data"][
+                    "streamCollection"
+                ]
                 return result["data"]["streamCollection"]
         else:
             raise ValueError("This connector is not connected to any stream")