Redis Sentinel Failover Testing Script

redis-failover.rb

require 'redis'
require 'redis-sentinel'
require 'timeout'
require 'logger'

class TestClient
  def initialize
    @sentinels = [
      { host: 'sentinel01.example.com', port: 26379 },
      { host: 'sentinel02.example.com', port: 26379 },
      { host: 'sentinel03.example.com', port: 26379 },
    ]

    @redis = Redis.new(
      master_name: 'your-master-name',
      sentinels: @sentinels,
      failover_reconnect_timeout: 60,
      logger: Logger.new(STDOUT)
    )
  end

  def test_connection
    input = Random.rand(10_000_000)
    output = nil

    @redis.set('foo', input)

    Timeout.timeout(0.1) do
      output = @redis.get('foo').to_i
    end

    return "ERROR: Incorrect response #{input} != #{output}" unless input == output
    return "Success (#{input} == #{output}) from #{@redis.id}"
  rescue Timeout::Error
    return 'ERROR: Timeout exceeded'
  end
end

test = TestClient.new

while true
  puts test.test_connection
end