Go HTTP Middleware学习2

使用第三方中间件

很多第三方库提供了不同类型的可重用的中间件组件,可以用在很多共同的功能:比如授权,登录,压缩响应头等.当使用Go开发一个web应用,可以使用这些第三方库应用到自己的项目中.

• 使用Gorilla Handlers

这个开发组件提供了很多net/http包下的handlers.
一个使用Gorilla的LoggingHandler和CompressHandler的例子

// Gorilla Handlers
package main

import (
    "fmt"
    "log"
    "net/http"
    "os"

    "github.com/gorilla/handlers"
)

func index(w http.ResponseWriter, r *http.Request) {
    log.Println("Execute index Handler")
    fmt.Fprintf(w, "Welcome!")

}

func about(w http.ResponseWriter, r *http.Request) {
    log.Println("Execute message Handler")
    fmt.Fprintf(w, "Message Go!")
}

func main() {
    indexHandler := http.HandlerFunc(index)
    aboutHandler := http.HandlerFunc(about)

    logFile, err := os.OpenFile("server.log", os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0666)
    if err != nil {
        panic(err)
    }

    http.Handle("/", handlers.LoggingHandler(logFile, handlers.CompressHandler(indexHandler)))
    http.Handle("about", handlers.LoggingHandler(logFile, handlers.CompressHandler(aboutHandler)))
    server := &http.Server{
        Addr: ":9090",
    }
    log.Println("Listening...")
    server.ListenAndServe()

}

//运行时访问日志信息server.log
::1 - - [29/Jul/2016:18:48:22 +0800] "GET / HTTP/1.1" 200 32
::1 - - [29/Jul/2016:18:48:25 +0800] "GET / HTTP/1.1" 200 32
::1 - - [29/Jul/2016:18:48:38 +0800] "GET /about HTTP/1.1" 200 32
::1 - - [02/Aug/2016:08:24:51 +0800] "GET / HTTP/1.1" 200 32
::1 - - [02/Aug/2016:08:24:51 +0800] "GET /favicon.ico HTTP/1.1" 200 32
::1 - - [02/Aug/2016:08:24:54 +0800] "GET / HTTP/1.1" 200 32
::1 - - [02/Aug/2016:08:24:54 +0800] "GET /favicon.ico HTTP/1.1" 200 32
::1 - - [02/Aug/2016:08:25:35 +0800] "GET /about HTTP/1.1" 200 32
::1 - - [02/Aug/2016:08:25:35 +0800] "GET /favicon.ico HTTP/1.1" 200 32

此外,还有很多的第三方库,可以到Github查找对应的文档,查阅. Alice;Negroni.

• 在中间件分享值

有时候,我们需要提供值给下一个handler或者在应用的handler和中间件的handler共享值.比如,当通过一个中间件的handler授权应用的权限,就需要在请求的handler环中提供用户的信息给下一个handler.

使用Gorilla上下文.

很多第三库允许在请求的生命周期里保存值,用来共享,Gorilla的context包是一个非常好的选择,用来在请求的生命周期里共享数据.

使用context的例子

package main

import (
    "fmt"
    "log"
    "net/http"

    "github.com/codegangsta/negroni"
    "github.com/gorilla/context"
)

func Authorize(w http.ResponseWriter, r *http.Request, next http.HandlerFunc) {
    token := r.Header.Get("X-AppToken")
    if token == "1l9o9v0e" {
        log.Printf("Authorized to the system!")
        context.Set(r, "user", "xiaoming")
        next(w, r)
    } else {
        http.Error(w, "Not Authorized", 401)
    }
}

func index(w http.ResponseWriter, r *http.Request) {
    user := context.Get(r, "user")
    fmt.Fprintf(w, "Welcome %s!", user)
}

func main() {
    mux := http.NewServeMux()
    mux.HandleFunc("/", index)
    n := negroni.Classic()
    n.Use(negroni.HandlerFunc(Authorize))
    n.UseHandler(mux)
    n.Run(":9090")
}

一个中间件的handler被创建,用来验证HTTP请求,HTTP头 的 "X-AppToken"从请求对象读取的验证token.这也是RESTful APIs验证的方式之一:客户端必须在请求头发送一个验证的token.还有的方式是通过session验证.
在这个程序,我们想传递用户名给下一个handler:应用的handler.因此在context对象设置了user的值,这个值在请求的生命周期里面都可以访问到.比如这里的index handler就访问这个user的值.