com.microsoft.azure.kusto.data.auth.CloudDependentTokenProviderBase Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kusto-data Show documentation

Kusto client library for executing queries and retrieving data

There is a newer version: 5.2.0

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

package com.microsoft.azure.kusto.data.auth;

import com.azure.core.http.HttpClient;
import com.microsoft.azure.kusto.data.instrumentation.SupplierTwoExceptions;
import com.microsoft.azure.kusto.data.exceptions.DataClientException;
import com.microsoft.azure.kusto.data.exceptions.DataServiceException;
import java.net.URISyntaxException;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

import com.microsoft.azure.kusto.data.instrumentation.MonitoredActivity;

import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import reactor.core.publisher.Mono;

public abstract class CloudDependentTokenProviderBase extends TokenProviderBase {
    private static final String ERROR_INVALID_SERVICE_RESOURCE_URL = "Error determining scope due to invalid Kusto Service Resource URL";
    protected final Set scopes = new HashSet<>();
    private boolean initialized = false;
    private CloudInfo cloudInfo;

    CloudDependentTokenProviderBase(@NotNull String clusterUrl, @Nullable HttpClient httpClient) throws URISyntaxException {
        super(clusterUrl, httpClient);
    }

    @Override
    final Mono initialize() {
        return Mono.fromCallable(() -> initialized)
                .flatMap(initialized -> {
                    if (initialized) {
                        return Mono.empty();
                    }
                    // trace retrieveCloudInfo
                    return MonitoredActivity.wrap(
                            CloudInfo.retrieveCloudInfoForClusterAsync(clusterUrl,
                                    httpClient),
                            "CloudDependentTokenProviderBase.retrieveCloudInfo", getTracingAttributes())
                            .flatMap(cloudInfo -> {
                                this.cloudInfo = cloudInfo;

                                try {
                                    initializeWithCloudInfo(cloudInfo);
                                } catch (DataClientException | DataServiceException e) {
                                    return Mono.error(e);
                                }
                                this.initialized = true;
                                return Mono.empty();
                            });
                });
    }

    protected void initializeWithCloudInfo(CloudInfo cloudInfo) throws DataClientException, DataServiceException {
        try {
            scopes.add(cloudInfo.determineScope());
        } catch (URISyntaxException e) {
            throw new DataServiceException(clusterUrl, ERROR_INVALID_SERVICE_RESOURCE_URL, e, true);
        }
    }

    @Override
    public Map getTracingAttributes() {
        Map attributes = super.getTracingAttributes();
        if (cloudInfo != null) {
            attributes.putAll(cloudInfo.getTracingAttributes());
        }
        attributes.put("http.url", clusterUrl);
        return attributes;
    }
}