mirror of
https://github.com/OpenNebula/one.git
synced 2026-11-08 21:16:39 +09:00
131 lines
3.9 KiB
Ruby
Executable File
131 lines
3.9 KiB
Ruby
Executable File
#!/usr/bin/env ruby
|
||
|
||
# -------------------------------------------------------------------------- #
|
||
# Copyright 2002-2012, OpenNebula Project Leads (OpenNebula.org) #
|
||
# #
|
||
# Licensed under the Apache License, Version 2.0 (the "License"); you may #
|
||
# not use this file except in compliance with the License. You may obtain #
|
||
# a copy of the License at #
|
||
# #
|
||
# http://www.apache.org/licenses/LICENSE-2.0 #
|
||
# #
|
||
# Unless required by applicable law or agreed to in writing, software #
|
||
# distributed under the License is distributed on an "AS IS" BASIS, #
|
||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. #
|
||
# See the License for the specific language governing permissions and #
|
||
# limitations under the License. #
|
||
#--------------------------------------------------------------------------- #
|
||
|
||
####################################################
|
||
# Script to implement host failure tolerance
|
||
# It can be set to
|
||
# -r resubmit VMs running in the host
|
||
# -d delete VMs running in the host
|
||
# Additional flags
|
||
# -f force resubmission of suspended VMs
|
||
# -p <n> avoid resubmission if host comes
|
||
# back after n monitoring cycles
|
||
####################################################
|
||
|
||
ONE_LOCATION=ENV["ONE_LOCATION"]
|
||
|
||
if !ONE_LOCATION
|
||
RUBY_LIB_LOCATION="/usr/lib/one/ruby"
|
||
VMDIR="/var/lib/one"
|
||
CONFIG_FILE="/var/lib/one/config"
|
||
else
|
||
RUBY_LIB_LOCATION=ONE_LOCATION+"/lib/ruby"
|
||
VMDIR=ONE_LOCATION+"/var"
|
||
CONFIG_FILE=ONE_LOCATION+"/var/config"
|
||
end
|
||
|
||
$: << RUBY_LIB_LOCATION
|
||
|
||
#require 'OpenNebula'
|
||
#include OpenNebula
|
||
|
||
require 'getoptlong'
|
||
|
||
if !(host_id=ARGV[0])
|
||
exit -1
|
||
end
|
||
|
||
mode = "-r" # By default, resubmit VMs
|
||
force = "n" # By default, don't resubmit/finalize suspended VMs
|
||
repeat = nil # By default, don't wait fo monitorization cycles"
|
||
|
||
opts = GetoptLong.new(
|
||
['--delete', '-d',GetoptLong::NO_ARGUMENT],
|
||
['--resubmit', '-r',GetoptLong::NO_ARGUMENT],
|
||
['--force', '-f',GetoptLong::NO_ARGUMENT],
|
||
['--pause', '-p',GetoptLong::REQUIRED_ARGUMENT]
|
||
)
|
||
|
||
begin
|
||
opts.each do |opt, arg|
|
||
case opt
|
||
when '--delete'
|
||
mode="-d"
|
||
when '--resubmit'
|
||
mode="-r"
|
||
when '--force'
|
||
force = "y"
|
||
when '--pause'
|
||
repeat = arg.to_i
|
||
end
|
||
end
|
||
rescue Exception => e
|
||
exit(-1)
|
||
end
|
||
|
||
begin
|
||
client = Client.new()
|
||
rescue Exception => e
|
||
puts "Error: #{e}"
|
||
exit -1
|
||
end
|
||
|
||
# Retrieve hostname
|
||
host = OpenNebula::Host.new_with_id(host_id, client)
|
||
exit -1 if OpenNebula.is_error?(host)
|
||
host.info
|
||
host_name = host.name
|
||
|
||
if repeat
|
||
# Retrieve host monitor interval
|
||
monitor_interval = nil
|
||
File.read(CONFIG_FILE).each{|line|
|
||
monitor_interval = line.split("=").last.to_i if /HOST_MONITORING_INTERVAL/=~line
|
||
}
|
||
# Sleep through the desired number of monitor interval
|
||
sleep (repeat * monitor_interval)
|
||
|
||
# If the host came back, exit! avoid duplicated VMs
|
||
exit 0 if host.state != 3
|
||
end
|
||
|
||
# Loop through all vms
|
||
vms = VirtualMachinePool.new(client)
|
||
exit -1 if OpenNebula.is_error?(vms)
|
||
|
||
vms.info_all
|
||
|
||
state = "STATE=3"
|
||
state += " or STATE=5" if force == "y"
|
||
|
||
vm_ids_array = vms.retrieve_elements("/VM_POOL/VM[#{state}]/HISTORY_RECORDS/HISTORY[HOSTNAME=\"#{host_name}\" and last()]/../../ID")
|
||
|
||
if vm_ids_array
|
||
vm_ids_array.each do |vm_id|
|
||
vm=OpenNebula::VirtualMachine.new_with_id(vm_id, client)
|
||
vm.info
|
||
|
||
if mode == "-r"
|
||
vm.resubmit
|
||
elsif mode == "-d"
|
||
vm.finalize
|
||
end
|
||
end
|
||
end
|
||
|