Added flagset support to factory

Author: chillaq

splitio/client/factory.py

@@ -350,7 +350,7 @@ def _build_in_memory_factory(api_key, cfg, sdk_url=None, events_url=None,  # pyl
     }
 
     storages = {
-        'splits': InMemorySplitStorage(),
+        'splits': InMemorySplitStorage(cfg['FlagSetsFilter'] if cfg['FlagSetsFilter'] is not None else []),
         'segments': InMemorySegmentStorage(),
         'impressions': InMemoryImpressionStorage(cfg['impressionsQueueSize'], telemetry_runtime_producer),
         'events': InMemoryEventStorage(cfg['eventsQueueSize'], telemetry_runtime_producer),

splitio/models/telemetry.py

@@ -41,6 +41,7 @@ class ConfigParams(Enum):
     EVENTS_QUEUE_SIZE = 'eventsQueueSize'
     IMPRESSIONS_MODE = 'impressionsMode'
     IMPRESSIONS_LISTENER = 'impressionListener'
+    FLAG_SETS = 'FlagSetsFilter'
 
 class ExtraConfig(Enum):
     """Extra config constants"""
@@ -671,6 +672,8 @@ def pop_update_from_sse(self, event):
         :rtype: int
         """
         with self._lock:
+            if self._update_from_sse.get(event.value) is None:
+                return 0
             update_from_sse = self._update_from_sse[event.value]
             self._update_from_sse[event.value] = 0
             return update_from_sse
@@ -826,6 +829,7 @@ def record_config(self, config, extra_config):
             self._impressions_mode = self._get_impressions_mode(config[ConfigParams.IMPRESSIONS_MODE.value])
             self._impression_listener = True if config[ConfigParams.IMPRESSIONS_LISTENER.value] is not None else False
             self._http_proxy = self._check_if_proxy_detected()
+            self._flag_sets = len(config[ConfigParams.FLAG_SETS.value]) if config[ConfigParams.FLAG_SETS.value] is not None else 0
 
     def record_active_and_redundant_factories(self, active_factory_count, redundant_factory_count):
         with self._lock: