com.microsoft.azure.eventhubs.AzureActiveDirectoryTokenProvider Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of azure-eventhubs Show documentation

Please note, a newer package azure-messaging-eventhubs for Azure Event Hubs is available at https://search.maven.org/artifact/com.azure/azure-messaging-eventhubs as of February 2020. While this package will continue to receive critical bug fixes, we strongly encourage you to upgrade. Read the migration guide at https://aka.ms/azsdk/java/migrate/eh for more details.

There is a newer version: 3.3.0

Show newest version

// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.

package com.microsoft.azure.eventhubs;

import java.text.ParseException;
import java.time.Duration;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.CompletionException;

import com.microsoft.azure.eventhubs.impl.ClientConstants;

public final class AzureActiveDirectoryTokenProvider implements ITokenProvider {
    private final AuthenticationCallback authCallback;
    private final String authority;
    private final Object authCallbackState;

    public AzureActiveDirectoryTokenProvider(
            final AuthenticationCallback authenticationCallback,
            final String authority,
            final Object state) {
        this.authCallbackState = state;
        this.authority = authority;
        this.authCallback = authenticationCallback;
    }

    @Override
    public CompletableFuture getToken(String resource, Duration timeout) {
        return this.authCallback.acquireToken(ClientConstants.EVENTHUBS_AUDIENCE, this.authority, this.authCallbackState)
                .thenApply((rawToken) -> {
                    try {
                        return new JsonSecurityToken(rawToken, resource);
                    } catch (ParseException e) {
                        throw new CompletionException(e);
                    }
                });
    }

    public interface AuthenticationCallback {
        CompletableFuture acquireToken(String audience, String authority, Object state);
    }
}