com.microsoft.azure.eventhubs.ManagedIdentityTokenProvider 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.

The newest version!

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

package com.microsoft.azure.eventhubs;

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

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

public class ManagedIdentityTokenProvider implements ITokenProvider {
    static final MSICredentials CREDENTIALS = new MSICredentials();
    
    @Override
    public CompletableFuture getToken(final String resource, final Duration timeout) {
        return CompletableFuture.supplyAsync(() -> {
            try {
                String rawToken = ManagedIdentityTokenProvider.CREDENTIALS.getToken(ClientConstants.EVENTHUBS_AUDIENCE);
                return new JsonSecurityToken(rawToken, resource);
            } catch (IOException | ParseException e) {
                throw new CompletionException(e);
            }
        });
    }
}