add sqlor context

This commit is contained in:
yumoqing 2020-05-06 11:48:26 +08:00
parent eec5507b5d
commit a18b9c381c
2 changed files with 9 additions and 1 deletions

View File

@ -2,6 +2,7 @@
import asyncio
from functools import wraps
import codecs
from contextlib import asynccontextmanager
from appPublic.myImport import myImport
from appPublic.dictObject import DictObject
@ -186,6 +187,14 @@ class DBPools:
async def freeSqlor(self,sor):
await self._releaseConn(sor.name,sor.conn,sor.cur)
@asynccontextmanager
async def sqlorContext(self,name):
sqlor = await self.getSqlor(name)
try:
yield sqlor
finally:
await self.freeSqlor(sqlor)
async def _aquireConn(self,dbname):
p = self._cpools.get(dbname)
if p == None:

View File

@ -455,5 +455,4 @@ class SQLor(object):
desc['validation'].append(idx)
return desc