mirror of
https://github.com/puppetlabs/vmpooler.git
synced 2026-01-26 01:58:41 -05:00
Revert "Consolidate scripts"
This reverts commit 8f18acfd2d91e1f35671f36552a669eb45754433.
This commit is contained in:
parent
485e6a0e4b
commit
bc20975fba
2 changed files with 98 additions and 76 deletions
96
vmware-host-pooler-api
Executable file
96
vmware-host-pooler-api
Executable file
|
|
@ -0,0 +1,96 @@
|
|||
#!/usr/bin/ruby
|
||||
|
||||
require 'json'
|
||||
require 'redis'
|
||||
require 'sinatra'
|
||||
require 'yaml'
|
||||
|
||||
$:.unshift(File.dirname(__FILE__))
|
||||
require 'lib/logger'
|
||||
require 'lib/require_relative'
|
||||
|
||||
logger = Logger.new
|
||||
|
||||
Dir.chdir(File.dirname(__FILE__))
|
||||
|
||||
# Load the configuration file
|
||||
config_file = File.expand_path('vmware-host-pooler.yaml')
|
||||
|
||||
pools = YAML.load_file(config_file)[:pools]
|
||||
|
||||
# Connect to Redis
|
||||
$redis = Redis.new
|
||||
|
||||
# Sinatra!
|
||||
get '/' do
|
||||
puts ''
|
||||
end
|
||||
|
||||
get '/status' do
|
||||
content_type :json
|
||||
|
||||
result = {}
|
||||
|
||||
pools.each do |pool|
|
||||
result[pool['name']] = {}
|
||||
result[pool['name']]['size'] = pool['size']
|
||||
[ 'pending', 'ready', 'completed' ].each do |queue|
|
||||
result[pool['name']][queue] = $redis.scard("vmware_host_pool__#{queue}__#{pool['name']}")
|
||||
end
|
||||
end
|
||||
|
||||
JSON.pretty_generate(result)
|
||||
end
|
||||
|
||||
get '/vm/:template' do
|
||||
content_type :json
|
||||
|
||||
result = {}
|
||||
result[params[:template]] = {}
|
||||
result[params[:template]]['hosts'] = $redis.smembers('vmware_host_pool__ready__'+params[:template])
|
||||
|
||||
JSON.pretty_generate(result)
|
||||
end
|
||||
|
||||
post '/vm/:template' do
|
||||
content_type :json
|
||||
|
||||
result = {}
|
||||
result[params[:template]] = {}
|
||||
|
||||
if ( ( ! params[:folder] ) or ( ! params[:pool] ))
|
||||
result[params[:template]]['error'] = 'You must specify a destination \'folder\' and \'pool\''
|
||||
else
|
||||
if ( $redis.scard('vmware_host_pool__ready__'+params[:template]) > 0 )
|
||||
vm = $redis.srandmember('vmware_host_pool__ready__'+params[:template])
|
||||
$redis.srem('vmware_host_pool__ready__'+params[:template], vm)
|
||||
$redis.sadd('vmware_host_pool__running__'+params[:template], vm)
|
||||
|
||||
logger.log('s', "[<] '#{vm}' moved to 'running' queue")
|
||||
|
||||
result[params[:template]]['ok'] = true
|
||||
result[params[:template]]['hostname'] = vm
|
||||
else
|
||||
result[params[:template]]['ok'] = false
|
||||
end
|
||||
end
|
||||
|
||||
JSON.pretty_generate(result)
|
||||
end
|
||||
|
||||
delete '/vm/:hostname' do
|
||||
content_type :json
|
||||
|
||||
result = {}
|
||||
|
||||
pools.each do |pool|
|
||||
if $redis.sismember('vmware_host_pool__running__'+pool['name'], params[:hostname])
|
||||
$redis.srem('vmware_host_pool__running__'+pool['name'], params[:hostname])
|
||||
$redis.sadd('vmware_host_pool__completed__'+pool['name'], params[:hostname])
|
||||
end
|
||||
end
|
||||
|
||||
result['ok'] = true
|
||||
JSON.pretty_generate(result)
|
||||
end
|
||||
|
||||
Loading…
Add table
Add a link
Reference in a new issue