Files
swift-mirror/test/Distributed/Runtime/distributed_actor_localSystem_distributedProtocol.swift
Konrad `ktoso` Malawski f48384bc7e [Distributed] Correct distributed accessor mangling for protocol calls
This is a crucial fix without which we can crash on some distributed
protocol declarations with @Resolvable. We cannot "just" use a String to
represent the "fake base" of the thunks, and must instead find the
$Target macro generated type and use it as the base of the thunk's
mangling.

Calls are made in such way that record for the protocol requirement:
`$s4main28GreeterDefinedSystemProtocolP5greetSSyYaKFTEHF` points at
`$$s4main29$GreeterDefinedSystemProtocolC5greetSSyYaKFTE` which makes a
dispatch through the _apropriate_ witness table.
And the record for the $witness named e.g.
`$s4main29$GreeterDefinedSystemProtocolC5greetSSyYaKFTEHF` points to
`$s4main28GreeterDefinedSystemProtocolPAA11Distributed01_F9ActorStubRzrlE5greetSSyYaKFTE`
which is an extension method: `distributed thunk (extension in main):main.GreeterDefinedSystemProtocol< where A: Distributed._DistributedActorStub>.greet() async throws -> Swift.String`,
this very specific design allows us to call the "right method" on the
recieving end of a remote call where we do not know the recipient type.
2024-12-03 14:59:15 +09:00

71 lines
2.0 KiB
Swift

// RUN: %empty-directory(%t)
// RUN: %target-build-swift -module-name main -j2 -parse-as-library -I %t %s -plugin-path %swift-plugin-dir -o %t/a.out
// RUN: %target-codesign %t/a.out
// RUN: %target-run %t/a.out | %FileCheck %s --color
// REQUIRES: executable_test
// REQUIRES: concurrency
// REQUIRES: distributed
// rdar://76038845
// UNSUPPORTED: use_os_stdlib
// UNSUPPORTED: back_deployment_runtime
// rdar://90373022
// UNSUPPORTED: OS=watchos
// rdar://125628060
// UNSUPPORTED: CPU=arm64e
import Distributed
@Resolvable
@available(SwiftStdlib 6.0, *)
protocol WorkerProtocol: DistributedActor where ActorSystem == LocalTestingDistributedActorSystem {
distributed func distributedMethod() -> String
distributed var distributedVariable: String { get }
distributed func genericMethod<E: Codable>(_ value: E) async -> E
}
@available(SwiftStdlib 6.0, *)
distributed actor Worker: WorkerProtocol {
distributed func distributedMethod() -> String {
"implemented method"
}
distributed var distributedVariable: String {
"implemented variable"
}
distributed func genericMethod<E: Codable>(_ value: E) async -> E {
return value
}
}
// ==== Execute ----------------------------------------------------------------
@available(SwiftStdlib 6.0, *)
func test_distributedVariable<DA: WorkerProtocol>(actor: DA) async throws -> String {
try await actor.distributedVariable
}
@available(SwiftStdlib 6.0, *)
@main struct Main {
static func main() async throws {
let system = LocalTestingDistributedActorSystem()
let actor: any WorkerProtocol = Worker(actorSystem: system)
let m = try await actor.distributedMethod()
print("m = \(m)") // CHECK: m = implemented method
// force a call through witness table
let v1 = try await test_distributedVariable(actor: actor)
print("v1 = \(v1)") // CHECK: v1 = implemented variable
let v2 = try await actor.distributedVariable
print("v2 = \(v2)") // CHECK: v2 = implemented variable
}
}