Search in sources :

Example 1 with MqttClient

use of org.thingsboard.mqtt.MqttClient in project thingsboard by thingsboard.

the class TbMqttNode method initClient.

protected MqttClient initClient(TbContext ctx) throws Exception {
    MqttClientConfig config = new MqttClientConfig(getSslContext());
    if (!StringUtils.isEmpty(this.mqttNodeConfiguration.getClientId())) {
        config.setClientId(this.mqttNodeConfiguration.isAppendClientIdSuffix() ? this.mqttNodeConfiguration.getClientId() + "_" + ctx.getServiceId() : this.mqttNodeConfiguration.getClientId());
    }
    config.setCleanSession(this.mqttNodeConfiguration.isCleanSession());
    prepareMqttClientConfig(config);
    MqttClient client = MqttClient.create(config, null);
    client.setEventLoop(ctx.getSharedEventLoop());
    Future<MqttConnectResult> connectFuture = client.connect(this.mqttNodeConfiguration.getHost(), this.mqttNodeConfiguration.getPort());
    MqttConnectResult result;
    try {
        result = connectFuture.get(this.mqttNodeConfiguration.getConnectTimeoutSec(), TimeUnit.SECONDS);
    } catch (TimeoutException ex) {
        connectFuture.cancel(true);
        client.disconnect();
        String hostPort = this.mqttNodeConfiguration.getHost() + ":" + this.mqttNodeConfiguration.getPort();
        throw new RuntimeException(String.format("Failed to connect to MQTT broker at %s.", hostPort));
    }
    if (!result.isSuccess()) {
        connectFuture.cancel(true);
        client.disconnect();
        String hostPort = this.mqttNodeConfiguration.getHost() + ":" + this.mqttNodeConfiguration.getPort();
        throw new RuntimeException(String.format("Failed to connect to MQTT broker at %s. Result code is: %s", hostPort, result.getReturnCode()));
    }
    return client;
}
Also used : MqttClient(org.thingsboard.mqtt.MqttClient) MqttConnectResult(org.thingsboard.mqtt.MqttConnectResult) MqttClientConfig(org.thingsboard.mqtt.MqttClientConfig) TimeoutException(java.util.concurrent.TimeoutException)

Example 2 with MqttClient

use of org.thingsboard.mqtt.MqttClient in project thingsboard by thingsboard.

the class MqttGatewayClientTest method getMqttClient.

private MqttClient getMqttClient(DeviceCredentials deviceCredentials, MqttMessageListener listener) throws InterruptedException, ExecutionException {
    MqttClientConfig clientConfig = new MqttClientConfig();
    clientConfig.setClientId("MQTT client from test");
    clientConfig.setUsername(deviceCredentials.getCredentialsId());
    MqttClient mqttClient = MqttClient.create(clientConfig, listener);
    mqttClient.connect("localhost", 1883).get();
    return mqttClient;
}
Also used : MqttClient(org.thingsboard.mqtt.MqttClient) MqttClientConfig(org.thingsboard.mqtt.MqttClientConfig)

Example 3 with MqttClient

use of org.thingsboard.mqtt.MqttClient in project thingsboard by thingsboard.

the class MqttClientTest method telemetryUpload.

@Test
public void telemetryUpload() throws Exception {
    restClient.login("tenant@thingsboard.org", "tenant");
    Device device = createDevice("mqtt_");
    DeviceCredentials deviceCredentials = restClient.getDeviceCredentialsByDeviceId(device.getId()).get();
    WsClient wsClient = subscribeToWebSocket(device.getId(), "LATEST_TELEMETRY", CmdsType.TS_SUB_CMDS);
    MqttClient mqttClient = getMqttClient(deviceCredentials, null);
    mqttClient.publish("v1/devices/me/telemetry", Unpooled.wrappedBuffer(createPayload().toString().getBytes())).get();
    WsTelemetryResponse actualLatestTelemetry = wsClient.getLastMessage();
    log.info("Received telemetry: {}", actualLatestTelemetry);
    wsClient.closeBlocking();
    Assert.assertEquals(4, actualLatestTelemetry.getData().size());
    Assert.assertEquals(Sets.newHashSet("booleanKey", "stringKey", "doubleKey", "longKey"), actualLatestTelemetry.getLatestValues().keySet());
    Assert.assertTrue(verify(actualLatestTelemetry, "booleanKey", Boolean.TRUE.toString()));
    Assert.assertTrue(verify(actualLatestTelemetry, "stringKey", "value1"));
    Assert.assertTrue(verify(actualLatestTelemetry, "doubleKey", Double.toString(42.0)));
    Assert.assertTrue(verify(actualLatestTelemetry, "longKey", Long.toString(73)));
    restClient.getRestTemplate().delete(HTTPS_URL + "/api/device/" + device.getId());
}
Also used : MqttClient(org.thingsboard.mqtt.MqttClient) Device(org.thingsboard.server.common.data.Device) WsTelemetryResponse(org.thingsboard.server.msa.mapper.WsTelemetryResponse) DeviceCredentials(org.thingsboard.server.common.data.security.DeviceCredentials) WsClient(org.thingsboard.server.msa.WsClient) AbstractContainerTest(org.thingsboard.server.msa.AbstractContainerTest)

Example 4 with MqttClient

use of org.thingsboard.mqtt.MqttClient in project thingsboard by thingsboard.

the class MqttClientTest method requestAttributeValuesFromServer.

@Test
public void requestAttributeValuesFromServer() throws Exception {
    restClient.login("tenant@thingsboard.org", "tenant");
    Device device = createDevice("mqtt_");
    DeviceCredentials deviceCredentials = restClient.getDeviceCredentialsByDeviceId(device.getId()).get();
    WsClient wsClient = subscribeToWebSocket(device.getId(), "CLIENT_SCOPE", CmdsType.ATTR_SUB_CMDS);
    MqttMessageListener listener = new MqttMessageListener();
    MqttClient mqttClient = getMqttClient(deviceCredentials, listener);
    // Add a new client attribute
    JsonObject clientAttributes = new JsonObject();
    String clientAttributeValue = RandomStringUtils.randomAlphanumeric(8);
    clientAttributes.addProperty("clientAttr", clientAttributeValue);
    mqttClient.publish("v1/devices/me/attributes", Unpooled.wrappedBuffer(clientAttributes.toString().getBytes())).get();
    WsTelemetryResponse actualLatestTelemetry = wsClient.getLastMessage();
    log.info("Received ws telemetry: {}", actualLatestTelemetry);
    wsClient.closeBlocking();
    Assert.assertEquals(1, actualLatestTelemetry.getData().size());
    Assert.assertEquals(Sets.newHashSet("clientAttr"), actualLatestTelemetry.getLatestValues().keySet());
    Assert.assertTrue(verify(actualLatestTelemetry, "clientAttr", clientAttributeValue));
    // Add a new shared attribute
    JsonObject sharedAttributes = new JsonObject();
    String sharedAttributeValue = RandomStringUtils.randomAlphanumeric(8);
    sharedAttributes.addProperty("sharedAttr", sharedAttributeValue);
    ResponseEntity sharedAttributesResponse = restClient.getRestTemplate().postForEntity(HTTPS_URL + "/api/plugins/telemetry/DEVICE/{deviceId}/SHARED_SCOPE", mapper.readTree(sharedAttributes.toString()), ResponseEntity.class, device.getId());
    Assert.assertTrue(sharedAttributesResponse.getStatusCode().is2xxSuccessful());
    // Subscribe to attributes response
    mqttClient.on("v1/devices/me/attributes/response/+", listener, MqttQoS.AT_LEAST_ONCE).get();
    // Wait until subscription is processed
    TimeUnit.SECONDS.sleep(3);
    // Request attributes
    JsonObject request = new JsonObject();
    request.addProperty("clientKeys", "clientAttr");
    request.addProperty("sharedKeys", "sharedAttr");
    mqttClient.publish("v1/devices/me/attributes/request/" + new Random().nextInt(100), Unpooled.wrappedBuffer(request.toString().getBytes())).get();
    MqttEvent event = listener.getEvents().poll(10, TimeUnit.SECONDS);
    AttributesResponse attributes = mapper.readValue(Objects.requireNonNull(event).getMessage(), AttributesResponse.class);
    log.info("Received telemetry: {}", attributes);
    Assert.assertEquals(1, attributes.getClient().size());
    Assert.assertEquals(clientAttributeValue, attributes.getClient().get("clientAttr"));
    Assert.assertEquals(1, attributes.getShared().size());
    Assert.assertEquals(sharedAttributeValue, attributes.getShared().get("sharedAttr"));
    restClient.getRestTemplate().delete(HTTPS_URL + "/api/device/" + device.getId());
}
Also used : MqttClient(org.thingsboard.mqtt.MqttClient) ResponseEntity(org.springframework.http.ResponseEntity) AttributesResponse(org.thingsboard.server.msa.mapper.AttributesResponse) Device(org.thingsboard.server.common.data.Device) WsTelemetryResponse(org.thingsboard.server.msa.mapper.WsTelemetryResponse) JsonObject(com.google.gson.JsonObject) DeviceCredentials(org.thingsboard.server.common.data.security.DeviceCredentials) WsClient(org.thingsboard.server.msa.WsClient) AbstractContainerTest(org.thingsboard.server.msa.AbstractContainerTest)

Example 5 with MqttClient

use of org.thingsboard.mqtt.MqttClient in project thingsboard by thingsboard.

the class MqttClientTest method deviceDeletedClosingSession.

@Test
public void deviceDeletedClosingSession() throws Exception {
    restClient.login("tenant@thingsboard.org", "tenant");
    String deviceForDeletingTestName = "Device for deleting notification test";
    Device device = createDevice(deviceForDeletingTestName);
    DeviceCredentials deviceCredentials = restClient.getDeviceCredentialsByDeviceId(device.getId()).get();
    MqttMessageListener listener = new MqttMessageListener();
    MqttClient mqttClient = getMqttClient(deviceCredentials, listener);
    restClient.deleteDevice(device.getId());
    TimeUnit.SECONDS.sleep(3);
    Assert.assertFalse(mqttClient.isConnected());
}
Also used : MqttClient(org.thingsboard.mqtt.MqttClient) Device(org.thingsboard.server.common.data.Device) DeviceCredentials(org.thingsboard.server.common.data.security.DeviceCredentials) AbstractContainerTest(org.thingsboard.server.msa.AbstractContainerTest)

Aggregations

MqttClient (org.thingsboard.mqtt.MqttClient)11 Device (org.thingsboard.server.common.data.Device)8 DeviceCredentials (org.thingsboard.server.common.data.security.DeviceCredentials)8 AbstractContainerTest (org.thingsboard.server.msa.AbstractContainerTest)8 JsonObject (com.google.gson.JsonObject)5 WsClient (org.thingsboard.server.msa.WsClient)4 WsTelemetryResponse (org.thingsboard.server.msa.mapper.WsTelemetryResponse)4 ResponseEntity (org.springframework.http.ResponseEntity)3 MqttClientConfig (org.thingsboard.mqtt.MqttClientConfig)3 JsonNode (com.fasterxml.jackson.databind.JsonNode)1 ListeningExecutorService (com.google.common.util.concurrent.ListeningExecutorService)1 IOException (java.io.IOException)1 TimeoutException (java.util.concurrent.TimeoutException)1 MqttConnectResult (org.thingsboard.mqtt.MqttConnectResult)1 RuleChainId (org.thingsboard.server.common.data.id.RuleChainId)1 RuleChain (org.thingsboard.server.common.data.rule.RuleChain)1 AttributesResponse (org.thingsboard.server.msa.mapper.AttributesResponse)1