move UseThenDisconnect into queueworker

This makes the tests pass while maintaining the same behavior.
This commit is contained in:
Jimmy Zelinskie 2015-09-21 13:34:12 -04:00
parent 98d6262a7f
commit 7c82e0b5b3
2 changed files with 19 additions and 17 deletions

View file

@ -8,8 +8,11 @@ from datetime import datetime, timedelta
from threading import Thread
from time import sleep
from app import app
from data.model import db
from data.queue import WorkQueue
from data.database import UseThenDisconnect
from workers.worker import Worker
@ -123,7 +126,8 @@ class QueueWorker(Worker):
logger.debug('No more work.')
def update_queue_metrics(self):
self._queue.update_metrics()
with UseThenDisconnect(app.config):
self._queue.update_metrics()
def mark_current_incomplete(self, restore_retry=False):
with self._current_item_lock: