def test_select():
pool = yield from create_pool(dsn)
with (yield from pool) as conn:
cur = yield from conn.cursor()
yield from cur.execute('select 1')
ret = yield from cur.fetchone()
print(ret)
def execute(self, *args, **kwargs):
if self.pg_pool is None:
self.pg_pool = yield from create_pool(DB_DSN, loop=self.loop, minsize=2, maxsize=10)
with (yield from self.pg_pool) as conn:
cur = yield from conn.cursor()
yield from cur.execute(*args, **kwargs)
return cur
def __init__(self, loop=None):
logger.info('Worker initialising...')
loop = loop or asyncio.get_event_loop()
self.loop = loop
logger.debug('Connecting to db: "%s"', DB_DSN)
self._pool = loop.run_until_complete(create_pool(dsn=DB_DSN, loop=loop, minsize=2, maxsize=10))
self.wkh2p_sema = Semaphore(value=MAX_WORKER_THREADS, loop=loop)
self.worker_sema = Semaphore(value=MAX_WORKER_JOBS, loop=loop)
self.redis = None
self.exc_info = None
def get_connection(dsn):
pool = pools.get(dsn, None)
if pool is None:
pool = pools[dsn] = yield from create_pool(dsn)
conn = yield from pool
return conn
def get_connection(dsn):
pool = pools.get(dsn, None)
if pool is None:
pool = pools[dsn] = yield from create_pool(dsn)
conn = yield from pool
return conn
def connection():
global pool
if pool is None:
pool = yield from create_pool(config.DSN)
conn = yield from pool
return conn
