如何获取Db.Query调用背后的实际SQL语句？实现Db.Translate类功能

嘿，这个需求我太懂了——调试的时候能直接看到最终拼好参数的SQL，简直能省超多时间！我给你分享几个可行的方案，帮你实现类似Db.Translate的功能：

1. 利用数据库驱动的日志/跟踪能力 #

如果你用的是PostgreSQL的pq驱动，其实可以直接在连接字符串里开启日志，比如：

db, err := sql.Open("postgres", "host=localhost user=postgres dbname=mydb password=secret log_statement=all")

开启log_statement=all后，驱动会把所有执行的SQL（包括替换参数后的版本）输出到日志里。不过这个是被动输出，如果你想在代码里主动拿到字符串，这个方法可能不太适合。

2. 自己实现参数替换逻辑（类似你要的Translate） #

如果想主动生成拼接好参数的SQL字符串，你可以自己写一个简单的Translate函数，针对常用的参数类型做处理，比如pq.Array：

import (
    "fmt"
    "github.com/lib/pq"
    "strings"
)

// Translate 模拟替换参数，生成最终SQL字符串（仅用于调试！）
func Translate(query string, args ...interface{}) string {
    processedQuery := query
    for idx, arg := range args {
        placeholder := fmt.Sprintf("$%d", idx+1)
        var valueStr string

        // 针对pq.Array做特殊处理
        if arr, ok := arg.(pq.Array); ok {
            elements := make([]string, len(arr))
            for i, elem := range arr {
                // 这里可以根据元素类型做更细致的处理，比如字符串加引号
                elements[i] = fmt.Sprintf("%v", elem)
            }
            valueStr = fmt.Sprintf("{%s}", strings.Join(elements, ","))
        } else {
            // 其他基础类型的简单处理，注意字符串类型要加单引号，这里简化了
            switch v := arg.(type) {
            case string:
                valueStr = fmt.Sprintf("'%s'", v)
            default:
                valueStr = fmt.Sprintf("%v", v)
            }
        }

        processedQuery = strings.ReplaceAll(processedQuery, placeholder, valueStr)
    }
    return processedQuery
}

⚠️ 注意：这个方法只能用于调试，绝对不能用它来生成生产环境执行的SQL——直接拼接参数会有SQL注入风险！生产环境一定要用Db.Query的参数化查询。

3. 借助第三方测试/调试库 #

如果是在测试场景下想捕获执行的SQL，可以用go-sqlmock这个库，它能模拟数据库连接，并且捕获所有执行的SQL语句和参数。比如：

import (
    "testing"
    "github.com/DATA-DOG/go-sqlmock"
)

func TestQuery(t *testing.T) {
    db, mock, err := sqlmock.New()
    if err != nil {
        t.Fatalf("failed to open mock db: %v", err)
    }
    defer db.Close()

    // 模拟查询
    mock.ExpectQuery(`SELECT * FROM table WHERE values = ANY\(\$1\)`).
        WithArgs(pq.Array([]int64{1,2,3})).
        WillReturnRows(sqlmock.NewRows([]string{"id"}).AddRow(1))

    // 执行你的查询逻辑
    rows, err := db.Query(`SELECT * FROM table WHERE values = ANY($1)`, pq.Array([]int64{1,2,3}))
    if err != nil {
        t.Fatalf("query failed: %v", err)
    }
    defer rows.Close()

    // 验证是否符合预期
    if err := mock.ExpectationsWereMet(); err != nil {
        t.Errorf("unfulfilled expectations: %v", err)
    }

    // 如果你想拿到执行的SQL，也可以通过mock的调用记录获取
    calls := mock.ExpectationsCalled()
    for _, call := range calls {
        fmt.Println("执行的SQL:", call.Query)
        fmt.Println("参数:", call.Arguments)
    }
}

这个方法适合测试时验证SQL是否正确，也能拿到完整的执行信息。

内容的提问来源于stack exchange，提问作者Victor