sql >> Databasteknik >  >> RDS >> PostgreSQL

Hur importerar jag rader till Postgresql från STDIN?

github.com/lib/pq paketdokument har faktiskt ett exempel hur man gör vad man vill. Här är den anpassade texten för hela programmet:

package main

import (
    "database/sql"
    "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)
    }
}

På min maskin importerar detta 1 000 000 poster på cirka 2 sekunder.



  1. PostgreSQL:formatera intervall som minuter

  2. SQL för att skapa matristabell

  3. Frågeuppsättningssortering:Anger kolumnsortering för django ORM-fråga

  4. Hur returnerar man ett värde från en lagrad procedur (inte funktion)?