golang 从零实现一个中间件(middleware)

最新推荐文章于 2024-08-14 21:36:42 发布

只会打野怪我咯

最新推荐文章于 2024-08-14 21:36:42 发布

阅读量6k

点赞数 2

分类专栏： GO GO语言测试实例文章标签： go middleware

本文链接：https://blog.csdn.net/u014270740/article/details/91401698

版权

本文介绍了在Golang中实现中间件的背景、代码示例及应用场景。通过示例展示了如何在HTTP接口服务中添加服务耗时处理日志，讨论了随着业务增长，中间件在简化代码维护方面的重要性。同时，提供了参考资源，如gin框架的中间件实现和《GO高级编程》。

摘要由CSDN通过智能技术生成

实现中间件的背景

先看如下代码

package main

func hello(wr http.ResponseWriter, r *http.Request) {
    wr.Write([]byte("hello"))
}

func main() {
    http.HandleFunc("/", hello)
    err := http.ListenAndServe(":8080", nil)
    ...
}

这是一个简单的HTTP接口服务

现在有了一个新需求，需要在接口上增加服务的耗时处理日志，我们对上面的程序进行少量修改

func hello(wr http.ResponseWriter, r *http.Request) {
    timeStart := time.Now()
    wr.Write([]byte("hello"))
    timeElapsed := time.Since(timeStart)
    fmt.Println(timeElapsed)
}

随着业务的逐渐增多后，路由可能变成了

package main

func helloHandler(wr http.ResponseWriter, r *http.Request) {
    // ...
}

func showInfoHandler(wr http.ResponseWriter, r *http.Request) {
    // ...
}

func showEmailHandler(wr http.ResponseWriter, r *http.Request) {
    // ...
}

func showFriendsHandler(wr http.ResponseWriter, r *http.Request) {
    timeStart := time.Now()
    wr.Write([]byte("your friends is tom and alex"))
    timeElapsed := time.Since(timeStart)
    logger.Println(timeElapsed)
}

func main() {
    http.HandleFunc("/", helloHandler)
    http.HandleFunc("/info/show", showInfoHandler)
    http.HandleFunc(