%% Copyright (c) 2013, Pedram Nimreezi <deadzen@deadzen.com>
|
%%
|
%% Permission to use, copy, modify, and/or distribute this software for any
|
%% purpose with or without fee is hereby granted, provided that the above
|
%% copyright notice and this permission notice appear in all copies.
|
%%
|
%% THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
%% WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
%% MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
%% ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
%% WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
%% ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
|
%% OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
|
-module(gr_param).
|
|
-behaviour(gen_server).
|
|
%% API
|
-export([start_link/1,
|
list/1, insert/2,
|
lookup/2, lookup_element/2,
|
info/1, info_size/1, transform/1]).
|
|
%% gen_server callbacks
|
-export([init/1,
|
handle_call/3,
|
handle_cast/2,
|
handle_info/2,
|
terminate/2,
|
code_change/3]).
|
|
-record(state, {table_id, waiting=[]}).
|
|
%%%===================================================================
|
%%% API
|
%%%===================================================================
|
list(Server) ->
|
case (catch gen_server:call(Server, list)) of
|
{'EXIT', _Reason} ->
|
list(gr_manager:wait_for_pid(Server));
|
Else -> Else
|
end.
|
|
info_size(Server) ->
|
case (catch gen_server:call(Server, info_size)) of
|
{'EXIT', _Reason} ->
|
info_size(gr_manager:wait_for_pid(Server));
|
Else -> Else
|
end.
|
|
insert(Server, Term) ->
|
case (catch gen_server:call(Server, {insert, Term})) of
|
{'EXIT', _Reason} ->
|
insert(gr_manager:wait_for_pid(Server), Term);
|
Else -> Else
|
end.
|
|
lookup(Server, Term) ->
|
case (catch gen_server:call(Server, {lookup, Term})) of
|
{'EXIT', _Reason} ->
|
lookup(gr_manager:wait_for_pid(Server), Term);
|
Else -> Else
|
end.
|
|
lookup_element(Server, Term) ->
|
case (catch gen_server:call(Server, {lookup_element, Term})) of
|
{'EXIT', _Reason} ->
|
lookup_element(gr_manager:wait_for_pid(Server), Term);
|
Else -> Else
|
end.
|
|
info(Server) ->
|
case (catch gen_server:call(Server, info)) of
|
{'EXIT', _Reason} ->
|
info(gr_manager:wait_for_pid(Server));
|
Else -> Else
|
end.
|
|
%% @doc Transform Term -> Key to Key -> Term
|
transform(Server) ->
|
case (catch gen_server:call(Server, transform)) of
|
{'EXIT', _Reason} ->
|
transform(gr_manager:wait_for_pid(Server));
|
Else -> Else
|
end.
|
|
%%--------------------------------------------------------------------
|
%% @doc
|
%% Starts the server
|
%%
|
%% @spec start_link(Name) -> {ok, Pid} | ignore | {error, Error}
|
%% @end
|
%%--------------------------------------------------------------------
|
start_link(Name) ->
|
gen_server:start_link({local, Name}, ?MODULE, [], []).
|
|
%%%===================================================================
|
%%% gen_server callbacks
|
%%%===================================================================
|
|
%%--------------------------------------------------------------------
|
%% @private
|
%% @doc
|
%% Initializes the server
|
%%
|
%% @spec init(Args) -> {ok, State} |
|
%% {ok, State, Timeout} |
|
%% ignore |
|
%% {stop, Reason}
|
%% @end
|
%%--------------------------------------------------------------------
|
init([]) ->
|
{ok, #state{}}.
|
|
%%--------------------------------------------------------------------
|
%% @private
|
%% @doc
|
%% Handling call messages
|
%%
|
%% @spec handle_call(Request, From, State) ->
|
%% {reply, Reply, State} |
|
%% {reply, Reply, State, Timeout} |
|
%% {noreply, State} |
|
%% {noreply, State, Timeout} |
|
%% {stop, Reason, Reply, State} |
|
%% {stop, Reason, State}
|
%% @end
|
%%--------------------------------------------------------------------
|
handle_call(Call, From, State) when is_atom(Call), Call =:= list;
|
Call =:= info; Call =:= info_size;
|
Call =:= transform ->
|
TableId = State#state.table_id,
|
Waiting = State#state.waiting,
|
case TableId of
|
undefined -> {noreply, State#state{waiting=[{Call, From}|Waiting]}};
|
_ when Call =:= list ->
|
{reply, handle_list(TableId), State};
|
_ when Call =:= info ->
|
{reply, handle_info(TableId), State};
|
_ when Call =:= info_size ->
|
{reply, handle_info_size(TableId), State};
|
_ when Call =:= transform ->
|
{reply, handle_transform(TableId), State}
|
end;
|
|
handle_call({Call, Term}, From, State) when is_atom(Call), Call =:= insert;
|
Call =:= lookup;
|
Call =:= lookup_element ->
|
TableId = State#state.table_id,
|
Waiting = State#state.waiting,
|
case TableId of
|
undefined ->
|
{noreply, State#state{waiting=[{{Call, Term}, From}|Waiting]}};
|
_ when Call =:= insert ->
|
{reply, handle_insert(TableId, Term), State};
|
_ when Call =:= lookup ->
|
{reply, handle_lookup(TableId, Term), State};
|
_ when Call =:= lookup_element ->
|
{reply, handle_lookup_element(TableId, Term), State}
|
end;
|
|
handle_call(_Request, _From, State) ->
|
Reply = {error, unhandled_message},
|
{reply, Reply, State}.
|
|
%%--------------------------------------------------------------------
|
%% @private
|
%% @doc
|
%% Handling cast messages
|
%%
|
%% @spec handle_cast(Msg, State) -> {noreply, State} |
|
%% {noreply, State, Timeout} |
|
%% {stop, Reason, State}
|
%% @end
|
%%--------------------------------------------------------------------
|
handle_cast(_Msg, State) ->
|
{noreply, State}.
|
|
%%--------------------------------------------------------------------
|
%% @private
|
%% @doc
|
%% Handling all non call/cast messages
|
%%
|
%% @spec handle_info(Info, State) -> {noreply, State} |
|
%% {noreply, State, Timeout} |
|
%% {stop, Reason, State}
|
%% @end
|
%%--------------------------------------------------------------------
|
handle_info({'ETS-TRANSFER', TableId, _Pid, _Data}, State) ->
|
_ = [ gen_server:reply(From, perform_call(TableId, Call))
|
|| {Call, From} <- State#state.waiting ],
|
{noreply, State#state{table_id=TableId, waiting=[]}};
|
handle_info(_Info, State) ->
|
{noreply, State}.
|
|
|
|
%%--------------------------------------------------------------------
|
%% @private
|
%% @doc
|
%% This function is called by a gen_server when it is about to
|
%% terminate. It should be the opposite of Module:init/1 and do any
|
%% necessary cleaning up. When it returns, the gen_server terminates
|
%% with Reason. The return value is ignored.
|
%%
|
%% @spec terminate(Reason, State) -> void()
|
%% @end
|
%%--------------------------------------------------------------------
|
terminate(_Reason, _State) ->
|
ok.
|
|
%%--------------------------------------------------------------------
|
%% @private
|
%% @doc
|
%% Convert process state when code is changed
|
%%
|
%% @spec code_change(OldVsn, State, Extra) -> {ok, NewState}
|
%% @end
|
%%--------------------------------------------------------------------
|
code_change(_OldVsn, State, _Extra) ->
|
{ok, State}.
|
|
%%%===================================================================
|
%%% Internal functions
|
%%%===================================================================
|
|
perform_call(TableId, Call) ->
|
case Call of
|
list ->
|
handle_list(TableId);
|
info ->
|
handle_info(TableId);
|
info_size ->
|
handle_info_size(TableId);
|
transform ->
|
handle_transform(TableId);
|
{insert, Term} ->
|
handle_insert(TableId, Term);
|
{lookup, Term} ->
|
handle_lookup(TableId, Term);
|
{lookup_element, Term} ->
|
handle_lookup_element(TableId, Term)
|
end.
|
|
|
handle_list(TableId) ->
|
ets:tab2list(TableId).
|
|
handle_info(TableId) ->
|
ets:info(TableId).
|
|
handle_info_size(TableId) ->
|
ets:info(TableId, size).
|
|
handle_transform(TableId) ->
|
ParamsList = [{K, V} || {V, K} <- ets:tab2list(TableId)],
|
ets:delete_all_objects(TableId),
|
ets:insert(TableId, ParamsList).
|
|
handle_insert(TableId, Term) ->
|
ets:insert(TableId, Term).
|
|
handle_lookup(TableId, Term) ->
|
ets:lookup(TableId, Term).
|
|
handle_lookup_element(TableId, Term) ->
|
ets:lookup_element(TableId, Term, 2).
|
