Class: Flare::Tools::Cli::Balance
- Inherits:
-
SubCommand
- Object
- SubCommand
- Flare::Tools::Cli::Balance
- Defined in:
- lib/flare/tools/cli/balance.rb
Constant Summary
Constants included from IndexServerConfig
IndexServerConfig::Entity, IndexServerConfig::FLARE_INDEX_SERVER, IndexServerConfig::FLARE_INDEX_SERVERS
Constants included from Util::Constant
Util::Constant::DefalutBwlimit, Util::Constant::DefaultIndexServerName, Util::Constant::DefaultIndexServerPort, Util::Constant::DefaultNodePort, Util::Constant::DefaultTimeout, Util::Constant::STATUS_NG, Util::Constant::STATUS_OK
Constants included from Flare::Tools::Common
Flare::Tools::Common::NodeListFormat, Flare::Tools::Common::NodeListHeader
Constants inherited from SubCommand
SubCommand::S_NG, SubCommand::S_OK
Constants included from Util::Interruption
Util::Interruption::InterruptionTargets
Instance Attribute Summary
Attributes included from Option
Instance Method Summary collapse
- #execute(config, args) ⇒ Object
-
#initialize ⇒ Balance
constructor
A new instance of Balance.
- #setup ⇒ Object
Methods included from Util::Logging
#debug, #error, #fatal, #info, logger, #puts, set_logger, #trace, #warn
Methods included from Flare::Tools::Common
#address_of_hostname, #fetch_cluster, #hostname_of_address, #nodekey_of, #string_of_nodelist, #user_confirmed, #wait_for_master_construction, #wait_for_servers, #wait_for_slave_construction
Methods included from Util::Conversion
Methods inherited from SubCommand
desc, #execute_subcommand, myname, #myname, to_s, to_sym, usage
Methods included from Util::Interruption
included, #initialize_interruption, #interrupt, #interrupt_, interrupt_all, #interrupted?, #interruptible, #interruptible?
Methods included from Option
#option_init, #parse_options, #set_option_dry_run, #set_option_force, #set_option_global, #set_option_index_server
Constructor Details
#initialize ⇒ Balance
Returns a new instance of Balance.
34 35 36 37 |
# File 'lib/flare/tools/cli/balance.rb', line 34 def initialize super @force = false end |
Instance Method Details
#execute(config, args) ⇒ Object
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 |
# File 'lib/flare/tools/cli/balance.rb', line 39 def execute(config, args) parse_index_server(config, args) return S_NG if args.empty? hosts = args.map {|x| x.to_s.split(':')} hosts.each do |x| if x.size != 3 puts "invalid argument '#{x.join(':')}'." return S_NG end end Flare::Tools::IndexServer.open(config[:index_server_hostname], config[:index_server_port], @timeout) do |s| cluster = Flare::Tools::Cluster.new(s.host, s.port, s.stats_nodes) hosts.each do |hostname,port,balance| balance = balance.to_i nodekey = nodekey_of hostname, port ipaddr = address_of_hostname(hostname) unless cluster.has_nodekey? nodekey error "unknown host: #{nodekey}" return S_NG end node = cluster.node_stat(nodekey) exec = false if @force exec = true elsif node['balance'].to_i == balance STDERR.puts "no need to change the balance of #{ipaddr}:#{port}." else interruptible do STDERR.print "updating node balance (node=#{ipaddr}:#{port}, balance=#{node['balance']} -> #{balance}) (y/n): " exec = true if gets.chomp.upcase == "Y" end end if exec s.set_role(hostname, port.to_i, node['role'], balance, node['partition']) unless @dry_run end end STDOUT.puts string_of_nodelist(s.stats_nodes, hosts.map {|x| "#{x[0]}:#{x[1]}"}) end return S_OK end |
#setup ⇒ Object
27 28 29 30 31 32 |
# File 'lib/flare/tools/cli/balance.rb', line 27 def setup super set_option_index_server set_option_dry_run set_option_force end |