pl.allegro.tech.hermes.consumers.consumer.oauth.OAuthSubscriptionHandlerFactory 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 org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import pl.allegro.tech.hermes.api.Subscription;
import pl.allegro.tech.hermes.api.SubscriptionName;
import pl.allegro.tech.hermes.domain.subscription.SubscriptionRepository;

import java.util.Optional;

public class OAuthSubscriptionHandlerFactory {

    private static final Logger logger = LoggerFactory.getLogger(OAuthSubscriptionHandlerFactory.class);

    private final SubscriptionRepository subscriptionRepository;

    private final OAuthAccessTokens accessTokens;

    private final OAuthTokenRequestRateLimiterFactory rateLimiterLoader;

    public OAuthSubscriptionHandlerFactory(SubscriptionRepository subscriptionRepository,
                                           OAuthAccessTokens accessTokens,
                                           OAuthTokenRequestRateLimiterFactory rateLimiterLoader) {
        this.subscriptionRepository = subscriptionRepository;
        this.accessTokens = accessTokens;
        this.rateLimiterLoader = rateLimiterLoader;
    }

    public Optional create(SubscriptionName subscriptionName) {
        Subscription subscription = subscriptionRepository.getSubscriptionDetails(subscriptionName);
        if (subscription.hasOAuthPolicy()) {
            try {
                String providerName = subscription.getOAuthPolicy().getProviderName();
                logger.info("Creating OAuth handler subscription {} using {} OAuth provider",
                        subscriptionName, providerName);
                OAuthTokenRequestRateLimiter rateLimiter = rateLimiterLoader.create(subscription);
                return Optional.of(new OAuthSubscriptionHandler(subscriptionName, providerName, accessTokens, rateLimiter));
            } catch (Exception e) {
                logger.error("Failed to create OAuth handler for subscription {}, {}",
                        subscriptionName.getQualifiedName(), e.getMessage());
            }
        }
        return Optional.empty();
    }
}