Add config property for Cassandra connection timeout (#679)

* Add config property for Cassandra connection timeout

* change connect timeout to duration

* update comment

Author: vitarb

common/cassandra/cassandraCluster.go

@@ -26,9 +26,8 @@ package cassandra
 
 import (
 	"crypto/tls"
-	"strings"
-
 	"github.com/gocql/gocql"
+	"strings"
 
 	"go.temporal.io/server/common/service/config"
 )
@@ -68,6 +67,9 @@ func NewCassandraCluster(cfg config.Cassandra) *gocql.ClusterConfig {
 	if cfg.MaxConns > 0 {
 		cluster.NumConns = cfg.MaxConns
 	}
+	if cfg.ConnectTimeout > 0 {
+		cluster.ConnectTimeout = cfg.ConnectTimeout
+	}
 
 	cluster.PoolConfig.HostSelectionPolicy = gocql.TokenAwareHostPolicy(gocql.RoundRobinHostPolicy())
 

common/persistence/cassandra/cassandraPersistenceTest.go

@@ -71,12 +71,13 @@ func NewTestCluster(keyspace, username, password, host string, port int, schemaD
 	}
 	result.schemaDir = schemaDir
 	result.cfg = config.Cassandra{
-		User:     username,
-		Password: password,
-		Hosts:    host,
-		Port:     port,
-		MaxConns: 2,
-		Keyspace: keyspace,
+		User:           username,
+		Password:       password,
+		Hosts:          host,
+		Port:           port,
+		MaxConns:       2,
+		ConnectTimeout: 600 * time.Millisecond,
+		Keyspace:       keyspace,
 	}
 	return &result
 }

common/service/config/config.go

@@ -230,6 +230,8 @@ type (
 		Datacenter string `yaml:"datacenter"`
 		// MaxConns is the max number of connections to this datastore for a single keyspace
 		MaxConns int `yaml:"maxConns"`
+		// ConnectTimeout is a timeout for initial dial to cassandra server (default: 600ms)
+		ConnectTimeout time.Duration `yaml:"connectTimeout"`
 		// TLS configuration
 		TLS *auth.TLS `yaml:"tls"`
 		// Consistency configuration (defaults to LOCAL_QUORUM / LOCAL_SERIAL for all stores if this field not set)