Go语言中捕获Writer/ResponseWriter输出为字符串的实用指南

package main

import (
    "bytes"
    "fmt"
    "io"
)

// 示例函数：接受一个io.Writer并写入内容
func renderTemplate(w io.Writer, name string) {
    fmt.Fprintf(w, "<h1>Welcome, %s!</h1>", name)
    fmt.Fprintln(w, "<p>This is a rendered page.</p>")
}

func main() {
    // 1. 创建一个bytes.Buffer实例
    // bytes.Buffer实现了io.Writer接口
    buf := new(bytes.Buffer)

    // 2. 将buf作为io.Writer传递给函数
    // 函数会将内容写入到buf中，而不是直接输出
    renderTemplate(buf, "Go Developer")

    // 3. 调用buf.String()方法获取写入的所有内容作为字符串
    capturedHTML := buf.String()
    fmt.Println("--- 捕获到的HTML内容 ---")
    fmt.Println(capturedHTML)
    fmt.Println("----------------------")

    // 此时，capturedHTML变量包含了renderTemplate函数写入的所有内容
    // 你可以将其存储到数据库、进行解析或打印
}

package main

import (
    "fmt"
    "net/http"
    "net/http/httptest"
)

// 示例HTTP处理器
func myHTTPHandler(w http.ResponseWriter, r *http.Request) {
    w.Header().Set("Content-Type", "text/plain; charset=utf-8")
    w.WriteHeader(http.StatusOK) // 设置HTTP状态码
    fmt.Fprintf(w, "Hello from HTTP handler! Path: %s", r.URL.Path)
}

func main() {
    // 1. 创建一个httptest.ResponseRecorder实例
    // ResponseRecorder实现了http.ResponseWriter接口
    recorder := httptest.NewRecorder()

    // 2. 创建一个模拟的HTTP请求
    // 在测试中，通常需要模拟一个请求对象
    req, err := http.NewRequest("GET", "/api/data", nil)
    if err != nil {
        fmt.Printf("创建请求失败: %v\n", err)
        return
    }

    // 3. 将recorder作为http.ResponseWriter传递给处理器
    // 处理器会将响应内容写入到recorder中
    myHTTPHandler(recorder, req)

    // 4. 获取捕获到的HTTP响应信息
    statusCode := recorder.Code
    responseHeader := recorder.Header()
    responseBody := recorder.Body.String() // recorder.Body 是一个 *bytes.Buffer

    fmt.Printf("--- 捕获到的HTTP响应信息 ---\n")
    fmt.Printf("HTTP状态码: %d\n", statusCode)
    fmt.Printf("响应头: %v\n", responseHeader)
    fmt.Printf("响应体: %q\n", responseBody)
    fmt.Println("--------------------------")

    // 此时，你可以对statusCode、responseHeader和responseBody进行断言或进一步处理
}