Grab the diag count from the target.

root-project · Jan 12, 2017 · bfe5e72 · bfe5e72
1 parent d02d88a
commit bfe5e72
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/lib/Interpreter/IncrementalParser.cpp b/lib/Interpreter/IncrementalParser.cpp
@@ -110,6 +110,11 @@ namespace {
     std::stack<bool> fIgnorePromptDiags;
     std::unique_ptr<DiagnosticConsumer> fOwnedTarget;
 
+    void SyncDiagCountWithTarget() {
+      NumWarnings = fOwnedTarget->getNumWarnings();
+      NumErrors = fOwnedTarget->getNumErrors();
+    }
+
   public:
     FilteringDiagConsumer(std::unique_ptr<DiagnosticConsumer>&& Target):
       ForwardingDiagnosticConsumer(*Target.get()),
@@ -122,14 +127,17 @@ namespace {
 
     void EndSourceFile() override {
       fOwnedTarget->EndSourceFile();
+      SyncDiagCountWithTarget();
     }
 
     void finish() override {
       fOwnedTarget->finish();
+      SyncDiagCountWithTarget();
     }
 
     void clear() override {
       fOwnedTarget->clear();
+      SyncDiagCountWithTarget();
     }
 
     void HandleDiagnostic(DiagnosticsEngine::Level DiagLevel,
@@ -153,6 +161,7 @@ namespace {
         }
       }
       ForwardingDiagnosticConsumer::HandleDiagnostic(DiagLevel, Info);
+      SyncDiagCountWithTarget();
     }
 
     void push(bool ignoreDiags) { fIgnorePromptDiags.push(ignoreDiags); }