puppetlabs/vmpooler
2
0
Fork 0
mirror of https://github.com/puppetlabs/vmpooler.git synced 2026-01-26 10:08:40 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

Each pool gets it's own vSphere object

Browse source
This commit is contained in:
Scott Schneider 2013-12-06 14:16:56 -08:00
parent e67d8a5d54
commit bac80108b2
1 changed files with 15 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

26
vmware-host-pooler
View file

@ -18,9 +18,8 @@ config = YAML.load_file(config_file)
pools = config[:pools] pools = config[:pools]
vsphere = config[:vsphere] vsphere = config[:vsphere]
# Load logger & vSphere helper libraries # Load logger library
$logger = Logger.new config[:config]['logfile'] $logger = Logger.new config[:config]['logfile']
$vsphere_helper = VsphereHelper.new
# Load Graphite helper library (if configured) # Load Graphite helper library (if configured)
if (defined? config[:config]['graphite']) if (defined? config[:config]['graphite'])
@ -31,6 +30,9 @@ end
# Connect to Redis # Connect to Redis
$redis = Redis.new $redis = Redis.new
# vSphere object
$vsphere = {}
# Our thread-tracker object # Our thread-tracker object
$threads = {} $threads = {}
@ -39,7 +41,7 @@ $threads = {}
# Check the state of a VM # Check the state of a VM
def check_pending_vm vm, pool, timeout def check_pending_vm vm, pool, timeout
Thread.new { Thread.new {
host = $vsphere_helper.find_vms(vm)[vm] host = $vsphere[pool].find_vms(vm)[vm]
if (host) if (host)
if ( if (
@ -73,7 +75,7 @@ end
def check_ready_vm vm, pool, ttl def check_ready_vm vm, pool, ttl
Thread.new { Thread.new {
host = $vsphere_helper.find_vms(vm)[vm] host = $vsphere[pool].find_vms(vm)[vm]
if (host) if (host)
if ( if (
@ -97,7 +99,7 @@ end
def check_running_vm vm, pool, ttl def check_running_vm vm, pool, ttl
Thread.new { Thread.new {
host = $vsphere_helper.find_vms(vm)[vm] host = $vsphere[pool].find_vms(vm)[vm]
if (host) if (host)
if ( if (
@ -133,7 +135,7 @@ def clone_vm template, pool, folder, datastore
end end
if templatefolders if templatefolders
vm[vm['template']] = $vsphere_helper.find_folder(templatefolders.join('/')).find(vm['template']) vm[vm['template']] = $vsphere[vm['template']].find_folder(templatefolders.join('/')).find(vm['template'])
else else
raise "Please provide a full path to the template" raise "Please provide a full path to the template"
end end
@ -158,8 +160,8 @@ def clone_vm template, pool, folder, datastore
# Put the VM in the specified folder and resource pool # Put the VM in the specified folder and resource pool
relocateSpec = RbVmomi::VIM.VirtualMachineRelocateSpec( relocateSpec = RbVmomi::VIM.VirtualMachineRelocateSpec(
:datastore => $vsphere_helper.find_datastore(datastore), :datastore => $vsphere[vm['template']].find_datastore(datastore),
:pool => $vsphere_helper.find_pool(pool), :pool => $vsphere[vm['template']].find_pool(pool),
:diskMoveType => :moveChildMostDiskBacking :diskMoveType => :moveChildMostDiskBacking
) )
@ -176,7 +178,7 @@ def clone_vm template, pool, folder, datastore
start = Time.now start = Time.now
vm[vm['template']].CloneVM_Task( vm[vm['template']].CloneVM_Task(
:folder => $vsphere_helper.find_folder(folder), :folder => $vsphere[vm['template']].find_folder(folder),
:name => vm['hostname'], :name => vm['hostname'],
:spec => spec :spec => spec
).wait_for_completion ).wait_for_completion
@ -193,7 +195,7 @@ def destroy_vm vm, pool
Thread.new { Thread.new {
$redis.srem('vmware_host_pool__completed__'+pool, vm) $redis.srem('vmware_host_pool__completed__'+pool, vm)
host = $vsphere_helper.find_vms(vm)[vm] host = $vsphere[pool].find_vms(vm)[vm]
if (host) if (host)
start = Time.now start = Time.now
@ -223,10 +225,12 @@ def check_pool pool
$redis.del('vmware_host_pool__pending__'+pool['name']) $redis.del('vmware_host_pool__pending__'+pool['name'])
$threads[pool['name']] = Thread.new { $threads[pool['name']] = Thread.new {
$vsphere[pool['name']] ||= VsphereHelper.new
loop do loop do
# INVENTORY # INVENTORY
inventory = {} inventory = {}
base = $vsphere_helper.find_pool(pool['pool']) base = $vsphere[pool['name']].find_pool(pool['pool'])
base.vm.each do |vm| base.vm.each do |vm|
if ( if (