in spring-cloud-previews/google-cloud-aiplatform-spring-starter/src/main/java/com/google/cloud/aiplatform/v1/spring/PipelineServiceSpringAutoConfiguration.java [119:404]
public PipelineServiceSettings pipelineServiceSettings(
@Qualifier("defaultPipelineServiceTransportChannelProvider")
TransportChannelProvider defaultTransportChannelProvider)
throws IOException {
PipelineServiceSettings.Builder clientSettingsBuilder = PipelineServiceSettings.newBuilder();
clientSettingsBuilder
.setCredentialsProvider(this.credentialsProvider)
.setTransportChannelProvider(defaultTransportChannelProvider)
.setEndpoint(PipelineServiceSettings.getDefaultEndpoint())
.setHeaderProvider(this.userAgentHeaderProvider());
if (this.clientProperties.getQuotaProjectId() != null) {
clientSettingsBuilder.setQuotaProjectId(this.clientProperties.getQuotaProjectId());
if (LOGGER.isTraceEnabled()) {
LOGGER.trace(
"Quota project id set to "
+ this.clientProperties.getQuotaProjectId()
+ ", this overrides project id from credentials.");
}
}
if (this.clientProperties.getExecutorThreadCount() != null) {
ExecutorProvider executorProvider =
PipelineServiceSettings.defaultExecutorProviderBuilder()
.setExecutorThreadCount(this.clientProperties.getExecutorThreadCount())
.build();
clientSettingsBuilder.setBackgroundExecutorProvider(executorProvider);
if (LOGGER.isTraceEnabled()) {
LOGGER.trace(
"Background executor thread count is "
+ this.clientProperties.getExecutorThreadCount());
}
}
Retry serviceRetry = clientProperties.getRetry();
if (serviceRetry != null) {
RetrySettings createTrainingPipelineRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.createTrainingPipelineSettings().getRetrySettings(),
serviceRetry);
clientSettingsBuilder
.createTrainingPipelineSettings()
.setRetrySettings(createTrainingPipelineRetrySettings);
RetrySettings getTrainingPipelineRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.getTrainingPipelineSettings().getRetrySettings(), serviceRetry);
clientSettingsBuilder
.getTrainingPipelineSettings()
.setRetrySettings(getTrainingPipelineRetrySettings);
RetrySettings listTrainingPipelinesRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.listTrainingPipelinesSettings().getRetrySettings(),
serviceRetry);
clientSettingsBuilder
.listTrainingPipelinesSettings()
.setRetrySettings(listTrainingPipelinesRetrySettings);
RetrySettings cancelTrainingPipelineRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.cancelTrainingPipelineSettings().getRetrySettings(),
serviceRetry);
clientSettingsBuilder
.cancelTrainingPipelineSettings()
.setRetrySettings(cancelTrainingPipelineRetrySettings);
RetrySettings createPipelineJobRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.createPipelineJobSettings().getRetrySettings(), serviceRetry);
clientSettingsBuilder
.createPipelineJobSettings()
.setRetrySettings(createPipelineJobRetrySettings);
RetrySettings getPipelineJobRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.getPipelineJobSettings().getRetrySettings(), serviceRetry);
clientSettingsBuilder.getPipelineJobSettings().setRetrySettings(getPipelineJobRetrySettings);
RetrySettings listPipelineJobsRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.listPipelineJobsSettings().getRetrySettings(), serviceRetry);
clientSettingsBuilder
.listPipelineJobsSettings()
.setRetrySettings(listPipelineJobsRetrySettings);
RetrySettings cancelPipelineJobRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.cancelPipelineJobSettings().getRetrySettings(), serviceRetry);
clientSettingsBuilder
.cancelPipelineJobSettings()
.setRetrySettings(cancelPipelineJobRetrySettings);
RetrySettings listLocationsRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.listLocationsSettings().getRetrySettings(), serviceRetry);
clientSettingsBuilder.listLocationsSettings().setRetrySettings(listLocationsRetrySettings);
RetrySettings getLocationRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.getLocationSettings().getRetrySettings(), serviceRetry);
clientSettingsBuilder.getLocationSettings().setRetrySettings(getLocationRetrySettings);
RetrySettings setIamPolicyRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.setIamPolicySettings().getRetrySettings(), serviceRetry);
clientSettingsBuilder.setIamPolicySettings().setRetrySettings(setIamPolicyRetrySettings);
RetrySettings getIamPolicyRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.getIamPolicySettings().getRetrySettings(), serviceRetry);
clientSettingsBuilder.getIamPolicySettings().setRetrySettings(getIamPolicyRetrySettings);
RetrySettings testIamPermissionsRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.testIamPermissionsSettings().getRetrySettings(), serviceRetry);
clientSettingsBuilder
.testIamPermissionsSettings()
.setRetrySettings(testIamPermissionsRetrySettings);
if (LOGGER.isTraceEnabled()) {
LOGGER.trace("Configured service-level retry settings from properties.");
}
}
Retry createTrainingPipelineRetry = clientProperties.getCreateTrainingPipelineRetry();
if (createTrainingPipelineRetry != null) {
RetrySettings createTrainingPipelineRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.createTrainingPipelineSettings().getRetrySettings(),
createTrainingPipelineRetry);
clientSettingsBuilder
.createTrainingPipelineSettings()
.setRetrySettings(createTrainingPipelineRetrySettings);
if (LOGGER.isTraceEnabled()) {
LOGGER.trace(
"Configured method-level retry settings for createTrainingPipeline from properties.");
}
}
Retry getTrainingPipelineRetry = clientProperties.getGetTrainingPipelineRetry();
if (getTrainingPipelineRetry != null) {
RetrySettings getTrainingPipelineRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.getTrainingPipelineSettings().getRetrySettings(),
getTrainingPipelineRetry);
clientSettingsBuilder
.getTrainingPipelineSettings()
.setRetrySettings(getTrainingPipelineRetrySettings);
if (LOGGER.isTraceEnabled()) {
LOGGER.trace(
"Configured method-level retry settings for getTrainingPipeline from properties.");
}
}
Retry listTrainingPipelinesRetry = clientProperties.getListTrainingPipelinesRetry();
if (listTrainingPipelinesRetry != null) {
RetrySettings listTrainingPipelinesRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.listTrainingPipelinesSettings().getRetrySettings(),
listTrainingPipelinesRetry);
clientSettingsBuilder
.listTrainingPipelinesSettings()
.setRetrySettings(listTrainingPipelinesRetrySettings);
if (LOGGER.isTraceEnabled()) {
LOGGER.trace(
"Configured method-level retry settings for listTrainingPipelines from properties.");
}
}
Retry cancelTrainingPipelineRetry = clientProperties.getCancelTrainingPipelineRetry();
if (cancelTrainingPipelineRetry != null) {
RetrySettings cancelTrainingPipelineRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.cancelTrainingPipelineSettings().getRetrySettings(),
cancelTrainingPipelineRetry);
clientSettingsBuilder
.cancelTrainingPipelineSettings()
.setRetrySettings(cancelTrainingPipelineRetrySettings);
if (LOGGER.isTraceEnabled()) {
LOGGER.trace(
"Configured method-level retry settings for cancelTrainingPipeline from properties.");
}
}
Retry createPipelineJobRetry = clientProperties.getCreatePipelineJobRetry();
if (createPipelineJobRetry != null) {
RetrySettings createPipelineJobRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.createPipelineJobSettings().getRetrySettings(),
createPipelineJobRetry);
clientSettingsBuilder
.createPipelineJobSettings()
.setRetrySettings(createPipelineJobRetrySettings);
if (LOGGER.isTraceEnabled()) {
LOGGER.trace(
"Configured method-level retry settings for createPipelineJob from properties.");
}
}
Retry getPipelineJobRetry = clientProperties.getGetPipelineJobRetry();
if (getPipelineJobRetry != null) {
RetrySettings getPipelineJobRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.getPipelineJobSettings().getRetrySettings(),
getPipelineJobRetry);
clientSettingsBuilder.getPipelineJobSettings().setRetrySettings(getPipelineJobRetrySettings);
if (LOGGER.isTraceEnabled()) {
LOGGER.trace("Configured method-level retry settings for getPipelineJob from properties.");
}
}
Retry listPipelineJobsRetry = clientProperties.getListPipelineJobsRetry();
if (listPipelineJobsRetry != null) {
RetrySettings listPipelineJobsRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.listPipelineJobsSettings().getRetrySettings(),
listPipelineJobsRetry);
clientSettingsBuilder
.listPipelineJobsSettings()
.setRetrySettings(listPipelineJobsRetrySettings);
if (LOGGER.isTraceEnabled()) {
LOGGER.trace(
"Configured method-level retry settings for listPipelineJobs from properties.");
}
}
Retry cancelPipelineJobRetry = clientProperties.getCancelPipelineJobRetry();
if (cancelPipelineJobRetry != null) {
RetrySettings cancelPipelineJobRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.cancelPipelineJobSettings().getRetrySettings(),
cancelPipelineJobRetry);
clientSettingsBuilder
.cancelPipelineJobSettings()
.setRetrySettings(cancelPipelineJobRetrySettings);
if (LOGGER.isTraceEnabled()) {
LOGGER.trace(
"Configured method-level retry settings for cancelPipelineJob from properties.");
}
}
Retry listLocationsRetry = clientProperties.getListLocationsRetry();
if (listLocationsRetry != null) {
RetrySettings listLocationsRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.listLocationsSettings().getRetrySettings(), listLocationsRetry);
clientSettingsBuilder.listLocationsSettings().setRetrySettings(listLocationsRetrySettings);
if (LOGGER.isTraceEnabled()) {
LOGGER.trace("Configured method-level retry settings for listLocations from properties.");
}
}
Retry getLocationRetry = clientProperties.getGetLocationRetry();
if (getLocationRetry != null) {
RetrySettings getLocationRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.getLocationSettings().getRetrySettings(), getLocationRetry);
clientSettingsBuilder.getLocationSettings().setRetrySettings(getLocationRetrySettings);
if (LOGGER.isTraceEnabled()) {
LOGGER.trace("Configured method-level retry settings for getLocation from properties.");
}
}
Retry setIamPolicyRetry = clientProperties.getSetIamPolicyRetry();
if (setIamPolicyRetry != null) {
RetrySettings setIamPolicyRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.setIamPolicySettings().getRetrySettings(), setIamPolicyRetry);
clientSettingsBuilder.setIamPolicySettings().setRetrySettings(setIamPolicyRetrySettings);
if (LOGGER.isTraceEnabled()) {
LOGGER.trace("Configured method-level retry settings for setIamPolicy from properties.");
}
}
Retry getIamPolicyRetry = clientProperties.getGetIamPolicyRetry();
if (getIamPolicyRetry != null) {
RetrySettings getIamPolicyRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.getIamPolicySettings().getRetrySettings(), getIamPolicyRetry);
clientSettingsBuilder.getIamPolicySettings().setRetrySettings(getIamPolicyRetrySettings);
if (LOGGER.isTraceEnabled()) {
LOGGER.trace("Configured method-level retry settings for getIamPolicy from properties.");
}
}
Retry testIamPermissionsRetry = clientProperties.getTestIamPermissionsRetry();
if (testIamPermissionsRetry != null) {
RetrySettings testIamPermissionsRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.testIamPermissionsSettings().getRetrySettings(),
testIamPermissionsRetry);
clientSettingsBuilder
.testIamPermissionsSettings()
.setRetrySettings(testIamPermissionsRetrySettings);
if (LOGGER.isTraceEnabled()) {
LOGGER.trace(
"Configured method-level retry settings for testIamPermissions from properties.");
}
}
return clientSettingsBuilder.build();
}