我应该回滚只包含 COPY 的事务吗？

哈喽！今天心血来潮给大家带来了《我应该回滚只包含 COPY 的事务吗？》，想必大家应该对Golang都不陌生吧，那么阅读本文就都不会很困难，以下内容主要涉及到，若是你正在学习Golang，千万别错过这篇文章~希望能帮助到你！

我在 go 后端使用 postgres copy。复制是事务内唯一的操作。如果失败我应该回滚吗？

func (pc *Postgres) Copy(records [][]interface{}) error {
    tx, err := pc.db.Begin()
    if err != nil {
        return errors.Wrap(err, "can't open transaction")
    }
    stmt, err := tx.Prepare(pq.CopyIn(pc.table, pc.columns...))
    if err != nil {
        return errors.Wrap(err, "can't prepare stmt")
    }
    for _, record := range records {
        if _, err := stmt.Exec(record...); err != nil {
            return errors.Wrap(err, "error exec record")
        }

    }

    if _, err = stmt.Exec(); err != nil {
        return errors.Wrap(err, "error exec stmt")
    }
    if err = stmt.Close(); err != nil {
        return errors.Wrap(err, "error close stmt")
    }
    if err = tx.Commit(); err != nil {
        return errors.Wrap(err, "error commit transaction")
    }
    return nil
}

据我了解，如果 \copy 失败，事务将被中止（链接）并回滚。

然而，在官方 lib/pq 示例中，我看到他们总是使用回滚（但他们有多个操作）。

有人可以指导我了解这些细微差别吗？

解决方案

看起来造成混乱的部分原因是 \COPY 与 COPY 不同。

\COPY（如 this question 中所引用）是 psql 中的一条命令，用于“执行前端（客户端）复制”。简单来说，您在计算机上运行 psql （基于终端的 postgresql 前端），并且可以在数据库和本地存储（或从您的计算机可访问）的文件之间进行 \copy 数据。此命令是 psql 的一部分，不能通过 lib/pq 使用。

COPY是在服务器上运行的postgres sql命令；您正在复制的文件“必须可由 postgresql 用户（服务器运行的用户 id）访问，并且必须从服务器的角度指定名称”。这就是您在应用程序中调用的内容（来自 pq docs“copyin 使用 copy from 内部”实现 here 和 here）

因此，正如 @mh-cbon 的 answer referenced 所说：

考虑：

tx, err := pc.db.Begin()
if err != nil {
    return errors.Wrap(err, "can't open transaction")
}
stmt, err := tx.Prepare(pq.CopyIn(pc.table, pc.columns...))
if err != nil {
    return errors.Wrap(err, "can't prepare stmt")
}

如果 prepare 失败，则说明您已经创建了一个事务，然后返回而不关闭它；这使得事务处于打开状态，即 not a good thing。添加 defer tx.rollback() 可确保不会发生这种情况。

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《我应该回滚只包含 COPY 的事务吗？》文章吧，也可关注golang学习网公众号了解相关技术文章。