循环为Go ServeMux绑定处理器失效问题及优化方案

错误原因 #

问题出在Go语言for循环的迭代变量复用机制和闭包的引用捕获特性上：

Go的for循环（遍历切片或map时）只会创建一个迭代变量，每次迭代都会覆盖这个变量的内容。如果在循环内为ServeMux绑定处理器时直接通过闭包引用该迭代变量，所有注册的处理器最终都会指向循环结束时迭代变量的最后值——因为闭包捕获的是变量的内存地址，而非当前迭代的数值副本。

典型错误代码示例：

type Route struct {
    Path    string
    Handler http.HandlerFunc
}

routes := []Route{
    {"/apple/auth", appleAuthHandler},
    {"/banana/auth", bananaAuthHandler},
    {"/banana/description", bananaDescHandler},
}

mux := http.NewServeMux()
for _, route := range routes {
    // 错误：闭包捕获迭代变量route的引用
    mux.HandleFunc(route.Path, func(w http.ResponseWriter, r *http.Request) {
        route.Handler(w, r)
    })
}

循环结束后，route变量保存切片最后一个元素的值，所有路由请求都会被对应处理器处理；若遍历顺序因map等因素变化，就会出现所有请求指向/banana/auth或/banana/description的情况。

避免冗余手动绑定的优化方案 #

无需手动逐个绑定，解决循环变量的引用捕获问题即可安全批量绑定路由：

1. 迭代内创建变量副本 #

在循环内部把迭代变量的值复制到新的局部变量，让闭包捕获这个独立的局部变量：

for _, route := range routes {
    currentRoute := route // 创建当前迭代的变量副本
    mux.HandleFunc(currentRoute.Path, func(w http.ResponseWriter, r *http.Request) {
        currentRoute.Handler(w, r)
    })
}

2. 直接传递处理器（无需闭包） #

如果不需要额外包装逻辑，直接将迭代变量的处理器传给HandleFunc即可。Go的函数类型是值传递，每次迭代都会传递当前函数值的副本：

for _, route := range routes {
    mux.HandleFunc(route.Path, route.Handler)
}

3. 使用辅助函数生成处理器 #

编写辅助函数接收处理器并返回绑定当前值的闭包，从根源避免引用捕获问题：

func wrapHandler(h http.HandlerFunc) http.HandlerFunc {
    // 函数参数h是当前迭代的处理器副本，闭包捕获该副本
    return func(w http.ResponseWriter, r *http.Request) {
        h(w, r)
    }
}

// 循环中使用
for _, route := range routes {
    mux.HandleFunc(route.Path, wrapHandler(route.Handler))
}

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