feat: add middlewares config for zrpc (#2766)

* feat: add middlewares config for zrpc

* chore: add tests

* chore: improve codecov

* chore: improve codecov

zrpc/internal/serverinterceptors/recoverinterceptor.go

@@ -0,0 +1,42 @@
+package serverinterceptors
+
+import (
+	"context"
+	"runtime/debug"
+
+	"github.com/zeromicro/go-zero/core/logx"
+	"google.golang.org/grpc"
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/status"
+)
+
+// StreamRecoverInterceptor catches panics in processing stream requests and recovers.
+func StreamRecoverInterceptor(svr interface{}, stream grpc.ServerStream, _ *grpc.StreamServerInfo,
+	handler grpc.StreamHandler) (err error) {
+	defer handleCrash(func(r interface{}) {
+		err = toPanicError(r)
+	})
+
+	return handler(svr, stream)
+}
+
+// UnaryRecoverInterceptor catches panics in processing unary requests and recovers.
+func UnaryRecoverInterceptor(ctx context.Context, req interface{}, _ *grpc.UnaryServerInfo,
+	handler grpc.UnaryHandler) (resp interface{}, err error) {
+	defer handleCrash(func(r interface{}) {
+		err = toPanicError(r)
+	})
+
+	return handler(ctx, req)
+}
+
+func handleCrash(handler func(interface{})) {
+	if r := recover(); r != nil {
+		handler(r)
+	}
+}
+
+func toPanicError(r interface{}) error {
+	logx.Errorf("%+v\n\n%s", r, debug.Stack())
+	return status.Errorf(codes.Internal, "panic: %v", r)
+}