add option to add connectTimeout in couchbase writer (apache#3958)

Author: arjun4084346

gobblin-modules/gobblin-couchbase/src/main/java/org/apache/gobblin/couchbase/writer/CouchbaseEnvironmentFactory.java

@@ -46,6 +46,7 @@ public static synchronized CouchbaseEnvironment getInstance(Config config)
     String sslTruststorePassword = ConfigUtils.getString(config, CouchbaseWriterConfigurationKeys.SSL_TRUSTSTORE_PASSWORD, "");
     Boolean certAuthEnabled = ConfigUtils.getBoolean(config, CouchbaseWriterConfigurationKeys.CERT_AUTH_ENABLED, false);
     Boolean dnsSrvEnabled = ConfigUtils.getBoolean(config, CouchbaseWriterConfigurationKeys.DNS_SRV_ENABLED, false);
+    Long connectTimeout = ConfigUtils.getLong(config, CouchbaseWriterConfigurationKeys.CONNECT_TIMEOUT, CouchbaseWriterConfigurationKeys.CONNECT_TIMEOUT_DEFAULT);
     Integer socketConnectTimeout = ConfigUtils.getInt(config, CouchbaseWriterConfigurationKeys.SOCKET_CONNECT_TIMEOUT,
         DefaultCouchbaseEnvironment.SOCKET_CONNECT_TIMEOUT);
 
@@ -56,6 +57,7 @@ public static synchronized CouchbaseEnvironment getInstance(Config config)
         .sslTruststoreFile(sslTruststoreFile)
         .sslTruststorePassword(sslTruststorePassword)
         .certAuthEnabled(certAuthEnabled)
+        .connectTimeout(connectTimeout)
         .dnsSrvEnabled(dnsSrvEnabled)
         .socketConnectTimeout(socketConnectTimeout);
 

gobblin-modules/gobblin-couchbase/src/main/java/org/apache/gobblin/couchbase/writer/CouchbaseWriterConfigurationKeys.java

@@ -53,6 +53,9 @@ public class CouchbaseWriterConfigurationKeys {
 
   public static final String OPERATION_TIMEOUT_MILLIS = prefix("operationTimeoutMillis");
   public static final long OPERATION_TIMEOUT_DEFAULT = 10000; // 10 second default timeout
+  public static final String CONNECT_TIMEOUT = prefix("connectTimeout");
+  // using the default value same as DefaultCouchbaseEnvironment.CONNECT_TIMEOUT which is private in java-client-2.7.6
+  public static final long CONNECT_TIMEOUT_DEFAULT = TimeUnit.SECONDS.toMillis(5);
 
   public static final String RETRIES_ENABLED = prefix("retriesEnabled");
   public static final boolean RETRIES_ENABLED_DEFAULT = false;