Modify error class name instead of error parameters

riyaverm-db · Aug 5, 2024 · d8f912b · d8f912b
1 parent 321940a
commit d8f912b
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/sql/catalyst/src/main/scala/org/apache/spark/sql/errors/QueryExecutionErrors.scala b/sql/catalyst/src/main/scala/org/apache/spark/sql/errors/QueryExecutionErrors.scala
@@ -2629,11 +2629,8 @@ private[sql] object QueryExecutionErrors extends QueryErrorsBase with ExecutionE
     stateStoreType: String,
     e: Throwable): Throwable = {
     new SparkException(
-      errorClass = "CANNOT_LOAD_STATE_STORE.OUT_OF_MEMORY",
-      messageParameters = Map(
-        "stateStoreType" -> stateStoreType,
-        "stateStoreId" -> stateStoreId
-        ),
+      errorClass = s"CANNOT_LOAD_STATE_STORE.${stateStoreType.toUpperCase()}_OUT_OF_MEMORY",
+      messageParameters = Map("stateStoreId" -> stateStoreId),
       cause = e
     )
   }