From 02067855777d39efc2f004bed23dea3a76d6ab8c Mon Sep 17 00:00:00 2001
From: turtled <turtled@emqtt.io>
Date: Mon, 5 Dec 2016 15:00:09 +0800
Subject: [PATCH] fix ws metrics

---
 src/emqttd_ws.erl        | 1 +
 src/emqttd_ws_client.erl | 1 +
 2 files changed, 2 insertions(+)

diff --git a/src/emqttd_ws.erl b/src/emqttd_ws.erl
index 58cb6dc43..aba43dfee 100644
--- a/src/emqttd_ws.erl
+++ b/src/emqttd_ws.erl
@@ -55,6 +55,7 @@ ws_loop([<<>>], State, _ReplyChannel) ->
 ws_loop(Data, State = #wsocket_state{peer = Peer, client_pid = ClientPid,
                                      parser_fun = ParserFun}, ReplyChannel) ->
     ?WSLOG(debug, Peer, "RECV ~p", [Data]),
+    emqttd_metrics:inc('bytes/received', size(Data)),
     case catch ParserFun(iolist_to_binary(Data)) of
         {more, NewParser} ->
             State#wsocket_state{parser_fun = NewParser};
diff --git a/src/emqttd_ws_client.erl b/src/emqttd_ws_client.erl
index 3ea9cdbdd..f9cce71d0 100644
--- a/src/emqttd_ws_client.erl
+++ b/src/emqttd_ws_client.erl
@@ -107,6 +107,7 @@ handle_cast({unsubscribe, Topics}, State) ->
         end, State);
 
 handle_cast({received, Packet}, State = #wsclient_state{peer = Peer, proto_state = ProtoState}) ->
+    emqttd_metrics:received(Packet),
     case emqttd_protocol:received(Packet, ProtoState) of
         {ok, ProtoState1} ->
             noreply(State#wsclient_state{proto_state = ProtoState1});