Examples with ToUsageStatsServiceMsg org.thingsboard.server.gen.transport.TransportProtos.ToUsageStatsServiceMsg used on opensource projects

Example 1 with ToUsageStatsServiceMsg

use of org.thingsboard.server.gen.transport.TransportProtos.ToUsageStatsServiceMsg in project thingsboard by thingsboard.

the class DefaultTbCoreConsumerService method launchUsageStatsConsumer.

private void launchUsageStatsConsumer() {
    usageStatsExecutor.submit(() -> {
        while (!stopped) {
            try {
                List<TbProtoQueueMsg<ToUsageStatsServiceMsg>> msgs = usageStatsConsumer.poll(getNotificationPollDuration());
                if (msgs.isEmpty()) {
                    continue;
                }
                ConcurrentMap<UUID, TbProtoQueueMsg<ToUsageStatsServiceMsg>> pendingMap = msgs.stream().collect(Collectors.toConcurrentMap(s -> UUID.randomUUID(), Function.identity()));
                CountDownLatch processingTimeoutLatch = new CountDownLatch(1);
                TbPackProcessingContext<TbProtoQueueMsg<ToUsageStatsServiceMsg>> ctx = new TbPackProcessingContext<>(processingTimeoutLatch, pendingMap, new ConcurrentHashMap<>());
                pendingMap.forEach((id, msg) -> {
                    log.trace("[{}] Creating usage stats callback for message: {}", id, msg.getValue());
                    TbCallback callback = new TbPackCallback<>(id, ctx);
                    try {
                        handleUsageStats(msg, callback);
                    } catch (Throwable e) {
                        log.warn("[{}] Failed to process usage stats: {}", id, msg, e);
                        callback.onFailure(e);
                    }
                });
                if (!processingTimeoutLatch.await(getNotificationPackProcessingTimeout(), TimeUnit.MILLISECONDS)) {
                    ctx.getAckMap().forEach((id, msg) -> log.warn("[{}] Timeout to process usage stats: {}", id, msg.getValue()));
                    ctx.getFailedMap().forEach((id, msg) -> log.warn("[{}] Failed to process usage stats: {}", id, msg.getValue()));
                }
                usageStatsConsumer.commit();
            } catch (Exception e) {
                if (!stopped) {
                    log.warn("Failed to obtain usage stats from queue.", e);
                    try {
                        Thread.sleep(getNotificationPollDuration());
                    } catch (InterruptedException e2) {
                        log.trace("Failed to wait until the server has capacity to handle new usage stats", e2);
                    }
                }
            }
        }
        log.info("TB Usage Stats Consumer stopped.");
    });
}

Example 2 with ToUsageStatsServiceMsg

use of org.thingsboard.server.gen.transport.TransportProtos.ToUsageStatsServiceMsg in project thingsboard by thingsboard.

the class DefaultTbApiUsageStateService method process.

@Override
public void process(TbProtoQueueMsg<ToUsageStatsServiceMsg> msg, TbCallback callback) {
    ToUsageStatsServiceMsg statsMsg = msg.getValue();
    TenantId tenantId = TenantId.fromUUID(new UUID(statsMsg.getTenantIdMSB(), statsMsg.getTenantIdLSB()));
    EntityId entityId;
    if (statsMsg.getCustomerIdMSB() != 0 && statsMsg.getCustomerIdLSB() != 0) {
        entityId = new CustomerId(new UUID(statsMsg.getCustomerIdMSB(), statsMsg.getCustomerIdLSB()));
    } else {
        entityId = tenantId;
    }
    processEntityUsageStats(tenantId, entityId, statsMsg.getValuesList());
    callback.onSuccess();
}