in spring-cloud-previews/google-cloud-orchestration-airflow-spring-starter/src/main/java/com/google/cloud/orchestration/airflow/service/v1/spring/EnvironmentsSpringAutoConfiguration.java [122:529]
public EnvironmentsSettings environmentsSettings(
@Qualifier("defaultEnvironmentsTransportChannelProvider")
TransportChannelProvider defaultTransportChannelProvider)
throws IOException {
EnvironmentsSettings.Builder clientSettingsBuilder;
if (this.clientProperties.getUseRest()) {
clientSettingsBuilder = EnvironmentsSettings.newHttpJsonBuilder();
if (LOGGER.isTraceEnabled()) {
LOGGER.trace("Using REST (HTTP/JSON) transport.");
}
} else {
clientSettingsBuilder = EnvironmentsSettings.newBuilder();
}
clientSettingsBuilder
.setCredentialsProvider(this.credentialsProvider)
.setTransportChannelProvider(defaultTransportChannelProvider)
.setEndpoint(EnvironmentsSettings.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 =
EnvironmentsSettings.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 getEnvironmentRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.getEnvironmentSettings().getRetrySettings(), serviceRetry);
clientSettingsBuilder.getEnvironmentSettings().setRetrySettings(getEnvironmentRetrySettings);
RetrySettings listEnvironmentsRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.listEnvironmentsSettings().getRetrySettings(), serviceRetry);
clientSettingsBuilder
.listEnvironmentsSettings()
.setRetrySettings(listEnvironmentsRetrySettings);
RetrySettings executeAirflowCommandRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.executeAirflowCommandSettings().getRetrySettings(),
serviceRetry);
clientSettingsBuilder
.executeAirflowCommandSettings()
.setRetrySettings(executeAirflowCommandRetrySettings);
RetrySettings stopAirflowCommandRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.stopAirflowCommandSettings().getRetrySettings(), serviceRetry);
clientSettingsBuilder
.stopAirflowCommandSettings()
.setRetrySettings(stopAirflowCommandRetrySettings);
RetrySettings pollAirflowCommandRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.pollAirflowCommandSettings().getRetrySettings(), serviceRetry);
clientSettingsBuilder
.pollAirflowCommandSettings()
.setRetrySettings(pollAirflowCommandRetrySettings);
RetrySettings listWorkloadsRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.listWorkloadsSettings().getRetrySettings(), serviceRetry);
clientSettingsBuilder.listWorkloadsSettings().setRetrySettings(listWorkloadsRetrySettings);
RetrySettings createUserWorkloadsSecretRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.createUserWorkloadsSecretSettings().getRetrySettings(),
serviceRetry);
clientSettingsBuilder
.createUserWorkloadsSecretSettings()
.setRetrySettings(createUserWorkloadsSecretRetrySettings);
RetrySettings getUserWorkloadsSecretRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.getUserWorkloadsSecretSettings().getRetrySettings(),
serviceRetry);
clientSettingsBuilder
.getUserWorkloadsSecretSettings()
.setRetrySettings(getUserWorkloadsSecretRetrySettings);
RetrySettings listUserWorkloadsSecretsRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.listUserWorkloadsSecretsSettings().getRetrySettings(),
serviceRetry);
clientSettingsBuilder
.listUserWorkloadsSecretsSettings()
.setRetrySettings(listUserWorkloadsSecretsRetrySettings);
RetrySettings updateUserWorkloadsSecretRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.updateUserWorkloadsSecretSettings().getRetrySettings(),
serviceRetry);
clientSettingsBuilder
.updateUserWorkloadsSecretSettings()
.setRetrySettings(updateUserWorkloadsSecretRetrySettings);
RetrySettings deleteUserWorkloadsSecretRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.deleteUserWorkloadsSecretSettings().getRetrySettings(),
serviceRetry);
clientSettingsBuilder
.deleteUserWorkloadsSecretSettings()
.setRetrySettings(deleteUserWorkloadsSecretRetrySettings);
RetrySettings createUserWorkloadsConfigMapRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.createUserWorkloadsConfigMapSettings().getRetrySettings(),
serviceRetry);
clientSettingsBuilder
.createUserWorkloadsConfigMapSettings()
.setRetrySettings(createUserWorkloadsConfigMapRetrySettings);
RetrySettings getUserWorkloadsConfigMapRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.getUserWorkloadsConfigMapSettings().getRetrySettings(),
serviceRetry);
clientSettingsBuilder
.getUserWorkloadsConfigMapSettings()
.setRetrySettings(getUserWorkloadsConfigMapRetrySettings);
RetrySettings listUserWorkloadsConfigMapsRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.listUserWorkloadsConfigMapsSettings().getRetrySettings(),
serviceRetry);
clientSettingsBuilder
.listUserWorkloadsConfigMapsSettings()
.setRetrySettings(listUserWorkloadsConfigMapsRetrySettings);
RetrySettings updateUserWorkloadsConfigMapRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.updateUserWorkloadsConfigMapSettings().getRetrySettings(),
serviceRetry);
clientSettingsBuilder
.updateUserWorkloadsConfigMapSettings()
.setRetrySettings(updateUserWorkloadsConfigMapRetrySettings);
RetrySettings deleteUserWorkloadsConfigMapRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.deleteUserWorkloadsConfigMapSettings().getRetrySettings(),
serviceRetry);
clientSettingsBuilder
.deleteUserWorkloadsConfigMapSettings()
.setRetrySettings(deleteUserWorkloadsConfigMapRetrySettings);
RetrySettings fetchDatabasePropertiesRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.fetchDatabasePropertiesSettings().getRetrySettings(),
serviceRetry);
clientSettingsBuilder
.fetchDatabasePropertiesSettings()
.setRetrySettings(fetchDatabasePropertiesRetrySettings);
if (LOGGER.isTraceEnabled()) {
LOGGER.trace("Configured service-level retry settings from properties.");
}
}
Retry getEnvironmentRetry = clientProperties.getGetEnvironmentRetry();
if (getEnvironmentRetry != null) {
RetrySettings getEnvironmentRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.getEnvironmentSettings().getRetrySettings(),
getEnvironmentRetry);
clientSettingsBuilder.getEnvironmentSettings().setRetrySettings(getEnvironmentRetrySettings);
if (LOGGER.isTraceEnabled()) {
LOGGER.trace("Configured method-level retry settings for getEnvironment from properties.");
}
}
Retry listEnvironmentsRetry = clientProperties.getListEnvironmentsRetry();
if (listEnvironmentsRetry != null) {
RetrySettings listEnvironmentsRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.listEnvironmentsSettings().getRetrySettings(),
listEnvironmentsRetry);
clientSettingsBuilder
.listEnvironmentsSettings()
.setRetrySettings(listEnvironmentsRetrySettings);
if (LOGGER.isTraceEnabled()) {
LOGGER.trace(
"Configured method-level retry settings for listEnvironments from properties.");
}
}
Retry executeAirflowCommandRetry = clientProperties.getExecuteAirflowCommandRetry();
if (executeAirflowCommandRetry != null) {
RetrySettings executeAirflowCommandRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.executeAirflowCommandSettings().getRetrySettings(),
executeAirflowCommandRetry);
clientSettingsBuilder
.executeAirflowCommandSettings()
.setRetrySettings(executeAirflowCommandRetrySettings);
if (LOGGER.isTraceEnabled()) {
LOGGER.trace(
"Configured method-level retry settings for executeAirflowCommand from properties.");
}
}
Retry stopAirflowCommandRetry = clientProperties.getStopAirflowCommandRetry();
if (stopAirflowCommandRetry != null) {
RetrySettings stopAirflowCommandRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.stopAirflowCommandSettings().getRetrySettings(),
stopAirflowCommandRetry);
clientSettingsBuilder
.stopAirflowCommandSettings()
.setRetrySettings(stopAirflowCommandRetrySettings);
if (LOGGER.isTraceEnabled()) {
LOGGER.trace(
"Configured method-level retry settings for stopAirflowCommand from properties.");
}
}
Retry pollAirflowCommandRetry = clientProperties.getPollAirflowCommandRetry();
if (pollAirflowCommandRetry != null) {
RetrySettings pollAirflowCommandRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.pollAirflowCommandSettings().getRetrySettings(),
pollAirflowCommandRetry);
clientSettingsBuilder
.pollAirflowCommandSettings()
.setRetrySettings(pollAirflowCommandRetrySettings);
if (LOGGER.isTraceEnabled()) {
LOGGER.trace(
"Configured method-level retry settings for pollAirflowCommand from properties.");
}
}
Retry listWorkloadsRetry = clientProperties.getListWorkloadsRetry();
if (listWorkloadsRetry != null) {
RetrySettings listWorkloadsRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.listWorkloadsSettings().getRetrySettings(), listWorkloadsRetry);
clientSettingsBuilder.listWorkloadsSettings().setRetrySettings(listWorkloadsRetrySettings);
if (LOGGER.isTraceEnabled()) {
LOGGER.trace("Configured method-level retry settings for listWorkloads from properties.");
}
}
Retry createUserWorkloadsSecretRetry = clientProperties.getCreateUserWorkloadsSecretRetry();
if (createUserWorkloadsSecretRetry != null) {
RetrySettings createUserWorkloadsSecretRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.createUserWorkloadsSecretSettings().getRetrySettings(),
createUserWorkloadsSecretRetry);
clientSettingsBuilder
.createUserWorkloadsSecretSettings()
.setRetrySettings(createUserWorkloadsSecretRetrySettings);
if (LOGGER.isTraceEnabled()) {
LOGGER.trace(
"Configured method-level retry settings for createUserWorkloadsSecret from properties.");
}
}
Retry getUserWorkloadsSecretRetry = clientProperties.getGetUserWorkloadsSecretRetry();
if (getUserWorkloadsSecretRetry != null) {
RetrySettings getUserWorkloadsSecretRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.getUserWorkloadsSecretSettings().getRetrySettings(),
getUserWorkloadsSecretRetry);
clientSettingsBuilder
.getUserWorkloadsSecretSettings()
.setRetrySettings(getUserWorkloadsSecretRetrySettings);
if (LOGGER.isTraceEnabled()) {
LOGGER.trace(
"Configured method-level retry settings for getUserWorkloadsSecret from properties.");
}
}
Retry listUserWorkloadsSecretsRetry = clientProperties.getListUserWorkloadsSecretsRetry();
if (listUserWorkloadsSecretsRetry != null) {
RetrySettings listUserWorkloadsSecretsRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.listUserWorkloadsSecretsSettings().getRetrySettings(),
listUserWorkloadsSecretsRetry);
clientSettingsBuilder
.listUserWorkloadsSecretsSettings()
.setRetrySettings(listUserWorkloadsSecretsRetrySettings);
if (LOGGER.isTraceEnabled()) {
LOGGER.trace(
"Configured method-level retry settings for listUserWorkloadsSecrets from properties.");
}
}
Retry updateUserWorkloadsSecretRetry = clientProperties.getUpdateUserWorkloadsSecretRetry();
if (updateUserWorkloadsSecretRetry != null) {
RetrySettings updateUserWorkloadsSecretRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.updateUserWorkloadsSecretSettings().getRetrySettings(),
updateUserWorkloadsSecretRetry);
clientSettingsBuilder
.updateUserWorkloadsSecretSettings()
.setRetrySettings(updateUserWorkloadsSecretRetrySettings);
if (LOGGER.isTraceEnabled()) {
LOGGER.trace(
"Configured method-level retry settings for updateUserWorkloadsSecret from properties.");
}
}
Retry deleteUserWorkloadsSecretRetry = clientProperties.getDeleteUserWorkloadsSecretRetry();
if (deleteUserWorkloadsSecretRetry != null) {
RetrySettings deleteUserWorkloadsSecretRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.deleteUserWorkloadsSecretSettings().getRetrySettings(),
deleteUserWorkloadsSecretRetry);
clientSettingsBuilder
.deleteUserWorkloadsSecretSettings()
.setRetrySettings(deleteUserWorkloadsSecretRetrySettings);
if (LOGGER.isTraceEnabled()) {
LOGGER.trace(
"Configured method-level retry settings for deleteUserWorkloadsSecret from properties.");
}
}
Retry createUserWorkloadsConfigMapRetry =
clientProperties.getCreateUserWorkloadsConfigMapRetry();
if (createUserWorkloadsConfigMapRetry != null) {
RetrySettings createUserWorkloadsConfigMapRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.createUserWorkloadsConfigMapSettings().getRetrySettings(),
createUserWorkloadsConfigMapRetry);
clientSettingsBuilder
.createUserWorkloadsConfigMapSettings()
.setRetrySettings(createUserWorkloadsConfigMapRetrySettings);
if (LOGGER.isTraceEnabled()) {
LOGGER.trace(
"Configured method-level retry settings for createUserWorkloadsConfigMap from properties.");
}
}
Retry getUserWorkloadsConfigMapRetry = clientProperties.getGetUserWorkloadsConfigMapRetry();
if (getUserWorkloadsConfigMapRetry != null) {
RetrySettings getUserWorkloadsConfigMapRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.getUserWorkloadsConfigMapSettings().getRetrySettings(),
getUserWorkloadsConfigMapRetry);
clientSettingsBuilder
.getUserWorkloadsConfigMapSettings()
.setRetrySettings(getUserWorkloadsConfigMapRetrySettings);
if (LOGGER.isTraceEnabled()) {
LOGGER.trace(
"Configured method-level retry settings for getUserWorkloadsConfigMap from properties.");
}
}
Retry listUserWorkloadsConfigMapsRetry = clientProperties.getListUserWorkloadsConfigMapsRetry();
if (listUserWorkloadsConfigMapsRetry != null) {
RetrySettings listUserWorkloadsConfigMapsRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.listUserWorkloadsConfigMapsSettings().getRetrySettings(),
listUserWorkloadsConfigMapsRetry);
clientSettingsBuilder
.listUserWorkloadsConfigMapsSettings()
.setRetrySettings(listUserWorkloadsConfigMapsRetrySettings);
if (LOGGER.isTraceEnabled()) {
LOGGER.trace(
"Configured method-level retry settings for listUserWorkloadsConfigMaps from properties.");
}
}
Retry updateUserWorkloadsConfigMapRetry =
clientProperties.getUpdateUserWorkloadsConfigMapRetry();
if (updateUserWorkloadsConfigMapRetry != null) {
RetrySettings updateUserWorkloadsConfigMapRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.updateUserWorkloadsConfigMapSettings().getRetrySettings(),
updateUserWorkloadsConfigMapRetry);
clientSettingsBuilder
.updateUserWorkloadsConfigMapSettings()
.setRetrySettings(updateUserWorkloadsConfigMapRetrySettings);
if (LOGGER.isTraceEnabled()) {
LOGGER.trace(
"Configured method-level retry settings for updateUserWorkloadsConfigMap from properties.");
}
}
Retry deleteUserWorkloadsConfigMapRetry =
clientProperties.getDeleteUserWorkloadsConfigMapRetry();
if (deleteUserWorkloadsConfigMapRetry != null) {
RetrySettings deleteUserWorkloadsConfigMapRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.deleteUserWorkloadsConfigMapSettings().getRetrySettings(),
deleteUserWorkloadsConfigMapRetry);
clientSettingsBuilder
.deleteUserWorkloadsConfigMapSettings()
.setRetrySettings(deleteUserWorkloadsConfigMapRetrySettings);
if (LOGGER.isTraceEnabled()) {
LOGGER.trace(
"Configured method-level retry settings for deleteUserWorkloadsConfigMap from properties.");
}
}
Retry fetchDatabasePropertiesRetry = clientProperties.getFetchDatabasePropertiesRetry();
if (fetchDatabasePropertiesRetry != null) {
RetrySettings fetchDatabasePropertiesRetrySettings =
RetryUtil.updateRetrySettings(
clientSettingsBuilder.fetchDatabasePropertiesSettings().getRetrySettings(),
fetchDatabasePropertiesRetry);
clientSettingsBuilder
.fetchDatabasePropertiesSettings()
.setRetrySettings(fetchDatabasePropertiesRetrySettings);
if (LOGGER.isTraceEnabled()) {
LOGGER.trace(
"Configured method-level retry settings for fetchDatabaseProperties from properties.");
}
}
return clientSettingsBuilder.build();
}