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

Example 1 with ApiUsageRecordKey

use of org.thingsboard.server.common.data.ApiUsageRecordKey in project thingsboard by thingsboard.

the class DefaultTbApiUsageStateService method updateProfileThresholds.

private void updateProfileThresholds(TenantId tenantId, ApiUsageStateId id, TenantProfileConfiguration oldData, TenantProfileConfiguration newData) {
    long ts = System.currentTimeMillis();
    List<TsKvEntry> profileThresholds = new ArrayList<>();
    for (ApiUsageRecordKey key : ApiUsageRecordKey.values()) {
        long newProfileThreshold = newData.getProfileThreshold(key);
        if (oldData == null || oldData.getProfileThreshold(key) != newProfileThreshold) {
            log.info("[{}] Updating profile threshold [{}]:[{}]", tenantId, key, newProfileThreshold);
            profileThresholds.add(new BasicTsKvEntry(ts, new LongDataEntry(key.getApiLimitKey(), newProfileThreshold)));
        }
    }
    if (!profileThresholds.isEmpty()) {
        tsWsService.saveAndNotifyInternal(tenantId, id, profileThresholds, VOID_CALLBACK);
    }
}

Example 2 with ApiUsageRecordKey

use of org.thingsboard.server.common.data.ApiUsageRecordKey in project thingsboard by thingsboard.

the class TenantApiUsageState method checkStateUpdatedDueToThreshold.

private Pair<ApiFeature, ApiUsageStateValue> checkStateUpdatedDueToThreshold(ApiFeature feature) {
    ApiUsageStateValue featureValue = ApiUsageStateValue.ENABLED;
    for (ApiUsageRecordKey recordKey : ApiUsageRecordKey.getKeys(feature)) {
        long value = get(recordKey);
        long threshold = getProfileThreshold(recordKey);
        long warnThreshold = getProfileWarnThreshold(recordKey);
        ApiUsageStateValue tmpValue;
        if (threshold == 0 || value == 0 || value < warnThreshold) {
            tmpValue = ApiUsageStateValue.ENABLED;
        } else if (value < threshold) {
            tmpValue = ApiUsageStateValue.WARNING;
        } else {
            tmpValue = ApiUsageStateValue.DISABLED;
        }
        featureValue = ApiUsageStateValue.toMoreRestricted(featureValue, tmpValue);
    }
    return setFeatureValue(feature, featureValue) ? Pair.of(feature, featureValue) : null;
}

Example 3 with ApiUsageRecordKey

use of org.thingsboard.server.common.data.ApiUsageRecordKey in project thingsboard by thingsboard.

the class BaseApiUsageState method setCycles.

public void setCycles(long currentCycleTs, long nextCycleTs) {
    this.currentCycleTs = currentCycleTs;
    this.nextCycleTs = nextCycleTs;
    for (ApiUsageRecordKey key : ApiUsageRecordKey.values()) {
        currentCycleValues.put(key, 0L);
    }
}

Example 4 with ApiUsageRecordKey

use of org.thingsboard.server.common.data.ApiUsageRecordKey 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 5 with ApiUsageRecordKey

use of org.thingsboard.server.common.data.ApiUsageRecordKey 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);
    }
}