Examples with RuleChain org.thingsboard.server.common.data.rule.RuleChain used on opensource projects

Example 31 with RuleChain

use of org.thingsboard.server.common.data.rule.RuleChain in project thingsboard by thingsboard.

the class BaseEdgeTest method verifyEdgeConnectionAndInitialData.

private void verifyEdgeConnectionAndInitialData() throws Exception {
    Assert.assertTrue(edgeImitator.waitForMessages());
    EdgeConfiguration configuration = edgeImitator.getConfiguration();
    Assert.assertNotNull(configuration);
    testAutoGeneratedCodeByProtobuf(configuration);
    Optional<DeviceUpdateMsg> deviceUpdateMsgOpt = edgeImitator.findMessageByType(DeviceUpdateMsg.class);
    Assert.assertTrue(deviceUpdateMsgOpt.isPresent());
    DeviceUpdateMsg deviceUpdateMsg = deviceUpdateMsgOpt.get();
    Assert.assertEquals(UpdateMsgType.ENTITY_CREATED_RPC_MESSAGE, deviceUpdateMsg.getMsgType());
    UUID deviceUUID = new UUID(deviceUpdateMsg.getIdMSB(), deviceUpdateMsg.getIdLSB());
    Device device = doGet("/api/device/" + deviceUUID.toString(), Device.class);
    Assert.assertNotNull(device);
    List<Device> edgeDevices = doGetTypedWithPageLink("/api/edge/" + edge.getUuidId() + "/devices?", new TypeReference<PageData<Device>>() {
    }, new PageLink(100)).getData();
    Assert.assertTrue(edgeDevices.contains(device));
    List<DeviceProfileUpdateMsg> deviceProfileUpdateMsgList = edgeImitator.findAllMessagesByType(DeviceProfileUpdateMsg.class);
    Assert.assertEquals(3, deviceProfileUpdateMsgList.size());
    Optional<DeviceProfileUpdateMsg> deviceProfileUpdateMsgOpt = deviceProfileUpdateMsgList.stream().filter(dfum -> CUSTOM_DEVICE_PROFILE_NAME.equals(dfum.getName())).findAny();
    Assert.assertTrue(deviceProfileUpdateMsgOpt.isPresent());
    DeviceProfileUpdateMsg deviceProfileUpdateMsg = deviceProfileUpdateMsgOpt.get();
    Assert.assertEquals(UpdateMsgType.ENTITY_CREATED_RPC_MESSAGE, deviceProfileUpdateMsg.getMsgType());
    UUID deviceProfileUUID = new UUID(deviceProfileUpdateMsg.getIdMSB(), deviceProfileUpdateMsg.getIdLSB());
    DeviceProfile deviceProfile = doGet("/api/deviceProfile/" + deviceProfileUUID.toString(), DeviceProfile.class);
    Assert.assertNotNull(deviceProfile);
    Assert.assertNotNull(deviceProfile.getProfileData());
    Assert.assertNotNull(deviceProfile.getProfileData().getAlarms());
    Assert.assertNotNull(deviceProfile.getProfileData().getAlarms().get(0).getClearRule());
    testAutoGeneratedCodeByProtobuf(deviceProfileUpdateMsg);
    Optional<AssetUpdateMsg> assetUpdateMsgOpt = edgeImitator.findMessageByType(AssetUpdateMsg.class);
    Assert.assertTrue(assetUpdateMsgOpt.isPresent());
    AssetUpdateMsg assetUpdateMsg = assetUpdateMsgOpt.get();
    Assert.assertEquals(UpdateMsgType.ENTITY_CREATED_RPC_MESSAGE, assetUpdateMsg.getMsgType());
    UUID assetUUID = new UUID(assetUpdateMsg.getIdMSB(), assetUpdateMsg.getIdLSB());
    Asset asset = doGet("/api/asset/" + assetUUID.toString(), Asset.class);
    Assert.assertNotNull(asset);
    List<Asset> edgeAssets = doGetTypedWithPageLink("/api/edge/" + edge.getUuidId() + "/assets?", new TypeReference<PageData<Asset>>() {
    }, new PageLink(100)).getData();
    Assert.assertTrue(edgeAssets.contains(asset));
    testAutoGeneratedCodeByProtobuf(assetUpdateMsg);
    Optional<RuleChainUpdateMsg> ruleChainUpdateMsgOpt = edgeImitator.findMessageByType(RuleChainUpdateMsg.class);
    Assert.assertTrue(ruleChainUpdateMsgOpt.isPresent());
    RuleChainUpdateMsg ruleChainUpdateMsg = ruleChainUpdateMsgOpt.get();
    Assert.assertEquals(UpdateMsgType.ENTITY_CREATED_RPC_MESSAGE, ruleChainUpdateMsg.getMsgType());
    UUID ruleChainUUID = new UUID(ruleChainUpdateMsg.getIdMSB(), ruleChainUpdateMsg.getIdLSB());
    RuleChain ruleChain = doGet("/api/ruleChain/" + ruleChainUUID.toString(), RuleChain.class);
    Assert.assertNotNull(ruleChain);
    List<RuleChain> edgeRuleChains = doGetTypedWithPageLink("/api/edge/" + edge.getUuidId() + "/ruleChains?", new TypeReference<PageData<RuleChain>>() {
    }, new PageLink(100)).getData();
    Assert.assertTrue(edgeRuleChains.contains(ruleChain));
    testAutoGeneratedCodeByProtobuf(ruleChainUpdateMsg);
    validateAdminSettings();
}

Example 32 with RuleChain

use of org.thingsboard.server.common.data.rule.RuleChain in project thingsboard by thingsboard.

the class BaseEdgeTest method testRuleChains.

@Test
public void testRuleChains() throws Exception {
    // 1
    edgeImitator.expectMessageAmount(2);
    RuleChain ruleChain = new RuleChain();
    ruleChain.setName("Edge Test Rule Chain");
    ruleChain.setType(RuleChainType.EDGE);
    RuleChain savedRuleChain = doPost("/api/ruleChain", ruleChain, RuleChain.class);
    doPost("/api/edge/" + edge.getUuidId() + "/ruleChain/" + savedRuleChain.getUuidId(), RuleChain.class);
    createRuleChainMetadata(savedRuleChain);
    Assert.assertTrue(edgeImitator.waitForMessages());
    Optional<RuleChainUpdateMsg> ruleChainUpdateMsgOpt = edgeImitator.findMessageByType(RuleChainUpdateMsg.class);
    Assert.assertTrue(ruleChainUpdateMsgOpt.isPresent());
    RuleChainUpdateMsg ruleChainUpdateMsg = ruleChainUpdateMsgOpt.get();
    Assert.assertTrue(UpdateMsgType.ENTITY_CREATED_RPC_MESSAGE.equals(ruleChainUpdateMsg.getMsgType()) || UpdateMsgType.ENTITY_UPDATED_RPC_MESSAGE.equals(ruleChainUpdateMsg.getMsgType()));
    Assert.assertEquals(ruleChainUpdateMsg.getIdMSB(), savedRuleChain.getUuidId().getMostSignificantBits());
    Assert.assertEquals(ruleChainUpdateMsg.getIdLSB(), savedRuleChain.getUuidId().getLeastSignificantBits());
    Assert.assertEquals(ruleChainUpdateMsg.getName(), savedRuleChain.getName());
    // 2
    testRuleChainMetadataRequestMsg(savedRuleChain.getId());
    // 3
    edgeImitator.expectMessageAmount(1);
    doDelete("/api/edge/" + edge.getUuidId() + "/ruleChain/" + savedRuleChain.getUuidId(), RuleChain.class);
    Assert.assertTrue(edgeImitator.waitForMessages());
    ruleChainUpdateMsgOpt = edgeImitator.findMessageByType(RuleChainUpdateMsg.class);
    Assert.assertTrue(ruleChainUpdateMsgOpt.isPresent());
    ruleChainUpdateMsg = ruleChainUpdateMsgOpt.get();
    Assert.assertEquals(UpdateMsgType.ENTITY_DELETED_RPC_MESSAGE, ruleChainUpdateMsg.getMsgType());
    Assert.assertEquals(ruleChainUpdateMsg.getIdMSB(), savedRuleChain.getUuidId().getMostSignificantBits());
    Assert.assertEquals(ruleChainUpdateMsg.getIdLSB(), savedRuleChain.getUuidId().getLeastSignificantBits());
    // 4
    edgeImitator.expectMessageAmount(1);
    doDelete("/api/ruleChain/" + savedRuleChain.getUuidId()).andExpect(status().isOk());
    Assert.assertFalse(edgeImitator.waitForMessages(1));
}

Example 33 with RuleChain

use of org.thingsboard.server.common.data.rule.RuleChain in project thingsboard by thingsboard.

the class RuleChainController method setEdgeTemplateRootRuleChain.

@ApiOperation(value = "Set Edge Template Root Rule Chain (setEdgeTemplateRootRuleChain)", notes = "Makes the rule chain to be root rule chain for any new edge that will be created. " + "Does not update root rule chain for already created edges. " + TENANT_AUTHORITY_PARAGRAPH)
@PreAuthorize("hasAnyAuthority('TENANT_ADMIN')")
@RequestMapping(value = "/ruleChain/{ruleChainId}/edgeTemplateRoot", method = RequestMethod.POST)
@ResponseBody
public RuleChain setEdgeTemplateRootRuleChain(@ApiParam(value = RULE_CHAIN_ID_PARAM_DESCRIPTION) @PathVariable(RULE_CHAIN_ID) String strRuleChainId) throws ThingsboardException {
    checkParameter(RULE_CHAIN_ID, strRuleChainId);
    try {
        RuleChainId ruleChainId = new RuleChainId(toUUID(strRuleChainId));
        RuleChain ruleChain = checkRuleChain(ruleChainId, Operation.WRITE);
        ruleChainService.setEdgeTemplateRootRuleChain(getTenantId(), ruleChainId);
        return ruleChain;
    } catch (Exception e) {
        logEntityAction(emptyId(EntityType.RULE_CHAIN), null, null, ActionType.UPDATED, e, strRuleChainId);
        throw handleException(e);
    }
}

Example 34 with RuleChain

use of org.thingsboard.server.common.data.rule.RuleChain in project thingsboard by thingsboard.

the class RuleChainController method deleteRuleChain.

@ApiOperation(value = "Delete rule chain (deleteRuleChain)", notes = "Deletes the rule chain. Referencing non-existing rule chain Id will cause an error. " + "Referencing rule chain that is used in the device profiles will cause an error." + TENANT_AUTHORITY_PARAGRAPH)
@PreAuthorize("hasAnyAuthority('TENANT_ADMIN')")
@RequestMapping(value = "/ruleChain/{ruleChainId}", method = RequestMethod.DELETE)
@ResponseStatus(value = HttpStatus.OK)
public void deleteRuleChain(@ApiParam(value = RULE_CHAIN_ID_PARAM_DESCRIPTION) @PathVariable(RULE_CHAIN_ID) String strRuleChainId) throws ThingsboardException {
    checkParameter(RULE_CHAIN_ID, strRuleChainId);
    try {
        RuleChainId ruleChainId = new RuleChainId(toUUID(strRuleChainId));
        RuleChain ruleChain = checkRuleChain(ruleChainId, Operation.DELETE);
        List<RuleNode> referencingRuleNodes = ruleChainService.getReferencingRuleChainNodes(getTenantId(), ruleChainId);
        Set<RuleChainId> referencingRuleChainIds = referencingRuleNodes.stream().map(RuleNode::getRuleChainId).collect(Collectors.toSet());
        List<EdgeId> relatedEdgeIds = null;
        if (RuleChainType.EDGE.equals(ruleChain.getType())) {
            relatedEdgeIds = findRelatedEdgeIds(getTenantId(), ruleChainId);
        }
        ruleChainService.deleteRuleChainById(getTenantId(), ruleChainId);
        referencingRuleChainIds.remove(ruleChain.getId());
        if (RuleChainType.CORE.equals(ruleChain.getType())) {
            referencingRuleChainIds.forEach(referencingRuleChainId -> tbClusterService.broadcastEntityStateChangeEvent(ruleChain.getTenantId(), referencingRuleChainId, ComponentLifecycleEvent.UPDATED));
            tbClusterService.broadcastEntityStateChangeEvent(ruleChain.getTenantId(), ruleChain.getId(), ComponentLifecycleEvent.DELETED);
        }
        logEntityAction(ruleChainId, ruleChain, null, ActionType.DELETED, null, strRuleChainId);
        if (RuleChainType.EDGE.equals(ruleChain.getType())) {
            sendDeleteNotificationMsg(ruleChain.getTenantId(), ruleChain.getId(), relatedEdgeIds);
        }
    } catch (Exception e) {
        logEntityAction(emptyId(EntityType.RULE_CHAIN), null, null, ActionType.DELETED, e, strRuleChainId);
        throw handleException(e);
    }
}

Example 35 with RuleChain

use of org.thingsboard.server.common.data.rule.RuleChain in project thingsboard by thingsboard.

the class RuleChainController method setAutoAssignToEdgeRuleChain.

@ApiOperation(value = "Set Auto Assign To Edge Rule Chain (setAutoAssignToEdgeRuleChain)", notes = "Makes the rule chain to be automatically assigned for any new edge that will be created. " + "Does not assign this rule chain for already created edges. " + TENANT_AUTHORITY_PARAGRAPH)
@PreAuthorize("hasAuthority('TENANT_ADMIN')")
@RequestMapping(value = "/ruleChain/{ruleChainId}/autoAssignToEdge", method = RequestMethod.POST)
@ResponseBody
public RuleChain setAutoAssignToEdgeRuleChain(@ApiParam(value = RULE_CHAIN_ID_PARAM_DESCRIPTION) @PathVariable(RULE_CHAIN_ID) String strRuleChainId) throws ThingsboardException {
    checkParameter(RULE_CHAIN_ID, strRuleChainId);
    try {
        RuleChainId ruleChainId = new RuleChainId(toUUID(strRuleChainId));
        RuleChain ruleChain = checkRuleChain(ruleChainId, Operation.WRITE);
        ruleChainService.setAutoAssignToEdgeRuleChain(getTenantId(), ruleChainId);
        return ruleChain;
    } catch (Exception e) {
        logEntityAction(emptyId(EntityType.RULE_CHAIN), null, null, ActionType.UPDATED, e, strRuleChainId);
        throw handleException(e);
    }
}