yuanbiao
/
emqx
1
0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

mqtt_topic

This commit is contained in:
Feng Lee 2016-08-21 16:21:45 +08:00
parent 0816ec9966
commit a1ed02499c
2 changed files with 22 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
include/emqttd.hrl
View File

@ -36,16 +36,6 @@
-define(SHARE, <<"$share/">>). %% Shared Topic
%%--------------------------------------------------------------------
%% MQTT Topic
%%--------------------------------------------------------------------
-record(mqtt_topic, {
topic :: binary(),
flags :: [retained | static]
}).
-type mqtt_topic() :: #mqtt_topic{}.
%%--------------------------------------------------------------------
%% PubSub
%%--------------------------------------------------------------------
@ -54,6 +44,17 @@
-define(PUBSUB(PS), (PS =:= publish orelse PS =:= subscribe)).
%%--------------------------------------------------------------------
%% MQTT Topic
%%--------------------------------------------------------------------
-record(mqtt_topic, {
topic :: binary(),
flags = [] :: [retained | static]
}).
-type(mqtt_topic() :: #mqtt_topic{}).
%%--------------------------------------------------------------------
%% MQTT Client
%%--------------------------------------------------------------------

21
src/emqttd_router.erl
View File

@ -53,20 +53,19 @@
%%--------------------------------------------------------------------
mnesia(boot) ->
ok = emqttd_mnesia:create_table(mqtt_topic, [
{ram_copies, [node()]},
{record_name, mqtt_topic},
{attributes, record_info(fields, mqtt_topic)}]),
ok = emqttd_mnesia:create_table(mqtt_route, [
{type, bag},
{ram_copies, [node()]},
{record_name, mqtt_route},
{attributes, record_info(fields, mqtt_route)}]),
ok = emqttd_mnesia:create_table(topic, [
{ram_copies, [node()]},
{record_name, mqtt_topic},
{attributes, record_info(fields, mqtt_topic)}]);
{attributes, record_info(fields, mqtt_route)}]);
mnesia(copy) ->
ok = emqttd_mnesia:copy_table(mqtt_route, ram_copies),
ok = emqttd_mnesia:copy_table(topic).
ok = emqttd_mnesia:copy_table(topic),
ok = emqttd_mnesia:copy_table(mqtt_route, ram_copies).
%%--------------------------------------------------------------------
%% Start the Router
@ -126,7 +125,8 @@ add_route_(Route = #mqtt_route{topic = Topic}) ->
true -> emqttd_trie:insert(Topic);
false -> ok
end,
mnesia:write(Route);
mnesia:write(Route),
mnesia:write(#mqtt_topic{topic = Topic});
Records ->
case lists:member(Route, Records) of
true -> ok;
@ -164,7 +164,8 @@ del_route_(Route = #mqtt_route{topic = Topic}) ->
case emqttd_topic:wildcard(Topic) of
true -> emqttd_trie:delete(Topic);
false -> ok
end;
end,
mnesia:delete({mqtt_topic, Topic});
_More ->
%% Remove route only
mnesia:delete_object(Route)