中间件是什么?Go 项目中引入中间件的目的和效果如何?本文详细介绍了Golang 中间件。
中间件是一种计算机 软件,可为 操作系统 提供的 软件应用程序 提供服务,以便于各个软件之间的沟通,特别是系统软件和应用软件。广泛用于 web 应用和面向服务的体系结构等。
纵观 GO 语言,中间件应用比较普遍,主要应用:
  • 记录对服务器发送的请求(request)
  • 处理服务器响应(response )
  • 请求和处理之间做一个权限认证工作
  • 远程调用
  • 安全
  • 等等
中间件处理程序是简单的http.Handler,它包装另一个http.Handler做请求的一些预处理和/或后处理。它被称为“中间件”,因为它位于 Go Web 服务器和实际处理程序之间的中间位置。
下面是一些中间件例子

记录日志中间件

package
 main


import
 (

"fmt"
"log"
"net/http"
)


funclogging(f http.HandlerFunc)http.HandlerFunc
 {

returnfunc(w http.ResponseWriter, r *http.Request)
 {

      log.Println(r.URL.Path)

      f(w, r)

   }

}

funcfoo(w http.ResponseWriter, r *http.Request)
 {

   fmt.Fprintln(w, 
"foo"
)

}


funcbar(w http.ResponseWriter, r *http.Request)
 {

   fmt.Fprintln(w, 
"bar"
)

}


funcmain()
 {

   http.HandleFunc(
"/foo"
, logging(foo))

   http.HandleFunc(
"/bar"
, logging(bar))

   http.ListenAndServe(
":8080"
nil
)

}

访问 http://localhost:8080/foo
返回结果
foo
将上面示例修改下,也可以实现相同的功能。
package
 main


import
 (

"fmt"
"log"
"net/http"
)


funcfoo(w http.ResponseWriter, r *http.Request)
 {

   fmt.Fprintln(w, 
"foo"
)

}

funcbar(w http.ResponseWriter, r *http.Request)
 {

   fmt.Fprintln(w, 
"bar"
)

}


funcloggingMiddleware(next http.Handler)http.Handler
 {

return
 http.HandlerFunc(
func(w http.ResponseWriter, r *http.Request)
 {

      log.Println(r.URL.Path)

      next.ServeHTTP(w, r)

   })

}


funcmain()
 {


   http.Handle(
"/foo"
, loggingMiddleware(http.HandlerFunc(foo)))

   http.Handle(
"/bar"
, loggingMiddleware(http.HandlerFunc(bar)))

   http.ListenAndServe(
":8080"
nil
)

}

访问 http://localhost:8080/foo
返回结果
foo

多中间件例子

package
 main


import
 (

"fmt"
"log"
"net/http"
"time"
)


type
 Middleware 
func(http.HandlerFunc)http.HandlerFunc

// Logging logs all requests with its path and the time it took to process
funcLogging()Middleware
 {


// Create a new Middleware
returnfunc(f http.HandlerFunc)http.HandlerFunc
 {


// Define the http.HandlerFunc
returnfunc(w http.ResponseWriter, r *http.Request)
 {


// Do middleware things
         start := time.Now()

deferfunc()
 { log.Println(r.URL.Path, time.Since(start)) }()


// Call the next middleware/handler in chain
         f(w, r)

      }

   }

}


// Method ensures that url can only be requested with a specific method, else returns a 400 Bad Request
funcMethod(m string)Middleware
 {


// Create a new Middleware
returnfunc(f http.HandlerFunc)http.HandlerFunc
 {


// Define the http.HandlerFunc
returnfunc(w http.ResponseWriter, r *http.Request)
 {


// Do middleware things
if
 r.Method != m {

            http.Error(w, http.StatusText(http.StatusBadRequest), http.StatusBadRequest)

return
         }


// Call the next middleware/handler in chain
         f(w, r)

      }

   }

}


// Chain applies middlewares to a http.HandlerFunc
funcChain(f http.HandlerFunc, middlewares ...Middleware)http.HandlerFunc
 {

for
 _, m := 
range
 middlewares {

      f = m(f)

   }

return
 f

}


funcHello(w http.ResponseWriter, r *http.Request)
 {

   fmt.Fprintln(w, 
"hello world"
)

}


funcmain()
 {

   http.HandleFunc(
"/"
, Chain(Hello, Method(
"GET"
), Logging()))

   http.ListenAndServe(
":8080"
nil
)

}

中间件本身只是将其http.HandlerFunc作为其参数之一,包装它并返回一个新http.HandlerFunc的服务器来调用。在这里,我们定义了一种新类型Middleware,最终可以更容易地将多个中间件链接在一起。
当然我们也可以改成如下形式
package
 main


import
 (

"fmt"
"log"
"net/http"
"time"
)


type
 Middleware 
func(http.Handler)http.Handler

funcHello(w http.ResponseWriter, r *http.Request)
 {

   fmt.Fprintln(w, 
"hello world"
)

}


funcChain(f http.Handler, mmap ...Middleware)http.Handler
 {

for
 _, m := 
range
 mmap {

      f = m(f)

   }

return
 f

}

funcMethod(m string)Middleware
 {

returnfunc(f http.Handler)http.Handler
 {

return
 http.HandlerFunc(
func(w http.ResponseWriter, r *http.Request)
 {

         log.Println(r.URL.Path)

if
 r.Method != m {

            http.Error(w, http.StatusText(http.StatusBadRequest), http.StatusBadRequest)

return
         }

         f.ServeHTTP(w, r)

      })

   }


}

funcLogging()Middleware
 {

returnfunc(f http.Handler)http.Handler
 {

return
 http.HandlerFunc(
func(w http.ResponseWriter, r *http.Request)
 {

//log.Println(r.URL.Path)
// Do middleware things
         start := time.Now()

deferfunc()
 { log.Println(r.URL.Path, time.Since(start)) }()

         f.ServeHTTP(w, r)

      })

   }

}


funcmain()
 {

   http.Handle(
"/"
, Chain(http.HandlerFunc(Hello), Method(
"GET"
), Logging()))

   http.ListenAndServe(
":8080"
nil
)

}

在 gin 框架下实现中间件

r := gin.Default() 创建带有默认中间件的路由,默认是包含 logger 和 recovery 中间件的

r :=gin.
new
()      创建带有没有中间件的路由

示例
package
 main


import
 (

"github.com/gin-gonic/gin"
"log"
"time"
)


funcLogger()gin.HandlerFunc
 {

returnfunc(c *gin.Context)
 {

      t := time.Now()

// Set example variable
      c.Set(
"example"
"12345"
)

// before request
      c.Next()

// after request
      latency := time.Since(t)

      log.Print(latency) 
//时间  0s
// access the status we are sending
      status := c.Writer.Status()

      log.Println(status) 
//状态 200
   }

}

funcmain()
 {

   r := gin.New()

   r.Use(Logger())


   r.GET(
"/test"
func(c *gin.Context)
 {

      example := c.MustGet(
"example"
).(
string
)


// it would print: "12345"
      log.Println(example)

   })


// Listen and serve on 0.0.0.0:8080
   r.Run(
":8080"
)

}

以上示例也可改为
package
 main


import
 (

"github.com/gin-gonic/gin"
"log"
"time"
)


funcLogger()gin.HandlerFunc
 {

returnfunc(c *gin.Context)
 {

      t := time.Now()

// Set example variable
      c.Set(
"example"
"12345"
)

// before request
      c.Next()

// after request
      latency := time.Since(t)

      log.Print(latency) 
//时间  0s
// access the status we are sending
      status := c.Writer.Status()

      log.Println(status) 
//状态 200
   }

}


funcmain()
 {

   r := gin.New()

   r.GET(
"/test"
, Logger(), 
func(c *gin.Context)
 {

      example := c.MustGet(
"example"
).(
string
)

// it would print: "12345"
      log.Println(example)

   })

// Listen and serve on 0.0.0.0:8080
   r.Run(
":8080"
)

}

即不用 r.use 添加中间件,直接将 Logger() 写到 r.GET 方法的参数里("/test"之后)。
更多 gin 中间件示例可参考 https://github.com/gin-gonic/gin
转自:guyan0319
segmentfault.com/a/1190000018819804
文章转载:Go开发大全

(版权归原作者所有,侵删)

点击下方“阅读原文”查看更多
继续阅读
阅读原文