Fixed setting rate limits (no limit) via service configuration callback

Author: samson0v

tb_device_mqtt.py

@@ -701,12 +701,14 @@ def on_service_configuration(self, _, response, *args, **kwargs):
         if service_config.get("rateLimit"):
             rate_limits_config = service_config.get("rateLimit")
 
-            if rate_limits_config.get('messages'):
-                self._messages_rate_limit.set_limit(rate_limits_config.get('messages'), percentage=80)
-            if rate_limits_config.get('telemetryMessages'):
-                self._telemetry_rate_limit.set_limit(rate_limits_config.get('telemetryMessages'), percentage=80)
-            if rate_limits_config.get('telemetryDataPoints'):
-                self._telemetry_dp_rate_limit.set_limit(rate_limits_config.get('telemetryDataPoints'), percentage=80)
+            messages_rate_limit_config = rate_limits_config['messages'] if rate_limits_config['messages'] else '0:0,'
+            self._messages_rate_limit.set_limit(messages_rate_limit_config)
+
+            telemetry_messages_rate_limit_config = rate_limits_config['telemetryMessages'] if rate_limits_config['telemetryMessages'] else '0:0,'
+            self._telemetry_rate_limit.set_limit(telemetry_messages_rate_limit_config)
+
+            telemetry_dp_rate_limit_config = rate_limits_config['telemetryDataPoints'] if rate_limits_config['telemetryDataPoints'] else '0:0,'
+            self._telemetry_dp_rate_limit.set_limit(telemetry_dp_rate_limit_config)
 
         if service_config.get('maxInflightMessages'):
             use_messages_rate_limit_factor = self._messages_rate_limit.has_limit()

tb_gateway_mqtt.py

@@ -301,11 +301,13 @@ def __on_service_configuration(self, _, response, *args, **kwargs):
         gateway_devices_rate_limit_config = service_config.pop('gatewayRateLimits', {})
         gateway_device_itself_rate_limit_config = service_config.pop('rateLimits', {})
 
-        if gateway_devices_rate_limit_config.get("messages"):
-            self._devices_connected_through_gateway_messages_rate_limit.set_limit(gateway_devices_rate_limit_config.get("messages"))
-        if gateway_devices_rate_limit_config.get('telemetryMessages'):
-            self._devices_connected_through_gateway_telemetry_messages_rate_limit.set_limit(gateway_devices_rate_limit_config.get('telemetryMessages'))
-        if gateway_devices_rate_limit_config.get('telemetryDataPoints'):
-            self._devices_connected_through_gateway_telemetry_datapoints_rate_limit.set_limit(gateway_devices_rate_limit_config.get('telemetryDataPoints'))
+        messages_rate_limit_config = gateway_devices_rate_limit_config['messages'] if gateway_devices_rate_limit_config['messages'] else '0:0,'
+        self._devices_connected_through_gateway_messages_rate_limit.set_limit(messages_rate_limit_config)
+
+        telemetry_messages_rate_limit_config = gateway_devices_rate_limit_config['telemetryMessages'] if gateway_devices_rate_limit_config['telemetryMessages'] else '0:0,'
+        self._devices_connected_through_gateway_telemetry_messages_rate_limit.set_limit(telemetry_messages_rate_limit_config)
+
+        telemetry_dp_rate_limit_config = gateway_devices_rate_limit_config['telemetryDataPoints'] if gateway_devices_rate_limit_config['telemetryDataPoints'] else '0:0,'
+        self._devices_connected_through_gateway_telemetry_datapoints_rate_limit.set_limit(telemetry_dp_rate_limit_config)
 
         super().on_service_configuration(_, {'rateLimit': gateway_device_itself_rate_limit_config, **service_config}, *args, **kwargs)