From 51c5ea1f96aea2eeaf4e75d71ef062c94870abce Mon Sep 17 00:00:00 2001
From: Feng <ery.lee@gmail.com>
Date: Sun, 17 Jan 2016 20:02:44 +0800
Subject: [PATCH] fix route, reverse_route stats

---
 src/emqttd_pubsub_sup.erl | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/emqttd_pubsub_sup.erl b/src/emqttd_pubsub_sup.erl
index c7185414b..592b8a7d8 100644
--- a/src/emqttd_pubsub_sup.erl
+++ b/src/emqttd_pubsub_sup.erl
@@ -34,7 +34,7 @@
 -define(CONCURRENCY_OPTS, [{read_concurrency, true}, {write_concurrency, true}]).
 
 %% API
--export([start_link/0]).
+-export([start_link/0, pubsub_pool/0]).
 
 %% Supervisor callbacks
 -export([init/1]).
@@ -42,6 +42,9 @@
 start_link() ->
     supervisor:start_link({local, ?MODULE}, ?MODULE, [emqttd_broker:env(pubsub)]).
 
+pubsub_pool() ->
+    hd([Pid|| {pubsub_pool, Pid, _, _} <- supervisor:which_children(?MODULE)]).
+
 init([Env]) ->
     %% Create tabs
     create_tab(route), create_tab(reverse_route),
@@ -83,6 +86,9 @@ pool_size(Env) ->
 setstats(route) ->
     emqttd_stats:setstat('routes/count', ets:info(route, size));
 
+setstats(reverse_route) ->
+    emqttd_stats:setstat('routes/reverse', ets:info(reverse_route, size));
+
 setstats(topic) ->
     emqttd_stats:setstats('topics/count', 'topics/max', mnesia:table_info(topic, size));