在这个小节你将会学习如果重写已存在的 Context 的方法。

    Context 是一个接口。然而你可能了解，当使用其它框架的时，即使它是一个接口，你也无法重写它。Iris 使用 app.ContextPool.Attach 方法连接你的实现到 context pool 中。

    1. 首先我们导入 github.com/kataras/iris/v12/context，这里需要它。
    2. 其次编写你自己的实现
    3. 然后添加 Do 和 Next 两个包级别函数
    4. 使用 Application 的 ContextPool 将其设置为应用于路由处理程序的Context实现。

    示例代码：

    请读注释：

    1. package main
    3. import (
    4.     "reflect"
    6.     "github.com/kataras/iris/v12"
    7.     // 1.
    8.     "github.com/kataras/iris/v12/context"
    9. )
    11. // 2.
    12. // Create your own custom Context, put any fields you'll need.
    13. type MyContext struct {
    14.     // Embed the `iris.Context` -  嵌入 iris.Context
    15.     // It's totally optional but you will need this if you  
    16.     // don't want to override all the context's methods!
    17.        // 这是可选的，但是你不想重写所有的 context的方法，需要这么做。
    18.     iris.Context
    19. }
    21. // Optionally: validate MyContext implements iris.Context on compile-time.
    22. // 可选的，在编译的时候验证是否实现 iris.Context
    23. var _ iris.Context = &MyContext{}
    25. // 3.
    26. func (ctx *MyContext) Do(handlers context.Handlers) {
    27.     context.Do(ctx, handlers)
    28. }
    1. // 3.
    2. func (ctx *MyContext) Next() {
    3.     context.Next(ctx)
    4. }
    6. // [Override any context's method you want here...]
    7. // Like the HTML below:
    9. func (ctx *MyContext) HTML(htmlContents string) (int, error) {
    10.     ctx.Application().Logger().Infof("Executing .HTML function from MyContext")
    12.     ctx.ContentType("text/html")
    13.     return ctx.WriteString(htmlContents)
    14. }
    16. func main() {
    17.     app := iris.New()
    19.     // 4.
    20.     app.ContextPool.Attach(func() iris.Context {
    21.         return &MyContext{
    22.             // If you use the embedded Context,
    23.             // call the `context.NewContext` to create one:
    24.             Context: context.NewContext(app),
    25.         }
    26.     })
    28.     // Register a view engine on .html files inside the ./view/** directory.
    29.     app.RegisterView(iris.HTML("./view", ".html"))
    31.     // Register your route, as you normally do
    32.     app.Handle("GET", "/", recordWhichContextForExample,
    33.     func(ctx iris.Context) {
    34.         // use the context's overridden HTML method.
    35.         ctx.HTML("<h1> Hello from my custom context's HTML! </h1>")
    36.     })
    38.     // This will be executed by the
    39.     // MyContext.Context embedded default context
    40.     // when MyContext is not directly define the View function by itself.
    41.     app.Handle("GET", "/hi/{firstname:alphabetical}",recordWhichContextForExample,
    42.     func(ctx iris.Context) {
    43.         firstname := ctx.Values().GetString("firstname")
    45.         ctx.ViewData("firstname", firstname)
    46.         ctx.Gzip(true)
    48.         ctx.View("hi.html")
    49.     })
    51.     app.Run(iris.Addr(":8080"))
    52. }
    54. // Should always print "($PATH) Handler is executing from 'MyContext'"
    55. func recordWhichContextForExample(ctx iris.Context) {
    56.     ctx.Application().Logger().Infof("(%s) Handler is executing from: '%s'",
    57.         ctx.Path(), reflect.TypeOf(ctx).Elem().Name())
    59.     ctx.Next()
    60. }