如何使用 Go 在 PostgreSQL 中高效执行批量更新？

使用 Go Query Builder 在 PostgreSQL 中进行批量更新

当需要更新 PostgreSQL 数据库中的多行时，传统方法是执行单独的 UPDATE 语句每行。然而，出于性能原因，通常需要执行单个 UPDATE 语句来一次性更新多行。

派生表方法

PostgreSQL 中批量更新的一种有效方法是利用派生表。此技术涉及创建一个包含所需更新的临时表，然后使用该表中的数据来更新目标表。

考虑以下场景，我们的目标是使用新的内容更新“table”表中的多行列“column_a”和“column_b”的值：

import (
    "database/sql"
    _"github.com/jackc/pgx/v4/stdlib" // PostgreSQL driver
)

func main() {
    db, err := sql.Open("pgx", "host=localhost port=5432 dbname=test user=postgres password=mysecretpassword")
    if err != nil {
        panic(err)
    }
    defer db.Close()

    // Prepare the derived table with updated values
    query := `UPDATE t
SET column_a = v.column_a,
    column_b = v.column_b
FROM (VALUES (1, 'FINISH', 1234),
             (2, 'UNFINISH', 3124)
     ) v(id, column_a, column_b)
WHERE v.id = t.id;`
    _, err = db.Exec(query)
    if err != nil {
        panic(err)
    }

    // Optionally, you can verify the updated rows
    rows, err := db.Query("SELECT * FROM table WHERE id IN (1, 2)")
    if err != nil {
        panic(err)
    }
    defer rows.Close()

    for rows.Next() {
        var id int
        var columnA, columnB string
        if err := rows.Scan(&id, &columnA, &columnB); err != nil {
            panic(err)
        }
        fmt.Printf("ID: %d, Column A: %s, Column B: %s\n", id, columnA, columnB)
    }
}

此方法可以灵活地在派生表中添加或删除行和列，而无需更改主查询。与显式指定多个 UPDATE 语句相比，它还有助于防止意外错误。

以上是如何使用 Go 在 PostgreSQL 中高效执行批量更新？的详细内容。更多信息请关注PHP中文网其他相关文章！