Examples with TbMsgMetaData org.thingsboard.server.common.msg.TbMsgMetaData used on opensource projects

Example 56 with TbMsgMetaData

use of org.thingsboard.server.common.msg.TbMsgMetaData in project thingsboard by thingsboard.

the class TbCreateRelationNodeTest method testDeleteCurrentRelationsCreateNewRelation.

@Test
public void testDeleteCurrentRelationsCreateNewRelation() throws TbNodeException {
    init(createRelationNodeConfigWithRemoveCurrentRelations());
    DeviceId deviceId = new DeviceId(Uuids.timeBased());
    AssetId assetId = new AssetId(Uuids.timeBased());
    Asset asset = new Asset();
    asset.setId(assetId);
    when(assetService.findAssetByTenantIdAndName(any(), eq("AssetName"))).thenReturn(asset);
    when(assetService.findAssetByIdAsync(any(), eq(assetId))).thenReturn(Futures.immediateFuture(asset));
    TbMsgMetaData metaData = new TbMsgMetaData();
    metaData.putValue("name", "AssetName");
    metaData.putValue("type", "AssetType");
    msg = TbMsg.newMsg(DataConstants.ENTITY_CREATED, deviceId, metaData, TbMsgDataType.JSON, "{}", ruleChainId, ruleNodeId);
    EntityRelation relation = new EntityRelation();
    when(ctx.getRelationService().findByToAndTypeAsync(any(), eq(msg.getOriginator()), eq(RELATION_TYPE_CONTAINS), eq(RelationTypeGroup.COMMON))).thenReturn(Futures.immediateFuture(Collections.singletonList(relation)));
    when(ctx.getRelationService().deleteRelationAsync(any(), eq(relation))).thenReturn(Futures.immediateFuture(true));
    when(ctx.getRelationService().checkRelation(any(), eq(assetId), eq(deviceId), eq(RELATION_TYPE_CONTAINS), eq(RelationTypeGroup.COMMON))).thenReturn(Futures.immediateFuture(false));
    when(ctx.getRelationService().saveRelationAsync(any(), eq(new EntityRelation(assetId, deviceId, RELATION_TYPE_CONTAINS, RelationTypeGroup.COMMON)))).thenReturn(Futures.immediateFuture(true));
    node.onMsg(ctx, msg);
    verify(ctx).tellNext(msg, TbRelationTypes.SUCCESS);
}

Example 57 with TbMsgMetaData

use of org.thingsboard.server.common.msg.TbMsgMetaData in project thingsboard by thingsboard.

the class DefaultTransportService method process.

@Override
public void process(TransportProtos.SessionInfoProto sessionInfo, TransportProtos.PostTelemetryMsg msg, TransportServiceCallback<Void> callback) {
    int dataPoints = 0;
    for (TransportProtos.TsKvListProto tsKv : msg.getTsKvListList()) {
        dataPoints += tsKv.getKvCount();
    }
    if (checkLimits(sessionInfo, msg, callback, dataPoints)) {
        reportActivityInternal(sessionInfo);
        TenantId tenantId = getTenantId(sessionInfo);
        DeviceId deviceId = new DeviceId(new UUID(sessionInfo.getDeviceIdMSB(), sessionInfo.getDeviceIdLSB()));
        CustomerId customerId = getCustomerId(sessionInfo);
        MsgPackCallback packCallback = new MsgPackCallback(msg.getTsKvListCount(), new ApiStatsProxyCallback<>(tenantId, customerId, dataPoints, callback));
        for (TransportProtos.TsKvListProto tsKv : msg.getTsKvListList()) {
            TbMsgMetaData metaData = new TbMsgMetaData();
            metaData.putValue("deviceName", sessionInfo.getDeviceName());
            metaData.putValue("deviceType", sessionInfo.getDeviceType());
            metaData.putValue("ts", tsKv.getTs() + "");
            JsonObject json = JsonUtils.getJsonObject(tsKv.getKvList());
            sendToRuleEngine(tenantId, deviceId, customerId, sessionInfo, json, metaData, SessionMsgType.POST_TELEMETRY_REQUEST, packCallback);
        }
    }
}

Example 58 with TbMsgMetaData

use of org.thingsboard.server.common.msg.TbMsgMetaData in project thingsboard by thingsboard.

the class DefaultTransportService method process.

@Override
public void process(TransportProtos.SessionInfoProto sessionInfo, TransportProtos.ToServerRpcRequestMsg msg, TransportServiceCallback<Void> callback) {
    if (checkLimits(sessionInfo, msg, callback)) {
        reportActivityInternal(sessionInfo);
        UUID sessionId = toSessionId(sessionInfo);
        TenantId tenantId = getTenantId(sessionInfo);
        DeviceId deviceId = getDeviceId(sessionInfo);
        JsonObject json = new JsonObject();
        json.addProperty("method", msg.getMethodName());
        json.add("params", JsonUtils.parse(msg.getParams()));
        TbMsgMetaData metaData = new TbMsgMetaData();
        metaData.putValue("deviceName", sessionInfo.getDeviceName());
        metaData.putValue("deviceType", sessionInfo.getDeviceType());
        metaData.putValue("requestId", Integer.toString(msg.getRequestId()));
        metaData.putValue("serviceId", serviceInfoProvider.getServiceId());
        metaData.putValue("sessionId", sessionId.toString());
        sendToRuleEngine(tenantId, deviceId, getCustomerId(sessionInfo), sessionInfo, json, metaData, SessionMsgType.TO_SERVER_RPC_REQUEST, new TransportTbQueueCallback(callback));
        String requestId = sessionId + "-" + msg.getRequestId();
        toServerRpcPendingMap.put(requestId, new RpcRequestMetadata(sessionId, msg.getRequestId()));
        scheduler.schedule(() -> processTimeout(requestId), clientSideRpcTimeout, TimeUnit.MILLISECONDS);
    }
}

Example 59 with TbMsgMetaData

use of org.thingsboard.server.common.msg.TbMsgMetaData in project thingsboard by thingsboard.

the class DefaultTransportService method process.

@Override
public void process(TransportProtos.SessionInfoProto sessionInfo, TransportProtos.PostAttributeMsg msg, TransportServiceCallback<Void> callback) {
    if (checkLimits(sessionInfo, msg, callback, msg.getKvCount())) {
        reportActivityInternal(sessionInfo);
        TenantId tenantId = getTenantId(sessionInfo);
        DeviceId deviceId = new DeviceId(new UUID(sessionInfo.getDeviceIdMSB(), sessionInfo.getDeviceIdLSB()));
        JsonObject json = JsonUtils.getJsonObject(msg.getKvList());
        TbMsgMetaData metaData = new TbMsgMetaData();
        metaData.putValue("deviceName", sessionInfo.getDeviceName());
        metaData.putValue("deviceType", sessionInfo.getDeviceType());
        metaData.putValue("notifyDevice", "false");
        CustomerId customerId = getCustomerId(sessionInfo);
        sendToRuleEngine(tenantId, deviceId, customerId, sessionInfo, json, metaData, SessionMsgType.POST_ATTRIBUTES_REQUEST, new TransportTbQueueCallback(new ApiStatsProxyCallback<>(tenantId, customerId, msg.getKvList().size(), callback)));
    }
}

Example 60 with TbMsgMetaData

use of org.thingsboard.server.common.msg.TbMsgMetaData in project thingsboard by thingsboard.

the class TbSnsNode method processPublishResult.

private TbMsg processPublishResult(TbContext ctx, TbMsg origMsg, PublishResult result) {
    TbMsgMetaData metaData = origMsg.getMetaData().copy();
    metaData.putValue(MESSAGE_ID, result.getMessageId());
    metaData.putValue(REQUEST_ID, result.getSdkResponseMetadata().getRequestId());
    return ctx.transformMsg(origMsg, origMsg.getType(), origMsg.getOriginator(), metaData, origMsg.getData());
}