initial import

core/rpc/serverinterceptors/crashinterceptor.go

@@ -0,0 +1,43 @@
+package serverinterceptors
+
+import (
+	"context"
+	"runtime/debug"
+
+	"zero/core/logx"
+
+	"google.golang.org/grpc"
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/status"
+)
+
+func StreamCrashInterceptor(srv interface{}, stream grpc.ServerStream, info *grpc.StreamServerInfo,
+	handler grpc.StreamHandler) (err error) {
+	defer handleCrash(func(r interface{}) {
+		err = toPanicError(r)
+	})
+
+	return handler(srv, stream)
+}
+
+func UnaryCrashInterceptor() grpc.UnaryServerInterceptor {
+	return func(ctx context.Context, req interface{}, info *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 %s", r, debug.Stack())
+	return status.Errorf(codes.Internal, "panic: %v", r)
+}