class Pooler

Public Class Methods

check_ondemandvm(verbose, request_id, url) click to toggle source
# File lib/vmfloaty/pooler.rb, line 71
def self.check_ondemandvm(verbose, request_id, url)
  conn = Http.get_conn(verbose, url)

  response = conn.get "ondemandvm/#{request_id}"
  res_body = JSON.parse(response.body)
  return res_body if response.status == 200

  return false if response.status == 202

  raise "HTTP #{response.status}: The request cannot be found, or an unknown error occurred" if response.status == 404

  false
end
delete(verbose, url, hosts, token, _user) click to toggle source
# File lib/vmfloaty/pooler.rb, line 128
def self.delete(verbose, url, hosts, token, _user)
  raise TokenError, 'Token provided was nil. Request cannot be made to delete vm' if token.nil?

  conn = Http.get_conn(verbose, url)

  conn.headers['X-AUTH-TOKEN'] = token if token

  response_body = {}

  hosts.each do |host|
    response = conn.delete "vm/#{host}"
    res_body = JSON.parse(response.body)
    response_body[host] = res_body
  end

  response_body
end
disk(verbose, url, hostname, token, disk) click to toggle source
# File lib/vmfloaty/pooler.rb, line 117
def self.disk(verbose, url, hostname, token, disk)
  raise TokenError, 'Token provided was nil. Request cannot be made to modify vm' if token.nil?

  conn = Http.get_conn(verbose, url)
  conn.headers['X-AUTH-TOKEN'] = token

  response = conn.post "vm/#{hostname}/disk/#{disk}"

  JSON.parse(response.body)
end
list(verbose, url, os_filter = nil) click to toggle source
# File lib/vmfloaty/pooler.rb, line 9
def self.list(verbose, url, os_filter = nil)
  conn = Http.get_conn(verbose, url)

  response = conn.get 'vm'
  response_body = JSON.parse(response.body)

  if os_filter
    response_body.select { |i| i[/#{os_filter}/] }
  else
    response_body
  end
end
list_active(verbose, url, token, _user) click to toggle source
# File lib/vmfloaty/pooler.rb, line 22
def self.list_active(verbose, url, token, _user)
  status = Auth.token_status(verbose, url, token)
  vms = []
  vms = status[token]['vms']['running'] if status[token] && status[token]['vms']
  vms
end
modify(verbose, url, hostname, token, modify_hash) click to toggle source
# File lib/vmfloaty/pooler.rb, line 85
def self.modify(verbose, url, hostname, token, modify_hash)
  raise TokenError, 'Token provided was nil. Request cannot be made to modify vm' if token.nil?

  modify_hash.each_key do |key|
    raise ModifyError, "Configured service type does not support modification of #{key}." unless %i[tags lifetime
                                                                                                    disk].include? key
  end

  conn = Http.get_conn(verbose, url)
  conn.headers['X-AUTH-TOKEN'] = token

  if modify_hash['disk']
    disk(verbose, url, hostname, token, modify_hash['disk'])
    modify_hash.delete 'disk'
  end

  response = conn.put do |req|
    req.url "vm/#{hostname}"
    req.body = modify_hash.to_json
  end

  res_body = JSON.parse(response.body)

  if res_body['ok']
    res_body
  elsif response.status == 401
    raise AuthError, "HTTP #{response.status}: The token provided could not authenticate to the pooler.\n#{res_body}"
  else
    raise ModifyError, "HTTP #{response.status}: Failed to modify VMs from the pooler vm/#{hostname}. #{res_body}"
  end
end
query(verbose, url, hostname) click to toggle source
# File lib/vmfloaty/pooler.rb, line 160
def self.query(verbose, url, hostname)
  conn = Http.get_conn(verbose, url)

  response = conn.get "vm/#{hostname}"
  JSON.parse(response.body)
end
retrieve(verbose, os_type, token, url, _user, _options, ondemand = nil, _continue = nil) click to toggle source
# File lib/vmfloaty/pooler.rb, line 29
def self.retrieve(verbose, os_type, token, url, _user, _options, ondemand = nil, _continue = nil)
  # NOTE:
  #   Developers can use `Utils.generate_os_hash` to
  #   generate the os_type param.
  conn = Http.get_conn(verbose, url)
  conn.headers['X-AUTH-TOKEN'] = token if token

  os_string = os_type.map { |os, num| Array(os) * num }.flatten.join('+')
  raise MissingParamError, 'No operating systems provided to obtain.' if os_string.empty?

  response = conn.post "vm/#{os_string}" unless ondemand
  response ||= conn.post "ondemandvm/#{os_string}"

  res_body = JSON.parse(response.body)

  if res_body['ok']
    res_body
  elsif response.status == 401
    raise AuthError, "HTTP #{response.status}: The token provided could not authenticate to the pooler.\n#{res_body}"
  elsif response.status == 403
    raise "HTTP #{response.status}: Failed to obtain VMs from the pooler at #{url}/vm/#{os_string}. Request exceeds the configured per pool maximum. #{res_body}"
  else
    unless ondemand
      raise "HTTP #{response.status}: Failed to obtain VMs from the pooler at #{url}/vm/#{os_string}. #{res_body}"
    end

    raise "HTTP #{response.status}: Failed to obtain VMs from the pooler at #{url}/ondemandvm/#{os_string}. #{res_body}"
  end
end
revert(verbose, url, hostname, token, snapshot_sha) click to toggle source
# File lib/vmfloaty/pooler.rb, line 177
def self.revert(verbose, url, hostname, token, snapshot_sha)
  raise TokenError, 'Token provided was nil. Request cannot be made to revert vm' if token.nil?

  conn = Http.get_conn(verbose, url)
  conn.headers['X-AUTH-TOKEN'] = token

  raise "Snapshot SHA provided was nil, could not revert #{hostname}" if snapshot_sha.nil?

  response = conn.post "vm/#{hostname}/snapshot/#{snapshot_sha}"
  JSON.parse(response.body)
end
snapshot(verbose, url, hostname, token) click to toggle source
# File lib/vmfloaty/pooler.rb, line 167
def self.snapshot(verbose, url, hostname, token)
  raise TokenError, 'Token provided was nil. Request cannot be made to snapshot vm' if token.nil?

  conn = Http.get_conn(verbose, url)
  conn.headers['X-AUTH-TOKEN'] = token

  response = conn.post "vm/#{hostname}/snapshot"
  JSON.parse(response.body)
end
status(verbose, url) click to toggle source
# File lib/vmfloaty/pooler.rb, line 146
def self.status(verbose, url)
  conn = Http.get_conn(verbose, url)

  response = conn.get 'status'
  JSON.parse(response.body)
end
summary(verbose, url) click to toggle source
# File lib/vmfloaty/pooler.rb, line 153
def self.summary(verbose, url)
  conn = Http.get_conn(verbose, url)

  response = conn.get 'summary'
  JSON.parse(response.body)
end
wait_for_request(verbose, request_id, url, timeout = 300) click to toggle source
# File lib/vmfloaty/pooler.rb, line 59
def self.wait_for_request(verbose, request_id, url, timeout = 300)
  start_time = Time.now
  while check_ondemandvm(verbose, request_id, url) == false
    return false if (Time.now - start_time).to_i > timeout

    FloatyLogger.info "waiting for request #{request_id} to be fulfilled"
    sleep 5
  end
  FloatyLogger.info 'The request has been fulfilled'
  check_ondemandvm(verbose, request_id, url)
end