Examples with TopicPartitionInfo org.thingsboard.server.common.msg.queue.TopicPartitionInfo used on opensource projects

Example 26 with TopicPartitionInfo

use of org.thingsboard.server.common.msg.queue.TopicPartitionInfo in project thingsboard by thingsboard.

the class DefaultTbApiUsageStateService method getOrFetchState.

BaseApiUsageState getOrFetchState(TenantId tenantId, EntityId entityId) {
    if (entityId == null || entityId.isNullUid()) {
        entityId = tenantId;
    }
    BaseApiUsageState state = myUsageStates.get(entityId);
    if (state != null) {
        return state;
    }
    ApiUsageState storedState = apiUsageStateService.findApiUsageStateByEntityId(entityId);
    if (storedState == null) {
        try {
            storedState = apiUsageStateService.createDefaultApiUsageState(tenantId, entityId);
        } catch (Exception e) {
            storedState = apiUsageStateService.findApiUsageStateByEntityId(entityId);
        }
    }
    if (entityId.getEntityType() == EntityType.TENANT) {
        if (!entityId.equals(TenantId.SYS_TENANT_ID)) {
            state = new TenantApiUsageState(tenantProfileCache.get((TenantId) entityId), storedState);
        } else {
            state = new TenantApiUsageState(storedState);
        }
    } else {
        state = new CustomerApiUsageState(storedState);
    }
    List<ApiUsageRecordKey> newCounts = new ArrayList<>();
    try {
        List<TsKvEntry> dbValues = tsService.findAllLatest(tenantId, storedState.getId()).get();
        for (ApiUsageRecordKey key : ApiUsageRecordKey.values()) {
            boolean cycleEntryFound = false;
            boolean hourlyEntryFound = false;
            for (TsKvEntry tsKvEntry : dbValues) {
                if (tsKvEntry.getKey().equals(key.getApiCountKey())) {
                    cycleEntryFound = true;
                    boolean oldCount = tsKvEntry.getTs() == state.getCurrentCycleTs();
                    state.put(key, oldCount ? tsKvEntry.getLongValue().get() : 0L);
                    if (!oldCount) {
                        newCounts.add(key);
                    }
                } else if (tsKvEntry.getKey().equals(key.getApiCountKey() + HOURLY)) {
                    hourlyEntryFound = true;
                    state.putHourly(key, tsKvEntry.getTs() == state.getCurrentHourTs() ? tsKvEntry.getLongValue().get() : 0L);
                }
                if (cycleEntryFound && hourlyEntryFound) {
                    break;
                }
            }
        }
        log.debug("[{}] Initialized state: {}", entityId, storedState);
        TopicPartitionInfo tpi = partitionService.resolve(ServiceType.TB_CORE, tenantId, entityId);
        if (tpi.isMyPartition()) {
            addEntityState(tpi, state);
        } else {
            otherUsageStates.put(entityId, state.getApiUsageState());
        }
        saveNewCounts(state, newCounts);
    } catch (InterruptedException | ExecutionException e) {
        log.warn("[{}] Failed to fetch api usage state from db.", tenantId, e);
    }
    return state;
}

Example 27 with TopicPartitionInfo

use of org.thingsboard.server.common.msg.queue.TopicPartitionInfo in project thingsboard by thingsboard.

the class DefaultTbContext method enqueueForTellNext.

@Override
public void enqueueForTellNext(TbMsg tbMsg, String relationType) {
    TopicPartitionInfo tpi = resolvePartition(tbMsg);
    enqueueForTellNext(tpi, tbMsg, Collections.singleton(relationType), null, null, null);
}

Example 28 with TopicPartitionInfo

use of org.thingsboard.server.common.msg.queue.TopicPartitionInfo in project thingsboard by thingsboard.

the class DefaultTbContext method enqueue.

@Override
public void enqueue(TbMsg tbMsg, String queueName, Runnable onSuccess, Consumer<Throwable> onFailure) {
    TopicPartitionInfo tpi = resolvePartition(tbMsg, queueName);
    enqueue(tpi, tbMsg, onFailure, onSuccess);
}

Example 29 with TopicPartitionInfo

use of org.thingsboard.server.common.msg.queue.TopicPartitionInfo in project thingsboard by thingsboard.

the class DefaultTbContext method enqueueForTellNext.

@Override
public void enqueueForTellNext(TbMsg tbMsg, String queueName, String relationType, Runnable onSuccess, Consumer<Throwable> onFailure) {
    TopicPartitionInfo tpi = resolvePartition(tbMsg, queueName);
    enqueueForTellNext(tpi, queueName, tbMsg, Collections.singleton(relationType), null, onSuccess, onFailure);
}

Example 30 with TopicPartitionInfo

use of org.thingsboard.server.common.msg.queue.TopicPartitionInfo in project thingsboard by thingsboard.

the class DefaultTbContext method enqueueForTellNext.

@Override
public void enqueueForTellNext(TbMsg tbMsg, Set<String> relationTypes, Runnable onSuccess, Consumer<Throwable> onFailure) {
    TopicPartitionInfo tpi = resolvePartition(tbMsg);
    enqueueForTellNext(tpi, tbMsg, relationTypes, null, onSuccess, onFailure);
}