TAB -> TABLE

This commit is contained in:
Ery Lee 2015-04-10 20:26:40 +08:00
parent 1a81a4cc68
commit cb6967cd46
5 changed files with 37 additions and 37 deletions

View File

@ -98,7 +98,7 @@ stop() ->
gen_server:call(?MODULE, stop).
init([AuthMods]) ->
ets:new(?AUTH_TABLE, [set, named_table, protected]),
ets:new(?AUTH_TABLE, [set, named_table, protected, {read_concurrency, true}]),
Modules = lists:map(
fun({Mod, Opts}) ->
AuthMod = authmod(Mod),

View File

@ -34,7 +34,7 @@
-define(SERVER, ?MODULE).
-define(BROKER_TAB, mqtt_broker).
-define(BROKER_TABLE, mqtt_broker).
%% API Function Exports
-export([start_link/1]).
@ -115,7 +115,7 @@ datetime() ->
%%------------------------------------------------------------------------------
-spec getstats() -> [{atom(), non_neg_integer()}].
getstats() ->
ets:tab2list(?BROKER_TAB).
ets:tab2list(?BROKER_TABLE).
%%------------------------------------------------------------------------------
%% @doc
@ -125,7 +125,7 @@ getstats() ->
%%------------------------------------------------------------------------------
-spec getstat(atom()) -> non_neg_integer() | undefined.
getstat(Name) ->
case ets:lookup(?BROKER_TAB, Name) of
case ets:lookup(?BROKER_TABLE, Name) of
[{Name, Val}] -> Val;
[] -> undefined
end.
@ -138,7 +138,7 @@ getstat(Name) ->
%%------------------------------------------------------------------------------
-spec setstat(atom(), pos_integer()) -> boolean().
setstat(Name, Val) ->
ets:insert(?BROKER_TAB, {Name, Val}).
ets:insert(?BROKER_TABLE, {Name, Val}).
%%%=============================================================================
%%% gen_server callbacks
@ -146,9 +146,9 @@ setstat(Name, Val) ->
init([Options]) ->
random:seed(now()),
ets:new(?BROKER_TAB, [set, public, named_table, {write_concurrency, true}]),
ets:new(?BROKER_TABLE, [set, public, named_table, {write_concurrency, true}]),
Topics = ?SYSTOP_CLIENTS ++ ?SYSTOP_SESSIONS ++ ?SYSTOP_PUBSUB,
[ets:insert(?BROKER_TAB, {Topic, 0}) || Topic <- Topics],
[ets:insert(?BROKER_TABLE, {Topic, 0}) || Topic <- Topics],
% Create $SYS Topics
[ok = create(systop(Topic)) || Topic <- ?SYSTOP_BROKERS],
[ok = create(systop(Topic)) || Topic <- Topics],
@ -175,7 +175,7 @@ handle_info(tick, State) ->
publish(systop(uptime), list_to_binary(uptime(State))),
publish(systop(datetime), list_to_binary(datetime())),
[publish(systop(Stat), i2b(Val))
|| {Stat, Val} <- ets:tab2list(?BROKER_TAB)],
|| {Stat, Val} <- ets:tab2list(?BROKER_TABLE)],
{noreply, tick(State), hibernate};
handle_info(_Info, State) ->

View File

@ -32,7 +32,7 @@
-define(SERVER, ?MODULE).
-define(CLIENT_TAB, mqtt_client).
-define(CLIENT_TABLE, mqtt_client).
%% API Exports
-export([start_link/0]).
@ -73,7 +73,7 @@ start_link() ->
%%------------------------------------------------------------------------------
-spec lookup(ClientId :: binary()) -> pid() | undefined.
lookup(ClientId) when is_binary(ClientId) ->
case ets:lookup(?CLIENT_TAB, ClientId) of
case ets:lookup(?CLIENT_TABLE, ClientId) of
[{_, Pid, _}] -> Pid;
[] -> undefined
end.
@ -113,11 +113,11 @@ getstats() ->
%%%=============================================================================
init([]) ->
ets:new(?CLIENT_TAB, [set, named_table, protected]),
ets:new(?CLIENT_TABLE, [set, named_table, protected]),
{ok, #state{}}.
handle_call({register, ClientId, Pid}, _From, State) ->
case ets:lookup(?CLIENT_TAB, ClientId) of
case ets:lookup(?CLIENT_TABLE, ClientId) of
[{_, Pid, _}] ->
lager:error("clientId '~s' has been registered with ~p", [ClientId, Pid]),
ignore;
@ -131,7 +131,7 @@ handle_call({register, ClientId, Pid}, _From, State) ->
{reply, ok, setstats(State)};
handle_call(getstats, _From, State = #state{max = Max}) ->
Stats = [{'clients/count', ets:info(?CLIENT_TAB, size)},
Stats = [{'clients/count', ets:info(?CLIENT_TABLE, size)},
{'clients/max', Max}],
{reply, Stats, State};
@ -139,10 +139,10 @@ handle_call(_Request, _From, State) ->
{reply, ok, State}.
handle_cast({unregister, ClientId, Pid}, State) ->
case ets:lookup(?CLIENT_TAB, ClientId) of
case ets:lookup(?CLIENT_TABLE, ClientId) of
[{_, Pid, MRef}] ->
erlang:demonitor(MRef, [flush]),
ets:delete(?CLIENT_TAB, ClientId);
ets:delete(?CLIENT_TABLE, ClientId);
[_] ->
ignore;
[] ->
@ -154,7 +154,7 @@ handle_cast(_Msg, State) ->
{noreply, State}.
handle_info({'DOWN', MRef, process, DownPid, _Reason}, State) ->
ets:match_delete(?CLIENT_TAB, {'_', DownPid, MRef}),
ets:match_delete(?CLIENT_TABLE, {'_', DownPid, MRef}),
{noreply, setstats(State)};
handle_info(_Info, State) ->
@ -171,10 +171,10 @@ code_change(_OldVsn, State, _Extra) ->
%%%=============================================================================
insert(ClientId, Pid) ->
ets:insert(?CLIENT_TAB, {ClientId, Pid, erlang:monitor(process, Pid)}).
ets:insert(?CLIENT_TABLE, {ClientId, Pid, erlang:monitor(process, Pid)}).
setstats(State = #state{max = Max}) ->
Count = ets:info(?CLIENT_TAB, size),
Count = ets:info(?CLIENT_TABLE, size),
emqttd_broker:setstat('clients/count', Count),
if
Count > Max ->

View File

@ -36,7 +36,7 @@
-define(SERVER, ?MODULE).
-define(METRIC_TAB, mqtt_broker_metric).
-define(METRIC_TABLE, mqtt_broker_metric).
%% API Function Exports
-export([start_link/1]).
@ -81,7 +81,7 @@ all() ->
{ok, Count} -> maps:put(Metric, Count+Val, Map);
error -> maps:put(Metric, Val, Map)
end
end, #{}, ?METRIC_TAB)).
end, #{}, ?METRIC_TABLE)).
%%------------------------------------------------------------------------------
%% @doc
@ -91,7 +91,7 @@ all() ->
%%------------------------------------------------------------------------------
-spec value(atom()) -> non_neg_integer().
value(Metric) ->
lists:sum(ets:select(?METRIC_TAB, [{{{Metric, '_'}, '$1'}, [], ['$1']}])).
lists:sum(ets:select(?METRIC_TABLE, [{{{Metric, '_'}, '$1'}, [], ['$1']}])).
%%------------------------------------------------------------------------------
%% @doc
@ -125,9 +125,9 @@ inc(Metric, Val) when is_atom(Metric) and is_integer(Val) ->
%%------------------------------------------------------------------------------
-spec inc(counter | gauge, atom(), pos_integer()) -> pos_integer().
inc(gauge, Metric, Val) ->
ets:update_counter(?METRIC_TAB, key(gauge, Metric), {2, Val});
ets:update_counter(?METRIC_TABLE, key(gauge, Metric), {2, Val});
inc(counter, Metric, Val) ->
ets:update_counter(?METRIC_TAB, key(counter, Metric), {2, Val}).
ets:update_counter(?METRIC_TABLE, key(counter, Metric), {2, Val}).
%%------------------------------------------------------------------------------
%% @doc
@ -147,7 +147,7 @@ dec(gauge, Metric) ->
%%------------------------------------------------------------------------------
-spec dec(gauge, atom(), pos_integer()) -> integer().
dec(gauge, Metric, Val) ->
ets:update_counter(?METRIC_TAB, key(gauge, Metric), {2, -Val}).
ets:update_counter(?METRIC_TABLE, key(gauge, Metric), {2, -Val}).
%%------------------------------------------------------------------------------
%% @doc
@ -158,7 +158,7 @@ dec(gauge, Metric, Val) ->
set(Metric, Val) when is_atom(Metric) ->
set(gauge, Metric, Val).
set(gauge, Metric, Val) ->
ets:insert(?METRIC_TAB, {key(gauge, Metric), Val}).
ets:insert(?METRIC_TABLE, {key(gauge, Metric), Val}).
%%------------------------------------------------------------------------------
%% @doc
@ -180,7 +180,7 @@ init([Options]) ->
random:seed(now()),
Metrics = ?SYSTOP_BYTES ++ ?SYSTOP_PACKETS ++ ?SYSTOP_MESSAGES,
% Create metrics table
ets:new(?METRIC_TAB, [set, public, named_table, {write_concurrency, true}]),
ets:new(?METRIC_TABLE, [set, public, named_table, {write_concurrency, true}]),
% Init metrics
[new_metric(Metric) || Metric <- Metrics],
% $SYS Topics for metrics
@ -223,11 +223,11 @@ publish(Topic, Payload) ->
emqttd_router:route(#mqtt_message{topic = Topic, payload = Payload}).
new_metric({gauge, Name}) ->
ets:insert(?METRIC_TAB, {{Name, 0}, 0});
ets:insert(?METRIC_TABLE, {{Name, 0}, 0});
new_metric({counter, Name}) ->
Schedulers = lists:seq(1, erlang:system_info(schedulers)),
[ets:insert(?METRIC_TAB, {{Name, I}, 0}) || I <- Schedulers].
[ets:insert(?METRIC_TABLE, {{Name, I}, 0}) || I <- Schedulers].
tick(State = #state{pub_interval = PubInterval}) ->
tick(PubInterval, State).

View File

@ -44,7 +44,7 @@
-define(SERVER, ?MODULE).
-define(SESSION_TAB, mqtt_session).
-define(SESSION_TABLE, mqtt_session).
%% API Function Exports
-export([start_link/0]).
@ -72,7 +72,7 @@ start_link() ->
%%------------------------------------------------------------------------------
-spec lookup_session(binary()) -> pid() | undefined.
lookup_session(ClientId) ->
case ets:lookup(?SESSION_TAB, ClientId) of
case ets:lookup(?SESSION_TABLE, ClientId) of
[{_, SessPid, _}] -> SessPid;
[] -> undefined
end.
@ -103,12 +103,12 @@ destroy_session(ClientId) ->
init([]) ->
process_flag(trap_exit, true),
ets:new(?SESSION_TAB, [set, protected, named_table]),
ets:new(?SESSION_TABLE, [set, protected, named_table]),
{ok, #state{}}.
handle_call({start_session, ClientId, ClientPid}, _From, State) ->
Reply =
case ets:lookup(?SESSION_TAB, ClientId) of
case ets:lookup(?SESSION_TABLE, ClientId) of
[{_, SessPid, _MRef}] ->
emqttd_session:resume(SessPid, ClientId, ClientPid),
{ok, SessPid};
@ -116,7 +116,7 @@ handle_call({start_session, ClientId, ClientPid}, _From, State) ->
case emqttd_session_sup:start_session(ClientId, ClientPid) of
{ok, SessPid} ->
MRef = erlang:monitor(process, SessPid),
ets:insert(?SESSION_TAB, {ClientId, SessPid, MRef}),
ets:insert(?SESSION_TABLE, {ClientId, SessPid, MRef}),
{ok, SessPid};
{error, Error} ->
{error, Error}
@ -125,11 +125,11 @@ handle_call({start_session, ClientId, ClientPid}, _From, State) ->
{reply, Reply, setstats(State)};
handle_call({destroy_session, ClientId}, _From, State) ->
case ets:lookup(?SESSION_TAB, ClientId) of
case ets:lookup(?SESSION_TABLE, ClientId) of
[{_, SessPid, MRef}] ->
erlang:demonitor(MRef),
emqttd_session:destroy(SessPid, ClientId),
ets:delete(?SESSION_TAB, ClientId);
ets:delete(?SESSION_TABLE, ClientId);
[] ->
ignore
end,
@ -142,7 +142,7 @@ handle_cast(_Msg, State) ->
{noreply, State}.
handle_info({'DOWN', MRef, process, DownPid, _Reason}, State) ->
ets:match_delete(?SESSION_TAB, {'_', DownPid, MRef}),
ets:match_delete(?SESSION_TABLE, {'_', DownPid, MRef}),
{noreply, setstats(State)};
handle_info(_Info, State) ->
@ -159,7 +159,7 @@ code_change(_OldVsn, State, _Extra) ->
%%%=============================================================================
setstats(State = #state{max = Max}) ->
Count = ets:info(?SESSION_TAB, size),
Count = ets:info(?SESSION_TABLE, size),
emqttd_broker:setstat('sessions/count', Count),
if
Count > Max ->