spring-cloud-previews/google-cloud-automl-spring-starter/src/main/java/com/google/cloud/automl/v1/spring/PredictionServiceSpringAutoConfiguration.java [66:203]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    matchIfMissing = true)
@EnableConfigurationProperties(PredictionServiceSpringProperties.class)
public class PredictionServiceSpringAutoConfiguration {
  private final PredictionServiceSpringProperties clientProperties;
  private final CredentialsProvider credentialsProvider;
  private static final Log LOGGER =
      LogFactory.getLog(PredictionServiceSpringAutoConfiguration.class);

  protected PredictionServiceSpringAutoConfiguration(
      PredictionServiceSpringProperties clientProperties, CredentialsProvider credentialsProvider)
      throws IOException {
    this.clientProperties = clientProperties;
    if (this.clientProperties.getCredentials().hasKey()) {
      if (LOGGER.isTraceEnabled()) {
        LOGGER.trace("Using credentials from PredictionService-specific configuration");
      }
      this.credentialsProvider =
          ((CredentialsProvider) new DefaultCredentialsProvider(this.clientProperties));
    } else {
      this.credentialsProvider = credentialsProvider;
    }
  }

  /**
   * Provides a default transport channel provider bean, corresponding to the client library's
   * default transport channel provider. If the library supports both GRPC and REST transport, and
   * the useRest property is configured, the HTTP/JSON transport provider will be used instead of
   * GRPC.
   *
   * @return a default transport channel provider.
   */
  @Bean
  @ConditionalOnMissingBean(name = "defaultPredictionServiceTransportChannelProvider")
  public TransportChannelProvider defaultPredictionServiceTransportChannelProvider() {
    if (this.clientProperties.getUseRest()) {
      return PredictionServiceSettings.defaultHttpJsonTransportProviderBuilder().build();
    }
    return PredictionServiceSettings.defaultTransportChannelProvider();
  }

  /**
   * Provides a PredictionServiceSettings bean configured to use a DefaultCredentialsProvider and
   * the client library's default transport channel provider
   * (defaultPredictionServiceTransportChannelProvider()). It also configures the quota project ID
   * and executor thread count, if provided through properties.
   *
   * <p>Retry settings are also configured from service-level and method-level properties specified
   * in PredictionServiceSpringProperties. Method-level properties will take precedence over
   * service-level properties if available, and client library defaults will be used if neither are
   * specified.
   *
   * @param defaultTransportChannelProvider TransportChannelProvider to use in the settings.
   * @return a {@link PredictionServiceSettings} bean configured with {@link
   *     TransportChannelProvider} bean.
   */
  @Bean
  @ConditionalOnMissingBean
  public PredictionServiceSettings predictionServiceSettings(
      @Qualifier("defaultPredictionServiceTransportChannelProvider")
          TransportChannelProvider defaultTransportChannelProvider)
      throws IOException {
    PredictionServiceSettings.Builder clientSettingsBuilder;
    if (this.clientProperties.getUseRest()) {
      clientSettingsBuilder = PredictionServiceSettings.newHttpJsonBuilder();
      if (LOGGER.isTraceEnabled()) {
        LOGGER.trace("Using REST (HTTP/JSON) transport.");
      }
    } else {
      clientSettingsBuilder = PredictionServiceSettings.newBuilder();
    }
    clientSettingsBuilder
        .setCredentialsProvider(this.credentialsProvider)
        .setTransportChannelProvider(defaultTransportChannelProvider)
        .setEndpoint(PredictionServiceSettings.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 =
          PredictionServiceSettings.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 predictRetrySettings =
          RetryUtil.updateRetrySettings(
              clientSettingsBuilder.predictSettings().getRetrySettings(), serviceRetry);
      clientSettingsBuilder.predictSettings().setRetrySettings(predictRetrySettings);

      if (LOGGER.isTraceEnabled()) {
        LOGGER.trace("Configured service-level retry settings from properties.");
      }
    }
    Retry predictRetry = clientProperties.getPredictRetry();
    if (predictRetry != null) {
      RetrySettings predictRetrySettings =
          RetryUtil.updateRetrySettings(
              clientSettingsBuilder.predictSettings().getRetrySettings(), predictRetry);
      clientSettingsBuilder.predictSettings().setRetrySettings(predictRetrySettings);
      if (LOGGER.isTraceEnabled()) {
        LOGGER.trace("Configured method-level retry settings for predict from properties.");
      }
    }
    return clientSettingsBuilder.build();
  }

  /**
   * Provides a PredictionServiceClient bean configured with PredictionServiceSettings.
   *
   * @param predictionServiceSettings settings to configure an instance of client bean.
   * @return a {@link PredictionServiceClient} bean configured with {@link
   *     PredictionServiceSettings}
   */
  @Bean
  @ConditionalOnMissingBean
  public PredictionServiceClient predictionServiceClient(
      PredictionServiceSettings predictionServiceSettings) throws IOException {
    return PredictionServiceClient.create(predictionServiceSettings);
  }

  private HeaderProvider userAgentHeaderProvider() {
    String springLibrary = "spring-autogen-prediction-service";
    String version = this.getClass().getPackage().getImplementationVersion();
    return () -> Collections.singletonMap("user-agent", springLibrary + "/" + version);
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



spring-cloud-previews/google-cloud-retail-spring-starter/src/main/java/com/google/cloud/retail/v2/spring/PredictionServiceSpringAutoConfiguration.java [66:203]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    matchIfMissing = true)
@EnableConfigurationProperties(PredictionServiceSpringProperties.class)
public class PredictionServiceSpringAutoConfiguration {
  private final PredictionServiceSpringProperties clientProperties;
  private final CredentialsProvider credentialsProvider;
  private static final Log LOGGER =
      LogFactory.getLog(PredictionServiceSpringAutoConfiguration.class);

  protected PredictionServiceSpringAutoConfiguration(
      PredictionServiceSpringProperties clientProperties, CredentialsProvider credentialsProvider)
      throws IOException {
    this.clientProperties = clientProperties;
    if (this.clientProperties.getCredentials().hasKey()) {
      if (LOGGER.isTraceEnabled()) {
        LOGGER.trace("Using credentials from PredictionService-specific configuration");
      }
      this.credentialsProvider =
          ((CredentialsProvider) new DefaultCredentialsProvider(this.clientProperties));
    } else {
      this.credentialsProvider = credentialsProvider;
    }
  }

  /**
   * Provides a default transport channel provider bean, corresponding to the client library's
   * default transport channel provider. If the library supports both GRPC and REST transport, and
   * the useRest property is configured, the HTTP/JSON transport provider will be used instead of
   * GRPC.
   *
   * @return a default transport channel provider.
   */
  @Bean
  @ConditionalOnMissingBean(name = "defaultPredictionServiceTransportChannelProvider")
  public TransportChannelProvider defaultPredictionServiceTransportChannelProvider() {
    if (this.clientProperties.getUseRest()) {
      return PredictionServiceSettings.defaultHttpJsonTransportProviderBuilder().build();
    }
    return PredictionServiceSettings.defaultTransportChannelProvider();
  }

  /**
   * Provides a PredictionServiceSettings bean configured to use a DefaultCredentialsProvider and
   * the client library's default transport channel provider
   * (defaultPredictionServiceTransportChannelProvider()). It also configures the quota project ID
   * and executor thread count, if provided through properties.
   *
   * <p>Retry settings are also configured from service-level and method-level properties specified
   * in PredictionServiceSpringProperties. Method-level properties will take precedence over
   * service-level properties if available, and client library defaults will be used if neither are
   * specified.
   *
   * @param defaultTransportChannelProvider TransportChannelProvider to use in the settings.
   * @return a {@link PredictionServiceSettings} bean configured with {@link
   *     TransportChannelProvider} bean.
   */
  @Bean
  @ConditionalOnMissingBean
  public PredictionServiceSettings predictionServiceSettings(
      @Qualifier("defaultPredictionServiceTransportChannelProvider")
          TransportChannelProvider defaultTransportChannelProvider)
      throws IOException {
    PredictionServiceSettings.Builder clientSettingsBuilder;
    if (this.clientProperties.getUseRest()) {
      clientSettingsBuilder = PredictionServiceSettings.newHttpJsonBuilder();
      if (LOGGER.isTraceEnabled()) {
        LOGGER.trace("Using REST (HTTP/JSON) transport.");
      }
    } else {
      clientSettingsBuilder = PredictionServiceSettings.newBuilder();
    }
    clientSettingsBuilder
        .setCredentialsProvider(this.credentialsProvider)
        .setTransportChannelProvider(defaultTransportChannelProvider)
        .setEndpoint(PredictionServiceSettings.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 =
          PredictionServiceSettings.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 predictRetrySettings =
          RetryUtil.updateRetrySettings(
              clientSettingsBuilder.predictSettings().getRetrySettings(), serviceRetry);
      clientSettingsBuilder.predictSettings().setRetrySettings(predictRetrySettings);

      if (LOGGER.isTraceEnabled()) {
        LOGGER.trace("Configured service-level retry settings from properties.");
      }
    }
    Retry predictRetry = clientProperties.getPredictRetry();
    if (predictRetry != null) {
      RetrySettings predictRetrySettings =
          RetryUtil.updateRetrySettings(
              clientSettingsBuilder.predictSettings().getRetrySettings(), predictRetry);
      clientSettingsBuilder.predictSettings().setRetrySettings(predictRetrySettings);
      if (LOGGER.isTraceEnabled()) {
        LOGGER.trace("Configured method-level retry settings for predict from properties.");
      }
    }
    return clientSettingsBuilder.build();
  }

  /**
   * Provides a PredictionServiceClient bean configured with PredictionServiceSettings.
   *
   * @param predictionServiceSettings settings to configure an instance of client bean.
   * @return a {@link PredictionServiceClient} bean configured with {@link
   *     PredictionServiceSettings}
   */
  @Bean
  @ConditionalOnMissingBean
  public PredictionServiceClient predictionServiceClient(
      PredictionServiceSettings predictionServiceSettings) throws IOException {
    return PredictionServiceClient.create(predictionServiceSettings);
  }

  private HeaderProvider userAgentHeaderProvider() {
    String springLibrary = "spring-autogen-prediction-service";
    String version = this.getClass().getPackage().getImplementationVersion();
    return () -> Collections.singletonMap("user-agent", springLibrary + "/" + version);
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -