Go语言中实现动态IN查询的指南

SELECT id, name FROM users WHERE id IN (1,2,3,4);

// 这种方式是错误的，不会按预期工作
ids := []int{1, 2, 3, 4}
rows, err := db.Query("SELECT id, name FROM users WHERE id IN (?)", ids)
// ...

package main

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

    _ "github.com/go-sql-driver/mysql" // 替换为你的数据库驱动
)

func main() {
    // 假设你已经配置好数据库连接
    db, err := sql.Open("mysql", "user:password@tcp(127.0.0.1:3306)/database")
    if err != nil {
        log.Fatal(err)
    }
    defer db.Close()

    // 示例：要查询的ID列表
    intIDs := []int{1, 2, 3, 4, 5}

    // 将int切片转换为interface{}切片
    // 这是为了满足db.Query的可变参数列表要求
    args := make([]interface{}, len(intIDs))
    for i, id := range intIDs {
        args[i] = id
    }

    // 动态生成IN子句的占位符字符串
    // 例如，如果len(intIDs)是4，marks会是"?,?,?,?"
    marks := strings.Repeat("?,", len(intIDs)-1) + "?"

    // 构建完整的SQL查询
    query := fmt.Sprintf("SELECT id, name FROM users WHERE id IN (%s)", marks)
    fmt.Printf("执行的SQL查询: %s\n", query)
    fmt.Printf("传递的参数: %v\n", args)

    // 执行查询
    rows, err := db.Query(query, args...) // 注意这里的args...，它将切片展开为独立的参数
    if err != nil {
        log.Fatal(err)
    }
    defer rows.Close()

    // 处理查询结果
    for rows.Next() {
        var id int
        var name string
        if err := rows.Scan(&id, &name); err != nil {
            log.Fatal(err)
        }
        fmt.Printf("ID: %d, Name: %s\n", id, name)
    }

    if err := rows.Err(); err != nil {
        log.Fatal(err)
    }

    // 示例：处理空切片的情况
    fmt.Println("\n--- 处理空切片 ---")
    emptyIntIDs := []int{}
    queryWithEmptyIDs(db, emptyIntIDs)

    // 示例：处理只有一个元素的切片
    fmt.Println("\n--- 处理单元素切片 ---")
    singleIntIDs := []int{10}
    queryWithEmptyIDs(db, singleIntIDs)
}

// 封装成一个函数，更好地处理空切片情况
func queryWithEmptyIDs(db *sql.DB, intIDs []int) {
    if len(intIDs) == 0 {
        fmt.Println("ID列表为空，不执行查询。")
        // 或者可以执行一个返回空结果集的查询，例如：
        // rows, err := db.Query("SELECT id, name FROM users WHERE 1=0")
        return
    }

    args := make([]interface{}, len(intIDs))
    for i, id := range intIDs {
        args[i] = id
    }

    // 如果只有一个元素，strings.Repeat("?,", 0) + "?" 结果是 "?"
    // 如果有多个元素，例如[1,2,3]，strings.Repeat("?,", 2) + "?" 结果是 "?,?,?"
    marks := strings.Repeat("?,", len(intIDs)-1) + "?"
    query := fmt.Sprintf("SELECT id, name FROM users WHERE id IN (%s)", marks)

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

    rows, err := db.Query(query, args...)
    if err != nil {
        log.Printf("查询错误: %v\n", err)
        return
    }
    defer rows.Close()

    found := false
    for rows.Next() {
        found = true
        var id int
        var name string
        if err := rows.Scan(&id, &name); err != nil {
            log.Printf("扫描结果错误: %v\n", err)
            return
        }
        fmt.Printf("ID: %d, Name: %s\n", id, name)
    }
    if !found {
        fmt.Println("未找到匹配结果。")
    }

    if err := rows.Err(); err != nil {
        log.Printf("迭代行错误: %v\n", err)
    }
}