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

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