valitydev/yandex-tank
14
0
Fork 0
mirror of https://github.com/valitydev/yandex-tank.git synced 2024-11-06 18:35:18 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

retry posting killer tasks if queue full

Browse Source
This commit is contained in:
Alexey Lavrenuke 2015-09-11 16:58:43 +03:00
parent d1dd93269d
commit 34e882afa6
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
yandextank/plugins/bfg/worker.py
View File

@ -98,8 +98,18 @@ Gun: {gun.__class__.__name__}
self.log.info(
"Feeded all data. Publishing %d killer tasks" % (
workers_count))
[self.task_queue.put(None, timeout=1) for _ in xrange(
0, workers_count)]
retry_delay = 1
for _ in range(5):
try:
[self.task_queue.put(None, timeout=1) for _ in xrange(
0, workers_count)]
break
except Full:
self.log.debug(
"Couldn't post killer tasks"
" because queue is full. Retrying in %ss", retry_delay)
time.sleep(retry_delay)
retry_delay *= 2
try:
self.log.info("Waiting for workers")