[C++] Add a way to retrieve stats

5 files changed, 127 insertions, 0 deletions

diff --git a/Makefile b/Makefile
index 7e523ec..5b0f4f7 100644
--- a/Makefile
+++ b/Makefile
@@ -28,6 +28,7 @@ CXX_SRCS:= \
 	ninja.cc \
 	parser.cc \
 	rule.cc \
+	stats.cc \
 	string_piece.cc \
 	string_pool.cc \
 	stringprintf.cc \
diff --git a/func.cc b/func.cc
index 306de80..a0ef199 100644
--- a/func.cc
+++ b/func.cc
@@ -30,6 +30,7 @@
 #include "eval.h"
 #include "log.h"
 #include "parser.h"
+#include "stats.h"
 #include "strutil.h"
 #include "symtab.h"
 #include "var.h"
@@ -242,6 +243,7 @@ void JoinFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
 }
 
 void WildcardFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
+  COLLECT_STATS("func wildcard time");
   shared_ptr<string> pat = args[0]->Eval(ev);
   if (ev->avoid_io()) {
     *s += "$(/bin/ls -d ";
@@ -411,6 +413,7 @@ void EvalFunc(const vector<Value*>& args, Evaluator* ev, string*) {
 }
 
 void ShellFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
+  COLLECT_STATS("func shell time");
   shared_ptr<string> cmd = args[0]->Eval(ev);
   if (ev->avoid_io()) {
     *s += "$(";
diff --git a/main.cc b/main.cc
index aa09bed..31a0d23 100644
--- a/main.cc
+++ b/main.cc
@@ -31,6 +31,7 @@
 #include "log.h"
 #include "ninja.h"
 #include "parser.h"
+#include "stats.h"
 #include "string_piece.h"
 #include "stringprintf.h"
 #include "strutil.h"
@@ -111,6 +112,8 @@ static void Init() {
 }
 
 static void Quit() {
+  ReportAllStats();
+
   QuitParser();
   QuitDepNodePool();
   QuitFuncTable();
diff --git a/stats.cc b/stats.cc
new file mode 100644
index 0000000..2d747a4
--- /dev/null
+++ b/stats.cc
@@ -0,0 +1,66 @@
+// Copyright 2015 Google Inc. All rights reserved
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+// +build ignore
+
+#include "stats.h"
+
+#include <vector>
+
+#include "log.h"
+#include "stringprintf.h"
+#include "time.h"
+
+namespace {
+
+vector<Stats*>* g_stats;
+
+}  // namespace
+
+Stats::Stats(const char* name)
+    : name_(name), start_time_(0), elapsed_(0) {
+  if (g_stats == NULL)
+    g_stats = new vector<Stats*>;
+  g_stats->push_back(this);
+}
+
+string Stats::String() const {
+  return StringPrintf("%s: %f", name_, elapsed_);
+}
+
+void Stats::Start() {
+  start_time_ = GetTime();
+}
+
+void Stats::End() {
+  elapsed_ += GetTime() - start_time_;
+}
+
+ScopedStatsRecorder::ScopedStatsRecorder(Stats* st)
+    : st_(st) {
+  st_->Start();
+}
+
+ScopedStatsRecorder::~ScopedStatsRecorder() {
+  st_->End();
+}
+
+void ReportAllStats() {
+  if (!g_stats)
+    return;
+  for (Stats* st : *g_stats) {
+    LOG_STAT("%s", st->String().c_str());
+  }
+  delete g_stats;
+}
diff --git a/stats.h b/stats.h
new file mode 100644
index 0000000..41525e1
--- /dev/null
+++ b/stats.h
@@ -0,0 +1,54 @@
+// Copyright 2015 Google Inc. All rights reserved
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#ifndef STATS_H_
+#define STATS_H_
+
+#include <string>
+
+using namespace std;
+
+class Stats {
+ public:
+  explicit Stats(const char* name);
+
+  string String() const;
+
+ private:
+  void Start();
+  void End();
+
+  friend class ScopedStatsRecorder;
+
+  const char* name_;
+  double start_time_;
+  double elapsed_;
+};
+
+class ScopedStatsRecorder {
+ public:
+  explicit ScopedStatsRecorder(Stats* st);
+  ~ScopedStatsRecorder();
+
+ private:
+  Stats* st_;
+};
+
+void ReportAllStats();
+
+#define COLLECT_STATS(name)                     \
+  static Stats stats(name);                     \
+  ScopedStatsRecorder ssr(&stats)
+
+#endif  // STATS_H_