[SPARK-51481] Add RuntimeConf actor

Author: dongjoon-hyun

Sources/SparkConnect/RuntimeConf.swift

@@ -0,0 +1,50 @@
+//
+// Licensed to the Apache Software Foundation (ASF) under one
+// or more contributor license agreements.  See the NOTICE file
+// distributed with this work for additional information
+// regarding copyright ownership.  The ASF licenses this file
+// to you under the Apache License, Version 2.0 (the
+// "License"); you may not use this file except in compliance
+// with the License.  You may obtain a copy of the License at
+//
+//  http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing,
+// software distributed under the License is distributed on an
+// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+// KIND, either express or implied.  See the License for the
+// specific language governing permissions and limitations
+// under the License.
+//
+
+/// User-facing configuration API, accessible through `SparkSession.conf`.
+public actor RuntimeConf {
+  private let client: SparkConnectClient
+
+  /// Create a `RuntimeConf` instance with the given client.
+  /// - Parameter client: A client to talk to the Spark Connect server.
+  init(_ client: SparkConnectClient) {
+    self.client = client
+  }
+
+  /// Set a new configuration.
+  /// - Parameters:
+  ///   - key: A string for the configuration key.
+  ///   - value: A string for the configuration value.
+  public func set(_ key: String, _ value: String) async throws {
+    _ = try await client.setConf(map: [key: value])
+  }
+
+  /// Get a configuration.
+  /// - Parameter key: A string for the configuration look-up.
+  /// - Returns: A string for the configuration.
+  public func get(_ key: String) async throws -> String {
+    return try await client.getConf(key)
+  }
+
+  /// Get all configurations.
+  /// - Returns: A map of configuration key-values.
+  public func getAll() async throws -> [String: String] {
+    return try await client.getConfAll()
+  }
+}

Tests/SparkConnectTests/RuntimeConfTests.swift

@@ -0,0 +1,58 @@
+//
+// Licensed to the Apache Software Foundation (ASF) under one
+// or more contributor license agreements.  See the NOTICE file
+// distributed with this work for additional information
+// regarding copyright ownership.  The ASF licenses this file
+// to you under the Apache License, Version 2.0 (the
+// "License"); you may not use this file except in compliance
+// with the License.  You may obtain a copy of the License at
+//
+//  http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing,
+// software distributed under the License is distributed on an
+// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+// KIND, either express or implied.  See the License for the
+// specific language governing permissions and limitations
+// under the License.
+//
+
+import Foundation
+import Testing
+
+@testable import SparkConnect
+
+/// A test suite for `RuntimeConf`
+@Suite(.serialized)
+struct RuntimeConfTests {
+  @Test
+  func get() async throws {
+    let client = SparkConnectClient(remote: "sc://localhost", user: "test")
+    _ = try await client.connect(UUID().uuidString)
+    let conf = RuntimeConf(client)
+    #expect(try await conf.get("spark.app.name") == "Spark Connect server")
+  }
+
+  @Test
+  func set() async throws {
+    let client = SparkConnectClient(remote: "sc://localhost", user: "test")
+    _ = try await client.connect(UUID().uuidString)
+    let conf = RuntimeConf(client)
+    try await conf.set("spark.test.key1", "value1")
+    #expect(try await conf.get("spark.test.key1") == "value1")
+  }
+
+  @Test
+  func getAll() async throws {
+    let client = SparkConnectClient(remote: "sc://localhost", user: "test")
+    _ = try await client.connect(UUID().uuidString)
+    let conf = RuntimeConf(client)
+    let map = try await conf.getAll()
+    print(map)
+    #expect(map.count > 0)
+    #expect(map["spark.app.id"] != nil)
+    #expect(map["spark.app.startTime"] != nil)
+    #expect(map["spark.executor.id"] == "driver")
+    #expect(map["spark.master"] != nil)
+  }
+}