Added metadata to messages

Author: samson0v

tb_device_mqtt.py

@@ -758,12 +758,12 @@ def __get_rate_limits_by_topic(self, topic, device=None, msg_rate_limit=None, dp
 
     def __send_publish_with_limitations(self, kwargs, timeout, device=None, msg_rate_limit: RateLimit = None,
                                         dp_rate_limit: RateLimit = None):
-        data_for_analysis = data = kwargs.get("payload")
+        data = kwargs.get("payload")
         if isinstance(data, str):
-            data_for_analysis = loads(data)
+            data = loads(data)
         datapoints = -1
         if dp_rate_limit.has_limit():
-            datapoints = self._count_datapoints_in_message(data_for_analysis, device=device)
+            datapoints = self._count_datapoints_in_message(data, device=device)
         payload = data
         if dp_rate_limit.has_limit() and datapoints >= 0 and dp_rate_limit.get_minimal_limit() < datapoints:
             log.debug("Rate limit is too low, cannot send message with %i datapoints, "
@@ -779,10 +779,12 @@ def __send_publish_with_limitations(self, kwargs, timeout, device=None, msg_rate
                 device_split_messages = self._split_message(device_data, dp_rate_limit.get_minimal_limit(),
                                                             self.max_payload_size)
                 split_messages = [
-                    {'message': {device: [split_message['data']]}, 'datapoints': split_message['datapoints']}
-                                  for split_message in device_split_messages]
+                    {'message': {device: [split_message['data']]}, 'datapoints': split_message['datapoints'],
+                     'metadata': split_message.get('metadata')} for split_message in device_split_messages]
+
             if len(split_messages) == 0:
                 log.debug("Cannot split message to smaller parts!")
+
             results = []
             for part in split_messages:
                 dp_rate_limit.increase_rate_limit_counter(part['datapoints'])
@@ -1036,6 +1038,7 @@ def _split_message(message_pack, max_size, max_payload_size):
                     values = message.get("values")
                 else:
                     values = message
+
                 values_data_keys = tuple(values.keys())
                 if len(values_data_keys) == 1:
                     if ts is not None:
@@ -1065,7 +1068,8 @@ def _split_message(message_pack, max_size, max_payload_size):
                         or current_data_key_index == len(values_data_keys) - 1) or len(
                             str(message_item_values_with_allowed_size)) >= max_payload_size:
                         if ts is not None:
-                            final_message_item['data'] = {"ts": ts, "values": message_item_values_with_allowed_size}
+                            final_message_item['data'] = {"ts": ts, "values": message_item_values_with_allowed_size,
+                                                          'metadata': message.get('metadata')}
                         else:
                             final_message_item['data'] = message_item_values_with_allowed_size
                         final_message_item['datapoints'] = len(message_item_values_with_allowed_size)