diff --git a/lib/redis_client/cluster_config.rb b/lib/redis_client/cluster_config.rb index fe33cfe..ca29a4d 100644 --- a/lib/redis_client/cluster_config.rb +++ b/lib/redis_client/cluster_config.rb @@ -69,10 +69,18 @@ def inspect "#<#{self.class.name} #{startup_nodes.values.map { |v| v.reject { |k| k == :command_builder } }}>" end + def connect_timeout + @client_config[:connect_timeout] || @client_config[:timeout] || ::RedisClient::Config::DEFAULT_TIMEOUT + end + def read_timeout @client_config[:read_timeout] || @client_config[:timeout] || ::RedisClient::Config::DEFAULT_TIMEOUT end + def write_timeout + @client_config[:write_timeout] || @client_config[:timeout] || ::RedisClient::Config::DEFAULT_TIMEOUT + end + def new_pool(size: 5, timeout: 5, **kwargs) @client_implementation.new( self, diff --git a/test/redis_client/test_cluster.rb b/test/redis_client/test_cluster.rb index 2b0b6e6..62f9a31 100644 --- a/test/redis_client/test_cluster.rb +++ b/test/redis_client/test_cluster.rb @@ -24,7 +24,9 @@ def teardown def test_config refute_nil @client.config + refute_nil @client.config.connect_timeout refute_nil @client.config.read_timeout + refute_nil @client.config.write_timeout end def test_inspect