使用第三方中间件
很多第三方库提供了不同类型的可重用的中间件组件,可以用在很多共同的功能:比如授权,登录,压缩响应头等.当使用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的值.