Remove DelayedDeclLists from PersistentParserState

Rather than parsing all delayed bodies for
`-dump-parse` once we finish parsing, tell the
parser not to delay any bodies. This then allows
us to remove `DelayedDeclLists` from
PersistentParserState.

Author: hamishknight

include/swift/Frontend/Frontend.h

@@ -609,7 +609,7 @@ class CompilerInstance {
   /// Parses the input file but does no type-checking or module imports.
   /// Note that this only supports parsing an invocation with a single file.
   void performParseOnly(bool EvaluateConditionals = false,
-                        bool ParseDelayedBodyOnEnd = false);
+                        bool CanDelayBodies = true);
 
   /// Parses and performs name binding on all input files.
   ///

include/swift/Parse/PersistentParserState.h

@@ -78,8 +78,6 @@ class PersistentParserState {
 
   std::unique_ptr<CodeCompletionDelayedDeclState> CodeCompletionDelayedDeclStat;
 
-  std::vector<IterableDeclContext *> DelayedDeclLists;
-
   /// The local context for all top-level code.
   TopLevelContext TopLevelCode;
 
@@ -112,10 +110,6 @@ class PersistentParserState {
     return std::move(CodeCompletionDelayedDeclStat);
   }
 
-  void delayDeclList(IterableDeclContext *D);
-
-  void parseAllDelayedDeclLists();
-
   TopLevelContext &getTopLevelContext() {
     return TopLevelCode;
   }

lib/Frontend/Frontend.cpp

@@ -1071,7 +1071,7 @@ SourceFile *CompilerInstance::createSourceFileForMainModule(
 }
 
 void CompilerInstance::performParseOnly(bool EvaluateConditionals,
-                                        bool ParseDelayedBodyOnEnd) {
+                                        bool CanDelayBodies) {
   const InputFileKind Kind = Invocation.getInputKind();
   ModuleDecl *const MainModule = getMainModule();
   Context->LoadedModules[MainModule->getName()] = MainModule;
@@ -1093,12 +1093,8 @@ void CompilerInstance::performParseOnly(bool EvaluateConditionals,
   }
 
   PersistentState = llvm::make_unique<PersistentParserState>();
-
-  SWIFT_DEFER {
-    if (ParseDelayedBodyOnEnd)
-      PersistentState->parseAllDelayedDeclLists();
-  };
   PersistentState->PerformConditionEvaluation = EvaluateConditionals;
+
   // Parse all the library files.
   for (auto BufferID : InputSourceCodeBufferIDs) {
     if (BufferID == MainBufferID)
@@ -1111,7 +1107,7 @@ void CompilerInstance::performParseOnly(bool EvaluateConditionals,
         BufferID);
 
     parseIntoSourceFileFull(*NextInput, BufferID, PersistentState.get(),
-                            /*DelayBodyParsing=*/!IsPrimary);
+                            /*DelayBodyParsing=*/!IsPrimary && CanDelayBodies);
   }
 
   // Now parse the main file.

lib/FrontendTool/FrontendTool.cpp

@@ -1262,11 +1262,12 @@ static bool performCompile(CompilerInstance &Instance,
     return compileLLVMIR(Invocation, Instance, Stats);
 
   if (FrontendOptions::shouldActionOnlyParse(Action)) {
-    bool ParseDelayedDeclListsOnEnd =
-      Action == FrontendOptions::ActionType::DumpParse;
+    // Disable delayed parsing of type and function bodies when we've been
+    // asked to dump the resulting AST.
+    bool CanDelayBodies = Action != FrontendOptions::ActionType::DumpParse;
     Instance.performParseOnly(/*EvaluateConditionals*/
                     Action == FrontendOptions::ActionType::EmitImportedModules,
-                              ParseDelayedDeclListsOnEnd);
+                              CanDelayBodies);
   } else if (Action == FrontendOptions::ActionType::ResolveImports) {
     Instance.performParseAndResolveImportsOnly();
   } else {

lib/Parse/ParseDecl.cpp

@@ -4441,8 +4441,6 @@ bool Parser::delayParsingDeclList(SourceLoc LBLoc, SourceLoc &RBLoc,
     RBLoc = Tok.getLoc();
     error = true;
   }
-
-  State->delayDeclList(IDC);
   return error;
 }
 

lib/Parse/PersistentParserState.cpp

@@ -50,12 +50,3 @@ void PersistentParserState::restoreCodeCompletionDelayedDeclState(
       ScopeInfo.saveCurrentScope(), other.StartOffset, other.EndOffset,
       other.PrevOffset));
 }
-
-void PersistentParserState::delayDeclList(IterableDeclContext *D) {
-  DelayedDeclLists.push_back(D);
-}
-
-void PersistentParserState::parseAllDelayedDeclLists() {
-  for (auto IDC : DelayedDeclLists)
-    IDC->loadAllMembers();
-}