Merge pull request #66839 from bnbarham/reuse-cursor-ast

[CursorInfo] Re-use already built ASTs ahead of currently building ones

Author: bnbarham

test/SourceKit/CursorInfo/cursor_after_edit.swift

@@ -8,4 +8,6 @@
 // RUN: %sourcekitd-test \
 // RUN:   -req=open -text-input %t/empty.swift %t/func.swift -- %t/func.swift == \
 // RUN:   -req=edit -offset=0 -length=0 -replace="func foo() {}" -req-opts=enablesyntaxmap=0,enablesubstructure=0,enablediagnostics=0 %t/func.swift -- %t/func.swift == \
-// RUN:   -req=cursor -offset=5 %t/func.swift -- %t/func.swift
+// RUN:   -req=cursor -offset=5 %t/func.swift -- %t/func.swift == \
+// RUN:   -req=edit -offset=0 -length=0 -replace="// some comment\n" -req-opts=enablesyntaxmap=0,enablesubstructure=0,enablediagnostics=0 %t/func.swift -- %t/func.swift == \
+// RUN:   -req=cursor -offset=21 %t/func.swift -- %t/func.swift

tools/SourceKit/lib/SwiftLang/SwiftASTManager.cpp

@@ -1240,30 +1240,61 @@ bool ASTBuildOperation::addConsumer(SwiftASTConsumerRef Consumer) {
   return true;
 }
 
+/// Returns a build operation that `Consumer` can use, in order of the
+/// following:
+///   1. The latest finished build operation that either exactly matches, or
+///      can be used with snapshots
+///   2. If none, the latest in-progress build operation with the same
+///      conditions
+///   3. `nullptr` otherwise
 ASTBuildOperationRef ASTProducer::getBuildOperationForConsumer(
     SwiftASTConsumerRef Consumer,
     IntrusiveRefCntPtr<llvm::vfs::FileSystem> FileSystem,
     SwiftASTManagerRef Mgr) {
+  ASTBuildOperationRef LatestUsableOp;
+  Statistic *StatCount = nullptr;
   for (auto &BuildOp : llvm::reverse(BuildOperations)) {
-    if (BuildOp->isCancelled()) {
+    if (BuildOp->isCancelled())
+      continue;
+
+    // No point checking for a match, we already have one - we're just looking
+    // for a finished operation that can be used with the file contents of
+    // `BuildOp` at this point (which we will prefer over an incomplete
+    // operation, whether that exactly matches or not).
+    if (LatestUsableOp && !BuildOp->isFinished())
+      continue;
+
+    // Check for an exact match
+    if (BuildOp->matchesSourceState(FileSystem)) {
+      LatestUsableOp = BuildOp;
+      StatCount = &Mgr->Impl.Stats->numASTCacheHits;
+      if (BuildOp->isFinished())
+        break;
       continue;
     }
+
+    // Check for whether the operation can be used taking into account
+    // snapshots
     std::vector<ImmutableTextSnapshotRef> Snapshots;
     Snapshots.reserve(BuildOp->getFileContents().size());
     for (auto &FileContent : BuildOp->getFileContents()) {
       if (FileContent.Snapshot) {
         Snapshots.push_back(FileContent.Snapshot);
       }
     }
-    if (BuildOp->matchesSourceState(FileSystem)) {
-      ++Mgr->Impl.Stats->numASTCacheHits;
-      return BuildOp;
-    } else if (Consumer->canUseASTWithSnapshots(Snapshots)) {
-      ++Mgr->Impl.Stats->numASTsUsedWithSnapshots;
-      return BuildOp;
+
+    if (Consumer->canUseASTWithSnapshots(Snapshots)) {
+      LatestUsableOp = BuildOp;
+      StatCount = &Mgr->Impl.Stats->numASTsUsedWithSnapshots;
+      if (BuildOp->isFinished())
+        break;
     }
   }
-  return nullptr;
+
+  if (StatCount) {
+    ++(*StatCount);
+  }
+  return LatestUsableOp;
 }
 
 void ASTProducer::enqueueConsumer(