Examples with PageData org.thingsboard.server.common.data.page.PageData used on opensource projects

Example 21 with PageData

use of org.thingsboard.server.common.data.page.PageData 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 22 with PageData

use of org.thingsboard.server.common.data.page.PageData in project thingsboard by thingsboard.

the class BaseEdgeTest method testSendAlarmToCloud.

@Test
public void testSendAlarmToCloud() throws Exception {
    Device device = saveDeviceOnCloudAndVerifyDeliveryToEdge();
    UplinkMsg.Builder uplinkMsgBuilder = UplinkMsg.newBuilder();
    AlarmUpdateMsg.Builder alarmUpdateMgBuilder = AlarmUpdateMsg.newBuilder();
    alarmUpdateMgBuilder.setName("alarm from edge");
    alarmUpdateMgBuilder.setStatus(AlarmStatus.ACTIVE_UNACK.name());
    alarmUpdateMgBuilder.setSeverity(AlarmSeverity.CRITICAL.name());
    alarmUpdateMgBuilder.setOriginatorName(device.getName());
    alarmUpdateMgBuilder.setOriginatorType(EntityType.DEVICE.name());
    testAutoGeneratedCodeByProtobuf(alarmUpdateMgBuilder);
    uplinkMsgBuilder.addAlarmUpdateMsg(alarmUpdateMgBuilder.build());
    testAutoGeneratedCodeByProtobuf(uplinkMsgBuilder);
    edgeImitator.expectResponsesAmount(1);
    edgeImitator.sendUplinkMsg(uplinkMsgBuilder.build());
    Assert.assertTrue(edgeImitator.waitForResponses());
    List<AlarmInfo> alarms = doGetTypedWithPageLink("/api/alarm/{entityType}/{entityId}?", new TypeReference<PageData<AlarmInfo>>() {
    }, new PageLink(100), device.getId().getEntityType().name(), device.getUuidId()).getData();
    Optional<AlarmInfo> foundAlarm = alarms.stream().filter(alarm -> alarm.getType().equals("alarm from edge")).findAny();
    Assert.assertTrue(foundAlarm.isPresent());
    AlarmInfo alarmInfo = foundAlarm.get();
    Assert.assertEquals(device.getId(), alarmInfo.getOriginator());
    Assert.assertEquals(AlarmStatus.ACTIVE_UNACK, alarmInfo.getStatus());
    Assert.assertEquals(AlarmSeverity.CRITICAL, alarmInfo.getSeverity());
}

Example 23 with PageData

use of org.thingsboard.server.common.data.page.PageData in project thingsboard by thingsboard.

the class TbAlarmDataSubCtx method doFetchAlarms.

private void doFetchAlarms() {
    AlarmDataUpdate update;
    if (!entitiesMap.isEmpty()) {
        long start = System.currentTimeMillis();
        PageData<AlarmData> alarms = alarmService.findAlarmDataByQueryForEntities(getTenantId(), query, getOrderedEntityIds());
        long end = System.currentTimeMillis();
        stats.getAlarmQueryInvocationCnt().incrementAndGet();
        stats.getAlarmQueryTimeSpent().addAndGet(end - start);
        alarms = setAndMergeAlarmsData(alarms);
        update = new AlarmDataUpdate(cmdId, alarms, null, maxEntitiesPerAlarmSubscription, data.getTotalElements());
    } else {
        update = new AlarmDataUpdate(cmdId, new PageData<>(), null, maxEntitiesPerAlarmSubscription, data.getTotalElements());
    }
    wsService.sendWsMsg(getSessionId(), update);
}

Example 24 with PageData

use of org.thingsboard.server.common.data.page.PageData in project thingsboard by thingsboard.

the class DefaultEntityQueryService method findAlarmDataByQuery.

@Override
public PageData<AlarmData> findAlarmDataByQuery(SecurityUser securityUser, AlarmDataQuery query) {
    EntityDataQuery entityDataQuery = this.buildEntityDataQuery(query);
    PageData<EntityData> entities = entityService.findEntityDataByQuery(securityUser.getTenantId(), securityUser.getCustomerId(), entityDataQuery);
    if (entities.getTotalElements() > 0) {
        LinkedHashMap<EntityId, EntityData> entitiesMap = new LinkedHashMap<>();
        for (EntityData entityData : entities.getData()) {
            entitiesMap.put(entityData.getEntityId(), entityData);
        }
        PageData<AlarmData> alarms = alarmService.findAlarmDataByQueryForEntities(securityUser.getTenantId(), query, entitiesMap.keySet());
        for (AlarmData alarmData : alarms.getData()) {
            EntityId entityId = alarmData.getEntityId();
            if (entityId != null) {
                EntityData entityData = entitiesMap.get(entityId);
                if (entityData != null) {
                    alarmData.getLatest().putAll(entityData.getLatest());
                }
            }
        }
        return alarms;
    } else {
        return new PageData<>();
    }
}

Example 25 with PageData

use of org.thingsboard.server.common.data.page.PageData in project thingsboard by thingsboard.

the class BaseDeviceControllerTest method testAssignDeviceToEdge.

@Test
public void testAssignDeviceToEdge() throws Exception {
    Edge edge = constructEdge("My edge", "default");
    Edge savedEdge = doPost("/api/edge", edge, Edge.class);
    Device device = new Device();
    device.setName("My device");
    device.setType("default");
    Device savedDevice = doPost("/api/device", device, Device.class);
    doPost("/api/edge/" + savedEdge.getId().getId().toString() + "/device/" + savedDevice.getId().getId().toString(), Device.class);
    PageData<Device> pageData = doGetTypedWithPageLink("/api/edge/" + savedEdge.getId().getId().toString() + "/devices?", new TypeReference<PageData<Device>>() {
    }, new PageLink(100));
    Assert.assertEquals(1, pageData.getData().size());
    doDelete("/api/edge/" + savedEdge.getId().getId().toString() + "/device/" + savedDevice.getId().getId().toString(), Device.class);
    pageData = doGetTypedWithPageLink("/api/edge/" + savedEdge.getId().getId().toString() + "/devices?", new TypeReference<PageData<Device>>() {
    }, new PageLink(100));
    Assert.assertEquals(0, pageData.getData().size());
}