[SPARK-51539] Refactor SparkConnectClient to use analyze helper function

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

This PR aims to refactor `SparkConnectClient` to use `analyze` helper function.

### Why are the changes needed?

To remove the duplication code pattern via newly added `private func analyze` like `Scala Spark Connect` client.

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

No, this is an internal refactoring.

### How was this patch tested?

Pass the CIs.

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

No.

Closes #21 from dongjoon-hyun/SPARK-51539.

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

Author: dongjoon-hyun

Sources/SparkConnect/SparkConnectClient.swift

@@ -64,12 +64,9 @@ public actor SparkConnectClient {
 
       self.sessionID = sessionID
       let service = SparkConnectService.Client(wrapping: client)
-      let version = AnalyzePlanRequest.SparkVersion()
-      var request = AnalyzePlanRequest()
-      request.clientType = clientType
-      request.userContext = userContext
-      request.sessionID = self.sessionID!
-      request.analyze = .sparkVersion(version)
+      let request = analyze(self.sessionID!, {
+        return OneOf_Analyze.sparkVersion(AnalyzePlanRequest.SparkVersion())
+      })
       let response = try await service.analyzePlan(request)
       return response
     }
@@ -243,13 +240,19 @@ public actor SparkConnectClient {
   func getAnalyzePlanRequest(_ sessionID: String, _ plan: Plan) async
     -> AnalyzePlanRequest
   {
+    return analyze(sessionID, {
+      var schema = AnalyzePlanRequest.Schema()
+      schema.plan = plan
+      return OneOf_Analyze.schema(schema)
+    })
+  }
+
+  private func analyze(_ sessionID: String, _ f: () -> OneOf_Analyze) -> AnalyzePlanRequest {
     var request = AnalyzePlanRequest()
     request.clientType = clientType
     request.userContext = userContext
     request.sessionID = self.sessionID!
-    var schema = AnalyzePlanRequest.Schema()
-    schema.plan = plan
-    request.analyze = .schema(schema)
+    request.analyze = f()
     return request
   }
 

Sources/SparkConnect/TypeAliases.swift

@@ -25,6 +25,7 @@ typealias Plan = Spark_Connect_Plan
 typealias Project = Spark_Connect_Project
 typealias KeyValue = Spark_Connect_KeyValue
 typealias Limit = Spark_Connect_Limit
+typealias OneOf_Analyze = AnalyzePlanRequest.OneOf_Analyze
 typealias Range = Spark_Connect_Range
 typealias Relation = Spark_Connect_Relation
 typealias SparkConnectService = Spark_Connect_SparkConnectService