yjbanov / barista2 Goto Github PK

diff --git a/api.cpp b/api.cpp
index d09e275..ff22fe7 100644
--- a/api.cpp
+++ b/api.cpp
@@ -30,7 +30,9 @@ bool _canUpdate(shared_ptr<RenderNode> node, shared_ptr<Node> configuration) {
   return node->GetConfiguration()->GetKey() == configuration->GetKey();
 }
 
-RenderNode::RenderNode(shared_ptr<Tree> tree) : _tree(tree) { }
+int RenderNode::__nodes = 0;
+
+RenderNode::RenderNode(shared_ptr<Tree> tree) : _tree(tree) { __nodes++; }
 
 void RenderNode::Update(shared_ptr<Node> newConfiguration, ElementUpdate& update) {
   assert(newConfiguration != nullptr);
diff --git a/api.h b/api.h
index 1d0567b..5f9c755 100644
--- a/api.h
+++ b/api.h
@@ -37,6 +37,7 @@ class Event;
 class Node {
  public:
   Node() { }
+
   virtual string GetKey() { return _key; }
   virtual void SetKey(string key) { _key = key; }
   virtual shared_ptr<RenderNode> Instantiate(shared_ptr<Tree> t) = 0;
@@ -49,6 +50,10 @@ typedef function<void(shared_ptr<RenderNode>)> RenderNodeVisitor;
 
 class RenderNode {
  public:
+  static int __nodes;
+
+  virtual ~RenderNode() { __nodes--; }
+
   RenderNode(shared_ptr<Tree> tree);
   virtual shared_ptr<Node> GetConfiguration() { return _configuration; }
   virtual shared_ptr<RenderParent> GetParent() { return _parent.lock(); }
diff --git a/test_giant.cpp b/test_giant.cpp
index 91fb48a..8340d60 100644
--- a/test_giant.cpp
+++ b/test_giant.cpp
@@ -56,6 +56,7 @@ TEST(TestBootstrapGiantApp)
     auto after_flip = system_clock::now();
     duration<double> delta = after_flip - before_flip;
     cout << "Flip #" << flip << " took: " << delta.count() * 1000 << "ms; tree size: " << html.size() << " chars" << endl;
+    cout << "Node: " << RenderNode::__nodes << endl;
   }
 END_TEST