src/redis-trib.rb

   1 #!/usr/bin/env ruby
   2
   3 require 'rubygems'
   4 require 'redis'
   5
   6 def xputs(s)
   7     printf s
   8     STDOUT.flush
   9 end
  10
  11 class ClusterNode
  12     def initialize(addr)
  13         s = addr.split(":")
  14         if s.length != 2
  15             puts "Invalid node name #{node}"
  16             exit 1
  17         end
  18         @host = s[0]
  19         @port = s[1]
  20     end
  21
  22     def to_s
  23         "#{@host}:#{@port}"
  24     end
  25
  26     def connect
  27         xputs "Connecting to node #{self}: "
  28         begin
  29             @r = Redis.new(:host => @ost, :port => @port)
  30             @r.ping
  31         rescue
  32             puts "ERROR"
  33             puts "Sorry, can't connect to node #{self}"
  34         end
  35         puts "OK"
  36     end
  37
  38     def assert_cluster
  39         info = @r.info
  40         if !info["cluster_enabled"] || info["cluster_enabled"].to_i == 0
  41             puts "Error: Node #{self} is not configured as a cluster node."
  42             exit 1
  43         end
  44     end
  45
  46     def r
  47         @r
  48     end
  49 end
  50
  51 class RedisTrib
  52     def check_arity(req_args, num_args)
  53         if ((req_args > 0 and num_args != req_args) ||
  54            (req_args < 0 and num_args < req_args.abs))
  55            puts "Wrong number of arguments for specified sub command"
  56            exit 1
  57         end
  58     end
  59
  60     def create_cluster
  61         puts "Creating cluster"
  62         ARGV[1..-1].each{|n|
  63             node = ClusterNode.new(n)
  64             node.connect
  65             node.assert_cluster
  66             # node.assert_empty
  67         }
  68     end
  69 end
  70
  71 COMMANDS={
  72     "create-cluster" => ["create_cluster", -2]
  73 }
  74
  75 # Sanity check
  76 if ARGV.length == 0
  77     puts "Usage: redis-trib <command> <arguments ...>"
  78     exit 1
  79 end
  80
  81 rt = RedisTrib.new
  82 cmd_spec = COMMANDS[ARGV[0].downcase]
  83 if !cmd_spec
  84     puts "Unknown redis-trib subcommand '#{ARGV[0]}'"
  85     exit 1
  86 end
  87 rt.check_arity(cmd_spec[1],ARGV.length)
  88
  89 # Dispatch
  90 rt.send(cmd_spec[0])