Golang进阶 —— 中间件的设计

package main

import (
    "net/http"
)

// 定义一个中间件处理函数
type Middleware func(http.HandlerFunc) http.HandlerFunc

// 应用中间件
func ApplyMiddleware(fn http.HandlerFunc, middlewares ...Middleware) http.HandlerFunc {
    for _, m := range middlewares {
        fn = m(fn)
    }
    return fn
}

// 示例中间件
func MiddlewareExample(next http.HandlerFunc) http.HandlerFunc {
    return func(w http.ResponseWriter, r *http.Request) {
        // 在处理请求之前执行的代码
        println("Before request handling")

        // 调用下一个中间件或处理函数
        next(w, r)

        // 在处理请求之后执行的代码
        println("After request handling")
    }
}

// 示例处理函数
func Handler(w http.ResponseWriter, r *http.Request) {
    w.Write([]byte("Hello, World!"))
}

func main() {
    // 创建一个服务 mux
    mux := http.NewServeMux()

    // 注册路由，应用中间件
    mux.HandleFunc("/", ApplyMiddleware(Handler, MiddlewareExample))

    // 启动HTTP服务器
    http.ListenAndServe(":8080", mux)
}