Examples with RuleNodeId org.thingsboard.server.common.data.id.RuleNodeId used on opensource projects

Example 16 with RuleNodeId

use of org.thingsboard.server.common.data.id.RuleNodeId 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 17 with RuleNodeId

use of org.thingsboard.server.common.data.id.RuleNodeId in project thingsboard by thingsboard.

the class TbMsg method fromBytes.

public static TbMsg fromBytes(String queueName, byte[] data, TbMsgCallback callback) {
    try {
        MsgProtos.TbMsgProto proto = MsgProtos.TbMsgProto.parseFrom(data);
        TbMsgMetaData metaData = new TbMsgMetaData(proto.getMetaData().getDataMap());
        EntityId entityId = EntityIdFactory.getByTypeAndUuid(proto.getEntityType(), new UUID(proto.getEntityIdMSB(), proto.getEntityIdLSB()));
        CustomerId customerId = null;
        RuleChainId ruleChainId = null;
        RuleNodeId ruleNodeId = null;
        if (proto.getCustomerIdMSB() != 0L && proto.getCustomerIdLSB() != 0L) {
            customerId = new CustomerId(new UUID(proto.getCustomerIdMSB(), proto.getCustomerIdLSB()));
        }
        if (proto.getRuleChainIdMSB() != 0L && proto.getRuleChainIdLSB() != 0L) {
            ruleChainId = new RuleChainId(new UUID(proto.getRuleChainIdMSB(), proto.getRuleChainIdLSB()));
        }
        if (proto.getRuleNodeIdMSB() != 0L && proto.getRuleNodeIdLSB() != 0L) {
            ruleNodeId = new RuleNodeId(new UUID(proto.getRuleNodeIdMSB(), proto.getRuleNodeIdLSB()));
        }
        TbMsgProcessingCtx ctx;
        if (proto.hasCtx()) {
            ctx = TbMsgProcessingCtx.fromProto(proto.getCtx());
        } else {
            // Backward compatibility with unprocessed messages fetched from queue after update.
            ctx = new TbMsgProcessingCtx(proto.getRuleNodeExecCounter());
        }
        TbMsgDataType dataType = TbMsgDataType.values()[proto.getDataType()];
        return new TbMsg(queueName, UUID.fromString(proto.getId()), proto.getTs(), proto.getType(), entityId, customerId, metaData, dataType, proto.getData(), ruleChainId, ruleNodeId, ctx, callback);
    } catch (InvalidProtocolBufferException e) {
        throw new IllegalStateException("Could not parse protobuf for TbMsg", e);
    }
}

Example 18 with RuleNodeId

use of org.thingsboard.server.common.data.id.RuleNodeId in project thingsboard by thingsboard.

the class AccessValidator method validateRule.

private void validateRule(final SecurityUser currentUser, Operation operation, EntityId entityId, FutureCallback<ValidationResult> callback) {
    if (currentUser.isCustomerUser()) {
        callback.onSuccess(ValidationResult.accessDenied(CUSTOMER_USER_IS_NOT_ALLOWED_TO_PERFORM_THIS_OPERATION));
    } else {
        ListenableFuture<RuleNode> ruleNodeFuture = ruleChainService.findRuleNodeByIdAsync(currentUser.getTenantId(), new RuleNodeId(entityId.getId()));
        Futures.addCallback(ruleNodeFuture, getCallback(callback, ruleNodeTmp -> {
            RuleNode ruleNode = ruleNodeTmp;
            if (ruleNode == null) {
                return ValidationResult.entityNotFound("Rule node with requested id wasn't found!");
            } else if (ruleNode.getRuleChainId() == null) {
                return ValidationResult.entityNotFound("Rule chain with requested node id wasn't found!");
            } else {
                // TODO: make async
                RuleChain ruleChain = ruleChainService.findRuleChainById(currentUser.getTenantId(), ruleNode.getRuleChainId());
                try {
                    accessControlService.checkPermission(currentUser, Resource.RULE_CHAIN, operation, ruleNode.getRuleChainId(), ruleChain);
                } catch (ThingsboardException e) {
                    return ValidationResult.accessDenied(e.getMessage());
                }
                return ValidationResult.ok(ruleNode);
            }
        }), executor);
    }
}

Example 19 with RuleNodeId

use of org.thingsboard.server.common.data.id.RuleNodeId in project thingsboard by thingsboard.

the class RuleChainMsgConstructorTest method createRuleNode.

private RuleNode createRuleNode(RuleChainId ruleChainId, String type, String name, JsonNode configuration, JsonNode additionalInfo) {
    RuleNode e = new RuleNode();
    e.setRuleChainId(ruleChainId);
    e.setType(type);
    e.setName(name);
    e.setDebugMode(false);
    e.setConfiguration(configuration);
    e.setAdditionalInfo(additionalInfo);
    e.setId(new RuleNodeId(UUID.randomUUID()));
    return e;
}

Example 20 with RuleNodeId

use of org.thingsboard.server.common.data.id.RuleNodeId in project thingsboard by thingsboard.

the class TbTransformMsgNodeTest method exceptionHandledCorrectly.

@Test
public void exceptionHandledCorrectly() throws TbNodeException, ScriptException {
    initWithScript();
    TbMsgMetaData metaData = new TbMsgMetaData();
    metaData.putValue("temp", "7");
    String rawJson = "{\"passed\": 5";
    RuleChainId ruleChainId = new RuleChainId(Uuids.timeBased());
    RuleNodeId ruleNodeId = new RuleNodeId(Uuids.timeBased());
    TbMsg msg = TbMsg.newMsg("USER", null, metaData, TbMsgDataType.JSON, rawJson, ruleChainId, ruleNodeId);
    when(scriptEngine.executeUpdateAsync(msg)).thenReturn(Futures.immediateFailedFuture(new IllegalStateException("error")));
    node.onMsg(ctx, msg);
    verifyError(msg, "error", IllegalStateException.class);
}