Max Petrov
2 years ago
committed by
GitHub
GitHub
24 changed files with 317 additions and 25 deletions
@ -0,0 +1,148 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.transport.mqtt.mqttv3.rpc; |
||||
|
|
||||
|
import io.netty.handler.codec.mqtt.MqttQoS; |
||||
|
import org.junit.Test; |
||||
|
import org.springframework.beans.factory.annotation.Value; |
||||
|
import org.thingsboard.common.util.JacksonUtil; |
||||
|
import org.thingsboard.server.common.data.TenantProfile; |
||||
|
import org.thingsboard.server.common.data.TransportPayloadType; |
||||
|
import org.thingsboard.server.common.data.tenant.profile.DefaultTenantProfileConfiguration; |
||||
|
import org.thingsboard.server.dao.service.DaoSqlTest; |
||||
|
import org.thingsboard.server.transport.mqtt.AbstractMqttIntegrationTest; |
||||
|
import org.thingsboard.server.transport.mqtt.MqttTestConfigProperties; |
||||
|
import org.thingsboard.server.transport.mqtt.limits.GatewaySessionLimits; |
||||
|
import org.thingsboard.server.transport.mqtt.limits.SessionLimits; |
||||
|
import org.thingsboard.server.transport.mqtt.mqttv3.MqttTestCallback; |
||||
|
import org.thingsboard.server.transport.mqtt.mqttv3.MqttTestClient; |
||||
|
import org.thingsboard.server.transport.mqtt.mqttv3.MqttTestSubscribeOnTopicCallback; |
||||
|
|
||||
|
import java.util.concurrent.TimeUnit; |
||||
|
|
||||
|
import static org.assertj.core.api.Assertions.assertThat; |
||||
|
import static org.junit.jupiter.api.Assertions.assertEquals; |
||||
|
import static org.junit.jupiter.api.Assertions.assertTrue; |
||||
|
import static org.thingsboard.server.common.data.device.profile.MqttTopics.DEVICE_RPC_REQUESTS_TOPIC; |
||||
|
import static org.thingsboard.server.common.data.device.profile.MqttTopics.DEVICE_RPC_RESPONSE_SUB_TOPIC; |
||||
|
import static org.thingsboard.server.common.data.device.profile.MqttTopics.DEVICE_RPC_RESPONSE_TOPIC; |
||||
|
|
||||
|
@DaoSqlTest |
||||
|
public class MqttClientSideRpcIntegrationTest extends AbstractMqttIntegrationTest { |
||||
|
|
||||
|
@Value("${transport.mqtt.netty.max_payload_size}") |
||||
|
private Integer maxPayloadSize; |
||||
|
|
||||
|
@Value("${transport.mqtt.msg_queue_size_per_device_limit}") |
||||
|
private int maxInflightMessages; |
||||
|
|
||||
|
@Test |
||||
|
public void getSessionLimitsRpcForDeviceTest() throws Exception { |
||||
|
loginSysAdmin(); |
||||
|
TenantProfile tenantProfile = doGet("/api/tenantProfile/" + tenantProfileId, TenantProfile.class); |
||||
|
DefaultTenantProfileConfiguration profileConfiguration = tenantProfile.getDefaultProfileConfiguration(); |
||||
|
|
||||
|
profileConfiguration.setTransportDeviceMsgRateLimit("20:600"); |
||||
|
profileConfiguration.setTransportDeviceTelemetryMsgRateLimit("10:600"); |
||||
|
profileConfiguration.setTransportDeviceTelemetryDataPointsRateLimit("40:600"); |
||||
|
|
||||
|
doPost("/api/tenantProfile", tenantProfile); |
||||
|
|
||||
|
var expectedLimits = new SessionLimits(); |
||||
|
var deviceLimits = new SessionLimits.SessionRateLimits(profileConfiguration.getTransportDeviceMsgRateLimit(), |
||||
|
profileConfiguration.getTransportDeviceTelemetryMsgRateLimit(), |
||||
|
profileConfiguration.getTransportDeviceTelemetryDataPointsRateLimit()); |
||||
|
expectedLimits.setRateLimits(deviceLimits); |
||||
|
expectedLimits.setMaxPayloadSize(maxPayloadSize); |
||||
|
expectedLimits.setMaxInflightMessages(maxInflightMessages); |
||||
|
|
||||
|
MqttTestConfigProperties configProperties = MqttTestConfigProperties.builder() |
||||
|
.deviceName("Test Get Service Configuration") |
||||
|
.transportPayloadType(TransportPayloadType.JSON) |
||||
|
.build(); |
||||
|
processBeforeTest(configProperties); |
||||
|
|
||||
|
MqttTestClient client = new MqttTestClient(); |
||||
|
client.connectAndWait(accessToken); |
||||
|
|
||||
|
MqttTestCallback callback = new MqttTestSubscribeOnTopicCallback(DEVICE_RPC_RESPONSE_TOPIC + "1"); |
||||
|
client.setCallback(callback); |
||||
|
client.subscribeAndWait(DEVICE_RPC_RESPONSE_SUB_TOPIC, MqttQoS.AT_MOST_ONCE); |
||||
|
|
||||
|
client.publishAndWait(DEVICE_RPC_REQUESTS_TOPIC + "1", "{\"method\":\"getSessionLimits\",\"params\":{}}".getBytes()); |
||||
|
|
||||
|
assertThat(callback.getSubscribeLatch().await(DEFAULT_WAIT_TIMEOUT_SECONDS, TimeUnit.SECONDS)) |
||||
|
.as("await callback").isTrue(); |
||||
|
|
||||
|
var payload = callback.getPayloadBytes(); |
||||
|
SessionLimits actualLimits = JacksonUtil.fromBytes(payload, SessionLimits.class); |
||||
|
assertEquals(expectedLimits, actualLimits); |
||||
|
|
||||
|
client.disconnect(); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void getSessionLimitsRpcForGatewayTest() throws Exception { |
||||
|
loginSysAdmin(); |
||||
|
TenantProfile tenantProfile = doGet("/api/tenantProfile/" + tenantProfileId, TenantProfile.class); |
||||
|
DefaultTenantProfileConfiguration profileConfiguration = tenantProfile.getDefaultProfileConfiguration(); |
||||
|
|
||||
|
profileConfiguration.setTransportGatewayMsgRateLimit("100:600"); |
||||
|
profileConfiguration.setTransportGatewayTelemetryMsgRateLimit("50:600"); |
||||
|
profileConfiguration.setTransportGatewayTelemetryDataPointsRateLimit("200:600"); |
||||
|
|
||||
|
profileConfiguration.setTransportGatewayDeviceMsgRateLimit("20:600"); |
||||
|
profileConfiguration.setTransportGatewayDeviceTelemetryMsgRateLimit("10:600"); |
||||
|
profileConfiguration.setTransportGatewayDeviceTelemetryDataPointsRateLimit("40:600"); |
||||
|
|
||||
|
doPost("/api/tenantProfile", tenantProfile); |
||||
|
|
||||
|
var expectedLimits = new GatewaySessionLimits(); |
||||
|
var gatewayLimits = new SessionLimits.SessionRateLimits(profileConfiguration.getTransportGatewayMsgRateLimit(), |
||||
|
profileConfiguration.getTransportGatewayTelemetryMsgRateLimit(), |
||||
|
profileConfiguration.getTransportGatewayTelemetryDataPointsRateLimit()); |
||||
|
var gatewayDeviceLimits = new SessionLimits.SessionRateLimits(profileConfiguration.getTransportGatewayDeviceMsgRateLimit(), |
||||
|
profileConfiguration.getTransportGatewayDeviceTelemetryMsgRateLimit(), |
||||
|
profileConfiguration.getTransportGatewayDeviceTelemetryDataPointsRateLimit()); |
||||
|
expectedLimits.setGatewayRateLimits(gatewayLimits); |
||||
|
expectedLimits.setRateLimits(gatewayDeviceLimits); |
||||
|
expectedLimits.setMaxPayloadSize(maxPayloadSize); |
||||
|
expectedLimits.setMaxInflightMessages(maxInflightMessages); |
||||
|
|
||||
|
MqttTestConfigProperties configProperties = MqttTestConfigProperties.builder() |
||||
|
.gatewayName("Test Get Service Configuration Gateway") |
||||
|
.transportPayloadType(TransportPayloadType.JSON) |
||||
|
.build(); |
||||
|
processBeforeTest(configProperties); |
||||
|
|
||||
|
MqttTestClient client = new MqttTestClient(); |
||||
|
client.connectAndWait(gatewayAccessToken); |
||||
|
|
||||
|
MqttTestCallback callback = new MqttTestSubscribeOnTopicCallback(DEVICE_RPC_RESPONSE_TOPIC + "1"); |
||||
|
client.setCallback(callback); |
||||
|
client.subscribeAndWait(DEVICE_RPC_RESPONSE_SUB_TOPIC, MqttQoS.AT_MOST_ONCE); |
||||
|
|
||||
|
client.publishAndWait(DEVICE_RPC_REQUESTS_TOPIC + "1", "{\"method\":\"getSessionLimits\",\"params\":{}}".getBytes()); |
||||
|
|
||||
|
assertTrue(callback.getSubscribeLatch().await(DEFAULT_WAIT_TIMEOUT_SECONDS, TimeUnit.SECONDS)); |
||||
|
|
||||
|
var payload = callback.getPayloadBytes(); |
||||
|
SessionLimits actualLimits = JacksonUtil.fromBytes(payload, GatewaySessionLimits.class); |
||||
|
assertEquals(expectedLimits, actualLimits); |
||||
|
|
||||
|
client.disconnect(); |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,25 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.transport.mqtt.limits; |
||||
|
|
||||
|
import lombok.Data; |
||||
|
|
||||
|
@Data |
||||
|
public class GatewaySessionLimits extends SessionLimits { |
||||
|
|
||||
|
private SessionRateLimits gatewayRateLimits; |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,29 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.transport.mqtt.limits; |
||||
|
|
||||
|
import lombok.Data; |
||||
|
import org.thingsboard.server.common.data.TransportPayloadType; |
||||
|
|
||||
|
@Data |
||||
|
public class SessionLimits { |
||||
|
|
||||
|
private int maxPayloadSize; |
||||
|
private int maxInflightMessages; |
||||
|
private SessionRateLimits rateLimits; |
||||
|
|
||||
|
public record SessionRateLimits(String messages, String telemetryMessages, String telemetryDataPoints) {} |
||||
|
} |
||||
Loading…
Reference in new issue