From d2f5c511567961aced4027804093248769df982e Mon Sep 17 00:00:00 2001
From: Suman Das <59254445+sumandas0@users.noreply.github.com>
Date: Wed, 11 Oct 2023 17:57:06 +0530
Subject: [PATCH] fix: Use the strings to validate insteaqd of class as it is
 breaking

---
 .../java/org/apache/atlas/GraphTransactionInterceptor.java     | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/repository/src/main/java/org/apache/atlas/GraphTransactionInterceptor.java b/repository/src/main/java/org/apache/atlas/GraphTransactionInterceptor.java
index 90e937e3f7..e34655a79b 100644
--- 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();