Impl WithGrpcStatusUnwrapper

Author: izumin5210

handlers.go

@@ -99,13 +99,10 @@ func WithStatusCodeMap(m map[int]codes.Code) interface {
 	StreamServerErrorHandler
 } {
 	return WithAppErrorHandler(func(c context.Context, err *apperrors.Error) error {
-		newCode := codes.Unknown
-		if s, ok := status.FromError(err.Err); ok {
-			newCode = s.Code()
-		} else if c, ok := m[err.StatusCode]; ok {
-			newCode = c
+		if c, ok := m[err.StatusCode]; ok {
+			return status.Error(c, err.Error())
 		}
-		return status.Error(newCode, err.Error())
+		return err
 	})
 }
 
@@ -118,3 +115,16 @@ func WithStatusCodeMapper(mapFn func(code int) codes.Code) interface {
 		return status.Error(mapFn(err.StatusCode), err.Error())
 	})
 }
+
+// WithGrpcStatusUnwrapper returns unwrapped error if this has a gRPC status.
+func WithGrpcStatusUnwrapper() interface {
+	UnaryServerErrorHandler
+	StreamServerErrorHandler
+} {
+	return WithAppErrorHandler(func(c context.Context, err *apperrors.Error) error {
+		if _, ok := status.FromError(err.Err); ok {
+			return err.Err
+		}
+		return err
+	})
+}

interceptors_test.go

@@ -287,15 +287,16 @@ func Test_UnaryServerInterceptor_WithStatusCodeMap_WhenUnknownCode(t *testing.T)
 	}
 }
 
-func Test_UnaryServerInterceptor_WithStatusCodeMap_WhenAlreadySet(t *testing.T) {
+func Test_UnaryServerInterceptor_WithGrpcStatusUnwrapper(t *testing.T) {
 	code := codes.Unauthenticated
 
 	ctx := errorstesting.CreateTestContext(t)
 	ctx.Service = &errorWithGrpcStatusService{Code: code}
 	ctx.AddUnaryServerInterceptor(
 		UnaryServerInterceptor(
+			WithGrpcStatusUnwrapper(),
 			WithStatusCodeMap(map[int]codes.Code{
-				int(code): codes.Aborted,
+				50: codes.Unavailable,
 			}),
 		),
 	)
@@ -319,6 +320,38 @@ func Test_UnaryServerInterceptor_WithStatusCodeMap_WhenAlreadySet(t *testing.T)
 	}
 }
 
+func Test_UnaryServerInterceptor_WithGrpcStatusUnwrapper_WithoutGrpcStatus(t *testing.T) {
+	code := 50
+	ctx := errorstesting.CreateTestContext(t)
+	ctx.Service = &errorWithStatusService{Code: code}
+	ctx.AddUnaryServerInterceptor(
+		UnaryServerInterceptor(
+			WithGrpcStatusUnwrapper(),
+			WithStatusCodeMap(map[int]codes.Code{
+				code: codes.Unauthenticated,
+			}),
+		),
+	)
+	ctx.Setup()
+	defer ctx.Teardown()
+
+	resp, err := ctx.Client.EmptyCall(context.Background(), &errorstesting.Empty{})
+
+	if resp != nil {
+		t.Error("The request should not return any responses")
+	}
+
+	if err == nil {
+		t.Error("The request should return an error")
+	}
+
+	if st, ok := status.FromError(err); !ok {
+		t.Error("Returned error should has status code")
+	} else if got, want := st.Code(), codes.Unauthenticated; got != want {
+		t.Errorf("Returned error had status code %v, want %v", got, want)
+	}
+}
+
 func Test_UnaryServerInterceptor_WithStatusCodeMapper(t *testing.T) {
 	code := 50
 	mappedCode := codes.Unavailable