Skip to content

Commit 221f6c1

Browse files
authored
Merge pull request #107 from puppetlabs/fix-jobid-uniqueness
(maint) Add more uniqueness to jobid and useful termination message
2 parents 3f8417d + 671623b commit 221f6c1

File tree

1 file changed

+13
-8
lines changed

1 file changed

+13
-8
lines changed

lib/vmfloaty/abs.rb

Lines changed: 13 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -248,7 +248,7 @@ def self.retrieve(verbose, os_types, token, url, user, config, _ondemand = nil)
248248
conn = Http.get_conn(verbose, url)
249249
conn.headers['X-AUTH-TOKEN'] = token if token
250250

251-
saved_job_id = DateTime.now.strftime('%Q')
251+
saved_job_id = user + "-" + DateTime.now.strftime('%Q')
252252
vmpooler_config = Utils.get_vmpooler_service_config(config['vmpooler_fallback'])
253253
req_obj = {
254254
:resources => os_types,
@@ -284,15 +284,20 @@ def self.retrieve(verbose, os_types, token, url, user, config, _ondemand = nil)
284284

285285
validate_queue_status_response(res.status, res.body, "Initial request", verbose)
286286

287-
(1..retries).each do |i|
288-
queue_place, res_body = check_queue(conn, saved_job_id, req_obj, verbose)
289-
return translated(res_body, saved_job_id) if res_body
287+
begin
288+
(1..retries).each do |i|
289+
queue_place, res_body = check_queue(conn, saved_job_id, req_obj, verbose)
290+
return translated(res_body, saved_job_id) if res_body
290291

291-
sleep_seconds = 10 if i >= 10
292-
sleep_seconds = i if i < 10
293-
FloatyLogger.info "Waiting #{sleep_seconds} seconds to check if ABS request has been filled. Queue Position: #{queue_place}... (x#{i})"
292+
sleep_seconds = 10 if i >= 10
293+
sleep_seconds = i if i < 10
294+
FloatyLogger.info "Waiting #{sleep_seconds} seconds to check if ABS request has been filled. Queue Position: #{queue_place}... (x#{i})"
294295

295-
sleep(sleep_seconds)
296+
sleep(sleep_seconds)
297+
end
298+
rescue SystemExit, Interrupt
299+
FloatyLogger.info "\n\nFloaty interrupted, you can query the state of your request via\n1) `floaty query #{saved_job_id}` or delete it via\n2) `floaty delete #{saved_job_id}`"
300+
exit 1
296301
end
297302
nil
298303
end

0 commit comments

Comments
 (0)