如何使用多种方法将中间件写入接口?

时间:2019-07-04 02:26:06

标签: go

我使用了不只一种方法的界面。我不知道如何编写一个中间件。

我在Google上进行搜索,但发现所有答案仅是一种方法的界面。我没有发现任何问题。而且我尝试编写一个演示,但无法正常工作。

package main

import (
    "fmt"
    "strconv"
)

type ti interface {
    Say(int) string
    Eat(string) int
}

type Middleware func(ti) ti

func GetMiddleWare(t ti) ti {
    var tm ti
    t.Say = func(i int) string {
        fmt.Println("arg is " + strconv.Itoa(i))
        var ret string
        defer func() {
            fmt.Println("ret is " + ret)
        }()
        ret = t.Say(i)
        return ret
    }
    t.Eat = func(s string) int {
        fmt.Println("arg is " + s)
        var ret int
        defer func() {
            fmt.Println("ret is " + strconv.Itoa(ret))
        }()
        ret = t.Eat(s)
        return ret
    }
    return tm
}

它不起作用

.\main.go:17:8: cannot assign to t.Say
.\main.go:26:8: cannot assign to t.Eat

那么,如何为具有多种方法的接口编写中间件?

1 个答案:

答案 0 :(得分:0)

定义一个包装值的类型。在该类型上实现接口方法。

// Middleware wraps the value t with logging.
type Middleware struct {
    t ti
}

func (m Middleware) Say(i int) string {
    fmt.Println("arg is " + strconv.Itoa(i))
    var ret string
    defer func() {
        fmt.Println("ret is " + ret)
    }()
    ret = m.t.Say(i)
    return ret
}

func (m Middleware) Eat(s string) int {
    fmt.Println("arg is " + s)
    var ret int
    defer func() {
        fmt.Println("ret is " + strconv.Itoa(ret))
    }()
    ret = m.t.Eat(s)
    return ret
}

func GetMiddleWare(t ti) ti {
    return Middleware{t}
}