pl.allegro.tech.hermes.consumers.consumer.oauth.OAuthTokenRequestRateLimiterFactory Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of hermes-consumers Show documentation

Fast and reliable message broker built on top of Kafka.

There is a newer version: 2.10.0

package pl.allegro.tech.hermes.consumers.consumer.oauth;

import com.google.common.base.Preconditions;
import pl.allegro.tech.hermes.api.OAuthProvider;
import pl.allegro.tech.hermes.api.Subscription;
import pl.allegro.tech.hermes.common.config.ConfigFactory;
import pl.allegro.tech.hermes.domain.oauth.OAuthProviderRepository;

import javax.inject.Inject;

import static pl.allegro.tech.hermes.common.config.Configs.OAUTH_PROVIDERS_TOKEN_REQUEST_RATE_LIMITER_RATE_REDUCTION_FACTOR;

public class OAuthTokenRequestRateLimiterFactory {

    private final OAuthProviderRepository oAuthProviderRepository;

    private final double rateReductionFactor;

    @Inject
    public OAuthTokenRequestRateLimiterFactory(OAuthProviderRepository oAuthProviderRepository,
                                               ConfigFactory configFactory) {
        this.oAuthProviderRepository = oAuthProviderRepository;
        this.rateReductionFactor = configFactory.getDoubleProperty(OAUTH_PROVIDERS_TOKEN_REQUEST_RATE_LIMITER_RATE_REDUCTION_FACTOR);

        Preconditions.checkArgument(rateReductionFactor >= 1,
                "Token request rate limiter rate reduction factor must be greater or equal to 1");
    }

    public OAuthTokenRequestRateLimiter create(Subscription subscription) {
        String providerName = subscription.getOAuthPolicy().getProviderName();
        OAuthProvider oAuthProvider = oAuthProviderRepository.getOAuthProviderDetails(providerName);
        double initialRate = delayToRate(oAuthProvider.getTokenRequestInitialDelay());
        double minimalRate = delayToRate(oAuthProvider.getTokenRequestMaxDelay());
        return new OAuthTokenRequestRateLimiter(initialRate, minimalRate, rateReductionFactor,
                oAuthProvider.getTokenRequestInitialDelay());
    }

    private double delayToRate(Integer delay) {
        return 1000.0 / delay;
    }
}