使用 libpq 的 COPY 协议维护自定义 PG 到 PG 连接

1. 引言

最近在我的开发工作中，需要在主节点上的 PG 后端与备节点上的其他 PG 后端之间维护一个自定义连接，以传输除现有 walsender/walreceiver 连接所传输的 WAL 数据之外的自定义数据。当然，我可以简单地创建一个新的独立后端并自己维护一个 socket 连接来传输自定义数据。从技术上讲，这是可行的，但也会带来一些问题。自定义连接的持久性、用户安全性、数据加密都需要额外处理。那么，为什么不直接使用 libpq 来为我们处理这些问题呢？本文将分享基于 PG14 使用 libpq COPY 协议维护自定义数据连接的经验。

2. 在 libpqwalreceiver.c 中创建新例程

该文件位于 src/backend/replication/libpqwalreceiver，并被编译为一个共享库 (.so)，其中包含与 libpq 库相关的例程，这些例程允许 PG 后端在不将 libpq 编译到后端代码中的情况下使用 libpq。当 PG 后端进程需要使用 libpq 时，它需要先使用 load_file() 调用加载共享库。

我的需求很简单，我需要的新例程是 connect、send、recv，类似于普通的 socket 交互。我这里没有定义 close 函数，因为我希望只要主节点和备节点在运行，连接就保持持久。当其中一个节点退出时，连接会在检测到对端断开后自动终止。

3. 连接例程

与用于复制的 libpqrcv_connect 例程不同，我的情况简单得多。我只需要让备节点连接到主节点，因此我可以简单地重用备节点的 primary_conninfo 配置参数来建立连接。这将触发主节点派生一个新的后端进程来服务这个连接。代码片段如下所示：

我还将 libpq socket 连接设置为使用阻塞 socket，并将 asyncStatus 设置为 PGASYNC_COPY_BOTH，以表示我将进行双向数据通信。

4. 发送例程

我的发送例程与用于复制的 libpqrcv_send 例程完全相同。两者都使用 PQputCopyData 来向主节点发送数据流。为了保持一致性，我对其进行了重命名。代码片段如下：

5. 接收例程

与用于复制的 libpqrcv_recv 例程也非常相似，代码几乎完全相同。只是根据我的需求，连接需要是同步连接。这意味着我的备节点在等待主节点响应时会阻塞。为了使接收同步，我必须将 PQgetCopyData 的第三个参数传递为 0。因此，如果你对异步连接没有问题，这个例程的代码也可以与 libpqrcv_recv 完全相同。

6. 让备节点发送自定义数据

现在我们已经为自己的目的创建了 libpq 包装例程，我们可以让备节点向主节点发送一些自定义数据并等待响应。注意，我发送了一个字母‘N’，后面跟着三个示例自定义数据：100、200、300。Libpq COPY 使用字母 d 来表示 COPY 命令，而我们在这里做的是在 d 命令中包装我们自己的命令。

StringInfoData buf_blk_request