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

  protected CompletionServiceSpringAutoConfiguration(
      CompletionServiceSpringProperties clientProperties, CredentialsProvider credentialsProvider)
      throws IOException {
    this.clientProperties = clientProperties;
    if (this.clientProperties.getCredentials().hasKey()) {
      if (LOGGER.isTraceEnabled()) {
        LOGGER.trace("Using credentials from CompletionService-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 = "defaultCompletionServiceTransportChannelProvider")
  public TransportChannelProvider defaultCompletionServiceTransportChannelProvider() {
    if (this.clientProperties.getUseRest()) {
      return CompletionServiceSettings.defaultHttpJsonTransportProviderBuilder().build();
    }
    return CompletionServiceSettings.defaultTransportChannelProvider();
  }

  /**
   * Provides a CompletionServiceSettings bean configured to use a DefaultCredentialsProvider and
   * the client library's default transport channel provider
   * (defaultCompletionServiceTransportChannelProvider()). 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 CompletionServiceSpringProperties. 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 CompletionServiceSettings} bean configured with {@link
   *     TransportChannelProvider} bean.
   */
  @Bean
  @ConditionalOnMissingBean
  public CompletionServiceSettings completionServiceSettings(
      @Qualifier("defaultCompletionServiceTransportChannelProvider")
          TransportChannelProvider defaultTransportChannelProvider)
      throws IOException {
    CompletionServiceSettings.Builder clientSettingsBuilder;
    if (this.clientProperties.getUseRest()) {
      clientSettingsBuilder = CompletionServiceSettings.newHttpJsonBuilder();
      if (LOGGER.isTraceEnabled()) {
        LOGGER.trace("Using REST (HTTP/JSON) transport.");
      }
    } else {
      clientSettingsBuilder = CompletionServiceSettings.newBuilder();
    }
    clientSettingsBuilder
        .setCredentialsProvider(this.credentialsProvider)
        .setTransportChannelProvider(defaultTransportChannelProvider)
        .setEndpoint(CompletionServiceSettings.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 =
          CompletionServiceSettings.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 completeQueryRetrySettings =
          RetryUtil.updateRetrySettings(
              clientSettingsBuilder.completeQuerySettings().getRetrySettings(), serviceRetry);
      clientSettingsBuilder.completeQuerySettings().setRetrySettings(completeQueryRetrySettings);

      if (LOGGER.isTraceEnabled()) {
        LOGGER.trace("Configured service-level retry settings from properties.");
      }
    }
    Retry completeQueryRetry = clientProperties.getCompleteQueryRetry();
    if (completeQueryRetry != null) {
      RetrySettings completeQueryRetrySettings =
          RetryUtil.updateRetrySettings(
              clientSettingsBuilder.completeQuerySettings().getRetrySettings(), completeQueryRetry);
      clientSettingsBuilder.completeQuerySettings().setRetrySettings(completeQueryRetrySettings);
      if (LOGGER.isTraceEnabled()) {
        LOGGER.trace("Configured method-level retry settings for completeQuery from properties.");
      }
    }
    return clientSettingsBuilder.build();
  }

  /**
   * Provides a CompletionServiceClient bean configured with CompletionServiceSettings.
   *
   * @param completionServiceSettings settings to configure an instance of client bean.
   * @return a {@link CompletionServiceClient} bean configured with {@link
   *     CompletionServiceSettings}
   */
  @Bean
  @ConditionalOnMissingBean
  public CompletionServiceClient completionServiceClient(
      CompletionServiceSettings completionServiceSettings) throws IOException {
    return CompletionServiceClient.create(completionServiceSettings);
  }

  private HeaderProvider userAgentHeaderProvider() {
    String springLibrary = "spring-autogen-completion-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/CompletionServiceSpringAutoConfiguration.java [66:203]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    matchIfMissing = true)
@EnableConfigurationProperties(CompletionServiceSpringProperties.class)
public class CompletionServiceSpringAutoConfiguration {
  private final CompletionServiceSpringProperties clientProperties;
  private final CredentialsProvider credentialsProvider;
  private static final Log LOGGER =
      LogFactory.getLog(CompletionServiceSpringAutoConfiguration.class);

  protected CompletionServiceSpringAutoConfiguration(
      CompletionServiceSpringProperties clientProperties, CredentialsProvider credentialsProvider)
      throws IOException {
    this.clientProperties = clientProperties;
    if (this.clientProperties.getCredentials().hasKey()) {
      if (LOGGER.isTraceEnabled()) {
        LOGGER.trace("Using credentials from CompletionService-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 = "defaultCompletionServiceTransportChannelProvider")
  public TransportChannelProvider defaultCompletionServiceTransportChannelProvider() {
    if (this.clientProperties.getUseRest()) {
      return CompletionServiceSettings.defaultHttpJsonTransportProviderBuilder().build();
    }
    return CompletionServiceSettings.defaultTransportChannelProvider();
  }

  /**
   * Provides a CompletionServiceSettings bean configured to use a DefaultCredentialsProvider and
   * the client library's default transport channel provider
   * (defaultCompletionServiceTransportChannelProvider()). 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 CompletionServiceSpringProperties. 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 CompletionServiceSettings} bean configured with {@link
   *     TransportChannelProvider} bean.
   */
  @Bean
  @ConditionalOnMissingBean
  public CompletionServiceSettings completionServiceSettings(
      @Qualifier("defaultCompletionServiceTransportChannelProvider")
          TransportChannelProvider defaultTransportChannelProvider)
      throws IOException {
    CompletionServiceSettings.Builder clientSettingsBuilder;
    if (this.clientProperties.getUseRest()) {
      clientSettingsBuilder = CompletionServiceSettings.newHttpJsonBuilder();
      if (LOGGER.isTraceEnabled()) {
        LOGGER.trace("Using REST (HTTP/JSON) transport.");
      }
    } else {
      clientSettingsBuilder = CompletionServiceSettings.newBuilder();
    }
    clientSettingsBuilder
        .setCredentialsProvider(this.credentialsProvider)
        .setTransportChannelProvider(defaultTransportChannelProvider)
        .setEndpoint(CompletionServiceSettings.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 =
          CompletionServiceSettings.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 completeQueryRetrySettings =
          RetryUtil.updateRetrySettings(
              clientSettingsBuilder.completeQuerySettings().getRetrySettings(), serviceRetry);
      clientSettingsBuilder.completeQuerySettings().setRetrySettings(completeQueryRetrySettings);

      if (LOGGER.isTraceEnabled()) {
        LOGGER.trace("Configured service-level retry settings from properties.");
      }
    }
    Retry completeQueryRetry = clientProperties.getCompleteQueryRetry();
    if (completeQueryRetry != null) {
      RetrySettings completeQueryRetrySettings =
          RetryUtil.updateRetrySettings(
              clientSettingsBuilder.completeQuerySettings().getRetrySettings(), completeQueryRetry);
      clientSettingsBuilder.completeQuerySettings().setRetrySettings(completeQueryRetrySettings);
      if (LOGGER.isTraceEnabled()) {
        LOGGER.trace("Configured method-level retry settings for completeQuery from properties.");
      }
    }
    return clientSettingsBuilder.build();
  }

  /**
   * Provides a CompletionServiceClient bean configured with CompletionServiceSettings.
   *
   * @param completionServiceSettings settings to configure an instance of client bean.
   * @return a {@link CompletionServiceClient} bean configured with {@link
   *     CompletionServiceSettings}
   */
  @Bean
  @ConditionalOnMissingBean
  public CompletionServiceClient completionServiceClient(
      CompletionServiceSettings completionServiceSettings) throws IOException {
    return CompletionServiceClient.create(completionServiceSettings);
  }

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