[SPARK-52318] Refactor SparkConnectError to simplify case names

### What changes were proposed in this pull request?

This PR aims to refactor `SparkConnectError` to simplify case names.

### Why are the changes needed?

We had better simplify case names by removing `Exception` postfix before adding more errors like `CatalogNotFound` and `SchemaNotFound`. In `Swift`, we use like the following with `SparkConnectError.` prefix.

```swift
- throw SparkConnectError.UnsupportedOperationException
+ throw SparkConnectError.UnsupportedOperation
```

### Does this PR introduce _any_ user-facing change?

Yes, but it's the exception names for 0.x release.

### How was this patch tested?

Pass the CIs.

### Was this patch authored or co-authored using generative AI tooling?

No.

Closes #177 from dongjoon-hyun/SPARK-52318.

Authored-by: Dongjoon Hyun <[email protected]>
Signed-off-by: Dongjoon Hyun <[email protected]>

Author: dongjoon-hyun

Sources/SparkConnect/DataFrame.swift

@@ -258,7 +258,7 @@ public actor DataFrame: Sendable {
     // SQLSTATE: 0A000
     // [UNSUPPORTED_CONNECT_FEATURE.RDD]
     // Feature is not supported in Spark Connect: Resilient Distributed Datasets (RDDs).
-    throw SparkConnectError.UnsupportedOperationException
+    throw SparkConnectError.UnsupportedOperation
   }
 
   /// Return an array of column name strings

Sources/SparkConnect/Extension.swift

@@ -213,7 +213,7 @@ extension YearMonthInterval {
     case 0: "year"
     case 1: "month"
     default:
-      throw SparkConnectError.InvalidTypeException
+      throw SparkConnectError.InvalidType
     }
   }
 
@@ -225,7 +225,7 @@ extension YearMonthInterval {
     } else if startFieldName < endFieldName {
       "interval \(startFieldName) to \(endFieldName)"
     } else {
-      throw SparkConnectError.InvalidTypeException
+      throw SparkConnectError.InvalidType
     }
     return interval
   }
@@ -239,7 +239,7 @@ extension DayTimeInterval {
     case 2: "minute"
     case 3: "second"
     default:
-      throw SparkConnectError.InvalidTypeException
+      throw SparkConnectError.InvalidType
     }
   }
 
@@ -251,7 +251,7 @@ extension DayTimeInterval {
     } else if startFieldName < endFieldName {
       "interval \(startFieldName) to \(endFieldName)"
     } else {
-      throw SparkConnectError.InvalidTypeException
+      throw SparkConnectError.InvalidType
     }
     return interval
   }
@@ -325,7 +325,7 @@ extension DataType {
       case .unparsed:
         self.unparsed.dataTypeString
       default:
-        throw SparkConnectError.InvalidTypeException
+        throw SparkConnectError.InvalidType
       }
     }
   }

Sources/SparkConnect/MergeIntoWriter.swift

@@ -187,7 +187,7 @@ public actor MergeIntoWriter {
       && self.mergeIntoTableCommand.notMatchedActions.count == 0
       && self.mergeIntoTableCommand.notMatchedBySourceActions.count == 0
     {
-      throw SparkConnectError.InvalidArgumentException
+      throw SparkConnectError.InvalidArgument
     }
     self.mergeIntoTableCommand.sourceTablePlan = await (self.df.getPlan() as! Plan).root
     self.mergeIntoTableCommand.withSchemaEvolution = self.schemaEvolution

Sources/SparkConnect/ProtoUtils.swift

@@ -30,10 +30,10 @@ public enum ProtoUtils {
     // because the Spark Connect job tag is also used as part of SparkContext job tag.
     // See SparkContext.throwIfInvalidTag and ExecuteHolderSessionTag
     if tag.isEmpty {
-      throw SparkConnectError.InvalidArgumentException
+      throw SparkConnectError.InvalidArgument
     }
     if tag.contains(SPARK_JOB_TAGS_SEP) {
-      throw SparkConnectError.InvalidArgumentException
+      throw SparkConnectError.InvalidArgument
     }
   }
 }

Sources/SparkConnect/Row.swift

@@ -45,7 +45,7 @@ public struct Row: Sendable, Equatable {
 
   public func get(_ i: Int) throws -> Sendable {
     if i < 0 || i >= self.length {
-      throw SparkConnectError.InvalidArgumentException
+      throw SparkConnectError.InvalidArgument
     }
     return values[i]
   }

Sources/SparkConnect/SparkConnectClient.swift

@@ -104,7 +104,7 @@ public actor SparkConnectClient {
     try await withGPRC { client in
       // To prevent server-side `INVALID_HANDLE.FORMAT (SQLSTATE: HY000)` exception.
       if UUID(uuidString: sessionID) == nil {
-        throw SparkConnectError.InvalidSessionIDException
+        throw SparkConnectError.InvalidSessionID
       }
 
       self.sessionID = sessionID
@@ -787,7 +787,7 @@ public actor SparkConnectClient {
       } catch let error as RPCError where error.code == .internalError {
         switch error.message {
         case let m where m.contains("UNSUPPORTED_DATATYPE") || m.contains("INVALID_IDENTIFIER"):
-          throw SparkConnectError.InvalidTypeException
+          throw SparkConnectError.InvalidType
         default:
           throw error
         }

Sources/SparkConnect/SparkConnectError.swift

@@ -19,8 +19,8 @@
 
 /// A enum for ``SparkConnect`` package errors
 public enum SparkConnectError: Error {
-  case UnsupportedOperationException
-  case InvalidArgumentException
-  case InvalidSessionIDException
-  case InvalidTypeException
+  case InvalidArgument
+  case InvalidSessionID
+  case InvalidType
+  case UnsupportedOperation
 }

Sources/SparkConnect/SparkFileUtils.swift

@@ -106,7 +106,7 @@ public enum SparkFileUtils {
     if fileManager.fileExists(atPath: url.path) {
       try fileManager.removeItem(at: url)
     } else {
-      throw SparkConnectError.InvalidArgumentException
+      throw SparkConnectError.InvalidArgument
     }
   }
 }

Sources/SparkConnect/SparkSession.swift

@@ -83,7 +83,7 @@ public actor SparkSession {
       // SQLSTATE: 0A000
       // [UNSUPPORTED_CONNECT_FEATURE.SESSION_SPARK_CONTEXT]
       // Feature is not supported in Spark Connect: Access to the SparkContext.
-      throw SparkConnectError.UnsupportedOperationException
+      throw SparkConnectError.UnsupportedOperation
     }
   }
 

Sources/SparkConnect/StreamingQueryManager.swift

@@ -135,7 +135,7 @@ public actor StreamingQueryManager {
     let response = try await self.sparkSession.client.executeStreamingQueryManagerCommand(command)
     let query = response.first!.streamingQueryManagerCommandResult.query
     guard query.hasID else {
-      throw SparkConnectError.InvalidArgumentException
+      throw SparkConnectError.InvalidArgument
     }
     return StreamingQuery(
       UUID(uuidString: query.id.id)!,
@@ -154,7 +154,7 @@ public actor StreamingQueryManager {
     var awaitAnyTerminationCommand = StreamingQueryManagerCommand.AwaitAnyTerminationCommand()
     if let timeoutMs {
       guard timeoutMs > 0 else {
-        throw SparkConnectError.InvalidArgumentException
+        throw SparkConnectError.InvalidArgument
       }
       awaitAnyTerminationCommand.timeoutMs = timeoutMs
     }