Merge pull request swiftlang#63482 from DougGregor/source-file-lookup-cache-fix

DougGregor · web-flow · commit d0635a69bf01 · 2023-02-06T22:07:59.000-08:00
diff --git a/lib/AST/Module.cpp b/lib/AST/Module.cpp
@@ -598,6 +598,9 @@ void ModuleDecl::updateSourceFileLocationMap() {
   std::sort(sourceFileLocationMap->allSourceFiles.begin(),
             sourceFileLocationMap->allSourceFiles.end(),
             SourceFileRangeComparison{&getASTContext().SourceMgr});
+
+  sourceFileLocationMap->numFiles = files.size();
+  sourceFileLocationMap->numAuxiliaryFiles = AuxiliaryFiles.size();
 }
 
 SourceFile *ModuleDecl::getSourceFileContainingLocation(SourceLoc loc) {

Original file line number	Diff line number	Diff line change
`@@ -598,6 +598,9 @@ void ModuleDecl::updateSourceFileLocationMap() {`
`598`	`598`	`std::sort(sourceFileLocationMap->allSourceFiles.begin(),`
`599`	`599`	`sourceFileLocationMap->allSourceFiles.end(),`
`600`	`600`	`SourceFileRangeComparison{&getASTContext().SourceMgr});`
	`601`	`+`
	`602`	`+ sourceFileLocationMap->numFiles = files.size();`
	`603`	`+ sourceFileLocationMap->numAuxiliaryFiles = AuxiliaryFiles.size();`
`601`	`604`	`}`
`602`	`605`
`603`	`606`	`SourceFile *ModuleDecl::getSourceFileContainingLocation(SourceLoc loc) {`