[Sema] Teach LocalDiscriminatorsRequest about runtime attribute initializers

xedin · xedin · commit 7a593ce1eaa1 · 2022-12-20T11:53:28.000-08:00
diff --git a/lib/Sema/TypeCheckStmt.cpp b/lib/Sema/TypeCheckStmt.cpp
@@ -337,6 +337,13 @@ unsigned LocalDiscriminatorsRequest::evaluate(
       node = initInfo.getInitFromProjectedValue();
       break;
     }
+  } else if (auto *runtimeAttrInit =
+                 dyn_cast<RuntimeAttributeInitializer>(dc)) {
+    auto *attachedTo = runtimeAttrInit->getAttachedToDecl();
+    auto generator = attachedTo->getRuntimeDiscoverableAttributeGenerator(
+        runtimeAttrInit->getAttr());
+    if (generator.second)
+      node = generator.first;
   }
 
   if (!node)

Original file line number	Diff line number	Diff line change
`@@ -337,6 +337,13 @@ unsigned LocalDiscriminatorsRequest::evaluate(`
`337`	`337`	`node = initInfo.getInitFromProjectedValue();`
`338`	`338`	`break;`
`339`	`339`	`}`
	`340`	`+ } else if (auto *runtimeAttrInit =`
	`341`	`+ dyn_cast<RuntimeAttributeInitializer>(dc)) {`
	`342`	`+ auto *attachedTo = runtimeAttrInit->getAttachedToDecl();`
	`343`	`+ auto generator = attachedTo->getRuntimeDiscoverableAttributeGenerator(`
	`344`	`+ runtimeAttrInit->getAttr());`
	`345`	`+ if (generator.second)`
	`346`	`+ node = generator.first;`
`340`	`347`	`}`
`341`	`348`
`342`	`349`	`if (!node)`