>  기사  >  백엔드 개발  >  Go를 사용하여 STDIN에서 PostgreSQL로 행을 가져오는 방법은 무엇입니까?

Go를 사용하여 STDIN에서 PostgreSQL로 행을 가져오는 방법은 무엇입니까?

Susan Sarandon
Susan Sarandon원래의
2024-10-24 16:06:01398검색

How to Import Rows into PostgreSQL from STDIN Using Go?

Go를 사용하여 STDIN에서 PostgreSQL로 행 가져오기

Go에서는 pq를 사용하여 표준 입력(STDIN)에서 PostgreSQL로 행을 가져올 수 있습니다. 패키지. 이 접근 방식은 중간 파일 없이 데이터베이스에 데이터를 직접 공급합니다.

STDIN에서 행을 직접 가져오려면 다음 단계를 따르세요.

  1. PostgreSQL 연결 열기 : sql.Open 함수를 사용하여 PostgreSQL 데이터베이스에 대한 연결을 설정합니다.
  2. 트랜잭션 시작: 여러 문을 그룹화하려면 Begin 함수를 사용하여 데이터베이스 트랜잭션을 시작합니다.
  3. 복사 문 준비: pq.CopyIn을 사용하여 데이터 복사를 위한 준비된 문을 만듭니다. 이 명령문은 가져올 대상 테이블 이름과 열을 지정합니다.
  4. 행 가져오기: 가져오려는 데이터의 각 행에 대해 준비된 명령문을 반복적으로 실행합니다. Exec 함수를 사용하여 각 행을 데이터베이스로 보냅니다.
  5. 복사 문 종료: 모든 행을 추가한 후 인수 없이 준비된 문을 다시 실행하여 가져오기 프로세스를 마무리합니다.
  6. Prepared 문 닫기: Close 메서드를 호출하여 준비된 문과 관련된 리소스를 해제합니다.
  7. 트랜잭션 커밋: 변경 사항을 영구적으로 적용하려면 트랜잭션을 커밋합니다.

예제 코드:

다음은 Go를 사용하여 STDIN에서 행 가져오기를 보여주는 코드 예입니다.

<code class="go">package main

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

    "github.com/lib/pq"
)

func main() {
    db, err := sql.Open("postgres", "host=localhost port=5432 user=postgres password=mysecret dbname=mydatabase")
    if err != nil {
        log.Fatal(err)
    }
    defer db.Close()

    rows := [][]string{
        {"Rob", "Pike"},
        {"Ken", "Thompson"},
        {"Robert", "Griesemer"},
    }

    txn, err := db.Begin()
    if err != nil {
        log.Fatal(err)
    }

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

    for _, r := range rows {
        if _, err = stmt.Exec(r[0], r[1]); err != nil {
            log.Fatal(err)
        }
    }

    if _, err = stmt.Exec(); err != nil {
        log.Fatal(err)
    }

    if err = stmt.Close(); err != nil {
        log.Fatal(err)
    }

    if err = txn.Commit(); err != nil {
        log.Fatal(err)
    }

    fmt.Println("Rows imported successfully.")
}</code>

다음 단계를 수행하고 pq 패키지를 활용하면 Go 프로그램의 STDIN에서 직접 PostgreSQL로 데이터를 효율적으로 가져올 수 있습니다.

위 내용은 Go를 사용하여 STDIN에서 PostgreSQL로 행을 가져오는 방법은 무엇입니까?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!

성명:
본 글의 내용은 네티즌들의 자발적인 기여로 작성되었으며, 저작권은 원저작자에게 있습니다. 본 사이트는 이에 상응하는 법적 책임을 지지 않습니다. 표절이나 침해가 의심되는 콘텐츠를 발견한 경우 admin@php.cn으로 문의하세요.