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

Implement 'pending' and 'ready' queues

Browse source
This commit is contained in:
Scott Schneider 2013-10-01 15:40:24 -07:00
parent 15841eb8e5
commit ba56310415
1 changed files with 33 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

41
vmware-host-pooler
View file

@ -34,25 +34,50 @@ $redis = Redis.new
# Update loop
loop do
pools.each do |pool|
inventory = {}
total = 0
# Locate the resource pool
datacenter = $vim.serviceInstance.find_datacenter
base = vsphere_helper.find_pool(pool['pool'])
# Count the number of VMs and correlate Redis inventory
inventory = {}
# Make sure all VMs in resource pool are accounted-for
base.vm.each do |vm|
if ! $redis.sismember('vmware_host_pool-'+pool['name'], vm['name'])
$redis.sadd('vmware_host_pool-'+pool['name'], vm['name'])
if (
(! $redis.sismember('vmware_host_pool__ready__'+pool['name'], vm['name'])) and
(! $redis.sismember('vmware_host_pool__pending__'+pool['name'], vm['name']))
)
$redis.sadd('vmware_host_pool__pending__'+pool['name'], vm['name'])
end
inventory[vm['name']] = 1
total = total + 1
end
$redis.smembers('vmware_host_pool-'+pool['name']).each do |vm|
# Check pending pool
$redis.smembers('vmware_host_pool__pending__'+pool['name']).each do |vm|
if (
(vsphere_helper.find_vms(vm)[vm]) and
(vsphere_helper.find_vms(vm)[vm].summary.guest.toolsRunningStatus == 'guestToolsRunning') and
(vsphere_helper.find_vms(vm)[vm].summary.guest.ipAddress != nil)
)
begin
Socket.getaddrinfo(vm, nil)
rescue
break
end
$redis.sadd('vmware_host_pool__ready__'+pool['name'], vm)
$redis.srem('vmware_host_pool__pending__'+pool['name'], vm)
puts '[>] '+vm+' moved to \'ready\' pool'
end
end
# Check ready pool
$redis.smembers('vmware_host_pool__ready__'+pool['name']).each do |vm|
if ! inventory[vm]
$redis.srem('vmware_host_pool-'+pool['name'], vm)
$redis.srem('vmware_host_pool__ready__'+pool['name'], vm)
end
end
@ -105,8 +130,8 @@ loop do
:spec => spec
).wait_for_completion
# Add VM to Redis inventory
$redis.sadd('vmware_host_pool-'+pool['name'], vm['hostname'])
# Add VM to Redis inventory ('pending' pool)
$redis.sadd('vmware_host_pool__pending__'+pool['name'], vm['hostname'])
}
end