move partition error

Author: eason-yuchen-liu

common/utils/src/main/resources/error/error-conditions.json

@@ -244,6 +244,11 @@
           "Error reading streaming state file of <fileToRead> does not exist. If the stream job is restarted with a new or updated state operation, please create a new checkpoint location or clear the existing checkpoint location."
         ]
       },
+      "SNAPSHOT_PARTITION_ID_NOT_FOUND" : {
+        "message" : [
+          "Partition id <snapshotPartitionId> not found for given state source."
+        ]
+      },
       "UNCATEGORIZED" : {
         "message" : [
           ""
@@ -3621,12 +3626,6 @@
     ],
     "sqlState" : "42K08"
   },
-  "SNAPSHOT_PARTITION_ID_NOT_FOUND" : {
-    "message" : [
-      "Partition id <snapshotPartitionId> not found for given state source."
-    ],
-    "sqlState" : "54054"
-  },
   "SORT_BY_WITHOUT_BUCKETING" : {
     "message" : [
       "sortBy must be used together with bucketBy."

sql/catalyst/src/main/scala/org/apache/spark/sql/errors/QueryExecutionErrors.scala

@@ -2200,7 +2200,7 @@ private[sql] object QueryExecutionErrors extends QueryErrorsBase with ExecutionE
 
   def snapshotPartitionNotFoundError(snapshotPartitionId : Long): Throwable = {
     new SparkException(
-      errorClass = "SNAPSHOT_PARTITION_ID_NOT_FOUND",
+      errorClass = "CANNOT_LOAD_STATE_STORE.SNAPSHOT_PARTITION_ID_NOT_FOUND",
       messageParameters = Map("snapshotPartitionId" -> snapshotPartitionId.toString()),
       cause = null)
   }

sql/core/src/test/scala/org/apache/spark/sql/execution/datasources/v2/state/StateDataSourceReadSuite.scala

@@ -1010,7 +1010,7 @@ abstract class StateDataSourceReadSuite[storeProvider <: StateStoreProvider]
       val exc = intercept[SparkException] {
         stateDfError.show()
       }
-      checkError(exc, "SNAPSHOT_PARTITION_ID_NOT_FOUND", "54054",
+      checkError(exc, "CANNOT_LOAD_STATE_STORE.SNAPSHOT_PARTITION_ID_NOT_FOUND", "58030",
         Map("snapshotPartitionId" -> spark.sessionState.conf.numShufflePartitions.toString))
     })
   }