使用golang框架中间件实现日志记录

package main

import (
    "log"
    "net/http"
)

func main() {
    // 创建一个新的 HTTP 路由器
    mux := http.NewServeMux()

    // 定义一个日志记录中间件
    mw := func(next http.Handler) http.Handler {
        return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
            // 在处理请求之前记录日志
            log.Printf("Received request: %s %s", r.Method, r.URL.Path)
            
            // 调用下一个处理器
            next.ServeHTTP(w, r)
            
            // 在处理请求之后记录日志
            log.Printf("Completed request: %s %s", r.Method, r.URL.Path)
        })
    }

    // 将中间件应用于所有 HTTP 路由
    mux.Use(mw)

    // 注册一个处理程序来处理 "/hello" 请求
    mux.HandleFunc("/hello", func(w http.ResponseWriter, r *http.Request) {
        w.Write([]byte("Hello world!"))
    })

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

go run main.go

2023/02/13 15:49:52 Received request: GET /hello
2023/02/13 15:49:52 Completed request: GET /hello