vbatts/quay
Archived
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

close connections after getting queue metrics

Browse source
This commit is contained in:
Jimmy Zelinskie 2015-09-21 12:21:39 -04:00
parent 0316f6e471
commit 98d6262a7f
1 changed files with 16 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

4
data/queue.py
View file

@ -1,6 +1,7 @@
from datetime import datetime, timedelta from datetime import datetime, timedelta
from data.database import QueueItem, db, db_for_update from app import app
from data.database import QueueItem, db, db_for_update, UseThenDisconnect
from util.morecollections import AttrDict from util.morecollections import AttrDict
@ -68,6 +69,7 @@ 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()