Go语言database/sql：高效构建和执行带有可变参数的IN查询

// 这种方式是无效的，因为占位符 '?' 期望的是单个值，而不是一个切片
// db.Query("SELECT id, name FROM users WHERE id IN (?)", []int{1,2,3,4})

package main

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

    _ "github.com/go-sql-driver/mysql" // 假设使用MySQL驱动，请根据实际情况替换
)

// InQueryExample 演示如何在Go中执行带有可变参数的IN查询
func InQueryExample(db *sql.DB, userIDs []int) ([]map[string]interface{}, error) {
    // 1. 处理空切片的情况
    if len(userIDs) == 0 {
        fmt.Println("用户ID列表为空，无需执行查询。")
        return nil, nil // 返回空结果，不执行数据库操作
    }

    // 2. 准备占位符字符串和参数
    // 将int切片转换为interface{}切片，以便作为可变参数传递给db.Query
    args := make([]interface{}, len(userIDs))
    for i, id := range userIDs {
        args[i] = id
    }

    // 生成占位符字符串，例如对于 [1,2,3,4] 会生成 "?,?,?,?"
    // strings.Repeat("?,", len(userIDs)) 会生成 "保有逗号的字符串，例如 "?,?,?,?,",
    // 然后通过切片操作移除末尾多余的逗号。
    placeholderStr := strings.Repeat("?,", len(userIDs))
    placeholderStr = placeholderStr[:len(placeholderStr)-1] // 移除最后一个逗号

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

    fmt.Printf("生成的SQL查询: %s\n", query)
    fmt.Printf("传递的参数: %v\n", args)

    // 3. 执行查询
    // 实际应用中，db会是一个有效的数据库连接
    if db == nil {
        return nil, fmt.Errorf("数据库连接未初始化，无法执行查询")
    }

    rows, err := db.Query(query, args...)
    if err != nil {
        return nil, fmt.Errorf("执行查询失败: %w", err)
    }
    defer rows.Close()

    var results []map[string]interface{}
    for rows.Next() {
        var id int
        var name string
        if err := rows.Scan(&id, &name); err != nil {
            return nil, fmt.Errorf("扫描结果失败: %w", err)
        }
        results = append(results, map[string]interface{}{"id": id, "name": name})
    }
    if err := rows.Err(); err != nil {
        return nil, fmt.Errorf("迭代结果集错误: %w", err)
    }

    return results, nil
}

func main() {
    // 模拟数据库连接 (在实际应用中，这里会建立真实的数据库连接)
    // 例如：
    // db, err := sql.Open("mysql", "user:password@tcp(127.0.0.1:3306)/dbname?parseTime=true")
    // if err != nil {
    //  log.Fatalf("无法连接数据库: %v", err)
    // }
    // defer db.Close()
    // 
    // 为了示例运行，我们这里使用一个 nil db，并仅打印SQL和参数
    var db *sql.DB 

    // 示例用户ID列表
    userIDs1 := []int{1, 2, 3, 4, 5}
    fmt.Println("--- 示例 1: 多个ID ---")
    res1, err := InQueryExample(db, userIDs1)
    if err != nil {
        log.Printf("示例1执行失败: %v", err)
    } else if res1 != nil {
        fmt.Printf("查询结果: %v\n", res1)
    }

    fmt.Println("\n--- 示例 2: 单个ID ---")
    userIDs2 := []int{10}
    res2, err := InQueryExample(db, userIDs2)
    if err != nil {
        log.Printf("示例2执行失败: %v", err)
    } else if res2 != nil {
        fmt.Printf("查询结果: %v\n", res2)
    }

    fmt.Println("\n--- 示例 3: 空ID列表 ---")
    userIDs3 := []int{}
    res3, err := InQueryExample(db, userIDs3)
    if err != nil {
        log.Printf("示例3执行失败: %v", err)
    } else if res3 != nil {
        fmt.Printf("查询结果: %v\n", res3)
    }
}