网站首页 返回列表像“草根”一样，紧贴着地面，低调的存在，冬去春来，枯荣无恙。

Middleware in Golang

2020-06-10 11:40:53 admin 1116

有时候我们需要统计web service接口的数据，比如记录日志、统计API调用时间、或者对HandleFunc进行错误处理，

这个时候，middleware就很有帮助。

## 最初版本

我们目前有个程序，监听8080端口，提供两个接口，

1. compute：进行计算，耗时在500ms-1000ms之间
    2. version：得到版本号，耗时在10ms-60ms之间

调用结果如下，

# curl http://localhost:8080/compute
    finish

## 增加日志

需要记录每次调用接口的名字。

有两种方式记录日志，

1. 在每一个HandleFunc第一行输出URI
    2. 增加中间件封装HandleFunc

对于第一种情况，修改Compute()函数为，

// Compute is compute process for doing task
    func Compute(w http.ResponseWriter, r *http.Request) {
        log.Printf(r.RequestURI)
    
        time.Sleep(time.Duration((500 + rand.Intn(500))) * time.Millisecond)
        w.Write([]byte("finish"))
    }

运行得到结果为，

2017/12/11 23:07:24 /compute

对于API少的情况，还比较适用。但是对于API接口比较多的情况，修改每一个函数就不太合适。并且当我们不光想统计URI信息时，还需要统计每一个调用接口的其他信息时，就需要修改每一处HandleFunc。

所以我们可以使用`Middleware`形式来完成。

我们希望使用middlware以后，对于main()函数变为，

func main() {
        http.HandleFunc("/compute", middleware(Compute))
        http.HandleFunc("/version", middleware(Version))
    
        log.Panic(http.ListenAndServe(":8080", nil))
    }

使用middlware对每一个http handle func进行封装，在middleware中进行相应的需求处理，比如日志记录、错误处理、用时统计等。

func middleware(fn func(w http.ResponseWriter, r *http.Request)) func(w http.ResponseWriter, r *http.Request) {
        return func(w http.ResponseWriter, r *http.Request) {
            begin := time.Now()
            defer func() {
                log.Printf("[%s] %s, time_used: %v", r.Method, r.URL.String(), time.Now().Sub(begin))
            }()
    
            fn(w, r)
        }
    }

`middlware`入参和返回值都为`func(w http.ResponseWriter, r *http.Request)`类型。

在`middleware`中，进行了调用接口URI的统计和用时统计。

完整的代码如下，

package main
    
    import (
        "log"
        "math/rand"
        "net/http"
        "time"
    )
    
    // Compute is compute process for doing task
    func Compute(w http.ResponseWriter, r *http.Request) {
        time.Sleep(time.Duration((500 + rand.Intn(500))) * time.Millisecond)
        w.Write([]byte("finish"))
    }
    
    // Version is getting version task
    func Version(w http.ResponseWriter, r *http.Request) {
        time.Sleep(time.Duration((10 + rand.Intn(50))) * time.Millisecond)
        w.Write([]byte("version"))
    }
    
    func middleware(fn func(w http.ResponseWriter, r *http.Request)) func(w http.ResponseWriter, r *http.Request) {
        return func(w http.ResponseWriter, r *http.Request) {
            begin := time.Now()
            defer func() {
                log.Printf("[%s] %s, time_used: %v", r.Method, r.URL.String(), time.Now().Sub(begin))
            }()
    
            fn(w, r)
        }
    }
    
    func main() {
        http.HandleFunc("/compute", middleware(Compute))
        http.HandleFunc("/version", middleware(Version))
    
        log.Panic(http.ListenAndServe(":8080", nil))
    }

测试

curl http://localhost:8080/compute
    # 2017/12/13 16:57:15 [GET] /compute, time_used: 583.357048ms
    
    curl http://localhost:8080/version
    # 2017/12/13 16:57:24 [GET] /version, time_used: 47.09209ms

## Source code in github.com

* [https://github.com/xpzouying/go-practice/blob/master/web_service_middleware_in_golang/main.go](https://link.jianshu.com?t=https://github.com/xpzouying/go-practice/blob/master/web_service_middleware_in_golang/main.go)

转载文章，原文链接： Middleware in Golang

关键字词golang Middleware

分享到：

上一篇：[译] HTTP Request Contexts & Go

下一篇：interface{}、类型断言

如需留言，请登录，没有账号？请注册

0 条评论 0 人参与

网站首页 返回列表像“草根”一样，紧贴着地面，低调的存在，冬去春来，枯荣无恙。

Middleware in Golang

zhou 3年前

zhou 3年前

zhou 3年前

zhou 3年前

武云 3年前

网站首页 返回列表 像“草根”一样，紧贴着地面，低调的存在，冬去春来，枯荣无恙。

Middleware in Golang

zhou 3年前

zhou 3年前

zhou 3年前

zhou 3年前

武云 3年前

网站首页返回列表像“草根”一样，紧贴着地面，低调的存在，冬去春来，枯荣无恙。