Kratos 异常恢复

2022-04-25 09:53 更新

异常恢复

Recovery 中间件用于异常恢复,服务出现异常的情况下,防止程序直接退出。

配置

Recovery 中间件中提供了两个配置方法 ​WithHandler()​ 和 ​WithLogger()​。

  • WithHandler()

func WithHandler(h HandlerFunc) Option {
    return func(o *options) {
        o.handler = h
    }
}

用于设置服务异常时可以使用自定义的 ​handler ​进行处理,例如投递异常信息到 sentry。

  • WithLogger()

func WithLogger(logger log.Logger) Option {
    return func(o *options) {
        o.logger = logger
    }
}

用于设置中间件打印日志时使用的 ​logger​。

使用方法

http

var opts = []http.ServerOption{
    http.Middleware(
        recovery.Recovery(
      recovery.WithLogger(log.DefaultLogger),
            recovery.WithHandler(func(ctx context.Context, req, err interface{}) error {
                    // do someting
                    return nil
            }),
    ),
    ),
}
srv := http.NewServer(opts...)

grpc

var opts = []grpc.ServerOption{
    grpc.Middleware(
        recovery.Recovery(
            recovery.WithLogger(log.DefaultLogger),
            recovery.WithHandler(func(ctx context.Context, req, err interface{}) error {
                // do someting
                return nil
            }),
        ),
    ),
}
srv := grpc.NewServer(opts...)


以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号