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

return ok | {error, any()}

This commit is contained in:
Feng 2016-03-11 23:43:41 +08:00
parent d9d7581013
commit 37fa9a2f9b
1 changed files with 16 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
src/emqttd_backend.erl
View File

@ -49,21 +49,20 @@ mnesia(copy) ->
%%-------------------------------------------------------------------- %%--------------------------------------------------------------------
%% @doc Add a static subscription manually. %% @doc Add a static subscription manually.
-spec add_subscription(mqtt_subscription()) -> {atom, ok}. -spec add_subscription(mqtt_subscription()) -> ok | {error, already_existed}.
add_subscription(Subscription = #mqtt_subscription{subid = SubId, topic = Topic}) -> add_subscription(Subscription = #mqtt_subscription{subid = SubId, topic = Topic}) ->
Pattern = match_pattern(SubId, Topic), Pattern = match_pattern(SubId, Topic),
mnesia:transaction( return(mnesia:transaction(fun() ->
fun() -> case mnesia:match_object(backend_subscription, Pattern, write) of
case mnesia:match_object(backend_subscription, Pattern, write) of [] ->
[] -> mnesia:write(backend_subscription, Subscription, write);
mnesia:write(backend_subscription, Subscription, write); [Subscription] ->
[Subscription] -> mnesia:abort(already_existed);
mnesia:abort({error, existed}); [Subscription1] -> %% QoS is different
[Subscription1] -> %% QoS is different mnesia:delete_object(backend_subscription, Subscription1, write),
mnesia:delete_object(backend_subscription, Subscription1, write), mnesia:write(backend_subscription, Subscription, write)
mnesia:write(backend_subscription, Subscription, write) end
end end)).
end).
%% @doc Lookup static subscriptions. %% @doc Lookup static subscriptions.
-spec lookup_subscriptions(binary()) -> list(mqtt_subscription()). -spec lookup_subscriptions(binary()) -> list(mqtt_subscription()).
@ -73,12 +72,12 @@ lookup_subscriptions(ClientId) when is_binary(ClientId) ->
%% @doc Delete static subscriptions by ClientId manually. %% @doc Delete static subscriptions by ClientId manually.
-spec del_subscriptions(binary()) -> ok. -spec del_subscriptions(binary()) -> ok.
del_subscriptions(ClientId) when is_binary(ClientId) -> del_subscriptions(ClientId) when is_binary(ClientId) ->
mnesia:transaction(fun mnesia:delete/1, [{backend_subscription, ClientId}]). return(mnesia:transaction(fun mnesia:delete/1, [{backend_subscription, ClientId}])).
%% @doc Delete a static subscription manually. %% @doc Delete a static subscription manually.
-spec del_subscription(binary(), binary()) -> ok. -spec del_subscription(binary(), binary()) -> ok.
del_subscription(ClientId, Topic) when is_binary(ClientId) andalso is_binary(Topic) -> del_subscription(ClientId, Topic) when is_binary(ClientId) andalso is_binary(Topic) ->
mnesia:transaction(fun del_subscription_/1, [match_pattern(ClientId, Topic)]). return(mnesia:transaction(fun del_subscription_/1, [match_pattern(ClientId, Topic)])).
del_subscription_(Pattern) -> del_subscription_(Pattern) ->
lists:foreach(fun(Subscription) -> lists:foreach(fun(Subscription) ->
@ -88,3 +87,5 @@ del_subscription_(Pattern) ->
match_pattern(SubId, Topic) -> match_pattern(SubId, Topic) ->
#mqtt_subscription{subid = SubId, topic = Topic, qos = '_'}. #mqtt_subscription{subid = SubId, topic = Topic, qos = '_'}.
return({atomic, ok}) -> ok;
return({aborted, Reason}) -> {error, Reason}.