[DistributedDecl] Add DistributedActorSystem to known SDK types

Author: ktoso

include/swift/AST/KnownSDKTypes.def

@@ -45,6 +45,7 @@ KNOWN_SDK_TYPE_DECL(Concurrency, TaskLocal, ClassDecl, 1)
 
 // Distributed actors
 KNOWN_SDK_TYPE_DECL(Distributed, DistributedActor, ProtocolDecl, 0)
+KNOWN_SDK_TYPE_DECL(Distributed, DistributedActorSystem, ProtocolDecl, 0)
 KNOWN_SDK_TYPE_DECL(Distributed, RemoteCallTarget, StructDecl, 0)
 
 // String processing

lib/AST/DistributedDecl.cpp

@@ -10,7 +10,7 @@
 //
 //===----------------------------------------------------------------------===//
 //
-//  This file implements the Decl class and subclasses.
+//  This file handles lookups related to distributed actor decls.
 //
 //===----------------------------------------------------------------------===//
 
@@ -89,7 +89,7 @@ Type swift::getDistributedActorSystemActorIDRequirementType(NominalTypeDecl *sys
   assert(!system->isDistributedActor());
   auto &ctx = system->getASTContext();
 
-  auto protocol = ctx.getProtocol(KnownProtocolKind::DistributedActorSystem);
+  auto protocol = ctx.getDistributedActorSystemDecl();
   if (!protocol)
     return Type();
 
@@ -128,7 +128,7 @@ Type ASTContext::getAssociatedTypeOfDistributedSystemOfActor(
   if (!actorSystemDecl)
     return Type();
 
-  auto actorSystemProtocol = ctx.getProtocol(KnownProtocolKind::DistributedActorSystem);
+  auto actorSystemProtocol = ctx.getDistributedActorSystemDecl();
   if (!actorSystemProtocol)
     return Type();
 
@@ -252,7 +252,7 @@ bool AbstractFunctionDecl::isDistributedActorSystemRemoteCall(bool isVoidReturn)
 
   // === Must be declared in a 'DistributedActorSystem' conforming type
   ProtocolDecl *systemProto =
-      C.getProtocol(KnownProtocolKind::DistributedActorSystem);
+      C.getDistributedActorSystemDecl();
 
   auto systemNominal = getDeclContext()->getSelfNominalTypeDecl();
   auto distSystemConformance = module->lookupConformance(
@@ -968,7 +968,7 @@ llvm::SmallPtrSet<ProtocolDecl *, 2>
 swift::extractDistributedSerializationRequirements(
     ASTContext &C, ArrayRef<Requirement> allRequirements) {
   llvm::SmallPtrSet<ProtocolDecl *, 2> serializationReqs;
-  auto systemProto = C.getProtocol(KnownProtocolKind::DistributedActorSystem);
+  auto systemProto = C.getDistributedActorSystemDecl();
   auto serializationReqAssocType =
       systemProto->getAssociatedType(C.Id_SerializationRequirement);
   auto systemSerializationReqTy = serializationReqAssocType->getInterfaceType();

test/Distributed/Runtime/distributed_actor_hop_to.swift

@@ -0,0 +1,46 @@
+// RUN: %empty-directory(%t)
+// RUN: %target-swift-frontend-emit-module -emit-module-path %t/FakeDistributedActorSystems.swiftmodule -module-name FakeDistributedActorSystems -disable-availability-checking %S/../Inputs/FakeDistributedActorSystems.swift
+// RUN: %target-build-swift -module-name main -Xfrontend -enable-experimental-distributed -Xfrontend -disable-availability-checking -j2 -parse-as-library -I %t %s %S/../Inputs/FakeDistributedActorSystems.swift -o %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
+
+// FIXME(distributed): Distributed actors currently have some issues on windows, isRemote always returns false. rdar://82593574
+// UNSUPPORTED: windows
+
+import _Distributed
+import FakeDistributedActorSystems
+
+typealias DefaultDistributedActorSystem = FakeRoundtripActorSystem
+
+protocol LifecycleWatch: DistributedActor where ActorSystem == FakeRoundtripActorSystem {
+}
+
+extension LifecycleWatch {
+  func watch() async throws {
+    // nothing here
+  }
+
+  distributed func test() async throws {
+    try await self.watch()
+  }
+}
+
+distributed actor Worker: LifecycleWatch {
+
+}
+
+@main struct Main {
+  static func main() async {
+    let worker: any LifecycleWatch = Worker(system: DefaultDistributedActorSystem())
+    try! await worker.test()
+
+    print("OK") // CHECK: OK
+  }
+}