feat : add exceptionResolver

rajadilipkolli · Mar 10, 2024 · f1c9143 · f1c9143
1 parent 5fd5eda
commit f1c9143
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 7 deletions.
diff --git a/...-graphql-webflux/src/main/java/com/example/graphql/exception/CustomExceptionResolver.java b/...-graphql-webflux/src/main/java/com/example/graphql/exception/CustomExceptionResolver.java
@@ -13,12 +13,16 @@ public class CustomExceptionResolver extends DataFetcherExceptionResolverAdapter
 
     @Override
     protected GraphQLError resolveToSingleError(Throwable ex, DataFetchingEnvironment env) {
-        ConstraintViolationException validationException = (ConstraintViolationException) ex;
-        return GraphqlErrorBuilder.newError()
-                .errorType(ErrorType.ValidationError)
-                .message(validationException.getMessage())
-                .path(env.getExecutionStepInfo().getPath())
-                .location(env.getField().getSourceLocation())
-                .build();
+
+        if (ex instanceof ConstraintViolationException) {
+            return GraphqlErrorBuilder.newError()
+                    .errorType(ErrorType.ValidationError)
+                    .message(ex.getMessage())
+                    .path(env.getExecutionStepInfo().getPath())
+                    .location(env.getField().getSourceLocation())
+                    .build();
+        } else {
+            return null;
+        }
     }
 }
diff --git a/...t-graphql-webmvc/src/main/java/com/example/graphql/exception/CustomExceptionResolver.java b/...t-graphql-webmvc/src/main/java/com/example/graphql/exception/CustomExceptionResolver.java
@@ -0,0 +1,28 @@
+package com.example.graphql.exception;
+
+import graphql.ErrorType;
+import graphql.GraphQLError;
+import graphql.GraphqlErrorBuilder;
+import graphql.schema.DataFetchingEnvironment;
+import jakarta.validation.ConstraintViolationException;
+import org.springframework.graphql.execution.DataFetcherExceptionResolverAdapter;
+import org.springframework.stereotype.Component;
+
+@Component
+public class CustomExceptionResolver extends DataFetcherExceptionResolverAdapter {
+
+    @Override
+    protected GraphQLError resolveToSingleError(Throwable ex, DataFetchingEnvironment env) {
+
+        if (ex instanceof ConstraintViolationException) {
+            return GraphqlErrorBuilder.newError()
+                    .errorType(ErrorType.ValidationError)
+                    .message(ex.getMessage())
+                    .path(env.getExecutionStepInfo().getPath())
+                    .location(env.getField().getSourceLocation())
+                    .build();
+        } else {
+            return null;
+        }
+    }
+}