>백엔드 개발 >Golang >Go를 사용하여 PostgreSQL에서 대량 업데이트를 수행하는 방법은 무엇입니까?

Go를 사용하여 PostgreSQL에서 대량 업데이트를 수행하는 방법은 무엇입니까?

Susan Sarandon
Susan Sarandon원래의
2024-12-13 11:32:11407검색

How to Perform Bulk Updates in PostgreSQL with Go?

PostgreSQL을 사용하여 단일 쿼리에서 여러 행 대량 업데이트

질문:

어떻게 할 수 있나요? PostgreSQL에서 단일 SQL 문을 사용하여 여러 행을 동시에 업데이트하고 하시겠습니까?

답변:

단일 쿼리를 사용하여 PostgreSQL에서 대량 업데이트를 수행하려면 파생 테이블을 사용하는 것이 좋습니다. 다음 예에서는 파생 테이블을 사용하여 여러 행을 업데이트하는 방법을 보여줍니다.

UPDATE t
SET column_a = v.column_a,
    column_b = v.column_b
FROM (VALUES (1, 'FINISH', 1234),
             (2, 'UNFINISH', 3124)
     ) v(id, column_a, column_b)
WHERE v.id = t.id;

Go 구현:

다음 Go 코드는 대량 업데이트 쿼리 실행을 보여줍니다.

import (
    "context"
    "database/sql"
    "fmt"

    _ "github.com/lib/pq" // PostgreSQL driver
)

func main() {
    // Open a database connection
    db, err := sql.Open("postgres", "user=postgres password=mypassword host=localhost port=5432 database=mydatabase")
    if err != nil {
        panic(err)
    }
    defer db.Close()

    // Create the derived table
    derivedTableQuery := `
CREATE TEMP TABLE BulkUpdate AS
SELECT 1 AS id, 'FINISH' AS column_a, 1234 AS column_b
UNION ALL
SELECT 2, 'UNFINISH', 3124
`
    if _, err := db.Exec(derivedTableQuery); err != nil {
        panic(err)
    }

    // Execute the bulk update query
    bulkUpdateQuery := `
UPDATE t
SET column_a = v.column_a,
    column_b = v.column_b
FROM BulkUpdate v
WHERE v.id = t.id
`

    // Create a context for the query execution
    ctx := context.Background()

    // Execute the bulk update query
    result, err := db.ExecContext(ctx, bulkUpdateQuery)
    if err != nil {
        panic(err)
    }

    // Retrieve the number of rows affected
    rowsAffected, err := result.RowsAffected()
    if err != nil {
        panic(err)
    }

    // Print the number of rows affected
    fmt.Printf("%d rows updated\n", rowsAffected)
}

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

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