Merge pull request #2972 from benlangmuir/index-finish-hook

Author: swift-ci

include/swift/Index/IndexDataConsumer.h

@@ -36,6 +36,8 @@ class IndexDataConsumer {
   virtual bool recordRelatedEntity(const IndexSymbol &symbol) = 0;
   virtual bool finishSourceEntity(SymbolKind kind, SymbolSubKindSet subKinds,
                                   SymbolRoleSet roles) = 0;
+
+  virtual void finish() {}
 };
 
 } // end namespace index

lib/Index/Index.cpp

@@ -1069,6 +1069,7 @@ void index::indexSourceFile(SourceFile *SF, StringRef hash,
   unsigned bufferID = SF->getBufferID().getValue();
   IndexSwiftASTWalker walker(consumer, SF->getASTContext(), bufferID);
   walker.visitModule(*SF->getParentModule(), hash);
+  consumer.finish();
 }
 
 void index::indexModule(ModuleDecl *module, StringRef hash,
@@ -1077,4 +1078,5 @@ void index::indexModule(ModuleDecl *module, StringRef hash,
   IndexSwiftASTWalker walker(consumer, module->getASTContext(),
                              /*bufferID*/ -1);
   walker.visitModule(*module, hash);
+  consumer.finish();
 }