如何在 Go 中使用 pgx 将数据批量插入到 Postgres 中：综合指南

在 Go 中使用 pgx 在 Postgres 中批量插入：综合解决方案

尝试在数据库中批量插入时，手动编写 SQL 语句可能会引入错误和性能瓶颈。利用 pgx 的 pgx.Conn.CopyFrom 功能提供了一种有效的解决方案，可以自动执行该过程。

手动 SQL 制作的问题

在提供的代码中，SQL 语句是通过连接字符串来构造的，这如果美元参数（$ 符号）的数量与传递给 conn.Exec 函数的参数数量不匹配，可能会导致错误。此外，大输入的字符串串联可能效率低下并导致内存问题。

利用 pgx 的 CopyFrom 方法

pgx 的 CopyFrom 方法通过利用 PostgreSQL 复制协议简化批量数据插入。它需要三个参数：

1. tableName：要插入的目标表的名称。
2. columnNames：指定要插入的列名称的字符串切片。
3. rowSrc：生成要插入的数据的 CopyFromSource 对象。

CopyFromSource 接口允许灵活指定数据源。它可以使用接口值的切片（如提供的示例中所示）、包含 CSV 数据的 strings.Reader 或自定义实现来实现。

代码示例

下面是修改后的代码片段演示了 CopyFrom 的使用：

<code class="go">rows := [][]interface{}{
    {"abc", 10},
    {"dns", 11},
    {"qwe", 12},
    {"dss", 13},
    {"xcmk", 14},
}

_, err := conn.CopyFrom(
    pgx.Identifier{"keys"},
    []string{"keyval", "lastval"},
    pgx.CopyFromRows(rows),
)</code>

此代码将有效地将行批量插入到“key”表中，与手动 SQL 制作相比，显着提高性能并减少出错的可能性。

以上是如何在 Go 中使用 pgx 将数据批量插入到 Postgres 中：综合指南的详细内容。更多信息请关注PHP中文网其他相关文章！