Replace emqttd_misc:inc_stats/1 with inc_stats/1

This commit is contained in:
Feng 2017-02-22 23:08:22 +08:00
parent 239cf1b5fc
commit 365bfb9e80
1 changed files with 5 additions and 3 deletions

View File

@ -301,7 +301,7 @@ init([CleanSess, {ClientId, Username}, ClientPid]) ->
created_at = os:timestamp()}, created_at = os:timestamp()},
emqttd_sm:register_session(ClientId, CleanSess, info(State)), emqttd_sm:register_session(ClientId, CleanSess, info(State)),
emqttd_hooks:run('session.created', [ClientId, Username]), emqttd_hooks:run('session.created', [ClientId, Username]),
{ok, emit_stats(State), hibernate, {backoff, 1000, 1000, 10000}, ?MODULE}. {ok, emit_stats(State), hibernate, {backoff, 1000, 1000, 10000}}.
init_stats(Keys) -> init_stats(Keys) ->
lists:foreach(fun(K) -> put(K, 0) end, Keys). lists:foreach(fun(K) -> put(K, 0) end, Keys).
@ -690,7 +690,7 @@ dispatch(Msg = #mqtt_message{qos = QoS},
end. end.
enqueue_msg(Msg, State = #state{mqueue = Q}) -> enqueue_msg(Msg, State = #state{mqueue = Q}) ->
emqttd_misc:inc_stats(enqueue_msg), inc_stats(enqueue_msg),
State#state{mqueue = emqttd_mqueue:in(Msg, Q)}. State#state{mqueue = emqttd_mqueue:in(Msg, Q)}.
%%-------------------------------------------------------------------- %%--------------------------------------------------------------------
@ -701,7 +701,7 @@ redeliver(Msg = #mqtt_message{qos = QoS}, State) ->
deliver(Msg#mqtt_message{dup = if QoS =:= ?QOS2 -> false; true -> true end}, State). deliver(Msg#mqtt_message{dup = if QoS =:= ?QOS2 -> false; true -> true end}, State).
deliver(Msg, #state{client_pid = Pid}) -> deliver(Msg, #state{client_pid = Pid}) ->
emqttd_misc:inc_stats(deliver_msg), inc_stats(deliver_msg),
Pid ! {deliver, Msg}. Pid ! {deliver, Msg}.
%%-------------------------------------------------------------------- %%--------------------------------------------------------------------
@ -793,6 +793,8 @@ emit_stats(State = #state{client_id = ClientId}) ->
emqttd_stats:set_session_stats(ClientId, stats(State)), emqttd_stats:set_session_stats(ClientId, stats(State)),
State. State.
inc_stats(Key) -> put(Key, get(Key) + 1).
%%-------------------------------------------------------------------- %%--------------------------------------------------------------------
%% Helper functions %% Helper functions
%%-------------------------------------------------------------------- %%--------------------------------------------------------------------