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

Support batch (multiple VM) requests

Browse source 
This PR allows a single API request to return multiple VM objects.  It
supports the following formats:

- POST <json> /vm/
  eg. 'curl -d '{"debian-7-i386":"3","debian-7-x86_64":"1"}' --url vmpooler/vm

- POST /vm/<pool>+<pool>+<pool>...
  eg. 'curl -d --url vmpooler/vm/debian-7-i386+debian-7-i386+debian-7-i386+debian-7-x86_64

Both commands listed above will return a JSON hash of 3 debian-7-i386 VMs and
1 debian-7-x86_64 VM:

  {
    "ok": true,
    "debian-7-i386": {
      "hostname": [
        "v49pwwk5yzg6oad",
        "ylghlgote5uso54",
        "wt0c4xovvulo7ge"
      ]
    },
    "debian-7-x86_64": {
      "hostname": "v3dkrulttp360fm"
    }
  }
This commit is contained in:
Scott Schneider 2014-06-09 15:15:05 -07:00
parent 61cdf3e2ad
commit d6b6f83fc7
1 changed files with 84 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

95
lib/vmpooler/api.rb
View file

@ -170,6 +170,54 @@ module Vmpooler
JSON.pretty_generate(result)
end
post '/vm/?' do
content_type :json
result = {}
available = 1
jdata = JSON.parse(request.body.read)
jdata.each do |template, count|
if ( $redis.scard('vmpooler__ready__'+template) < count.to_i )
available = 0
end
end
if ( available == 1 )
result['ok'] = true
jdata.each do |template, count|
result[template] ||= {}
count.to_i.times do |i|
vm = $redis.spop('vmpooler__ready__'+template)
unless (vm.nil?)
$redis.sadd('vmpooler__running__'+template, vm)
$redis.hset('vmpooler__active__'+template, vm, Time.now)
result[template] ||= {}
if ( result[template]['hostname'] )
result[template]['hostname'] = [result[template]['hostname']] if ! result[template]['hostname'].is_a?(Array)
result[template]['hostname'].push(vm)
else
result[template]['hostname'] = vm
end
else
result['ok'] = false
end
end
end
else
result['ok'] = false
end
JSON.pretty_generate(result)
end
get '/vm/:template/?' do
content_type :json
@ -184,22 +232,47 @@ module Vmpooler
content_type :json
result = {}
result[params[:template]] = {}
request = {}
if ( $redis.scard('vmpooler__ready__'+params[:template]) > 0 )
vm = $redis.spop('vmpooler__ready__'+params[:template])
params[:template].split('+').each do |template|
request[template] ||= 0
request[template] = request[template] + 1
end
unless (vm.nil?)
$redis.sadd('vmpooler__running__'+params[:template], vm)
$redis.hset('vmpooler__active__'+params[:template], vm, Time.now)
available = 1
result[params[:template]]['ok'] = true
result[params[:template]]['hostname'] = vm
else
result[params[:template]]['ok'] = false
request.keys.each do |template|
if ( $redis.scard('vmpooler__ready__'+template) < request[template] )
available = 0
end
end
if ( available == 1 )
result['ok'] = true
params[:template].split('+').each do |template|
result[template] ||= {}
vm = $redis.spop('vmpooler__ready__'+template)
unless (vm.nil?)
$redis.sadd('vmpooler__running__'+template, vm)
$redis.hset('vmpooler__active__'+template, vm, Time.now)
result[template] ||= {}
if ( result[template]['hostname'] )
result[template]['hostname'] = [result[template]['hostname']] if ! result[template]['hostname'].is_a?(Array)
result[template]['hostname'].push(vm)
else
result[template]['hostname'] = vm
end
else
result['ok'] = false
end
end
else
result[params[:template]]['ok'] = false
result['ok'] = false
end
JSON.pretty_generate(result)