在 double with 语句中使用 psycopg2 执行多个 SQL 查询

Question

是否可以使用 psycopg2 在双

with

游标打开语句中传递多个查询（首先打开连接，然后游标）？

例如替换：

import psycopg2

def connector():
    return psycopg2.connect(**DB_DICT_PARAMS)

########

sql_update1 = ("UPDATE table SET array = %s::varchar[], "
               "array_created = true, timestamp = now() AT TIME ZONE 'UTC' "
               "WHERE id = %s")

sql_update2 = ("UPDATE table SET json_field = %s "
               "WHERE id = %s")

with connector() as conn:
    with conn.cursor() as curs:
        curs.execute(sql_update1, [stringArray, ID])

with connector() as conn:
    with conn.cursor() as curs:
        curs.execute(sql_update2, [jsonString, ID])

作者：

#(...)

sql_update1 = ("UPDATE table SET array = %s::varchar[], "
               "array_created = true, timestamp = now() AT TIME ZONE 'UTC' "
               "WHERE id = %s")

sql_update2 = ("UPDATE table SET json_field = %s "
               "WHERE id = %s")

with connector() as conn:
    with conn.cursor() as curs:
        curs.execute(sql_update1, [stringArray, ID])
        curs.execute(sql_update2, [jsonString, ID])

如果第二个查询需要第一个查询之前完成怎么办，如果不需要怎么办？
在所示的情况下，他们肯定会更新数据库中的相同记录（即行），但不会更新相同的字段（即属性或列）。

这是精确授权的，因为两个 SQL 语句是按顺序提交的，即第一个先完成。然后，之后且仅在之后，第二个被执行。？

或者它实际上是被禁止的，因为它们可以并行执行，每个查询在任何时刻 t 都不知道另一个查询的状态？

数据库中没有花哨的触发器或过程。我们先简单点吧。
（请注意，我故意在这里编写了两个查询，其中一个查询就完全适合，但情况并非总是如此，因为在将其他一些结果保存到数据库中的同一记录之前，某些计算仍在进行中）。

Answer 1

如果您希望它们同时执行，只需将它们放在同一个字符串中，并用分号分隔即可。我有点生疏，但我认为以下应该有效：

sql_updates = ("UPDATE table SET array = %s::varchar[], "
               "array_created = true, timestamp = now() AT TIME ZONE 'UTC' "
               "WHERE id = %s;"
               "UPDATE table SET json_field = %s "
               "WHERE id = %s;")

with connector() as conn:
    with conn.cursor() as curs:
        curs.execute(sql_updates, [stringArray, ID, jsonString, ID])

Answer 2

最好避免这种情况：

with connector() as conn:
    with conn.cursor() as curs:
        curs.execute(sql_update1, [stringArray, ID])

with connector() as conn:
    with conn.cursor() as curs:
        curs.execute(sql_update2, [jsonString, ID])

与执行查询相比，打开数据库连接相当慢，因此重用它比为每个查询打开一个新连接要好得多。如果您的程序是脚本，通常您只需在启动时打开连接并在退出时关闭它。

但是，如果您的程序在查询之间等待很长时间，并且有许多实例在运行，那么最好关闭连接，以免在 postgres 服务器上无所事事地消耗宝贵的 RAM。这在客户端/服务器应用程序中很常见，其中客户端主要等待用户输入。如果有很多客户端，您还可以使用连接池，它提供了两全其美的优点，但代价是增加了一些额外的复杂性。但如果这只是一个脚本，则无需为此烦恼。

with connector() as conn:
    with conn.cursor() as curs:
        curs.execute(sql_update1, [stringArray, ID])
        curs.execute(sql_update2, [jsonString, ID])

这样会更快。您不需要构建新的游标，可以重复使用同一个游标。请注意，如果您在重用游标之前不获取第一个查询的结果，那么在执行第二个查询后您将无法执行此操作，因为游标仅存储最后一个查询的结果。由于这些是更新，因此没有结果，除非您想检查行计数以查看它是否确实更新了行。

如果第二个查询需要第一个查询之前完成怎么办，如果不需要怎么办？

不在乎。 execute() 在返回之前处理整个查询，因此当 python 到达下一段代码时，查询已完成。

这是精确授权的，因为两个 SQL 语句是按顺序提交的，即第一个先完成。然后，之后且仅在之后，第二个被执行。？

是的，它们会按照您给出的顺序执行，一个接一个。

注意“承诺”这个词并不理想。如果您已打开自动提交，那么它会在每次查询后提交，结束事务。这并不理想，因为您希望控制事务：如果您要插入几行相互依赖的行，通常最好要么全部插入，要么根本不插入。此外，提交需要确保数据安全地写入存储，这很慢。在事务内对插入或更新进行分组要快得多。

关于错误处理，如果其中一个语句抛出异常（例如外键冲突），那么整个事务将被丢弃，并且 postgres 将不会接受新命令，直到您的代码确认它正在通过发送回滚来处理错误。因此，您可以将所有查询放在 try 块中，并在 except 子句中在其他错误处理代码中发出回滚。这样，要么所有内容都会提交，要么全部回滚，并且您的数据库保持干净。

或者它实际上是被禁止的，因为它们可以并行执行，每个查询在任何时刻 t 都不知道另一个查询的状态？

如果你想并行执行多个查询，例如，因为一个查询需要一段时间，并且你想在运行其他查询的同时执行它，那么你需要多个数据库连接，当然每个数据库连接需要一个Python线程，因为execute()是阻塞。不常用。