asyncpg -连接到服务器并创建数据库(如果不存在

Question

有没有一种方法可以像在Hibernate中那样使用异步can在PostgreSQL服务器中创建数据库，使用：

jdbc:mysql://localhost:3306/db_name?createDatabaseIfNotExist=true

uri中的标志。有没有一种方法可以在不连接数据库的情况下连接到服务器并执行sql语句？

Answer 1

这在asyncpg中没有捷径，但在您的代码中很容易实现。每个PostgreSQL集群都有一个template1数据库，如果您要连接的原始数据库不存在，则可以使用该数据库发出CREATE DATABASE语句。CREATE DATABASE通常需要特殊权限，因此在连接到template1 (通常是postgres)时需要使用超级用户帐户。

示例：

import asyncio
import asyncpg

async def connect_create_if_not_exists(user, database):
    try:
        conn = await asyncpg.connect(user=user, database=database)
    except asyncpg.InvalidCatalogNameError:
        # Database does not exist, create it.
        sys_conn = await asyncpg.connect(
            database='template1',
            user='postgres'
        )
        await sys_conn.execute(
            f'CREATE DATABASE "{database}" OWNER "{user}"'
        )
        await sys_conn.close()

        # Connect to the newly created database.
        conn = await asyncpg.connect(user=user, database=database)

    return conn


asyncio.get_event_loop().run_until_complete(
    connect_create_if_not_exists(user='elvis', database='new-database')
)