Преглед изворни кода

Merge pull request #795 from emqtt/emq20

ws metrics
Feng Lee пре 9 година
родитељ
комит
d2bb3925a0
2 измењених фајлова са 2 додато и 0 уклоњено
  1. 1 0
      src/emqttd_ws.erl
  2. 1 0
      src/emqttd_ws_client.erl

+ 1 - 0
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};

+ 1 - 0
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});