./ct_report/coverage/mod_roster_mnesia.COVER.html

1 %%%----------------------------------------------------------------------
2 %%% File : mod_roster_mnesia.erl
3 %%% Author : Michał Piotrowski <michal.piotrowski@erlang-solutions.com>
4 %%% Purpose : mod_roster mnesia backend
5 %%%
6 %%%
7 %%% ejabberd, Copyright (C) 2002-2014 ProcessOne
8 %%% MongooseIM, Copyright (C) 2015 Erlang Solutions Ltd.
9 %%%
10 %%%----------------------------------------------------------------------
11 -module(mod_roster_mnesia).
12
13 -include("mod_roster.hrl").
14
15 -behaviour(mod_roster_backend).
16
17 %% API
18 -export([init/2,
19 transaction/2,
20 read_roster_version/3,
21 write_roster_version/5,
22 get_roster/3,
23 get_roster_entry/6,
24 get_subscription_lists/3,
25 roster_subscribe_t/2,
26 update_roster_t/2,
27 del_roster_t/4,
28 remove_user_t/3]).
29
30 -spec init(mongooseim:host_type(), gen_mod:module_opts()) -> ok.
31 init(_HostType, _Opts) ->
32 436 mongoose_mnesia:create_table(roster,
33 [{disc_copies, [node()]},
34 {attributes, record_info(fields, roster)}]),
35 436 mongoose_mnesia:create_table(roster_version,
36 [{disc_copies, [node()]},
37 {attributes, record_info(fields, roster_version)}]),
38 436 mnesia:add_table_index(roster, us),
39 436 mnesia:add_table_index(roster_version, us),
40 436 ok.
41
42 -spec transaction(mongooseim:host_type(), fun(() -> any())) ->
43 {aborted, any()} | {atomic, any()}.
44 transaction(_HostType, F) ->
45 3760 mnesia:transaction(F).
46
47 -spec read_roster_version(mongooseim:host_type(), jid:luser(), jid:lserver()) ->
48 binary() | error.
49 read_roster_version(_HostType, LUser, LServer) ->
50 4 US = {LUser, LServer},
51 4 case mnesia:dirty_read(roster_version, US) of
52 3 [#roster_version{version = V}] -> V;
53 1 [] -> error
54 end.
55
56 -spec write_roster_version(mongooseim:host_type(), jid:luser(), jid:lserver(),
57 mod_roster:transaction_state(), mod_roster:version()) -> ok.
58 write_roster_version(_HostType, LUser, LServer, TransactionState, Ver) ->
59 2 write(#roster_version{us = {LUser, LServer}, version = Ver}, TransactionState).
60
61 -spec get_roster(mongooseim:host_type(), jid:luser(), jid:lserver()) -> [mod_roster:roster()].
62 get_roster(_HostType, LUser, LServer) ->
63 3172 US = {LUser, LServer},
64 3172 case catch mnesia:dirty_index_read(roster, US, #roster.us) of
65 3172 Items when is_list(Items)-> Items;
66 
:-(
 _ -> []
67 end.
68
69 -spec get_roster_entry(mongooseim:host_type(), jid:luser(), jid:lserver(), mod_roster:contact(),
70 mod_roster:transaction_state(), mod_roster:entry_format()) ->
71 mod_roster:roster() | does_not_exist.
72 get_roster_entry(_HostType, LUser, LServer, LJID, TransactionState, _Format) ->
73 921 LowerJID = jid:to_lower(LJID),
74 921 case read({roster, {{LUser, LServer}, LowerJID}}, TransactionState) of
75 [] ->
76 277 does_not_exist;
77 [I] ->
78 644 I#roster{jid = LJID, xs = []}
79 end.
80
81 -spec get_subscription_lists(mongoose_acc:t(), jid:luser(), jid:lserver()) -> [mod_roster:roster()].
82 get_subscription_lists(_, LUser, LServer) ->
83 3055 US = {LUser, LServer},
84 3055 case catch mnesia:dirty_index_read(roster, US, #roster.us) of
85 3055 Items when is_list(Items) -> Items;
86 
:-(
 _ -> []
87 end.
88
89 -spec roster_subscribe_t(mongooseim:host_type(), mod_roster:roster()) -> ok.
90 roster_subscribe_t(_HostType, Item) ->
91 608 mnesia:write(Item).
92
93 -spec update_roster_t(mongooseim:host_type(), mod_roster:roster()) -> ok.
94 update_roster_t(_HostType, Item) ->
95 120 mnesia:write(Item).
96
97 -spec del_roster_t(mongooseim:host_type(), jid:luser(), jid:lserver(), mod_roster:contact()) -> ok.
98 del_roster_t(_HostType, LUser, LServer, LJID) ->
99 27 mnesia:delete({roster, {{LUser, LServer}, LJID}}).
100
101 -spec remove_user_t(mongooseim:host_type(), jid:luser(), jid:lserver()) -> ok.
102 remove_user_t(_HostType, LUser, LServer) ->
103 3064 US = {LUser, LServer},
104 3064 lists:foreach(fun (R) -> mnesia:delete_object(R) end,
105 mnesia:index_read(roster, US, #roster.us)),
106 2994 ok.
107
108 %% Helpers
109
110 1 write(Record, in_transaction) -> mnesia:write(Record);
111 1 write(Record, no_transaction) -> mnesia:dirty_write(Record).
112
113 779 read(TabKey, in_transaction) -> mnesia:read(TabKey);
114 142 read(TabKey, no_transaction) -> mnesia:dirty_read(TabKey).
Line Hits Source