Golang grpc：如何从恐慌中恢复grpc服务器？

Question

我有一个这样的处理程序：

func (daemon *Daemon) List(item string) (map[string][]string, error) {
      panic("this is a panic")
      ...

我只是想恢复这种潜在的恐慌，所以在我的 grpc 服务器中我写道：

// Serve serves gRPC request by goroutines
func (s *ServerRPC) Serve(addr string) error {
    defer func() {
        if err := recover(); err != nil {
            glog.Errorf("Recovered from err: ", err)
        }
    }()

    l, err := net.Listen("tcp", addr)
    if err != nil {
        glog.Fatalf("Failed to listen %s: %v", addr, err)
        return err
    }

    return s.server.Serve(l)
}

但是它没有捕捉到恐慌，我以为是因为我的恐慌发生在子 goroutines 中？ 如果是这样，我如何才能从正常崩溃中恢复我的 grpc 服务器？

Answer 1

看一下Golang gRPC Middlewares包中的恢复拦截器。 它从请求 Goroutine 中的恐慌中恢复，并将它们作为内部 gRPC 错误返回。

myServer := grpc.NewServer(
    grpc.StreamInterceptor(grpc_middleware.ChainStreamServer(
        grpc_zap.StreamServerInterceptor(zapLogger),
        grpc_auth.StreamServerInterceptor(myAuthFunction),
        grpc_recovery.StreamServerInterceptor(),
    )),
    grpc.UnaryInterceptor(grpc_middleware.ChainUnaryServer(
        grpc_zap.UnaryServerInterceptor(zapLogger),
        grpc_auth.UnaryServerInterceptor(myAuthFunction),
        grpc_recovery.UnaryServerInterceptor(),
    )),
)