com.snowflake.kafka.connector.internal.SnowflakeIngestionServiceFactory Maven / Gradle / Ivy

Go to download
package com.snowflake.kafka.connector.internal;

import com.snowflake.kafka.connector.internal.telemetry.SnowflakeTelemetryService;
import java.security.PrivateKey;
import javax.annotation.Nullable;

/** A factory to create {@link SnowflakeIngestionService} */
public class SnowflakeIngestionServiceFactory {

  public static SnowflakeIngestionServiceBuilder builder(
      String accountName,
      String userName,
      String host,
      int port,
      String connectionScheme,
      String stageName,
      String pipeName,
      PrivateKey privateKey,
      String userAgentSuffix,
      SnowflakeTelemetryService telemetry) {
    return new SnowflakeIngestionServiceBuilder(
        accountName,
        userName,
        host,
        port,
        connectionScheme,
        stageName,
        pipeName,
        privateKey,
        userAgentSuffix,
        telemetry);
  }

  /** Builder class to create instance of {@link SnowflakeIngestionService} */
  static class SnowflakeIngestionServiceBuilder {
    private final SnowflakeIngestionService service;

    private SnowflakeIngestionServiceBuilder(
        String accountName,
        String userName,
        String host,
        int port,
        String connectionScheme,
        String stageName,
        String pipeName,
        PrivateKey privateKey,
        String userAgentSuffix,
        @Nullable SnowflakeTelemetryService telemetry) {
      this.service =
          new SnowflakeIngestionServiceV1(
              accountName,
              userName,
              host,
              port,
              connectionScheme,
              stageName,
              pipeName,
              privateKey,
              userAgentSuffix,
              telemetry);
    }

    SnowflakeIngestionServiceBuilder setTelemetry(SnowflakeTelemetryService telemetry) {
      service.setTelemetry(telemetry);
      return this;
    }

    SnowflakeIngestionService build() {
      return service;
    }
  }
}