异常恢复

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...)