Examples with LongDataEntry org.thingsboard.server.common.data.kv.LongDataEntry used on opensource projects

Example 11 with LongDataEntry

use of org.thingsboard.server.common.data.kv.LongDataEntry in project thingsboard by thingsboard.

the class DefaultTbApiUsageStateService method saveNewCounts.

private void saveNewCounts(BaseApiUsageState state, List<ApiUsageRecordKey> keys) {
    List<TsKvEntry> counts = keys.stream().map(key -> new BasicTsKvEntry(state.getCurrentCycleTs(), new LongDataEntry(key.getApiCountKey(), 0L))).collect(Collectors.toList());
    tsWsService.saveAndNotifyInternal(state.getTenantId(), state.getApiUsageState().getId(), counts, VOID_CALLBACK);
}

Example 12 with LongDataEntry

use of org.thingsboard.server.common.data.kv.LongDataEntry in project thingsboard by thingsboard.

the class DefaultTbApiUsageStateService method processEntityUsageStats.

private void processEntityUsageStats(TenantId tenantId, EntityId entityId, List<UsageStatsKVProto> values) {
    if (deletedEntities.contains(entityId))
        return;
    BaseApiUsageState usageState;
    List<TsKvEntry> updatedEntries;
    Map<ApiFeature, ApiUsageStateValue> result;
    updateLock.lock();
    try {
        usageState = getOrFetchState(tenantId, entityId);
        long ts = usageState.getCurrentCycleTs();
        long hourTs = usageState.getCurrentHourTs();
        long newHourTs = SchedulerUtils.getStartOfCurrentHour();
        if (newHourTs != hourTs) {
            usageState.setHour(newHourTs);
        }
        updatedEntries = new ArrayList<>(ApiUsageRecordKey.values().length);
        Set<ApiFeature> apiFeatures = new HashSet<>();
        for (UsageStatsKVProto kvProto : values) {
            ApiUsageRecordKey recordKey = ApiUsageRecordKey.valueOf(kvProto.getKey());
            long newValue = usageState.add(recordKey, kvProto.getValue());
            updatedEntries.add(new BasicTsKvEntry(ts, new LongDataEntry(recordKey.getApiCountKey(), newValue)));
            long newHourlyValue = usageState.addToHourly(recordKey, kvProto.getValue());
            updatedEntries.add(new BasicTsKvEntry(newHourTs, new LongDataEntry(recordKey.getApiCountKey() + HOURLY, newHourlyValue)));
            apiFeatures.add(recordKey.getApiFeature());
        }
        if (usageState.getEntityType() == EntityType.TENANT && !usageState.getEntityId().equals(TenantId.SYS_TENANT_ID)) {
            result = ((TenantApiUsageState) usageState).checkStateUpdatedDueToThreshold(apiFeatures);
        } else {
            result = Collections.emptyMap();
        }
    } finally {
        updateLock.unlock();
    }
    tsWsService.saveAndNotifyInternal(tenantId, usageState.getApiUsageState().getId(), updatedEntries, VOID_CALLBACK);
    if (!result.isEmpty()) {
        persistAndNotify(usageState, result);
    }
}

Example 13 with LongDataEntry

use of org.thingsboard.server.common.data.kv.LongDataEntry in project thingsboard by thingsboard.

the class DefaultOtaPackageStateService method updateAttributes.

private void updateAttributes(Device device, OtaPackageInfo otaPackage, long ts, TenantId tenantId, DeviceId deviceId, OtaPackageType otaPackageType) {
    List<AttributeKvEntry> attributes = new ArrayList<>();
    List<String> attrToRemove = new ArrayList<>();
    attributes.add(new BaseAttributeKvEntry(ts, new StringDataEntry(getAttributeKey(otaPackageType, TITLE), otaPackage.getTitle())));
    attributes.add(new BaseAttributeKvEntry(ts, new StringDataEntry(getAttributeKey(otaPackageType, VERSION), otaPackage.getVersion())));
    if (StringUtils.isNotEmpty(otaPackage.getTag())) {
        attributes.add(new BaseAttributeKvEntry(ts, new StringDataEntry(getAttributeKey(otaPackageType, TAG), otaPackage.getTag())));
    } else {
        attrToRemove.add(getAttributeKey(otaPackageType, TAG));
    }
    if (otaPackage.hasUrl()) {
        attributes.add(new BaseAttributeKvEntry(ts, new StringDataEntry(getAttributeKey(otaPackageType, URL), otaPackage.getUrl())));
        if (otaPackage.getDataSize() == null) {
            attrToRemove.add(getAttributeKey(otaPackageType, SIZE));
        } else {
            attributes.add(new BaseAttributeKvEntry(ts, new LongDataEntry(getAttributeKey(otaPackageType, SIZE), otaPackage.getDataSize())));
        }
        if (otaPackage.getChecksumAlgorithm() != null) {
            attrToRemove.add(getAttributeKey(otaPackageType, CHECKSUM_ALGORITHM));
        } else {
            attributes.add(new BaseAttributeKvEntry(ts, new StringDataEntry(getAttributeKey(otaPackageType, CHECKSUM_ALGORITHM), otaPackage.getChecksumAlgorithm().name())));
        }
        if (StringUtils.isEmpty(otaPackage.getChecksum())) {
            attrToRemove.add(getAttributeKey(otaPackageType, CHECKSUM));
        } else {
            attributes.add(new BaseAttributeKvEntry(ts, new StringDataEntry(getAttributeKey(otaPackageType, CHECKSUM), otaPackage.getChecksum())));
        }
    } else {
        attributes.add(new BaseAttributeKvEntry(ts, new LongDataEntry(getAttributeKey(otaPackageType, SIZE), otaPackage.getDataSize())));
        attributes.add(new BaseAttributeKvEntry(ts, new StringDataEntry(getAttributeKey(otaPackageType, CHECKSUM_ALGORITHM), otaPackage.getChecksumAlgorithm().name())));
        attributes.add(new BaseAttributeKvEntry(ts, new StringDataEntry(getAttributeKey(otaPackageType, CHECKSUM), otaPackage.getChecksum())));
        attrToRemove.add(getAttributeKey(otaPackageType, URL));
    }
    remove(device, otaPackageType, attrToRemove);
    telemetryService.saveAndNotify(tenantId, deviceId, DataConstants.SHARED_SCOPE, attributes, new FutureCallback<>() {

        @Override
        public void onSuccess(@Nullable Void tmp) {
            log.trace("[{}] Success save attributes with target firmware!", deviceId);
        }

        @Override
        public void onFailure(Throwable t) {
            log.error("[{}] Failed to save attributes with target firmware!", deviceId, t);
        }
    });
}

Example 14 with LongDataEntry

use of org.thingsboard.server.common.data.kv.LongDataEntry in project thingsboard by thingsboard.

the class TbSubscriptionUtils method getKvEntry.

private static KvEntry getKvEntry(KeyValueProto proto) {
    KvEntry entry = null;
    DataType type = DataType.values()[proto.getType().getNumber()];
    switch(type) {
        case BOOLEAN:
            entry = new BooleanDataEntry(proto.getKey(), proto.getBoolV());
            break;
        case LONG:
            entry = new LongDataEntry(proto.getKey(), proto.getLongV());
            break;
        case DOUBLE:
            entry = new DoubleDataEntry(proto.getKey(), proto.getDoubleV());
            break;
        case STRING:
            entry = new StringDataEntry(proto.getKey(), proto.getStringV());
            break;
        case JSON:
            entry = new JsonDataEntry(proto.getKey(), proto.getJsonV());
            break;
    }
    return entry;
}

Example 15 with LongDataEntry

use of org.thingsboard.server.common.data.kv.LongDataEntry in project thingsboard by thingsboard.

the class DefaultRuleEngineStatisticsService method reportQueueStats.

@Override
public void reportQueueStats(long ts, TbRuleEngineConsumerStats ruleEngineStats) {
    String queueName = ruleEngineStats.getQueueName();
    ruleEngineStats.getTenantStats().forEach((id, stats) -> {
        TenantId tenantId = TenantId.fromUUID(id);
        try {
            AssetId serviceAssetId = getServiceAssetId(tenantId, queueName);
            if (stats.getTotalMsgCounter().get() > 0) {
                List<TsKvEntry> tsList = stats.getCounters().entrySet().stream().map(kv -> new BasicTsKvEntry(ts, new LongDataEntry(kv.getKey(), (long) kv.getValue().get()))).collect(Collectors.toList());
                if (!tsList.isEmpty()) {
                    tsService.saveAndNotifyInternal(tenantId, serviceAssetId, tsList, CALLBACK);
                }
            }
        } catch (DataValidationException e) {
            if (!e.getMessage().equalsIgnoreCase("Asset is referencing to non-existent tenant!")) {
                throw e;
            }
        }
    });
    ruleEngineStats.getTenantExceptions().forEach((tenantId, e) -> {
        TsKvEntry tsKv = new BasicTsKvEntry(e.getTs(), new JsonDataEntry("ruleEngineException", e.toJsonString()));
        try {
            tsService.saveAndNotifyInternal(tenantId, getServiceAssetId(tenantId, queueName), Collections.singletonList(tsKv), CALLBACK);
        } catch (DataValidationException e2) {
            if (!e2.getMessage().equalsIgnoreCase("Asset is referencing to non-existent tenant!")) {
                throw e2;
            }
        }
    });
}