Skip to content

Commit

Permalink
CodeGen from PR 3168 in test-repo-billy/azure-rest-api-specs
Browse files Browse the repository at this point in the history 
Merge bd17a55da84491db851285461c49917f63d53fc2 into 26085c41a71f253272b512056461ddfcab0d9778
  • Loading branch information
SDKAuto committed Dec 26, 2023
1 parent 242c183 commit ac081ae
Show file tree
Hide file tree
Showing 64 changed files with 14,715 additions and 1,331 deletions.
894 changes: 646 additions & 248 deletions ...ting/src/main/java/com/azure/developer/loadtesting/LoadTestAdministrationAsyncClient.java
View file

Large diffs are not rendered by default.

780 changes: 564 additions & 216 deletions ...adtesting/src/main/java/com/azure/developer/loadtesting/LoadTestAdministrationClient.java
View file

Large diffs are not rendered by default.

189 changes: 105 additions & 84 deletions ...ng/src/main/java/com/azure/developer/loadtesting/LoadTestAdministrationClientBuilder.java
View file
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
// Code generated by Microsoft (R) AutoRest Code Generator.
// Code generated by Microsoft (R) TypeSpec Code Generator.
package com.azure.developer.loadtesting;

import com.azure.core.annotation.Generated;
Expand All @@ -11,6 +11,7 @@
import com.azure.core.client.traits.TokenCredentialTrait;
import com.azure.core.credential.TokenCredential;
import com.azure.core.http.HttpClient;
import com.azure.core.http.HttpHeaderName;
import com.azure.core.http.HttpHeaders;
import com.azure.core.http.HttpPipeline;
import com.azure.core.http.HttpPipelineBuilder;
Expand All @@ -19,7 +20,6 @@
import com.azure.core.http.policy.AddHeadersFromContextPolicy;
import com.azure.core.http.policy.AddHeadersPolicy;
import com.azure.core.http.policy.BearerTokenAuthenticationPolicy;
import com.azure.core.http.policy.CookiePolicy;
import com.azure.core.http.policy.HttpLogOptions;
import com.azure.core.http.policy.HttpLoggingPolicy;
import com.azure.core.http.policy.HttpPipelinePolicy;
Expand All @@ -32,34 +32,37 @@
import com.azure.core.util.Configuration;
import com.azure.core.util.CoreUtils;
import com.azure.core.util.builder.ClientBuilderUtil;
import com.azure.core.util.logging.ClientLogger;
import com.azure.core.util.serializer.JacksonAdapter;
import com.azure.developer.loadtesting.implementation.LoadTestingClientImpl;
import com.azure.developer.loadtesting.implementation.LoadTestAdministrationClientImpl;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
import java.util.Objects;

/** A builder for creating a new instance of the LoadTestAdministrationClient type. */
@ServiceClientBuilder(serviceClients = {LoadTestAdministrationClient.class, LoadTestAdministrationAsyncClient.class})
/**
* A builder for creating a new instance of the LoadTestAdministrationClient type.
*/
@ServiceClientBuilder(serviceClients = { LoadTestAdministrationClient.class, LoadTestAdministrationAsyncClient.class })
public final class LoadTestAdministrationClientBuilder
implements HttpTrait<LoadTestAdministrationClientBuilder>,
ConfigurationTrait<LoadTestAdministrationClientBuilder>,
TokenCredentialTrait<LoadTestAdministrationClientBuilder>,
EndpointTrait<LoadTestAdministrationClientBuilder> {
implements HttpTrait<LoadTestAdministrationClientBuilder>, ConfigurationTrait<LoadTestAdministrationClientBuilder>,
TokenCredentialTrait<LoadTestAdministrationClientBuilder>, EndpointTrait<LoadTestAdministrationClientBuilder> {

@Generated private static final String SDK_NAME = "name";

@Generated private static final String SDK_VERSION = "version";
@Generated
private static final String SDK_NAME = "name";

@Generated
private static final String[] DEFAULT_SCOPES = new String[] {"https://cnt-prod.loadtesting.azure.com/.default"};
private static final String SDK_VERSION = "version";

@Generated
private final Map<String, String> properties = CoreUtils.getProperties("azure-developer-loadtesting.properties");
private static final String[] DEFAULT_SCOPES = new String[] { "https://cnt-prod.loadtesting.azure.com/.default" };

@Generated private final List<HttpPipelinePolicy> pipelinePolicies;
@Generated
private final List<HttpPipelinePolicy> pipelinePolicies;

/** Create an instance of the LoadTestAdministrationClientBuilder. */
/**
* Create an instance of the LoadTestAdministrationClientBuilder.
*/
@Generated
public LoadTestAdministrationClientBuilder() {
this.pipelinePolicies = new ArrayList<>();
Expand All @@ -68,22 +71,31 @@ public LoadTestAdministrationClientBuilder() {
/*
* The HTTP pipeline to send requests through.
*/
@Generated private HttpPipeline pipeline;
@Generated
private HttpPipeline pipeline;

/** {@inheritDoc}. */
/**
* {@inheritDoc}.
*/
@Generated
@Override
public LoadTestAdministrationClientBuilder pipeline(HttpPipeline pipeline) {
if (this.pipeline != null && pipeline == null) {
LOGGER.info("HttpPipeline is being set to 'null' when it was previously configured.");
}
this.pipeline = pipeline;
return this;
}

/*
* The HTTP client used to send the request.
*/
@Generated private HttpClient httpClient;
@Generated
private HttpClient httpClient;

/** {@inheritDoc}. */
/**
* {@inheritDoc}.
*/
@Generated
@Override
public LoadTestAdministrationClientBuilder httpClient(HttpClient httpClient) {
Expand All @@ -94,9 +106,12 @@ public LoadTestAdministrationClientBuilder httpClient(HttpClient httpClient) {
/*
* The logging configuration for HTTP requests and responses.
*/
@Generated private HttpLogOptions httpLogOptions;
@Generated
private HttpLogOptions httpLogOptions;

/** {@inheritDoc}. */
/**
* {@inheritDoc}.
*/
@Generated
@Override
public LoadTestAdministrationClientBuilder httpLogOptions(HttpLogOptions httpLogOptions) {
Expand All @@ -105,12 +120,14 @@ public LoadTestAdministrationClientBuilder httpLogOptions(HttpLogOptions httpLog
}

/*
* The client options such as application ID and custom headers to set on a
* request.
* The client options such as application ID and custom headers to set on a request.
*/
@Generated private ClientOptions clientOptions;
@Generated
private ClientOptions clientOptions;

/** {@inheritDoc}. */
/**
* {@inheritDoc}.
*/
@Generated
@Override
public LoadTestAdministrationClientBuilder clientOptions(ClientOptions clientOptions) {
Expand All @@ -121,31 +138,39 @@ public LoadTestAdministrationClientBuilder clientOptions(ClientOptions clientOpt
/*
* The retry options to configure retry policy for failed requests.
*/
@Generated private RetryOptions retryOptions;
@Generated
private RetryOptions retryOptions;

/** {@inheritDoc}. */
/**
* {@inheritDoc}.
*/
@Generated
@Override
public LoadTestAdministrationClientBuilder retryOptions(RetryOptions retryOptions) {
this.retryOptions = retryOptions;
return this;
}

/** {@inheritDoc}. */
/**
* {@inheritDoc}.
*/
@Generated
@Override
public LoadTestAdministrationClientBuilder addPolicy(HttpPipelinePolicy customPolicy) {
Objects.requireNonNull(customPolicy, "'customPolicy' cannot be null.");
pipelinePolicies.add(customPolicy);
return this;
}

/*
* The configuration store that is used during construction of the service
* client.
* The configuration store that is used during construction of the service client.
*/
@Generated private Configuration configuration;
@Generated
private Configuration configuration;

/** {@inheritDoc}. */
/**
* {@inheritDoc}.
*/
@Generated
@Override
public LoadTestAdministrationClientBuilder configuration(Configuration configuration) {
Expand All @@ -156,9 +181,12 @@ public LoadTestAdministrationClientBuilder configuration(Configuration configura
/*
* The TokenCredential used for authentication.
*/
@Generated private TokenCredential tokenCredential;
@Generated
private TokenCredential tokenCredential;

/** {@inheritDoc}. */
/**
* {@inheritDoc}.
*/
@Generated
@Override
public LoadTestAdministrationClientBuilder credential(TokenCredential tokenCredential) {
Expand All @@ -169,9 +197,12 @@ public LoadTestAdministrationClientBuilder credential(TokenCredential tokenCrede
/*
* The service endpoint
*/
@Generated private String endpoint;
@Generated
private String endpoint;

/** {@inheritDoc}. */
/**
* {@inheritDoc}.
*/
@Generated
@Override
public LoadTestAdministrationClientBuilder endpoint(String endpoint) {
Expand All @@ -182,7 +213,8 @@ public LoadTestAdministrationClientBuilder endpoint(String endpoint) {
/*
* Service version
*/
@Generated private LoadTestingServiceVersion serviceVersion;
@Generated
private LoadTestingServiceVersion serviceVersion;

/**
* Sets Service version.
Expand All @@ -197,10 +229,10 @@ public LoadTestAdministrationClientBuilder serviceVersion(LoadTestingServiceVers
}

/*
* The retry policy that will attempt to retry failed requests, if
* applicable.
* The retry policy that will attempt to retry failed requests, if applicable.
*/
@Generated private RetryPolicy retryPolicy;
@Generated
private RetryPolicy retryPolicy;

/**
* Sets The retry policy that will attempt to retry failed requests, if applicable.
Expand All @@ -215,69 +247,53 @@ public LoadTestAdministrationClientBuilder retryPolicy(RetryPolicy retryPolicy)
}

/**
* Builds an instance of LoadTestingClientImpl with the provided parameters.
* Builds an instance of LoadTestAdministrationClientImpl with the provided parameters.
*
* @return an instance of LoadTestingClientImpl.
* @return an instance of LoadTestAdministrationClientImpl.
*/
@Generated
private LoadTestingClientImpl buildInnerClient() {
if (pipeline == null) {
this.pipeline = createHttpPipeline();
}
if (serviceVersion == null) {
this.serviceVersion = LoadTestingServiceVersion.getLatest();
}
LoadTestingClientImpl client =
new LoadTestingClientImpl(
pipeline, JacksonAdapter.createDefaultSerializerAdapter(), endpoint, serviceVersion);
private LoadTestAdministrationClientImpl buildInnerClient() {
HttpPipeline localPipeline = (pipeline != null) ? pipeline : createHttpPipeline();
LoadTestingServiceVersion localServiceVersion
= (serviceVersion != null) ? serviceVersion : LoadTestingServiceVersion.getLatest();
LoadTestAdministrationClientImpl client = new LoadTestAdministrationClientImpl(localPipeline,
JacksonAdapter.createDefaultSerializerAdapter(), this.endpoint, localServiceVersion);
return client;
}

@Generated
private HttpPipeline createHttpPipeline() {
Configuration buildConfiguration =
(configuration == null) ? Configuration.getGlobalConfiguration() : configuration;
if (httpLogOptions == null) {
httpLogOptions = new HttpLogOptions();
}
if (clientOptions == null) {
clientOptions = new ClientOptions();
}
Configuration buildConfiguration
= (configuration == null) ? Configuration.getGlobalConfiguration() : configuration;
HttpLogOptions localHttpLogOptions = this.httpLogOptions == null ? new HttpLogOptions() : this.httpLogOptions;
ClientOptions localClientOptions = this.clientOptions == null ? new ClientOptions() : this.clientOptions;
List<HttpPipelinePolicy> policies = new ArrayList<>();
String clientName = properties.getOrDefault(SDK_NAME, "UnknownName");
String clientVersion = properties.getOrDefault(SDK_VERSION, "UnknownVersion");
String applicationId = CoreUtils.getApplicationId(clientOptions, httpLogOptions);
String clientName = PROPERTIES.getOrDefault(SDK_NAME, "UnknownName");
String clientVersion = PROPERTIES.getOrDefault(SDK_VERSION, "UnknownVersion");
String applicationId = CoreUtils.getApplicationId(localClientOptions, localHttpLogOptions);
policies.add(new UserAgentPolicy(applicationId, clientName, clientVersion, buildConfiguration));
policies.add(new RequestIdPolicy());
policies.add(new AddHeadersFromContextPolicy());
HttpHeaders headers = new HttpHeaders();
clientOptions.getHeaders().forEach(header -> headers.set(header.getName(), header.getValue()));
localClientOptions.getHeaders()
.forEach(header -> headers.set(HttpHeaderName.fromString(header.getName()), header.getValue()));
if (headers.getSize() > 0) {
policies.add(new AddHeadersPolicy(headers));
}
policies.addAll(
this.pipelinePolicies.stream()
.filter(p -> p.getPipelinePosition() == HttpPipelinePosition.PER_CALL)
.collect(Collectors.toList()));
this.pipelinePolicies.stream().filter(p -> p.getPipelinePosition() == HttpPipelinePosition.PER_CALL)
.forEach(p -> policies.add(p));
HttpPolicyProviders.addBeforeRetryPolicies(policies);
policies.add(ClientBuilderUtil.validateAndGetRetryPolicy(retryPolicy, retryOptions, new RetryPolicy()));
policies.add(new AddDatePolicy());
policies.add(new CookiePolicy());
if (tokenCredential != null) {
policies.add(new BearerTokenAuthenticationPolicy(tokenCredential, DEFAULT_SCOPES));
}
policies.addAll(
this.pipelinePolicies.stream()
.filter(p -> p.getPipelinePosition() == HttpPipelinePosition.PER_RETRY)
.collect(Collectors.toList()));
this.pipelinePolicies.stream().filter(p -> p.getPipelinePosition() == HttpPipelinePosition.PER_RETRY)
.forEach(p -> policies.add(p));
HttpPolicyProviders.addAfterRetryPolicies(policies);
policies.add(new HttpLoggingPolicy(httpLogOptions));
HttpPipeline httpPipeline =
new HttpPipelineBuilder()
.policies(policies.toArray(new HttpPipelinePolicy[0]))
.httpClient(httpClient)
.clientOptions(clientOptions)
.build();
HttpPipeline httpPipeline = new HttpPipelineBuilder().policies(policies.toArray(new HttpPipelinePolicy[0]))
.httpClient(httpClient).clientOptions(localClientOptions).build();
return httpPipeline;
}

Expand All @@ -288,7 +304,7 @@ private HttpPipeline createHttpPipeline() {
*/
@Generated
public LoadTestAdministrationAsyncClient buildAsyncClient() {
return new LoadTestAdministrationAsyncClient(buildInnerClient().getLoadTestAdministrations());
return new LoadTestAdministrationAsyncClient(buildInnerClient());
}

/**
Expand All @@ -298,7 +314,12 @@ public LoadTestAdministrationAsyncClient buildAsyncClient() {
*/
@Generated
public LoadTestAdministrationClient buildClient() {
return new LoadTestAdministrationClient(
new LoadTestAdministrationAsyncClient(buildInnerClient().getLoadTestAdministrations()));
return new LoadTestAdministrationClient(new LoadTestAdministrationAsyncClient(buildInnerClient()));
}

@Generated
private static final Map<String, String> PROPERTIES
= CoreUtils.getProperties("azure-developer-loadtesting.properties");

private static final ClientLogger LOGGER = new ClientLogger(LoadTestAdministrationClientBuilder.class);
}
Loading

0 comments on commit ac081ae

Please sign in to comment.