Go语言中lib/pq驱动与PostgreSQL SQL占位符的正确使用指南

package main

import (
    "database/sql"
    "fmt"
    _ "github.com/lib/pq" // 导入PostgreSQL驱动，下划线表示只导入包进行初始化，不直接使用其导出成员
    "log"
)

func main() {
    // 数据库连接字符串示例。请根据您的实际情况修改。
    // 注意：生产环境中，敏感信息如密码不应硬编码，应通过环境变量或配置管理。
    connStr := "user=postgres password=your_password dbname=your_db host=localhost sslmode=disable"

    // 打开数据库连接
    db, err := sql.Open("postgres", connStr)
    if err != nil {
        log.Fatalf("无法打开数据库连接: %v", err)
    }
    defer db.Close() // 确保在函数结束时关闭数据库连接

    // 尝试ping数据库以确认连接是否成功
    err = db.Ping()
    if err != nil {
        log.Fatalf("无法连接到PostgreSQL数据库: %v", err)
    }
    fmt.Println("成功连接到PostgreSQL数据库！")

    // 假设我们有一个名为 'things' 的表，结构为 (id INT PRIMARY KEY, thing VARCHAR(255))
    // 如果表不存在，可以先创建它：
    // _, err = db.Exec(`CREATE TABLE IF NOT EXISTS things (id SERIAL PRIMARY KEY, thing VARCHAR(255) UNIQUE)`)
    // if err != nil {
    //  log.Fatalf("创建表失败: %v", err)
    // }
    // fmt.Println("表 'things' 已确保存在。")

    // 声明用于查询和插入的变量
    var thingName string = "example_item"
    var id int

    // --- 错误示例 (如果直接运行会报错，此处注释掉以避免程序中断) ---
    // 以下代码如果使用 '?' 占位符，PostgreSQL将抛出语法错误。
    // queryWrong := "SELECT id FROM things WHERE thing = ?"
    // err = db.QueryRow(queryWrong, thingName).Scan(&id)
    // if err != nil {
    //     fmt.Printf("错误示例：查询失败 (%s) - %v\n", queryWrong, err)
    // }

    // --- 正确使用PostgreSQL的占位符 $1 进行查询 ---
    fmt.Println("\n--- 正确查询示例 ---")
    queryCorrect := "SELECT id FROM things WHERE thing = $1"
    err = db.QueryRow(queryCorrect, thingName).Scan(&id)
    if err != nil {
        if err == sql.ErrNoRows {
            fmt.Printf("未找到名为 '%s' 的记录。尝试插入新记录...\n", thingName)
            // 如果未找到，我们插入一条记录
            insertQuery := "INSERT INTO things (thing) VALUES ($1) RETURNING id"
            err = db.QueryRow(insertQuery, thingName).Scan(&id)
            if err != nil {
                log.Fatalf("插入记录失败: %v", err)
            }
            fmt.Printf("成功插入新记录：thing='%s', 分配的ID=%d\n", thingName, id)
        } else {
            log.Fatalf("查询失败: %v\n", err)
        }
    } else {
        fmt.Printf("找到记录：thing='%s', ID=%d\n", thingName, id)
    }

    // --- 多个参数的插入/更新示例 ---
    fmt.Println("\n--- 多个参数示例 ---")
    newThingID := 1001
    anotherThingName := "another_item"

    // 使用 $1, $2 等占位符
    // ON CONFLICT (id) DO UPDATE SET ... 是PostgreSQL的UPSERT语法
    upsertQuery := `
        INSERT INTO things (id, thing) VALUES ($1, $2)
        ON CONFLICT (id) DO UPDATE SET thing = EXCLUDED.thing
        RETURNING id
    `
    var returnedID int
    err = db.QueryRow(upsertQuery, newThingID, anotherThingName).Scan(&returnedID)
    if err != nil {
        log.Fatalf("插入/更新记录失败: %v", err)
    }
    fmt.Printf("成功插入/更新记录：ID=%d, thing='%s'，返回的ID=%d\n", newThingID, anotherThingName, returnedID)

    // 查询所有记录以验证
    fmt.Println("\n--- 查询所有记录 ---")
    rows, err := db.Query("SELECT id, thing FROM things ORDER BY id")
    if err != nil {
        log.Fatalf("查询所有记录失败: %v", err)
    }
    defer rows.Close()

    for rows.Next() {
        var currentID int
        var currentThing string
        if err := rows.Scan(&currentID, &currentThing); err != nil {
            log.Fatalf("扫描行数据失败: %v", err)
        }
        fmt.Printf("ID: %d, Thing: %s\n", currentID, currentThing)
    }
    if err = rows.Err(); err != nil {
        log.Fatalf("遍历行时发生错误: %v", err)
    }
}