自定义服务

服务即是被注入到处理器中的参数. 你可以映射一个服务到 全局 或者 请求 的级别.

全局映射

因为 Macaron 实现了 inject.Injector 的接口, 那么映射一个服务就变得非常简单:

  1. db := &MyDatabase{}
  2. m := macaron.Classic()
  3. m.Map(db) // Service will be available to all handlers as *MyDatabase
  4. m.Get("/", func(db *MyDatabase) {
  5.     // Operations with db.
  6. })
  7. m.Run()

请求级别的映射

映射在请求级别的服务可以通过 *macaron.Context 来完成:

  1. func MyCustomLoggerHandler(ctx *macaron.Context) {
  2.     logger := &MyCustomLogger{ctx.Req}
  3.     ctx.Map(logger) // mapped as *MyCustomLogger
  4. }
  5.  
  6. func main() {
  7.     //...
  8.     m.Get("/", MyCustomLoggerHandler, func(logger *MyCustomLogger) {
  9.         // Operations with logger.
  10.     })
  11.     m.Get("/panic", func(logger *MyCustomLogger) {
  12.         // This will panic because no logger service maps to this request.
  13.     })
  14.     //...
  15. }

映射值到接口

关于服务最强悍的地方之一就是它能够映射服务到接口. 例如说, 假设你想要覆盖 http.ResponseWriter 成为一个对象, 那么你可以封装它并包含你自己的额外操作, 你可以如下这样来编写你的处理器:

  1. func WrapResponseWriter(ctx *macaron.Context) {
  2.     rw := NewSpecialResponseWriter(ctx.Resp)
  3.     // override ResponseWriter with our wrapper ResponseWriter
  4.     ctx.MapTo(rw, (*http.ResponseWriter)(nil)) 
  5. }

如此一来，您不仅可以修改自定义的实现而不对客户代码做任何修改，还可以允许对于相同类型的服务使用多种实现。