Extended MetricsParser to handle incremental parsing #10

mcserep · Jul 12, 2018 · 7affe06 · 7affe06
1 parent 8f85585
commit 7affe06
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/plugins/metrics/parser/include/metricsparser/metricsparser.h b/plugins/metrics/parser/include/metricsparser/metricsparser.h
@@ -17,6 +17,7 @@ class MetricsParser : public AbstractParser
   MetricsParser(ParserContext& ctx_);
   virtual std::vector<std::string> getDependentParsers() const override;  
   virtual bool parse() override;
+  virtual void preparse() override;
 
 private:
   util::DirIterCallback getParserCallback();

diff --git a/plugins/metrics/parser/src/metricsparser.cpp b/plugins/metrics/parser/src/metricsparser.cpp
@@ -51,6 +51,26 @@ std::vector<std::string> MetricsParser::getDependentParsers() const
   return std::vector<std::string>{};
 }
 
+void MetricsParser::preparse()
+{
+  if(!_fileIdCache.empty())
+  {
+    (util::OdbTransaction(_ctx.db))([this] {
+      for(const model::File& file
+        : _ctx.db->query<model::File>(
+        odb::query<model::File>::id.in_range(_fileIdCache.begin(), _fileIdCache.end())))
+      {
+        if(_ctx.fileStatus[file.path] == cc::parser::IncrementalStatus::DELETED ||
+           _ctx.fileStatus[file.path] == cc::parser::IncrementalStatus::MODIFIED)
+        {
+          _ctx.db->erase_query<model::Metrics>(odb::query<model::Metrics>::file == file.id);
+          _fileIdCache.erase(file.id);
+        }
+      }
+    });
+  }
+}
+
 bool MetricsParser::parse()
 {
   for(std::string path : _ctx.options["input"].as<std::vector<std::string>>())