redis-proxy 0.0.1

data/.gitignore

@@ -0,0 +1,2 @@
+*.swp
+*.swo

data/lib/redis-proxy.rb

@@ -0,0 +1,20 @@
+require 'eventmachine'
+require 'redis-proxy/redis_connection'
+require 'redis-proxy/redis_proxy_server'
+
+class RedisProxy
+
+  def self.start(redis_config, host = "0.0.0.0", port = 6379, options = {})
+    EventMachine.epoll
+    EventMachine.run do
+      puts "Starting up redis proxy server"
+      EventMachine.start_server host, port, RedisProxyServer, redis_config
+    end
+  end
+
+  def self.stop
+    puts "Terminating redis proxy server"
+    EventMachine.stop
+  end
+
+end

data/lib/redis-proxy/redis_connection.rb

@@ -0,0 +1,67 @@
+class RedisConnection < EventMachine::Connection
+
+  def initialize(config, client, request)
+    @@nodes = config[:nodes] || []
+    @@reconnect_limit = config[:reconnect_limit]
+    @@reconnect_delay = config[:reconnect_delay]
+    @node_id = 0;
+    @last_alive_node = 0;
+    @reconnect_count = 0;
+
+    @client = client
+    @request = request
+  end
+
+  def connection_completed
+    puts "RedisConnection: connected to redis node: #{@node_id}"
+    send_data "*1\r\n$4\r\nINFO\r\n"
+  end
+
+  def receive_data(data)
+    if data =~ /role:master/
+      puts "RedisConnection: connected to a master node, proxying"
+      @last_alive_node = @node_id
+      @reconnect_count = 0
+      EventMachine.enable_proxy self, @client
+
+      send_data @request
+    else
+      puts "RedisConnection: connected to a slave node, reconnecting"
+      close_connection
+    end
+  end
+
+  def proxy_target_unbound
+    puts "RedisConnection: proxy target unbound"
+    close_connection
+  end
+
+  def unbind
+    puts "RedisConnection: connection unbound"
+
+    EventMachine.disable_proxy self
+
+    unless @client_unbind
+      @node_id = (@node_id + 1) % @@nodes.count
+      @reconnect_count += 1
+
+      if @reconnect_count == @@reconnect_limit
+        @client.close_connection_after_writing
+      else
+        if @node_id == @last_alive_node
+          puts "RedisConnection: finished one reconnect cycle, sleeping before trying again"
+          sleep @@reconnect_delay
+        end
+        reconnect @@nodes[@node_id][:host], @@nodes[@node_id][:port]
+      end
+    end
+  end
+
+  def client_unbind
+    puts "RedisConnection: client unbind"
+    EventMachine.disable_proxy self
+    @client_unbind = true
+    close_connection
+  end
+
+end

data/lib/redis-proxy/redis_proxy_server.rb

@@ -0,0 +1,25 @@
+class RedisProxyServer < EventMachine::Connection
+
+  def initialize(redis_config)
+    @@redis_config = redis_config
+    @@default_node = redis_config[:nodes].first
+  end
+
+  def receive_data(data)
+
+    if @proxy.nil?
+      (@buffer ||= "") << data
+      if @buffer =~ /\r\n/
+        @proxy = EventMachine.connect @@default_node[:host], @@default_node[:port], RedisConnection, @@redis_config, self, data
+      end
+    else
+      @proxy.send_data data
+    end
+  end
+
+  def unbind
+    puts "RedisProxy: unbind"
+    @proxy.client_unbind
+  end
+
+end

data/redis-proxy.gemspec

@@ -0,0 +1,12 @@
+Gem::Specification.new do |s|
+  s.name     = "redis-proxy"
+  s.summary  = "A simple redis proxy that handles reconnection and sentinel failover"
+  s.version  = "0.0.1"
+  s.author   = "Nan Zhong"
+  s.email    = "nan@nine27.com"
+  s.homepage = 'https://github.com/nanzhong/redis-proxy'
+
+  s.files    = `git ls-files`.split($\)
+
+  s.add_dependency "eventmachine"
+end

metadata

@@ -0,0 +1,65 @@
+--- !ruby/object:Gem::Specification
+name: redis-proxy
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Nan Zhong
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-11-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: eventmachine
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: 
+email: nan@nine27.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- lib/redis-proxy.rb
+- lib/redis-proxy/redis_connection.rb
+- lib/redis-proxy/redis_proxy_server.rb
+- redis-proxy.gemspec
+homepage: https://github.com/nanzhong/redis-proxy
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: A simple redis proxy that handles reconnection and sentinel failover
+test_files: []