Go语言database/sql包中处理动态SQL IN 查询的实践指南

package main

import (
    "database/sql"
    "fmt"
    "strings"

    _ "github.com/go-sql-driver/mysql" // 假设使用MySQL驱动，请根据实际数据库选择
)

func main() {
    // 模拟数据库连接
    // 在实际应用中，请替换为您的数据库连接字符串
    // 例如："user:password@tcp(127.0.0.1:3306)/database_name?charset=utf8mb4&parseTime=True&loc=Local"
    db, err := sql.Open("mysql", "root:password@tcp(127.0.0.1:3306)/testdb") // 请替换为您的实际连接信息
    if err != nil {
        fmt.Printf("Error connecting to database: %v\n", err)
        return
    }
    defer db.Close()

    // 验证数据库连接
    err = db.Ping()
    if err != nil {
        fmt.Printf("Error pinging database: %v\n", err)
        return
    }
    fmt.Println("Successfully connected to database!")

    // 待查询的ID列表
    ids := []int{1, 2, 3, 4}
    // ids := []int{} // 尝试空切片情况

    // 1. 处理空切片的情况
    // 这是非常重要的，避免生成无效SQL或运行时错误
    if len(ids) == 0 {
        fmt.Println("ID列表为空，不执行查询。")
        // 根据业务逻辑，可以选择返回空结果集或执行其他操作
        return
    }

    // 2. 将 []int 转换为 []interface{}
    // 这是必需的，因为 db.Query 接受 ...interface{} 类型的可变参数
    args := make([]interface{}, len(ids))
    for i, id := range ids {
        args[i] = id
    }

    // 3. 动态生成占位符字符串
    // 例如，对于长度为4的切片，生成 "?,?,?,?"
    // strings.Repeat("?,", len(ids)-1) 会生成 "断开连接,断开连接,断开连接,"
    // 再加上 "?" 即可得到 "断开连接,断开连接,断开连接,?"
    placeholders := strings.Repeat("?,", len(ids)-1) + "?"

    // 4. 构建完整的SQL查询语句
    query := fmt.Sprintf("SELECT id, name FROM users WHERE id IN (%s)", placeholders)
    fmt.Println("Generated SQL query:", query)

    // 5. 执行查询
    // 注意这里的 args... 展开操作，将 []interface{} 切片中的元素作为独立的参数传递
    rows, err := db.Query(query, args...)
    if err != nil {
        fmt.Printf("Error executing query: %v\n", err)
        return
    }
    defer rows.Close()

    // 处理查询结果
    fmt.Println("Query Results:")
    for rows.Next() {
        var id int
        var name string
        if err := rows.Scan(&id, &name); err != nil {
            fmt.Printf("Error scanning row: %v\n", err)
            continue
        }
        fmt.Printf("  User ID: %d, Name: %s\n", id, name)
    }

    if err := rows.Err(); err != nil {
        fmt.Printf("Error iterating rows: %v\n", err)
    }
}