fix: Use the strings to validate insteaqd of class as it is breaking

atlanhq · Oct 11, 2023 · d2f5c51 · d2f5c51
1 parent a4a1c61
commit d2f5c51
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/repository/src/main/java/org/apache/atlas/GraphTransactionInterceptor.java b/repository/src/main/java/org/apache/atlas/GraphTransactionInterceptor.java
@@ -17,7 +17,6 @@
 
 package org.apache.atlas;
 
-import com.datastax.oss.driver.api.core.servererrors.InvalidQueryException;
 import com.google.common.annotations.VisibleForTesting;
 import org.aopalliance.intercept.MethodInterceptor;
 import org.aopalliance.intercept.MethodInvocation;
@@ -174,7 +173,7 @@ public boolean checkForBatchTooLargeError(Throwable t) {
             String message = currentCause.getMessage();
             if (message != null &&
                     message.contains("Batch too large") &&
-                    currentCause.getClass().equals(InvalidQueryException.class)) {
+                    currentCause.getClass().getSimpleName().equals("InvalidQueryException")) {
                 return true;
             }
             currentCause = currentCause.getCause();