Add isolation

Author: glbrntt

Sources/GRPCCore/GRPCClient.swift

@@ -386,11 +386,13 @@ public final class GRPCClient: Sendable {
 public func withGRPCClient<Result: Sendable>(
   transport: some ClientTransport,
   interceptors: [any ClientInterceptor] = [],
+  isolation: isolated (any Actor)? = #isolation,
   handleClient: (GRPCClient) async throws -> Result
 ) async throws -> Result {
   try await withGRPCClient(
     transport: transport,
     interceptorPipeline: interceptors.map { .apply($0, to: .all) },
+    isolation: isolation,
     handleClient: handleClient
   )
 }
@@ -410,6 +412,7 @@ public func withGRPCClient<Result: Sendable>(
 public func withGRPCClient<Result: Sendable>(
   transport: some ClientTransport,
   interceptorPipeline: [ClientInterceptorPipelineOperation],
+  isolation: isolated (any Actor)? = #isolation,
   handleClient: (GRPCClient) async throws -> Result
 ) async throws -> Result {
   try await withThrowingDiscardingTaskGroup { group in

Sources/GRPCCore/GRPCServer.swift

@@ -258,12 +258,14 @@ public func withGRPCServer<Result: Sendable>(
   transport: any ServerTransport,
   services: [any RegistrableRPCService],
   interceptors: [any ServerInterceptor] = [],
+  isolation: isolated (any Actor)? = #isolation,
   handleServer: (GRPCServer) async throws -> Result
 ) async throws -> Result {
   try await withGRPCServer(
     transport: transport,
     services: services,
     interceptorPipeline: interceptors.map { .apply($0, to: .all) },
+    isolation: isolation,
     handleServer: handleServer
   )
 }
@@ -285,6 +287,7 @@ public func withGRPCServer<Result: Sendable>(
   transport: any ServerTransport,
   services: [any RegistrableRPCService],
   interceptorPipeline: [ServerInterceptorPipelineOperation],
+  isolation: isolated (any Actor)? = #isolation,
   handleServer: (GRPCServer) async throws -> Result
 ) async throws -> Result {
   return try await withThrowingDiscardingTaskGroup { group in

Tests/GRPCInProcessTransportTests/ClientServerWithMethods.swift

@@ -0,0 +1,44 @@
+/*
+ * Copyright 2024, gRPC Authors All rights reserved.
+ *
+ * Licensed 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 GRPCCore
+import GRPCInProcessTransport
+import Testing
+
+@Suite("withGRPCServer / withGRPCClient")
+struct WithMethods {
+  @Test("Actor isolation")
+  func actorIsolation() async throws {
+    let testActor = TestActor()
+    #expect(await !testActor.hasRun)
+    try await testActor.run()
+    #expect(await testActor.hasRun)
+  }
+}
+
+fileprivate actor TestActor {
+  private(set) var hasRun = false
+
+  func run() async throws {
+    let inProcess = InProcessTransport()
+
+    try await withGRPCServer(transport: inProcess.server, services: []) { server in
+      try await withGRPCClient(transport: inProcess.client) { client in
+        self.hasRun = true
+      }
+    }
+  }
+}