1
2
3
4
5
6
7
8
9
10
11
12
13
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
|
#!/usr/bin/env ruby
require 'mcollective'
require 'pp'
oparser = MCollective::Optionparser.new({:verbose => true}, "filter")
options = oparser.parse{|parser, options|
parser.define_head "Control the mcollective daemon"
parser.banner = "Usage: mc-controller [options] command"
parser.separator ""
parser.separator "command can be one of:"
parser.separator ""
parser.separator " stats - retrieve statistics from the mcollectived"
parser.separator " reload_agents - reloads all agents"
parser.separator " reload_agent - reloads an agent, needs an argument"
parser.separator " exit - terminates the mcollectived"
parser.separator ""
parser.on('--arg', '--argument ARGUMENT', 'Argument to pass to agent') do |v|
options[:argument] = v
end
}
if ARGV.length > 0
command = ARGV.shift
else
STDERR.puts("Please specify a command and optional arguments")
exit 1
end
STDOUT.sync = true
received = 0
expected = Array.new
def printstats(id, stats)
printf("%30s> total=%d replies=%d valid=%d invalid=%d filtered=%d passed=%d\n", id, stats[:total], stats[:replies], stats[:validated],
stats[:unvalidated], stats[:filtered], stats[:passed])
end
# if we're asking for restarting an agent, make sure we have
# an agent name in argument and set appropriate filters
if command == "reload_agent"
unless options[:argument]
puts("Please specify an agent to relaod with --argument")
exit 1
end
options[:filter]["agent"] << options[:argument]
end
begin
client = MCollective::Client.new(options[:config])
client.options = options
command += " #{options[:argument]}" if options[:argument]
stats = client.discovered_req(command, "mcollective") do |resp|
next if resp == nil
case command
when "stats"
printstats(resp[:senderid], resp[:body][:stats])
when /reload_agent/
printf("%30s> %s\n", resp[:senderid], resp[:body])
else
if options[:verbose]
puts("#{resp[:senderid]}>")
pp resp[:body]
else
puts if c % 4 == 1
print("#{resp[:senderid]} ")
end
end
end
client.disconnect
rescue Exception => e
STDERR.puts "Could not call remote agent: #{e}"
exit 1
end
client.display_stats(stats, false, "mcollectived controller summary")
# vi:tabstop=4:expandtab:ai
|