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

问题描述

是否可以使用 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[],ID])
        curs.execute(sql_update2,ID])

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

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

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

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

解决方法

最好避免这种情况：

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,ID])
        curs.execute(sql_update2,ID])

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

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

别在意。 execute() 在返回之前处理了整个查询，所以当 python 到达下一段代码时，查询就完成了。

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

是的

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

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

,

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

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,ID,jsonString,ID])