如何在 Go 中將行從 STDIN 匯入到 PostgreSQL？

從STDIN 導入行到PostgreSQL

在Python 中，subprocess 模組提供了一種使用COPY 將資料從STDIN 寫入Postgregre 的Postgre便捷方法命令。然而，在 Go 中實現相同的功能需要不同的方法。

github.com/lib/pq 套件的官方文件提供如何使用 CopyIn 函數直接從 STDIN 匯入記錄的範例。以下是所提供程式碼的改編版本：

<code class="go">package main

import (
    "database/sql"
    "fmt"
    "log"

    "github.com/lib/pq"
)

func main() {
    records := [][]string{
        {"Rob", "Pike"},
        {"Ken", "Thompson"},
        {"Robert", "Griesemer"},
    }

    db, err := sql.Open("postgres", "dbname=postgres user=postgres password=postgres")
    if err != nil {
        log.Fatalf("open: %v", err)
    }
    if err = db.Ping(); err != nil {
        log.Fatalf("open ping: %v", err)
    }
    defer db.Close()

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

    stmt, err := txn.Prepare(pq.CopyIn("test", "first_name", "last_name"))
    if err != nil {
        log.Fatalf("prepare: %v", err)
    }

    for _, r := range records {
        _, err = stmt.Exec(r[0], r[1])
        if err != nil {
            log.Fatalf("exec: %v", err)
        }
    }

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

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

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

    log.Println("Records successfully imported")
}</code>

此修訂後的程式碼使用交易來確保導入過程中的原子性、一致性、隔離性和持久性 (ACID) 屬性。 CopyIn 函數可讓您指定目標表名稱以及將插入資料的列。

然後使用準備好的語句將提供的範例資料插入到「測試」表中。準備好的語句透過預先編譯 SQL 查詢來提高效能，減少與解析和規劃相關的開銷。

插入所有記錄後，交易將被提交，資料將永久儲存在資料庫中。

以上是如何在 Go 中將行從 STDIN 匯入到 PostgreSQL？的詳細內容。更多資訊請關注PHP中文網其他相關文章！