【Go】sqlxのNamedStmtでIN句の条件数を動的に指定する

sqlxでIN句を含むクエリをNamedStmtで扱う方法のサンプルコードです。
EXISTS句にも応用可能です。

サンプルコード

main.go

func main() {
    userIds := []int{1, 2, 3, 4, 5}
    age := 20
    select(ids, createdAt)
}

select.go

var db *sqlx.DB

func select(userIds []int, age int) ([]User, error) {

    // SQLの雛形
    originQuery := `SELECT * FROM USER WHERE USER_ID IN (:USER_ID) AND AGE > :AGE`

    // バインドする値
    arg := map[string]interface{}{
        "USER_ID":  userIds,
        "AGE": age,
    }

    // NamedStmtでIN句を使える形式にする
    query, params := formatQuery(&originQuery, &arg)

    // Select文を実行
    namedStmt, err := db.PrepareNamedContext(context.Background(), *query)
    if err != nil {
        return nil, err
    }
    defer namedStmt.Close()

    var users []User
    if err := namedStmt.SelectContext(context.Background(), &sqlxSamples, *params); err != nil {
        return nil, err
    }

    return users, nil
}

format.go

func formatQuery(q *string, a *map[string]interface{}) (*string, *map[string]interface{}) {

    query, args, _ := sqlx.Named(*q, *a)
    fmt.Println(query) // SELECT * FROM USER WHERE USER_ID IN (?) AND AGE > ?
    fmt.Println(args)  // [[1 2 3 4 5] 20]

    query, args, _ = sqlx.In(query, args...)
    fmt.Println(query) // SELECT * FROM USER WHERE USER_ID IN (?, ?, ?, ?, ?) AND AGE > ?
    fmt.Println(args)  // [1 2 3 4 5 20]

    query = sqlx.Rebind(sqlx.NAMED, query)
    params := map[string]interface{}{}
    for i, arg := range args {
        key := fmt.Sprintf("arg%d", i+1)
        params[key] = arg
    }
    fmt.Println(query)  // SELECT * FROM USER WHERE USER_ID IN (:arg1, :arg2, :arg3, :arg4, :arg5) AND AGE > :arg6
    fmt.Println(params) // map[arg1:1 arg2:2 arg3:3 arg4:4 arg5:5 arg6:20]

    return &query, &params
}