解决PostgreSQL DO块无法返回数据的方案（libpqxx+C++场景）

关于DO块的返回限制 #

是的，PostgreSQL的DO块确实无法返回任何数据。它的设计目的仅用于执行匿名PL/pgSQL代码块（比如执行插入、更新操作），不支持返回结果集或单个值。

解决思路 #

一、直接用libpqxx传递text[]数组参数（推荐，避免绕DO块） #

libpqxx原生支持数组参数传递，有两种常用方式：

1. 使用pqxx::array构造数组（类型安全）

适合复杂数组场景，能避免字符串拼接的潜在问题：

#include <pqxx/pqxx>

int main() {
    pqxx::connection conn("dbname=your_db user=your_user");
    
    // 构造第一个text[]数组
    pqxx::array arr1;
    arr1.push_back("value1");
    arr1.push_back("value2");
    arr1.push_back("value3");
    
    // 构造第二个text[]数组（直接初始化）
    pqxx::array arr2{"foo", "bar", "baz"};
    
    pqxx::work txn(conn);
    // 调用PostgreSQL函数并获取返回的ID
    auto result = txn.exec_params(
        "SELECT your_insert_function($1, $2)",
        arr1,
        arr2
    );
    
    // 提取新行ID（假设函数返回单个ID值）
    int new_row_id = result[0][0].as<int>();
    txn.commit();
    
    return 0;
}

2. 直接传递PostgreSQL格式的数组字符串（简单场景）

如果数组元素不含逗号、引号等特殊字符，可以直接用字符串构造数组：

pqxx::work txn(conn);
auto result = txn.exec_params(
    "SELECT your_insert_function($1::text[], $2::text[])",
    "{a,b,c}",  // 对应text[]数组
    "{x,y,z}"
);
int new_row_id = result[0][0].as<int>();
txn.commit();

二、替代DO块获取插入ID的方案 #

如果暂时需要保留数组构造逻辑，可放弃DO块，改用以下方式：

1. 用WITH子句构造数组并调用函数

直接在SQL中构造数组，同时调用函数并返回ID：

pqxx::work txn(conn);
// 构造数组并调用函数，返回新ID
auto result = txn.exec(R"(
    WITH array_data AS (
        SELECT ARRAY['val1','val2'] AS arr_param1, ARRAY['val3','val4'] AS arr_param2
    )
    SELECT your_insert_function(arr_param1, arr_param2) AS new_id FROM array_data
)");
int new_row_id = result[0]["new_id"].as<int>();
txn.commit();

2. 直接执行带RETURNING的插入语句

如果函数核心是插入数据，可跳过函数，直接构造数组并执行INSERT...RETURNING：

pqxx::work txn(conn);
auto result = txn.exec_params(
    "INSERT INTO your_table(col1, col2) VALUES ($1::text[], $2::text[]) RETURNING id",
    "{a,b,c}",
    "{x,y,z}"
);
int new_row_id = result[0]["id"].as<int>();
txn.commit();

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