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

Example 51 with TbMsg

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

the class RuleChainActorMessageProcessor method onQueueToRuleEngineMsg.

void onQueueToRuleEngineMsg(QueueToRuleEngineMsg envelope) {
    TbMsg msg = envelope.getMsg();
    if (!checkMsgValid(msg)) {
        return;
    }
    log.trace("[{}][{}] Processing message [{}]: {}", entityId, firstId, msg.getId(), msg);
    if (envelope.getRelationTypes() == null || envelope.getRelationTypes().isEmpty()) {
        onTellNext(msg, true);
    } else {
        onTellNext(msg, envelope.getMsg().getRuleNodeId(), envelope.getRelationTypes(), envelope.getFailureMessage());
    }
}

Example 52 with TbMsg

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

the class DefaultTbContext method enqueueForTellNext.

private void enqueueForTellNext(TopicPartitionInfo tpi, String queueName, TbMsg source, Set<String> relationTypes, String failureMessage, Runnable onSuccess, Consumer<Throwable> onFailure) {
    if (!source.isValid()) {
        log.trace("[{}] Skip invalid message: {}", getTenantId(), source);
        onFailure.accept(new IllegalArgumentException("Source message is no longer valid!"));
        return;
    }
    RuleChainId ruleChainId = nodeCtx.getSelf().getRuleChainId();
    RuleNodeId ruleNodeId = nodeCtx.getSelf().getId();
    TbMsg tbMsg = TbMsg.newMsg(source, queueName, ruleChainId, ruleNodeId);
    TransportProtos.ToRuleEngineMsg.Builder msg = TransportProtos.ToRuleEngineMsg.newBuilder().setTenantIdMSB(getTenantId().getId().getMostSignificantBits()).setTenantIdLSB(getTenantId().getId().getLeastSignificantBits()).setTbMsg(TbMsg.toByteString(tbMsg)).addAllRelationTypes(relationTypes);
    if (failureMessage != null) {
        msg.setFailureMessage(failureMessage);
    }
    if (nodeCtx.getSelf().isDebugMode()) {
        relationTypes.forEach(relationType -> mainCtx.persistDebugOutput(nodeCtx.getTenantId(), nodeCtx.getSelf().getId(), tbMsg, relationType, null, failureMessage));
    }
    mainCtx.getClusterService().pushMsgToRuleEngine(tpi, tbMsg.getId(), msg.build(), new SimpleTbQueueCallback(onSuccess, onFailure));
}

Example 53 with TbMsg

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

the class RuleNodeActorMessageProcessor method onRuleToSelfMsg.

public void onRuleToSelfMsg(RuleNodeToSelfMsg msg) throws Exception {
    checkComponentStateActive(msg.getMsg());
    TbMsg tbMsg = msg.getMsg();
    int ruleNodeCount = tbMsg.getAndIncrementRuleNodeCounter();
    int maxRuleNodeExecutionsPerMessage = getTenantProfileConfiguration().getMaxRuleNodeExecsPerMessage();
    if (maxRuleNodeExecutionsPerMessage == 0 || ruleNodeCount < maxRuleNodeExecutionsPerMessage) {
        apiUsageClient.report(tenantId, tbMsg.getCustomerId(), ApiUsageRecordKey.RE_EXEC_COUNT);
        if (ruleNode.isDebugMode()) {
            systemContext.persistDebugInput(tenantId, entityId, msg.getMsg(), "Self");
        }
        try {
            tbNode.onMsg(defaultCtx, msg.getMsg());
        } catch (Exception e) {
            defaultCtx.tellFailure(msg.getMsg(), e);
        }
    } else {
        tbMsg.getCallback().onFailure(new RuleNodeException("Message is processed by more then " + maxRuleNodeExecutionsPerMessage + " rule nodes!", ruleChainName, ruleNode));
    }
}

Example 54 with TbMsg

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

the class DefaultTbRuleEngineConsumerService method forwardToRuleEngineActor.

private void forwardToRuleEngineActor(String queueName, TenantId tenantId, ToRuleEngineMsg toRuleEngineMsg, TbMsgCallback callback) {
    TbMsg tbMsg = TbMsg.fromBytes(queueName, toRuleEngineMsg.getTbMsg().toByteArray(), callback);
    QueueToRuleEngineMsg msg;
    ProtocolStringList relationTypesList = toRuleEngineMsg.getRelationTypesList();
    Set<String> relationTypes = null;
    if (relationTypesList != null) {
        if (relationTypesList.size() == 1) {
            relationTypes = Collections.singleton(relationTypesList.get(0));
        } else {
            relationTypes = new HashSet<>(relationTypesList);
        }
    }
    msg = new QueueToRuleEngineMsg(tenantId, tbMsg, relationTypes, toRuleEngineMsg.getFailureMessage());
    actorContext.tell(msg);
}

Example 55 with TbMsg

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

the class DefaultDeviceStateService method pushRuleEngineMessage.

private void pushRuleEngineMessage(DeviceStateData stateData, String msgType) {
    DeviceState state = stateData.getState();
    try {
        String data;
        if (msgType.equals(CONNECT_EVENT)) {
            ObjectNode stateNode = JacksonUtil.convertValue(state, ObjectNode.class);
            stateNode.remove(ACTIVITY_STATE);
            data = JacksonUtil.toString(stateNode);
        } else {
            data = JacksonUtil.toString(state);
        }
        TbMsgMetaData md = stateData.getMetaData().copy();
        if (!persistToTelemetry) {
            md.putValue(DataConstants.SCOPE, SERVER_SCOPE);
        }
        TbMsg tbMsg = TbMsg.newMsg(msgType, stateData.getDeviceId(), stateData.getCustomerId(), md, TbMsgDataType.JSON, data);
        clusterService.pushMsgToRuleEngine(stateData.getTenantId(), stateData.getDeviceId(), tbMsg, null);
    } catch (Exception e) {
        log.warn("[{}] Failed to push inactivity alarm: {}", stateData.getDeviceId(), state, e);
    }
}