Merge pull request #41082 from xedin/dist-decode-in-accessor

[Distributed] Augment accessor to be able to decode arguments instead of accepting buffer

Author: xedin

include/swift/AST/ASTContext.h

@@ -1345,6 +1345,14 @@ class ASTContext final {
   /// alternative specified via the -entry-point-function-name frontend flag.
   std::string getEntryPointFunctionName() const;
 
+  /// Find the concrete invocation decoder associated with the given actor.
+  NominalTypeDecl *
+  getDistributedActorInvocationDecoder(NominalTypeDecl *);
+
+  /// Find `decodeNextArgument<T>(type: T.Type) -> T` method associated with
+  /// invocation decoder of the given distributed actor.
+  FuncDecl *getDistributedActorArgumentDecodingMethod(NominalTypeDecl *);
+
 private:
   friend Decl;
 

include/swift/AST/KnownIdentifiers.def

@@ -289,7 +289,9 @@ IDENTIFIER(system)
 IDENTIFIER(target)
 IDENTIFIER(throwing)
 IDENTIFIER(using)
+IDENTIFIER(InvocationDecoder)
 IDENTIFIER(whenLocal)
+IDENTIFIER(decodeNextArgument)
 
 #undef IDENTIFIER
 #undef IDENTIFIER_

include/swift/AST/TypeCheckRequests.h

@@ -1091,6 +1091,43 @@ class GetDistributedActorIDPropertyRequest :
     bool isCached() const { return true; }
 };
 
+/// Obtain the invocation decoder associated with the given distributed actor.
+class GetDistributedActorInvocationDecoderRequest :
+  public SimpleRequest<GetDistributedActorInvocationDecoderRequest,
+                       NominalTypeDecl *(NominalTypeDecl *),
+                       RequestFlags::Cached> {
+public:
+  using SimpleRequest::SimpleRequest;
+
+private:
+  friend SimpleRequest;
+
+  NominalTypeDecl *evaluate(Evaluator &evaluator, NominalTypeDecl *actor) const;
+
+public:
+  // Caching
+  bool isCached() const { return true; }
+};
+
+/// Obtain the method that could be used to decode argument values passed
+/// to a particular actor invocation type.
+class GetDistributedActorArgumentDecodingMethodRequest :
+  public SimpleRequest<GetDistributedActorArgumentDecodingMethodRequest,
+                       FuncDecl *(NominalTypeDecl *),
+                       RequestFlags::Cached> {
+public:
+  using SimpleRequest::SimpleRequest;
+
+private:
+  friend SimpleRequest;
+
+  FuncDecl *evaluate(Evaluator &evaluator, NominalTypeDecl *actor) const;
+
+public:
+  // Caching
+  bool isCached() const { return true; }
+};
+
 /// Retrieve the static "shared" property within a global actor that provides
 /// the actor instance representing the global actor.
 ///

include/swift/AST/TypeCheckerTypeIDZone.def

@@ -112,6 +112,12 @@ SWIFT_REQUEST(TypeChecker, GetDistributedActorIDPropertyRequest, VarDecl *(Nomin
               Cached, NoLocationInfo)
 SWIFT_REQUEST(TypeChecker, GetDistributedActorSystemPropertyRequest, VarDecl *(NominalTypeDecl *),
               Cached, NoLocationInfo)
+SWIFT_REQUEST(TypeChecker, GetDistributedActorInvocationDecoderRequest,
+              NominalTypeDecl *(NominalTypeDecl *),
+              Cached, NoLocationInfo)
+SWIFT_REQUEST(TypeChecker, GetDistributedActorArgumentDecodingMethodRequest,
+              FuncDecl *(NominalTypeDecl *),
+              Cache, NoLocationInfo)
 SWIFT_REQUEST(TypeChecker, GlobalActorInstanceRequest,
               VarDecl *(NominalTypeDecl *),
               Cached, NoLocationInfo)