sql >> Database >  >> RDS >> PostgreSQL

Hoe importeer ik rijen naar Postgresql vanuit STDIN?

De github.com/lib/pq pakketdocumenten hebben eigenlijk een voorbeeld van hoe u kunt doen wat u wilt. Hier is de aangepaste tekst van het hele programma:

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)
    }
}

Op mijn machine importeert dit 1 000 000 records in ongeveer 2 seconden.



  1. Aangepaste zoekopdracht met ManyToMany

  2. Grote (95Mb) JSON-array opsplitsen in kleinere brokken?

  3. Kan een DB niet verbinden met AWS

  4. Variabelen in opgeslagen procedure MySQL