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

Merge 22f0daa92b into ed22ed6e08

Browse source
This commit is contained in:
Branan Purvine-Riley 2014-02-03 21:31:22 -08:00
commit ede77883ff
1 changed files with 25 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

46
vmware-host-pooler
View file

@ -5,6 +5,8 @@ require 'redis'
require 'time'
require 'yaml'
require 'thread/pool'
$:.unshift(File.dirname(__FILE__))
require 'lib/logger'
require 'lib/require_relative'
@ -40,11 +42,11 @@ $vsphere = {}
# Our thread-tracker object
$threads = {}
$vcenter_pool = Thread.pool config[:config]['num_vmware_connections'] || 4
# Check the state of a VM
def check_pending_vm vm, pool, timeout
Thread.new {
$vcenter_pool.process {
host = $vsphere[pool].find_vms(vm)[vm]
if (host)
@ -78,7 +80,7 @@ def check_pending_vm vm, pool, timeout
end
def check_ready_vm vm, pool, ttl
Thread.new {
$vcenter_pool.process {
host = $vsphere[pool].find_vms(vm)[vm]
if (host)
@ -106,7 +108,7 @@ def check_ready_vm vm, pool, ttl
end
def check_running_vm vm, pool, ttl
Thread.new {
$vcenter_pool.process {
host = $vsphere[pool].find_vms(vm)[vm]
if (host)
@ -134,14 +136,21 @@ end
# Clone a VM
def clone_vm template, pool, folder, datastore
Thread.new {
vm = {}
vm = {}
if template =~ /\//
templatefolders = template.split('/')
vm['template'] = templatefolders.pop
end
if template =~ /\//
templatefolders = template.split('/')
vm['template'] = templatefolders.pop
end
# Generate a randomized hostname
o = [('a'..'z'),('0'..'9')].map{|r| r.to_a}.flatten
vm['hostname'] = o[rand(25)]+(0...14).map{o[rand(o.length)]}.join
# Add VM to Redis inventory ('pending' pool)
$redis.sadd('vmware_host_pool__pending__'+vm['template'], vm['hostname'])
$vcenter_pool.process {
if templatefolders
vm[vm['template']] = $vsphere[vm['template']].find_folder(templatefolders.join('/')).find(vm['template'])
else
@ -152,13 +161,6 @@ def clone_vm template, pool, folder, datastore
raise "Unable to find template '#{vm['template']}'!"
end
# Generate a randomized hostname
o = [('a'..'z'),('0'..'9')].map{|r| r.to_a}.flatten
vm['hostname'] = o[rand(25)]+(0...14).map{o[rand(o.length)]}.join
# Add VM to Redis inventory ('pending' pool)
$redis.sadd('vmware_host_pool__pending__'+vm['template'], vm['hostname'])
# Annotate with creation time, origin template, etc.
configSpec = RbVmomi::VIM.VirtualMachineConfigSpec(
:annotation =>
@ -210,10 +212,12 @@ end
# Destroy a VM
def destroy_vm vm, pool
Thread.new {
$redis.srem('vmware_host_pool__completed__'+pool, vm)
$redis.hdel('vmware_host_pool__active__'+pool, vm)
# Removing from redis before we kick off the asynchronous task
# prevents a nasty race condition that can lead to multiple delete
# tasks starting for the same VM.
$redis.srem('vmware_host_pool__completed__'+pool, vm)
$redis.hdel('vmware_host_pool__active__'+pool, vm)
$vcenter_pool.process {
host = $vsphere[pool].find_vms(vm)[vm]
if (host)