com.azure.spring.cloud.service.implementation.appconfiguration.ConfigurationClientBuilderFactory Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of spring-cloud-azure-service Show documentation
Spring Cloud Azure Service
There is a newer version: 6.0.0-beta.4
Show newest version
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.

package com.azure.spring.cloud.service.implementation.appconfiguration;

import com.azure.core.credential.TokenCredential;
import com.azure.core.http.HttpClient;
import com.azure.core.http.HttpPipeline;
import com.azure.core.http.policy.HttpLogOptions;
import com.azure.core.http.policy.HttpPipelinePolicy;
import com.azure.core.http.policy.RetryPolicy;
import com.azure.core.util.ClientOptions;
import com.azure.core.util.Configuration;
import com.azure.data.appconfiguration.ConfigurationClientBuilder;
import com.azure.spring.cloud.core.implementation.credential.descriptor.AuthenticationDescriptor;
import com.azure.spring.cloud.core.implementation.credential.descriptor.TokenAuthenticationDescriptor;
import com.azure.spring.cloud.core.implementation.factory.AbstractAzureHttpClientBuilderFactory;
import com.azure.spring.cloud.core.implementation.properties.PropertyMapper;
import com.azure.spring.cloud.core.properties.AzureProperties;

import java.util.Arrays;
import java.util.List;
import java.util.function.BiConsumer;

/**
 * Azure App Configuration client builder factory, it builds the {@link ConfigurationClientBuilder}.
 */
public class ConfigurationClientBuilderFactory extends AbstractAzureHttpClientBuilderFactory {

    private final ConfigurationClientProperties configurationClientProperties;


    /**
     * Create a {@link ConfigurationClientBuilderFactory} instance with a {@link ConfigurationClientProperties}.
     * @param configurationClientProperties the properties for the configuration client.
     */
    public ConfigurationClientBuilderFactory(ConfigurationClientProperties configurationClientProperties) {
        this.configurationClientProperties = configurationClientProperties;
    }

    @Override
    protected BiConsumer consumeClientOptions() {
        return ConfigurationClientBuilder::clientOptions;
    }

    @Override
    protected BiConsumer consumeHttpClient() {
        return ConfigurationClientBuilder::httpClient;
    }

    @Override
    protected BiConsumer consumeHttpPipelinePolicy() {
        return ConfigurationClientBuilder::addPolicy;
    }

    @Override
    protected BiConsumer consumeHttpPipeline() {
        return ConfigurationClientBuilder::pipeline;
    }

    @Override
    protected BiConsumer consumeHttpLogOptions() {
        return ConfigurationClientBuilder::httpLogOptions;
    }

    @Override
    protected ConfigurationClientBuilder createBuilderInstance() {
        return new ConfigurationClientBuilder();
    }

    @Override
    protected AzureProperties getAzureProperties() {
        return this.configurationClientProperties;
    }

    @Override
    protected List> getAuthenticationDescriptors(ConfigurationClientBuilder builder) {
        return Arrays.asList(
            new TokenAuthenticationDescriptor(this.tokenCredentialResolver, builder::credential)
        );
    }

    @Override
    protected void configureService(ConfigurationClientBuilder builder) {
        PropertyMapper map = new PropertyMapper();
        map.from(configurationClientProperties.getEndpoint()).to(builder::endpoint);
        map.from(configurationClientProperties.getServiceVersion()).to(builder::serviceVersion);
    }

    @Override
    protected BiConsumer consumeConfiguration() {
        return ConfigurationClientBuilder::configuration;
    }

    @Override
    protected BiConsumer consumeDefaultTokenCredential() {
        return ConfigurationClientBuilder::credential;
    }

    @Override
    protected BiConsumer consumeConnectionString() {
        return ConfigurationClientBuilder::connectionString;
    }

    @Override
    protected BiConsumer consumeRetryPolicy() {
        return ConfigurationClientBuilder::retryPolicy;
    }
}