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 connection(): global pool if pool is None: pool = yield from create_pool(config.DSN) conn = yield from pool return conn