diff options
Diffstat (limited to 'src/ejabberd_sm_mnesia.erl')
| -rw-r--r-- | src/ejabberd_sm_mnesia.erl | 119 |
1 files changed, 68 insertions, 51 deletions
diff --git a/src/ejabberd_sm_mnesia.erl b/src/ejabberd_sm_mnesia.erl index b900da315..9626ba3d6 100644 --- a/src/ejabberd_sm_mnesia.erl +++ b/src/ejabberd_sm_mnesia.erl @@ -1,11 +1,27 @@ %%%------------------------------------------------------------------- -%%% @author Evgeny Khramtsov <ekhramtsov@process-one.net> -%%% @copyright (C) 2015-2016, Evgeny Khramtsov -%%% @doc -%%% -%%% @end +%%% File : ejabberd_sm_mnesia.erl +%%% Author : Evgeny Khramtsov <ekhramtsov@process-one.net> %%% Created : 9 Mar 2015 by Evgeny Khramtsov <ekhramtsov@process-one.net> -%%%------------------------------------------------------------------- +%%% +%%% +%%% ejabberd, Copyright (C) 2002-2019 ProcessOne +%%% +%%% This program is free software; you can redistribute it and/or +%%% modify it under the terms of the GNU General Public License as +%%% published by the Free Software Foundation; either version 2 of the +%%% License, or (at your option) any later version. +%%% +%%% This program is distributed in the hope that it will be useful, +%%% but WITHOUT ANY WARRANTY; without even the implied warranty of +%%% MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +%%% General Public License for more details. +%%% +%%% You should have received a copy of the GNU General Public License along +%%% with this program; if not, write to the Free Software Foundation, Inc., +%%% 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. +%%% +%%%---------------------------------------------------------------------- + -module(ejabberd_sm_mnesia). -behaviour(gen_server). @@ -13,20 +29,19 @@ %% API -export([init/0, + use_cache/1, set_session/1, - delete_session/4, + delete_session/1, get_sessions/0, get_sessions/1, - get_sessions/2, - get_sessions/3]). + get_sessions/2]). %% gen_server callbacks -export([init/1, handle_call/3, handle_cast/2, handle_info/2, - terminate/2, code_change/3]). + terminate/2, code_change/3, start_link/0]). --include("ejabberd.hrl"). -include("ejabberd_sm.hrl"). --include("jlib.hrl"). +-include("logger.hrl"). -include_lib("stdlib/include/ms_transform.hrl"). -record(state, {}). @@ -36,27 +51,28 @@ %%%=================================================================== -spec init() -> ok | {error, any()}. init() -> - case gen_server:start_link({local, ?MODULE}, ?MODULE, [], []) of - {ok, _Pid} -> - ok; - Err -> - Err + Spec = {?MODULE, {?MODULE, start_link, []}, + transient, 5000, worker, [?MODULE]}, + case supervisor:start_child(ejabberd_backend_sup, Spec) of + {ok, _Pid} -> ok; + Err -> Err end. +-spec start_link() -> {ok, pid()} | {error, any()}. +start_link() -> + gen_server:start_link({local, ?MODULE}, ?MODULE, [], []). + +-spec use_cache(binary()) -> boolean(). +use_cache(_LServer) -> + false. + -spec set_session(#session{}) -> ok. set_session(Session) -> mnesia:dirty_write(Session). --spec delete_session(binary(), binary(), binary(), sid()) -> - {ok, #session{}} | {error, notfound}. -delete_session(_LUser, _LServer, _LResource, SID) -> - case mnesia:dirty_read(session, SID) of - [Session] -> - mnesia:dirty_delete(session, SID), - {ok, Session}; - [] -> - {error, notfound} - end. +-spec delete_session(#session{}) -> ok. +delete_session(#session{sid = SID}) -> + mnesia:dirty_delete(session, SID). -spec get_sessions() -> [#session{}]. get_sessions() -> @@ -68,48 +84,49 @@ get_sessions(LServer) -> [{#session{usr = '$1', _ = '_'}, [{'==', {element, 2, '$1'}, LServer}], ['$_']}]). --spec get_sessions(binary(), binary()) -> [#session{}]. +-spec get_sessions(binary(), binary()) -> {ok, [#session{}]}. get_sessions(LUser, LServer) -> - mnesia:dirty_index_read(session, {LUser, LServer}, #session.us). - --spec get_sessions(binary(), binary(), binary()) -> [#session{}]. -get_sessions(LUser, LServer, LResource) -> - mnesia:dirty_index_read(session, {LUser, LServer, LResource}, #session.usr). + {ok, mnesia:dirty_index_read(session, {LUser, LServer}, #session.us)}. %%%=================================================================== %%% gen_server callbacks %%%=================================================================== init([]) -> update_tables(), - mnesia:create_table(session, + ejabberd_mnesia:create(?MODULE, session, [{ram_copies, [node()]}, - {attributes, record_info(fields, session)}]), - mnesia:create_table(session_counter, + {attributes, record_info(fields, session)}, + {index, [usr,us]}]), + ejabberd_mnesia:create(?MODULE, session_counter, [{ram_copies, [node()]}, {attributes, record_info(fields, session_counter)}]), - mnesia:add_table_index(session, usr), - mnesia:add_table_index(session, us), - mnesia:add_table_copy(session, node(), ram_copies), - mnesia:add_table_copy(session_counter, node(), ram_copies), mnesia:subscribe(system), {ok, #state{}}. -handle_call(_Request, _From, State) -> - Reply = ok, - {reply, Reply, State}. +handle_call(Request, From, State) -> + ?WARNING_MSG("Unexpected call from ~p: ~p", [From, Request]), + {noreply, State}. -handle_cast(_Msg, State) -> +handle_cast(Msg, State) -> + ?WARNING_MSG("Unexpected cast: ~p", [Msg]), {noreply, State}. handle_info({mnesia_system_event, {mnesia_down, Node}}, State) -> - ets:select_delete( - session, - ets:fun2ms( - fun(#session{sid = {_, Pid}}) -> - node(Pid) == Node - end)), + Sessions = + ets:select( + session, + ets:fun2ms( + fun(#session{sid = {_, Pid}} = S) + when node(Pid) == Node -> + S + end)), + lists:foreach( + fun(S) -> + mnesia:dirty_delete_object(S) + end, Sessions), {noreply, State}; -handle_info(_Info, State) -> +handle_info(Info, State) -> + ?WARNING_MSG("Unexpected info: ~p", [Info]), {noreply, State}. terminate(_Reason, _State) -> |