diff --git a/cassandra/config.go b/cassandra/config.go index 74102716..3732c565 100644 --- a/cassandra/config.go +++ b/cassandra/config.go @@ -36,12 +36,14 @@ type Config struct { // ConfigDefault is the default config var ConfigDefault = Config{ - Hosts: []string{"localhost:9042"}, - Keyspace: "gofiber", - Table: "kv_store", - Consistency: gocql.Quorum, - Reset: false, - Expiration: 10 * time.Minute, + Hosts: []string{"localhost:9042"}, + Keyspace: "gofiber", + Table: "kv_store", + Consistency: gocql.Quorum, + Reset: false, + Expiration: 10 * time.Minute, + MaxRetries: 3, + ConnectTimeout: 5 * time.Second, } // configDefault applies `ConfigDefault` values to a user‑supplied Config. @@ -78,5 +80,13 @@ func configDefault(config ...Config) Config { cfg.Expiration = 0 } + if cfg.MaxRetries == 0 { + cfg.MaxRetries = ConfigDefault.MaxRetries + } + + if cfg.ConnectTimeout == 0 { + cfg.ConnectTimeout = ConfigDefault.ConnectTimeout + } + return cfg }