polish

Author: chillaq

splitio/engine/telemetry.py

@@ -6,7 +6,7 @@
 _LOGGER = logging.getLogger(__name__)
 
 from  splitio.storage.inmemmory import InMemoryTelemetryStorage
-from splitio.models.telemetry import CounterConstants
+from splitio.models.telemetry import CounterConstants, UpdateFromSSE
 
 class TelemetryStorageProducer(object):
     """Telemetry storage producer class."""
@@ -300,6 +300,7 @@ def pop_formatted_stats(self):
             'iDr': self.get_impressions_stats(CounterConstants.IMPRESSIONS_DROPPED),
             'eQ': self.get_events_stats(CounterConstants.EVENTS_QUEUED),
             'eD': self.get_events_stats(CounterConstants.EVENTS_DROPPED),
+            'ufs': {event.value: self.pop_update_from_sse(event) for event in UpdateFromSSE},
             'lS': {'sp': last_synchronization['split'],
                       'se': last_synchronization['segment'],
                       'im': last_synchronization['impression'],

splitio/sync/telemetry.py

@@ -61,8 +61,7 @@ def _build_stats(self):
         merged_dict = {
             'spC': self._feature_flag_storage.get_splits_count(),
             'seC': self._segment_storage.get_segments_count(),
-            'skC': self._segment_storage.get_segments_keys_count(),
-            'ufs': {event.value: self._telemetry_runtime_consumer.pop_update_from_sse(event) for event in UpdateFromSSE}
+            'skC': self._segment_storage.get_segments_keys_count()
         }
         merged_dict.update(self._telemetry_runtime_consumer.pop_formatted_stats())
         merged_dict.update(self._telemetry_evaluation_consumer.pop_formatted_stats())