时间戳

首页 > 解决方案 > 如何使用多个方法将中间件写入接口？

go - 如何使用多个方法将中间件写入接口？

问题描述

我得到了一个具有不止一种方法的界面。我想知道如何为它编写一个中间件。

我在谷歌中寻找，但发现所有答案都是针对只有一种方法的接口。我没有发现任何问题。我尝试编写一个演示，但它不起作用。

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

那么，如何为具有多个方法的接口编写中间件呢？

标签： go

解决方案

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

// 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}
}

推荐阅读