如何调用PostgreSQL函数且无需等待执行完成？

要实现Python调用PostgreSQL函数后无需等待执行完成，核心是让数据库在后台独立执行函数，而非让psycopg2保持连接等待结果。以下是两种可靠的实现方式：

方法一：使用dblink扩展（原生支持，仅需启用内置扩展） #

dblink可创建跨连接的数据库调用，通过它发起独立连接执行耗时函数，当前连接发送请求后立即断开，无需等待结果返回。

步骤1：启用dblink扩展 #

先在PostgreSQL中执行以下SQL启用扩展：

CREATE EXTENSION IF NOT EXISTS dblink;

步骤2：修改Python代码 #

替换原有函数调用逻辑为dblink异步执行：

import psycopg2
import configs

conn = psycopg2.connect(
    user=configs.USERNAME,
    password=configs.PASSWORD,
    host=configs.HOST,
    port=configs.PORT,
    dbname=configs.DATABASE
)
cur = conn.cursor()

# 通过dblink发起异步函数调用
cur.execute("""
-- 建立临时连接
SELECT dblink_connect('async_conn', 'dbname=' || current_database());
-- 发送函数执行请求（无需等待返回）
SELECT dblink_send_query('async_conn', 'SELECT wrangler_func(''new_data'')');
-- 断开临时连接
SELECT dblink_disconnect('async_conn');
""")
conn.commit()

# 立即关闭连接，无需等待函数执行完成
cur.close()
conn.close()

方法二：使用pg_background扩展（更简洁的异步执行方案） #

pg_background是PostgreSQL的第三方扩展，专门用于后台执行SQL语句，安装后可直接调用实现异步执行。

步骤1：安装并启用扩展 #

• 根据PostgreSQL版本和操作系统安装扩展（例如Debian/Ubuntu：sudo apt install postgresql-xx-pg-background，xx为你的PostgreSQL版本号）
• 执行SQL启用扩展：

CREATE EXTENSION IF NOT EXISTS pg_background;

步骤2：修改Python代码 #

直接调用pg_background函数发起后台执行：

import psycopg2
import configs

conn = psycopg2.connect(
    user=configs.USERNAME,
    password=configs.PASSWORD,
    host=configs.HOST,
    port=configs.PORT,
    dbname=configs.DATABASE
)
cur = conn.cursor()

# 发起后台函数执行，立即返回任务ID
cur.execute("SELECT pg_background('SELECT wrangler_func(''new_data'')');")
conn.commit()

# 直接关闭连接即可
cur.close()
conn.close()

注意事项 #

• 若使用Python线程/进程异步调用，主进程退出时子线程/进程会被终止，无法保证函数执行完成，因此数据库端异步方案更可靠。
• 可通过以下SQL查询后台函数的执行状态：

  SELECT * FROM pg_stat_activity WHERE query LIKE '%wrangler_func%';
  

内容的提问来源于stack exchange，提问作者user18140022