Go에서 pgx를 사용하여 Postgres에 데이터를 대량 삽입하는 방법: 종합 가이드

Go에서 pgx를 사용하여 Postgres에 대량 삽입: 포괄적인 솔루션

데이터베이스에 대량 삽입을 시도할 때 SQL 문을 수동으로 작성하면 오류와 성능 병목 현상이 발생할 수 있습니다. pgx의 pgx.Conn.CopyFrom 기능을 활용하면 프로세스를 자동화하는 효율적인 솔루션이 제공됩니다.

수동 SQL 작성 문제

제공된 코드에서 SQL 문은 문자열을 연결하여 구성됩니다. 달러 매개변수($ 기호)의 수가 conn.Exec 함수에 전달된 인수 수와 일치하지 않으면 오류가 발생할 수 있습니다. 또한 대규모 입력에 대한 문자열 연결은 비효율적이며 메모리 문제로 이어질 수 있습니다.

pgx의 CopyFrom 메서드 활용

pgx의 CopyFrom 메서드는 PostgreSQL 복사 프로토콜을 활용하여 대량 데이터 삽입을 단순화합니다. 세 가지 인수를 사용합니다:

1. tableName: 삽입할 대상 테이블의 이름.
2. columnNames: 삽입할 열의 이름을 지정하는 문자열 조각.
3. rowSrc: 삽입할 데이터를 생성하는 CopyFromSource 개체.

CopyFromSource 인터페이스를 사용하면 데이터 소스를 유연하게 지정할 수 있습니다. 제공된 예제와 같이 인터페이스 값 조각, CSV 데이터가 포함된 strings.Reader 또는 사용자 정의 구현을 사용하여 구현할 수 있습니다.

코드 예

아래는 CopyFrom 사용을 보여주는 수정된 코드 조각:

<code class="go">rows := [][]interface{}{
    {"abc", 10},
    {"dns", 11},
    {"qwe", 12},
    {"dss", 13},
    {"xcmk", 14},
}

_, err := conn.CopyFrom(
    pgx.Identifier{"keys"},
    []string{"keyval", "lastval"},
    pgx.CopyFromRows(rows),
)</code>

이 코드는 "키" 테이블에 행을 효율적으로 대량 삽입하여 수동 SQL 작성에 비해 성능을 크게 향상시키고 오류 가능성을 줄입니다.

위 내용은 Go에서 pgx를 사용하여 Postgres에 데이터를 대량 삽입하는 방법: 종합 가이드의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!