Module: RedisStat::Option

Defined in:
lib/redis-stat/option.rb

Constant Summary collapse

DEFAULT = 
{
  :hosts      => ['127.0.0.1:6379'],
  :interval   => 2,
  :count      => nil,
  :csv_file   => nil,
  :csv_output => false,
  :style      => :unicode
}

Class Method Summary collapse

Class Method Details

.parse(argv) ⇒ Object 



14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
 
# File 'lib/redis-stat/option.rb', line 14

def self.parse argv
  argv = argv.reject { |e| e == '-h' }

  options = DEFAULT.dup
  op = ::OptionParser.new { |opts|
    opts.banner = "usage: redis-stat [HOST[:PORT] ...] [INTERVAL [COUNT]]"
    opts.separator ''

    opts.on('-a', '--auth=PASSWORD', 'Password') do |v|
      options[:auth] = v
    end

    opts.on('-v', '--verbose', 'Show more info') do |v|
      options[:verbose] = v
    end

    opts.on('--style=STYLE', 'Output style: unicode|ascii') do |v|
      options[:style] = v.downcase.to_sym
    end

    opts.on('--no-color', 'Suppress ANSI color codes') do |v|
      options[:mono] = true
    end

    opts.on('--csv[=CSV_FILE]', 'Print or save the result in CSV') do |v|
      if v
        options[:csv_file] = v
      else
        options[:csv_output] = true
      end
    end

    opts.on('--es=ELASTICSEARCH_URL', 'Send results to ElasticSearch: [http://]HOST[:PORT][/INDEX]') do |v|
      options[:es] = RedisStat::ElasticsearchSink.parse_url v
    end

    opts.separator ''

    opts.on('--server[=PORT]', "Launch redis-stat web server (default port: #{RedisStat::DEFAULT_SERVER_PORT})") do |v|
      options[:server_port] = v || RedisStat::DEFAULT_SERVER_PORT
    end

    opts.on('--daemon', "Daemonize redis-stat. Must be used with --server option.") do |v|
      options[:daemon] = true
      if RUBY_PLATFORM == 'java'
        raise ArgumentError.new("Sorry. Daemonization is not supported in JRuby.")
      end
    end

    opts.separator ''

    opts.on('--version', 'Show version') do
      puts RedisStat::VERSION
      exit 0
    end

    opts.on_tail('--help', 'Show this message') do
      puts opts
      exit 0
    end
  }

  begin
    op.parse! argv

    is_number   = lambda { |str| str =~ /^([0-9]\.?[0-9]*)$|^([1-9][0-9]*)$/ }

    numbers, hosts = argv.partition { |e| is_number.call e }
    interval, count = numbers.map(&:to_f)

    options[:interval] = interval if interval
    options[:count]    = count if count
    options[:hosts]    = hosts unless hosts.empty?

    validate options

    return options
  rescue SystemExit => e
    exit e.status
  rescue Exception => e
    puts e.to_s
    puts op
    exit 1
  end
end

.validate(options) ⇒ Object 



100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
 
# File 'lib/redis-stat/option.rb', line 100

def self.validate options
  options[:interval].tap do |interval|
    unless interval.is_a?(Numeric) && interval > 0
      raise ArgumentError.new("Invalid interval: #{interval}")
    end
  end

  options[:count].tap do |count|
    unless count.nil? || (count.is_a?(Numeric) && count > 0)
      raise ArgumentError.new("Invalid count: #{count}")
    end
  end

  options[:hosts].tap do |hosts|
    if hosts.empty?
      raise ArgumentError.new("Redis host not given")
    end

    hosts.each do |host|
      host, port = host.split(':')
      if port
        port = port.to_i
        unless port > 0 && port < 65536
          raise ArgumentError.new("Invalid port: #{port}")
        end
      end
    end
  end

  options[:style].tap do |style|
    unless [:unicode, :ascii].include?(style)
      raise ArgumentError.new("Invalid style")
    end
  end

  if options[:daemon] && options[:server_port].nil?
    raise ArgumentError.new("--daemon option must be used with --server option")
  end
end