Examples with TbRateLimitsException org.thingsboard.server.common.msg.tools.TbRateLimitsException used on opensource projects

Example 1 with TbRateLimitsException

use of org.thingsboard.server.common.msg.tools.TbRateLimitsException in project thingsboard by thingsboard.

the class RateLimitProcessingFilter method doFilter.

@Override
public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException {
    SecurityUser user = getCurrentUser();
    if (user != null && !user.isSystemAdmin()) {
        if (perTenantLimitsEnabled) {
            TbRateLimits rateLimits = perTenantLimits.computeIfAbsent(user.getTenantId(), id -> new TbRateLimits(perTenantLimitsConfiguration));
            if (!rateLimits.tryConsume()) {
                errorResponseHandler.handle(new TbRateLimitsException(EntityType.TENANT), (HttpServletResponse) response);
                return;
            }
        }
        if (perCustomerLimitsEnabled && user.isCustomerUser()) {
            TbRateLimits rateLimits = perCustomerLimits.computeIfAbsent(user.getCustomerId(), id -> new TbRateLimits(perCustomerLimitsConfiguration));
            if (!rateLimits.tryConsume()) {
                errorResponseHandler.handle(new TbRateLimitsException(EntityType.CUSTOMER), (HttpServletResponse) response);
                return;
            }
        }
    }
    chain.doFilter(request, response);
}

Example 2 with TbRateLimitsException

use of org.thingsboard.server.common.msg.tools.TbRateLimitsException in project thingsboard by thingsboard.

the class DefaultTransportService method checkLimits.

private boolean checkLimits(TransportProtos.SessionInfoProto sessionInfo, Object msg, TransportServiceCallback<?> callback, int dataPoints) {
    if (log.isTraceEnabled()) {
        log.trace("[{}] Processing msg: {}", toSessionId(sessionInfo), msg);
    }
    TenantId tenantId = TenantId.fromUUID(new UUID(sessionInfo.getTenantIdMSB(), sessionInfo.getTenantIdLSB()));
    DeviceId deviceId = new DeviceId(new UUID(sessionInfo.getDeviceIdMSB(), sessionInfo.getDeviceIdLSB()));
    EntityType rateLimitedEntityType = rateLimitService.checkLimits(tenantId, deviceId, dataPoints);
    if (rateLimitedEntityType == null) {
        return true;
    } else {
        if (callback != null) {
            callback.onError(new TbRateLimitsException(rateLimitedEntityType));
        }
        return false;
    }
}