如何将行从 STDIN 导入到 Postgresql？ [重复]

在 Go 中，您可以使用 pq 包将行从 STDIN 导入到 PostgreSQL。以下是分步解决方案：

第 1 步：准备数据库连接

import (
    "database/sql"

    "github.com/lib/pq"
)

db, err := sql.Open("postgres", "dbname=mydb user=myuser password=mypassword")
if err != nil {
    log.Fatalf("open: %v", err)
}

第2步：开始事务

txn, err := db.Begin()
if err != nil {
    log.Fatalf("begin: %v", err)
}

第 3 步：准备复制语句

使用 pq.CopyIn() 创建准备好的语句。

stmt, err := txn.Prepare(pq.CopyIn("test_table", "column1", "column2", ...))
if err != nil {
    log.Fatalf("prepare: %v", err)
}

第 4 步：执行复制语句

迭代数据并对每一行执行 stmt.Exec()。

for _, row := range rows {
    _, err = stmt.Exec(row.Column1, row.Column2, ...)
    if err != nil {
        log.Fatalf("exec: %v", err)
    }
}

第五步：执行最终复制语句

_, err = stmt.Exec()
if err != nil {
    log.Fatalf("exec: %v", err)
}

第6步：关闭语句并提交交易

stmt.Close()
err = txn.Commit()
if err != nil {
    log.Fatalf("commit: %v", err)
}

此代码将有效地将行从 STDIN 导入到 PostgreSQL 表。