ruby library client is not Redis-rb merged with RubyRedis "engine" by Brian McKinney

[redis.git] / client-libraries / ruby / lib / dist_redis.rb

require 'redis'
require 'hash_ring'
class DistRedis
  attr_reader :ring
  def initialize(opts={})
    hosts = []
    
    db = opts[:db] || nil
    timeout = opts[:timeout] || nil 

    raise Error, "No hosts given" unless opts[:hosts]

    opts[:hosts].each do |h|
      host, port = h.split(':')
      hosts << Redis.new(:host => host, :port => port, :db => db, :timeout => timeout, :db => db)
    end

    @ring = HashRing.new hosts 
  end
  
  def node_for_key(key)
    if key =~ /\{(.*)?\}/
      key = $1
    end
    @ring.get_node(key)
  end
  
  def add_server(server)
    server, port = server.split(':')
    @ring.add_node Redis.new(:host => server, :port => port)
  end
  
  def method_missing(sym, *args, &blk)
    if redis = node_for_key(args.first.to_s)
      redis.send sym, *args, &blk
    else
      super
    end
  end
  
  def keys(glob)
    keyz = []
    @ring.nodes.each do |red|
      keyz.concat red.keys(glob)
    end
    keyz
  end
  
  def save
    @ring.nodes.each do |red|
      red.save
    end
  end
  
  def bgsave
    @ring.nodes.each do |red|
      red.bgsave
    end
  end
  
  def quit
    @ring.nodes.each do |red|
      red.quit
    end
  end
  
  def delete_cloud!
    @ring.nodes.each do |red|
      red.keys("*").each do |key|
        red.delete key
      end  
    end
  end
  
end


if __FILE__ == $0

r = DistRedis.new 'localhost:6379', 'localhost:6380', 'localhost:6381', 'localhost:6382'
  r['urmom'] = 'urmom'
  r['urdad'] = 'urdad'
  r['urmom1'] = 'urmom1'
  r['urdad1'] = 'urdad1'
  r['urmom2'] = 'urmom2'
  r['urdad2'] = 'urdad2'
  r['urmom3'] = 'urmom3'
  r['urdad3'] = 'urdad3'
  p r['urmom']
  p r['urdad']
  p r['urmom1']
  p r['urdad1']
  p r['urmom2']
  p r['urdad2']
  p r['urmom3']
  p r['urdad3']
  
  r.push_tail 'listor', 'foo1'
  r.push_tail 'listor', 'foo2'
  r.push_tail 'listor', 'foo3'
  r.push_tail 'listor', 'foo4'
  r.push_tail 'listor', 'foo5'
  
  p r.pop_tail('listor')
  p r.pop_tail('listor')
  p r.pop_tail('listor')
  p r.pop_tail('listor')
  p r.pop_tail('listor')
  
  puts "key distribution:"
  
  r.ring.nodes.each do |red|
    p [red.port, red.keys("*")]
  end
  r.delete_cloud!
  p r.keys('*')

end