在GoZero框架中，可以很容易地添加和使用中间件。以下是如何添加和使用中间件的示例。
首先，在你的服务中定义一个全局的中间件管理器：
var (
Greeter = zrpc.NewServer(
zrpc.Address(":9000"),
zrpc.Timeout(time.Second*3),
)
)然后，你可以添加GoZero框架内建的中间件，比如日志、超时、限流等：
Greeter.Use(
zrpc.Logger(),
zrpc.Recovery(),
zrpc.Timeout(time.Second*3),
zrpc.RateLimit(zrpc.RateLimitOption{
Frequency: 3,
Duration: time.Second * 10,
}),
)你也可以自定义中间件。自定义中间件需要实现 znet.HandlerFunc 接口：
func MyMiddleware(fn znet.HandlerFunc) znet.HandlerFunc {
return func(ctx context.Context, req ziface.IRequest) {
// 在请求处理前执行的逻辑
fmt.Println("Before request handling")
// 调用下一个中间件或最终的处理函数
fn(ctx, req)
// 在请求处理后执行的逻辑
fmt.Println("After request handling")
}
}然后，将自定义的中间件添加到服务中：
Greeter.Use(MyMiddleware)完整示例代码：
package main
import (
"context"
"fmt"
"time"
"github.com/zeromicro/go-zero/zrpc"
"github.com/zeromicro/go-zero/zrpc/internal/znet"
"github.com/zeromicro/go-zero/zrpc/internal/ziface"
)
var (
Greeter = zrpc.NewServer(
zrpc.Address(":9000"),
zrpc.Timeout(time.Second*3),
)
)
func MyMiddleware(fn znet.HandlerFunc) znet.HandlerFunc {
return func(ctx context.Context, req ziface.IRequest) {
// 在请求处理前执行的逻辑
fmt.Println("Before request handling")
// 调用下一个中间件或最终的处理函数
fn(ctx, req)
// 在请求处理后执行的逻辑
fmt.Println("After request handling")
}
}
func main() {
Greeter.Use(
zrpc.Logger(),
zrpc.Recovery(),
zrpc.Timeout(time.Second*3),
zrpc.RateLimit(zrpc.RateLimitOption{
Frequency: 3,
Duration: time.Second * 10,
}),
MyMiddleware,
)
// ... 其他服务启动代码
}在这个示例中，我们定义了一个名为 Greeter 的RPC服务，并向它添加了内建的中间件和自定义的中间件。这样，每个请求在处理前后都会执行相应的逻辑。