Set connection timeout for cassandra tool (#1277)

temporalio · Feb 10, 2021 · cf20471 · cf20471
1 parent 677e0d1
commit cf20471
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tools/cassandra/cqlclient.go b/tools/cassandra/cqlclient.go
@@ -114,6 +114,7 @@ func NewCassandraCluster(cfg *config.Cassandra, timeoutSeconds int) (*gocql.Clus
 	}
 
 	timeout := time.Duration(timeoutSeconds) * time.Second
+	clusterCfg.ConnectTimeout = timeout
 	clusterCfg.Timeout = timeout
 	clusterCfg.ProtoVersion = cqlProtoVersion
 	clusterCfg.Consistency = cfg.Consistency.GetConsistency()