Class: Hotdog::Commands::Down

Inherits:
BaseCommand show all
Defined in:
lib/hotdog/commands/down.rb

Constant Summary

Constants inherited from BaseCommand

BaseCommand::MASK_DATABASE, BaseCommand::MASK_QUERY, BaseCommand::PERSISTENT_DB

Instance Attribute Summary

Attributes inherited from BaseCommand

#application, #logger, #options

Instance Method Summary collapse

Methods inherited from BaseCommand

#execute, #fixed_string?, #initialize, #parse_options, #reload

Constructor Details

This class inherits a constructor from Hotdog::Commands::BaseCommand

Instance Method Details

#define_options(optparse, options = {}) ⇒ Object 



8
9
10
11
12
13
14
15
16
17
 
# File 'lib/hotdog/commands/down.rb', line 8

def define_options(optparse, options={})
  options[:downtime] = 86400
  options[:start] = Time.new
  optparse.on("--downtime DURATION") do |v|
    options[:downtime] = v.to_i
  end
  optparse.on("--start TIME") do |v|
    options[:start] = Time.parse(v)
  end
end

#run(args = [], options = {}) ⇒ Object 



19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
 
# File 'lib/hotdog/commands/down.rb', line 19

def run(args=[], options={})
  args.each do |arg|
    if arg.index(":").nil?
      scope = "host:#{arg}"
    else
      scope = arg
    end
    code, schedule = dog.schedule_downtime(scope, :start => options[:start].to_i, :end => (options[:start]+options[:downtime]).to_i)
    logger.debug("dog.schedule_donwtime(%s, :start => %s, :end => %s) #==> [%s, %s]" % [scope.inspect, options[:start].to_i, (options[:start]+options[:downtime]).to_i, code.inspect, schedule.inspect])
    if code.to_i / 100 != 2
      raise("dog.schedule_downtime(%s, ...) returns [%s, %s]" % [scope.inspect, code.inspect, schedule.inspect])
    end
  end

  # Remove persistent.db to schedule update on next invocation
  if @db
    close_db(@db)
    FileUtils.rm_f(File.join(options[:confdir], PERSISTENT_DB))
  end
end