Class: Beaker::Vmpooler
- Inherits:
-
Hypervisor
- Object
- Hypervisor
- Beaker::Vmpooler
- Defined in:
- lib/beaker/hypervisor/vmpooler.rb
Constant Summary collapse
- SSH_EXCEPTIONS =
[ SocketError, Timeout::Error, Errno::ETIMEDOUT, Errno::EHOSTDOWN, Errno::EHOSTUNREACH, Errno::ECONNREFUSED, Errno::ECONNRESET, Errno::ENETUNREACH, ]
Constants inherited from Hypervisor
Constants included from HostPrebuiltSteps
HostPrebuiltSteps::APT_CFG, HostPrebuiltSteps::CUMULUS_PACKAGES, HostPrebuiltSteps::DEBIAN_PACKAGES, HostPrebuiltSteps::ETC_HOSTS_PATH, HostPrebuiltSteps::ETC_HOSTS_PATH_SOLARIS, HostPrebuiltSteps::IPS_PKG_REPO, HostPrebuiltSteps::NTPSERVER, HostPrebuiltSteps::PSWINDOWS_PACKAGES, HostPrebuiltSteps::ROOT_KEYS_SCRIPT, HostPrebuiltSteps::ROOT_KEYS_SYNC_CMD, HostPrebuiltSteps::SLEEPWAIT, HostPrebuiltSteps::SLES_PACKAGES, HostPrebuiltSteps::TRIES, HostPrebuiltSteps::UNIX_PACKAGES, HostPrebuiltSteps::WINDOWS_PACKAGES
Instance Method Summary collapse
- #check_url(url) ⇒ Object
- #cleanup ⇒ Object
- #get_template_url(pooling_api, template) ⇒ Object
-
#initialize(vmpooler_hosts, options) ⇒ Vmpooler
constructor
A new instance of Vmpooler.
- #provision ⇒ Object
Methods inherited from Hypervisor
#configure, create, #generate_host_name, #proxy_package_manager, #validate
Methods included from HostPrebuiltSteps
#add_el_extras, #additive_hash_merge, #apt_get_update, #check_and_install_packages_if_needed, #construct_env, #copy_file_to_remote, #copy_ssh_to_root, #disable_iptables, #disable_se_linux, #echo_on_host, #enable_root_login, #epel_info_for, #get_domain_name, #get_ip, #hack_etc_hosts, #package_proxy, #proxy_config, #set_env, #set_etc_hosts, #sync_root_keys, #timesync, #validate_host
Methods included from DSL::Patterns
Constructor Details
#initialize(vmpooler_hosts, options) ⇒ Vmpooler
Returns a new instance of Vmpooler.
18 19 20 21 22 |
# File 'lib/beaker/hypervisor/vmpooler.rb', line 18 def initialize(vmpooler_hosts, ) @options = @logger = [:logger] @hosts = vmpooler_hosts end |
Instance Method Details
#check_url(url) ⇒ Object
24 25 26 27 28 29 30 31 |
# File 'lib/beaker/hypervisor/vmpooler.rb', line 24 def check_url url begin URI.parse(url) rescue return false end true end |
#cleanup ⇒ Object
111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 |
# File 'lib/beaker/hypervisor/vmpooler.rb', line 111 def cleanup vm_names = @hosts.map {|h| h['vmhostname'] }.compact if @hosts.length != vm_names.length @logger.warn "Some hosts did not have vmhostname set correctly! This likely means VM provisioning was not successful" end start = Time.now vm_names.each do |name| @logger.notify "Handing '#{name}' back to vmpooler for VM destruction" uri = URI.parse(get_template_url(@options['pooling_api'], name)) http = Net::HTTP.new( uri.host, uri.port ) request = Net::HTTP::Delete.new(uri.request_uri) begin response = http.request(request) rescue *SSH_EXCEPTIONS => e report_and_raise(@logger, e, 'Vmpooler.cleanup (http.request)') end end @logger.notify "Spent %.2f seconds cleaning up" % (Time.now - start) end |
#get_template_url(pooling_api, template) ⇒ Object
33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
# File 'lib/beaker/hypervisor/vmpooler.rb', line 33 def get_template_url pooling_api, template if not check_url(pooling_api) raise ArgumentError, "Invalid pooling_api URL: #{pooling_api}" end scheme = '' if not URI.parse(pooling_api).scheme scheme = 'http://' end #check that you have a valid uri template_url = scheme + pooling_api + '/vm/' + template if not check_url(template_url) raise ArgumentError, "Invalid full template URL: #{template_url}" end template_url end |
#provision ⇒ Object
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 99 100 101 102 103 104 105 106 107 108 109 |
# File 'lib/beaker/hypervisor/vmpooler.rb', line 49 def provision request_payload = {} start = Time.now @hosts.each_with_index do |h, i| if not h['template'] raise ArgumentError, "You must specify a template name for #{h}" end if h['template'] =~ /\// templatefolders = h['template'].split('/') h['template'] = templatefolders.pop end request_payload[h['template']] = (request_payload[h['template']].to_i + 1).to_s end @logger.notify "Requesting VM set from vmpooler" last_wait, wait = 0, 1 waited = 0 #the amount of time we've spent waiting for this host to provision begin uri = URI.parse(@options['pooling_api'] + '/vm/') http = Net::HTTP.new(uri.host, uri.port) request = Net::HTTP::Post.new(uri.request_uri) request.body = request_payload.to_json response = http.request(request) parsed_response = JSON.parse(response.body) if parsed_response['ok'] domain = parsed_response['domain'] @hosts.each_with_index do |h, i| if parsed_response[h['template']]['hostname'].is_a?(Array) hostname = parsed_response[h['template']]['hostname'].shift else hostname = parsed_response[h['template']]['hostname'] end h['vmhostname'] = domain ? "#{hostname}.#{domain}" : hostname @logger.notify "Using available host '#{h['vmhostname']}' (#{h.name})" end else raise "Vmpooler.provision - requested host set not available" end rescue JSON::ParserError, RuntimeError, *SSH_EXCEPTIONS => e if waited <= @options[:timeout].to_i @logger.debug("Retrying provision for vmpooler host after waiting #{wait} second(s) (failed with #{e.class})") sleep wait waited += wait last_wait, wait = wait, last_wait + wait retry end report_and_raise(@logger, e, 'Vmpooler.provision') end @logger.notify 'Spent %.2f seconds grabbing VMs' % (Time.now - start) end |