Merge pull request #518 from jzelinskie/fixmysqlssl
move UseThenDisconnect into queueworker
This commit is contained in:
commit
2ff77df946
2 changed files with 19 additions and 17 deletions
|
@ -1,7 +1,6 @@
|
||||||
from datetime import datetime, timedelta
|
from datetime import datetime, timedelta
|
||||||
|
|
||||||
from app import app
|
from data.database import QueueItem, db, db_for_update
|
||||||
from data.database import QueueItem, db, db_for_update, UseThenDisconnect
|
|
||||||
from util.morecollections import AttrDict
|
from util.morecollections import AttrDict
|
||||||
|
|
||||||
|
|
||||||
|
@ -69,7 +68,6 @@ class WorkQueue(object):
|
||||||
return db_for_update(QueueItem.select().where(QueueItem.id == queue_id)).get()
|
return db_for_update(QueueItem.select().where(QueueItem.id == queue_id)).get()
|
||||||
|
|
||||||
def get_metrics(self, require_transaction=True):
|
def get_metrics(self, require_transaction=True):
|
||||||
with UseThenDisconnect(app.config):
|
|
||||||
guard = self._transaction_factory(db) if require_transaction else NoopWith()
|
guard = self._transaction_factory(db) if require_transaction else NoopWith()
|
||||||
with guard:
|
with guard:
|
||||||
now = datetime.utcnow()
|
now = datetime.utcnow()
|
||||||
|
|
|
@ -8,8 +8,11 @@ from datetime import datetime, timedelta
|
||||||
from threading import Thread
|
from threading import Thread
|
||||||
from time import sleep
|
from time import sleep
|
||||||
|
|
||||||
|
from app import app
|
||||||
|
|
||||||
from data.model import db
|
from data.model import db
|
||||||
from data.queue import WorkQueue
|
from data.queue import WorkQueue
|
||||||
|
from data.database import UseThenDisconnect
|
||||||
|
|
||||||
from workers.worker import Worker
|
from workers.worker import Worker
|
||||||
|
|
||||||
|
@ -123,6 +126,7 @@ class QueueWorker(Worker):
|
||||||
logger.debug('No more work.')
|
logger.debug('No more work.')
|
||||||
|
|
||||||
def update_queue_metrics(self):
|
def update_queue_metrics(self):
|
||||||
|
with UseThenDisconnect(app.config):
|
||||||
self._queue.update_metrics()
|
self._queue.update_metrics()
|
||||||
|
|
||||||
def mark_current_incomplete(self, restore_retry=False):
|
def mark_current_incomplete(self, restore_retry=False):
|
||||||
|
|
Reference in a new issue