程序化调用Gorilla Mux处理器并处理mux.Vars()

package main

import (
    "fmt"
    "io/ioutil"
    "log"
    "net/http"
    "net/http/httptest"
    "strings"

    "github.com/gorilla/mux"
)

// userHandler 是一个依赖于 mux.Vars() 的处理器
func userHandler(w http.ResponseWriter, r *http.Request) {
    vars := mux.Vars(r)
    userID := vars["id"] // 从请求上下文中获取路径变量
    if userID == "" {
        http.Error(w, "User ID not found", http.StatusBadRequest)
        return
    }
    fmt.Fprintf(w, "查询用户ID: %s", userID)
}

// invokeUserHandler 是另一个处理器，它程序化地调用 userHandler
func invokeUserHandler(w http.ResponseWriter, r *http.Request) {
    // 1. 定义要模拟访问的目标URL
    targetURL := "/users/456" // 模拟访问用户ID为456的路径

    // 2. 创建一个模拟的 HTTP 请求
    // "GET" 方法，目标URL，nil 表示没有请求体
    req := httptest.NewRequest("GET", targetURL, nil)

    // 3. 创建一个响应记录器，用于捕获目标处理器的输出
    rr := httptest.NewRecorder()

    // 4. 获取或创建 Mux 路由器实例
    // 在实际应用中，你可能已经有一个全局的路由器实例
    // 这里为了示例完整性，我们重新创建一个并注册 userHandler
    router := mux.NewRouter()
    router.HandleFunc("/users/{id}", userHandler).Methods("GET")

    // 5. 调用路由器的 ServeHTTP 方法
    // 这是核心步骤，Mux 路由器会根据 req 匹配路由，解析 {id} 变量，
    // 并将其存储到 req 的上下文中，然后调用 userHandler。
    router.ServeHTTP(rr, req)

    // 6. 处理从 userHandler 捕获到的响应
    if rr.Code == http.StatusOK {
        // 读取响应体
        bodyBytes, err := ioutil.ReadAll(rr.Result().Body)
        if err != nil {
            http.Error(w, "无法读取内部处理器响应", http.StatusInternalServerError)
            return
        }
        responseBody := string(bodyBytes)
        fmt.Fprintf(w, "成功程序化调用 userHandler。\n状态码: %d\n响应体: %s", rr.Code, responseBody)
    } else {
        // 如果目标处理器返回非200状态码
        fmt.Fprintf(w, "程序化调用 userHandler 失败。\n状态码: %d\n错误信息: %s", rr.Code, rr.Body.String())
    }
}

func main() {
    mainRouter := mux.NewRouter()
    mainRouter.HandleFunc("/users/{id}", userHandler).Methods("GET")
    mainRouter.HandleFunc("/invoke-user", invokeUserHandler).Methods("GET") // 注册调用器处理器

    fmt.Println("服务器启动，监听 :8080")
    // 为了演示，这里不实际启动HTTP服务器，但在真实场景中会这样：
    // log.Fatal(http.ListenAndServe(":8080", mainRouter))

    // 模拟访问 invoke-user 处理器来触发程序化调用
    fmt.Println("\n--- 模拟访问 /invoke-user ---")
    req := httptest.NewRequest("GET", "/invoke-user", nil)
    rr := httptest.NewRecorder()
    mainRouter.ServeHTTP(rr, req)
    fmt.Printf("外部调用 /invoke-user 的响应：\n状态码: %d\n响应体:\n%s\n", rr.Code, rr.Body.String())

    fmt.Println("\n--- 模拟直接访问 /users/123 ---")
    req2 := httptest.NewRequest("GET", "/users/123", nil)
    rr2 := httptest.NewRecorder()
    mainRouter.ServeHTTP(rr2, req2)
    fmt.Printf("外部调用 /users/123 的响应：\n状态码: %d\n响应体:\n%s\n", rr2.Code, rr2.Body.String())
}