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

golang 网络编程（6）中间件

2020-06-10 02:46:12 admin 642

golang_real.jpg

在前面的分享  
[golang 网络编程（5）中间件](https://www.jianshu.com/p/877492d3bcc8)  
我们通过学习自己动手写了两个拦截器，输出日志和简单权限校验。这也是我们中间件通常要做的事。其实很多时候我们不需要自己做过多事，想喝咖啡我们还有必要去种、烘焙、研磨吗？直接麦当劳了。

不过我们简单了解一下中间件实现原理还是很有必要的。先拿日志系统举例吧。通常我们的日志需要以文件形式保存下来，而不是在控制台输出。  
我们会用到 `"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 about handler")
        fmt.Fprintf(w,"about")
    }

我们先定义两个路由器，分别在浏览器输出文本。

logFile , err := os.OpenFile("server.log",os.O_WRONLY|os.O_CREATE|os.O_APPEND,0666)
        // logFile, err := os.OpenFile("server.log", os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0666)

这里通过 go 语言提供 os 包来创建一个用于保存日志的文件`server.log`

http.Handle("/", handlers.LoggingHandler(logFile, handlers.CompressHandler(indexHandler))) 
        http.Handle("/about", handlers.LoggingHandler(logFile, handlers.CompressHandler( aboutHandler)))

这里使用了`handlers`提供的`LoggingHandler`和`CompressHandler`来将日志进行压缩后输入到文件 logFile
中，这里压缩是使用 gzip 或 Deflate 对请求返回进行压缩。

完整代码如下

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 about handler")
        fmt.Fprintf(w,"about")
    }
    
    func iconHandler(w http.ResponseWriter, r *http.Request) {
    }
    
    func main(){
        http.HandleFunc("/favicon.icon",iconHandler)
    
        indexHandler := http.HandlerFunc(index)
        aboutHandler := http.HandlerFunc(about)
    
        logFile, err := os.OpenFile("server.log",os.O_WRONLY|os.O_CREATE|os.O_APPEND,0666)
        // 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: ":8080",
        }
        log.Println("Listening...")
        server.ListenAndServe()
    }

::1 - - [24/Apr/2019:05:39:31 +0800] "GET / HTTP/1.1" 200 31
    ::1 - - [24/Apr/2019:05:39:31 +0800] "GET /favicon.ico HTTP/1.1" 200 31
    ::1 - - [24/Apr/2019:05:39:38 +0800] "GET /about HTTP/1.1" 200 31
    ::1 - - [24/Apr/2019:05:39:38 +0800] "GET /favicon.ico HTTP/1.1" 200 31

这一次我们又引入了`github.com/justinas/alice`这个第三方包，使用 alice 原因是让我们代码看起来更优雅。

commonHandlers := alice.New(loggingHandler, handlers.CompressHandler)

对我们日志进行整合封装。

package main
    
    import(
        "io"
        "log"
        "net/http"
        "os"
    
        "github.com/justinas/alice"
        "github.com/gorilla/handlers"
    )
    
    func loggingHandler(next http.Handler) http.Handler{
        
        logFile, err := os.OpenFile("server.log",os.O_WRONLY|os.O_CREATE|os.O_APPEND,0666)
    
        if err != nil{
            panic(err)
        }
    
        return handlers.LoggingHandler(logFile,next)
    }
    
    func index(w http.ResponseWriter, r *http.Request){
        w.Header().Set(
            "Content-Type",
            "text/html",
        )
        io.WriteString(
            w,
            `<doctype html>
            <html>
                <head>
                    <title>Index</title>
                </head>
                <body>
                    Hello Zidea!
                </body>
            </html>`, )
    }
    
    func about(w http.ResponseWriter, r *http.Request){
        w.Header().Set(
            "Content-Type",
            "text/html",
        )
        io.WriteString(
            w,
            `<doctype html>
            <html>
                <head>
                    <title>About</title>
                </head>
                <body>
                    about HTTP Middleware
                </body>
            </html>`, )
    }
    
    func iconHandler(w http.ResponseWriter, r *http.Request) {
        // http.ServeFile(w, r, "./favicon.ico")
    }
    
    func main(){
        http.HandleFunc("/favicon.ico", iconHandler)
        indexHandler := http.HandlerFunc(index)
        aboutHandler := http.HandlerFunc(about)
        commonHandlers := alice.New(loggingHandler, handlers.CompressHandler)
        http.Handle("/", commonHandlers.ThenFunc(indexHandler))
        http.Handle("/about", commonHandlers.ThenFunc(aboutHandler))
    
        server := &http.Server{
            Addr: ":8080",
        }
        log.Println("Listening...")
        server.ListenAndServe()
    }

::1 - - [24/Apr/2019:05:53:01 +0800] "GET /about HTTP/1.1" 200 110
    ::1 - - [24/Apr/2019:05:53:05 +0800] "GET /welcome HTTP/1.1" 200 103

th-9.jpeg

转载文章，原文链接： golang 网络编程（6）中间件

关键字词golang 中间件

分享到：

上一篇：go 语言中 RPC

下一篇：golang 基础(31) 锁

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

0 条评论 0 人参与

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

golang 网络编程（6）中间件

zhou 2年前

zhou 2年前

zhou 2年前

zhou 2年前

武云 2年前

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

golang 网络编程（6）中间件

zhou 2年前

zhou 2年前

zhou 2年前

zhou 2年前

武云 2年前

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