自定义服务

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

全局映射

因为 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. }

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