| 1 : |
agomez |
1 |
require 'rubygems'
|
| 2 : |
|
|
require 'restfully'
|
| 3 : |
|
|
require 'restfully/addons/bonfire'
|
| 4 : |
|
|
|
| 5 : |
|
|
##Editable parameters
|
| 6 : |
|
|
SIZES = "1,2" #Comma separated list of volume sizes in GB
|
| 7 : |
|
|
#LOCATION = "de-hlrs"
|
| 8 : |
|
|
LOCATION = "fr-inria"
|
| 9 : |
|
|
#LOCATION = "uk-epcc"
|
| 10 : |
|
|
##
|
| 11 : |
|
|
|
| 12 : |
|
|
EXPERIMENT_NAME = "Build storages"
|
| 13 : |
|
|
EXPERIMENT_DESCRIPTION = "Build storages test"
|
| 14 : |
|
|
EXPERIMENT_WALLTIME = 2*3600
|
| 15 : |
|
|
MASTER_IMAGE_NAME = "BonFIRE Debian Squeeze 2G v3"
|
| 16 : |
|
|
WAN_NAME = "BonFIRE WAN"
|
| 17 : |
|
|
|
| 18 : |
|
|
session = Restfully::Session.new(
|
| 19 : |
|
|
:configuration_file => "~/.restfully/api.bonfire-project.eu",
|
| 20 : |
|
|
:cache => false,
|
| 21 : |
|
|
:gateway => "ssh.fr-inria.bonfire-project.eu",
|
| 22 : |
|
|
:keys => ["~/.ssh/id_rsa"]
|
| 23 : |
|
|
)
|
| 24 : |
|
|
session.logger.level = Logger::INFO
|
| 25 : |
|
|
#session.logger.level = Logger::DEBUG
|
| 26 : |
|
|
experiment = nil
|
| 27 : |
|
|
volumes = []
|
| 28 : |
|
|
begin
|
| 29 : |
|
|
session.logger.info "Deploying experiment..."
|
| 30 : |
|
|
experiment = session.root.experiments.submit(
|
| 31 : |
|
|
:name => EXPERIMENT_NAME,
|
| 32 : |
|
|
:description => EXPERIMENT_DESCRIPTION + " - " + Time.now.to_s,
|
| 33 : |
|
|
# :status => "waiting",
|
| 34 : |
|
|
:walltime => EXPERIMENT_WALLTIME
|
| 35 : |
|
|
)
|
| 36 : |
|
|
|
| 37 : |
|
|
location = session.root.locations[:"#{LOCATION}"]
|
| 38 : |
|
|
session.logger.info "Chosen location is: #{location['name']}"
|
| 39 : |
|
|
fail "Can't select the machine location" if location.nil?
|
| 40 : |
|
|
|
| 41 : |
|
|
def create_volume(session, location, size)
|
| 42 : |
|
|
session.logger.info "Deploying volume on #{location['name']}..."
|
| 43 : |
|
|
return location.storages.find{|storage|
|
| 44 : |
|
|
storage['name'] == "volume-#{size}GB-empty"
|
| 45 : |
|
|
} || location.storages.submit(
|
| 46 : |
|
|
:name => "volume-#{size}GB-empty",
|
| 47 : |
|
|
:size => size*1024,
|
| 48 : |
|
|
:type => "DATABLOCK",
|
| 49 : |
|
|
:group => "private",
|
| 50 : |
|
|
:fstype => "ext3",
|
| 51 : |
|
|
:public => "NO",
|
| 52 : |
|
|
:persistent => "YES"
|
| 53 : |
|
|
)
|
| 54 : |
|
|
end
|
| 55 : |
|
|
|
| 56 : |
|
|
SIZES.split(',').each{|size| volumes.push(create_volume(session,location,size.to_i))}
|
| 57 : |
|
|
volumes.each{|storage|
|
| 58 : |
|
|
storage.update(:persistent => "YES")
|
| 59 : |
|
|
storage.reload}
|
| 60 : |
|
|
|
| 61 : |
|
|
session.logger.info "Launching machine..."
|
| 62 : |
|
|
|
| 63 : |
|
|
disks = [{:storage => location.storages.find{|s| s['name'] == MASTER_IMAGE_NAME },:type => "OS"}]
|
| 64 : |
|
|
volumes.each{|storage| disks.push({:storage => storage,:type => "DATABLOCK"})}
|
| 65 : |
|
|
#pp disks
|
| 66 : |
|
|
machine = experiment.computes.submit(
|
| 67 : |
|
|
:name => "machine-experiment#{experiment['id']}",
|
| 68 : |
|
|
:instance_type => "small",
|
| 69 : |
|
|
:disk => disks,
|
| 70 : |
|
|
:nic => [
|
| 71 : |
|
|
{:network => location.networks.find{|n| n['name'] == WAN_NAME}}
|
| 72 : |
|
|
],
|
| 73 : |
|
|
:location => location,
|
| 74 : |
|
|
:context => {}
|
| 75 : |
|
|
)
|
| 76 : |
|
|
hostname = "#{machine['name']}-#{machine['id']}"
|
| 77 : |
|
|
session.logger.info "Machine: #{hostname} #{machine['nic'][0]['ip']}"
|
| 78 : |
|
|
|
| 79 : |
|
|
session.logger.info "Starting experiment..."
|
| 80 : |
|
|
experiment.update(:status => "running")
|
| 81 : |
|
|
|
| 82 : |
|
|
session.logger.info "Checking if machine is ready..."
|
| 83 : |
|
|
until [machine].all?{|vm|
|
| 84 : |
|
|
vm.reload['state'] == 'RUNNING' && vm.ssh.accessible?
|
| 85 : |
|
|
} do
|
| 86 : |
|
|
fail "Machine has failed" if [machine].any?{|vm| vm['state'] == 'FAILED'}
|
| 87 : |
|
|
session.logger.info "Machine is not ready. Waiting..."
|
| 88 : |
|
|
sleep 20
|
| 89 : |
|
|
end
|
| 90 : |
|
|
|
| 91 : |
|
|
session.logger.info "Machine is ready"
|
| 92 : |
|
|
#Fill volumes with zeros
|
| 93 : |
|
|
machine.ssh do |ssh|
|
| 94 : |
|
|
('e'..'i').to_a[0,volumes.size].each do |letter|
|
| 95 : |
|
|
session.logger.info "Formating device xvd#{letter}"
|
| 96 : |
|
|
output = ssh.exec!("blockdev --getsize64 /dev/xvd#{letter}")
|
| 97 : |
|
|
session.logger.info output unless output.nil?
|
| 98 : |
|
|
output = ssh.exec!("dd if=/dev/zero of=/dev/xvd#{letter} bs=1M")
|
| 99 : |
|
|
session.logger.info output unless output.nil?
|
| 100 : |
|
|
end
|
| 101 : |
|
|
end
|
| 102 : |
|
|
session.logger.info "Devices formatted!"
|
| 103 : |
|
|
|
| 104 : |
|
|
machine.update(:state => 'SHUTDOWN')
|
| 105 : |
|
|
|
| 106 : |
|
|
until machine.reload['state'] == 'DONE' do
|
| 107 : |
|
|
session.logger.info "Machine is #{machine['state']}."
|
| 108 : |
|
|
sleep 20
|
| 109 : |
|
|
end
|
| 110 : |
|
|
|
| 111 : |
|
|
session.logger.info "Disable persistence of disks"
|
| 112 : |
|
|
volumes.each{|storage| storage.update(:persistent => "NO") }
|
| 113 : |
|
|
|
| 114 : |
|
|
session.logger.info "Deleting experiment"
|
| 115 : |
|
|
experiment.delete
|
| 116 : |
|
|
|
| 117 : |
|
|
session.logger.info "Volumes available:"
|
| 118 : |
|
|
volumes.each{|storage| session.logger.info storage['name']}
|
| 119 : |
|
|
session.logger.warn "Experiment terminated!"
|
| 120 : |
|
|
|
| 121 : |
|
|
rescue Exception => e
|
| 122 : |
|
|
session.logger.error "#{e.class.name}: #{e.message}"
|
| 123 : |
|
|
session.logger.error e.backtrace.join("\n")
|
| 124 : |
|
|
|
| 125 : |
|
|
session.logger.warn "Cleaning up in 30 seconds. Hit CTRL-C now to keep your VMs..."
|
| 126 : |
|
|
sleep 30
|
| 127 : |
|
|
experiment.delete unless experiment.nil?
|
| 128 : |
|
|
sleep 30
|
| 129 : |
|
|
volumes.each{|storage| storage.update(:persistent => "NO") }
|
| 130 : |
|
|
volumes.each{|storage| storage.delete}
|
| 131 : |
|
|
end
|
Parent Directory
Revision Log