异步asyncpg

# -*- coding: utf-8 -*-
# !/usr/bin/env python
# Software: PyCharm
# __author__ == "YU HAIPENG"
# fileName: asyncpostgres.py
# Month: 六月
# time: 2021/6/21 18:33
""" 官方文档 https://www.postgresql.org/docs/devel/functions-json.html """
from functools import wraps
from asyncpg import create_pool
from asyncio import Lock

create_pool_look = Lock()


def init_decorator():
    def decorator(func):
        @wraps(func)
        async def wrapper(self, *args, **kwargs):
            transaction = kwargs.pop("transaction", None) or self.transaction
            await self.create_pool(**self.params)
            async with self._pool.acquire(timeout=self.acquire_timeout) as con:
                if transaction:
                    isolation = kwargs.pop("isolation", None)
                    readonly = kwargs.pop("readonly", False)
                    deferrable = kwargs.pop("deferrable", False)
                    async with con.transaction(isolation=isolation, readonly=readonly, deferrable=deferrable):
                        n = await func(self, __conn__=con, *args, **kwargs)
                else:
                    n = await func(self, __conn__=con, *args, **kwargs)
            return n

        return wrapper

    return decorator


class AsyncPostGre:

    def __init__(
            self,
            dsn=None, *, min_size=10, max_size=10,
            max_queries=50000, max_inactive_connection_lifetime=300.0, setup=None, init=None,
            loop=None, host=None, port=None, user=None, password=None, command_timeout=360,
            database=None, transaction=False, acquire_timeout=360, **connect_kwargs):
        # dsn = "postgres://user:pass@host:port/database?option=value"
        """
        @param dsn:
        @param min_size:
        @param max_size:
        @param max_queries:
        @param max_inactive_connection_lifetime:
        @param setup:
        @param init:
        @param loop:
        @param host:
        @param port:
        @param user:
        @param password:
     
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值