averello/swift-mirror
1
0
Fork 0
mirror of https://github.com/apple/swift.git synced 2025-12-14 20:36:38 +01:00
Code Issues Packages Projects Releases Wiki Activity

[Distributed] Implementing ad-hoc protocol requirements

Browse Source
This commit is contained in:
Konrad `ktoso` Malawski
2022-01-11 21:44:58 +09:00
parent 78862575e4
commit 0ffc68b5ef
26 changed files with 1815 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

63
lib/Sema/CodeSynthesisDistributedActor.cpp
View File

@@ -112,6 +112,7 @@ synthesizeRemoteFuncStubBody(AbstractFunctionDecl *func, void *context) {
return { body, /*isTypeChecked=*/true };
}
// FIXME(distributed): remove this, won't be needed with the new remoteCall
static Identifier makeRemoteFuncIdentifier(FuncDecl* distributedFunc) {
auto &C = distributedFunc->getASTContext();
assert(distributedFunc->isDistributed());
@@ -197,7 +198,7 @@ static AbstractFunctionDecl *addImplicitDistributedActorRemoteFunction(
//
// The "derived" mechanisms are not really geared towards emitting for
// what already has a witness.
static ValueDecl *addImplicitDistributedActorIDProperty(
static VarDecl *addImplicitDistributedActorIDProperty(
// DeclContext *parentDC,
NominalTypeDecl *nominal) {
if (!nominal || !nominal->isDistributedActor())
@@ -241,6 +242,43 @@ static ValueDecl *addImplicitDistributedActorIDProperty(
return propDecl;
}
/******************************************************************************/
/************ LOCATING AD-HOC PROTOCOL REQUIREMENT IMPLS **********************/
/******************************************************************************/
AbstractFunctionDecl*
GetDistributedActorSystemRemoteCallFunctionRequest::evaluate(
Evaluator &evaluator, NominalTypeDecl *decl) const {
auto &C = decl->getASTContext();
// It would be nice to check if this is a DistributedActorSystem
// "conforming" type, but we can't do this as we invoke this function WHILE
// deciding if the type conforms or not;
// Not via `ensureDistributedModuleLoaded` to avoid generating a warning,
// we won't be emitting the offending decl after all.
if (!C.getLoadedModule(C.Id_Distributed)) {
return nullptr;
}
AbstractFunctionDecl *remoteCallFunc = nullptr;
for (auto value : decl->lookupDirect(C.Id_remoteCall)) {
auto func = dyn_cast<AbstractFunctionDecl>(value);
if (func && func->isDistributedActorSystemRemoteCall()) {
remoteCallFunc = func;
break;
}
}
if (!remoteCallFunc) {
fprintf(stderr, "[%s:%d] (%s) NOT FOUND\n", __FILE__, __LINE__, __FUNCTION__);
decl->dump();
fprintf(stderr, "[%s:%d] (%s) --------------------!!!!!\n", __FILE__, __LINE__, __FUNCTION__);
}
return remoteCallFunc;
}
/******************************************************************************/
/************************ SYNTHESIS ENTRY POINT *******************************/
/******************************************************************************/
@@ -261,7 +299,7 @@ AbstractFunctionDecl *GetDistributedRemoteFuncRequest::evaluate(
return addImplicitDistributedActorRemoteFunction(DC, func);
}
ValueDecl *GetDistributedActorIDPropertyRequest::evaluate(
VarDecl *GetDistributedActorIDPropertyRequest::evaluate(
Evaluator &evaluator, NominalTypeDecl *actor) const {
if (!actor->isDistributedActor())
return nullptr;
@@ -275,3 +313,24 @@ ValueDecl *GetDistributedActorIDPropertyRequest::evaluate(
return addImplicitDistributedActorIDProperty(actor);
}
VarDecl *GetDistributedActorSystemPropertyRequest::evaluate(
Evaluator &evaluator, NominalTypeDecl *actor) const {
if (!actor->isDistributedActor())
return nullptr;
auto &C = actor->getASTContext();
// not via `ensureDistributedModuleLoaded` to avoid generating a warning,
// we won't be emitting the offending decl after all.
if (!C.getLoadedModule(C.Id_Distributed))
return nullptr;
for (auto system : actor->lookupDirect(C.Id_actorSystem)) {
// TODO(distributed): may need to check conformance here?
return dyn_cast<VarDecl>(system);
}
return nullptr;
}