-
Notifications
You must be signed in to change notification settings - Fork 56
/
rfc4627_jsonrpc_registry.erl
109 lines (94 loc) · 3.81 KB
/
rfc4627_jsonrpc_registry.erl
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
%% JSON-RPC service registry
%%---------------------------------------------------------------------------
%% @author Tony Garnock-Jones <tonyg@kcbbs.gen.nz>
%% @author LShift Ltd. <query@lshift.net>
%% @copyright 2007, 2008 Tony Garnock-Jones and LShift Ltd.
%% @license
%%
%% Permission is hereby granted, free of charge, to any person
%% obtaining a copy of this software and associated documentation
%% files (the "Software"), to deal in the Software without
%% restriction, including without limitation the rights to use, copy,
%% modify, merge, publish, distribute, sublicense, and/or sell copies
%% of the Software, and to permit persons to whom the Software is
%% furnished to do so, subject to the following conditions:
%%
%% The above copyright notice and this permission notice shall be
%% included in all copies or substantial portions of the Software.
%%
%% THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
%% EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
%% MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
%% NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
%% BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
%% ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
%% CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
%% SOFTWARE.
%%---------------------------------------------------------------------------
%% @since 1.2.0
%% @private
%% @doc JSON-RPC service registry implementation.
%%
%% Started and managed by functions defined in module {@link rfc4627_jsonrpc}.
-module(rfc4627_jsonrpc_registry).
-include("rfc4627_jsonrpc.hrl").
-behaviour(gen_server).
-export([init/1, terminate/2, code_change/3, handle_call/3, handle_cast/2, handle_info/2]).
-define(TABLE_NAME, rfc4627_jsonrpc_registry).
%% @doc gen_server behaviour callback.
init(_Args) ->
case mnesia:create_table(?TABLE_NAME, [{attributes,[key, value]}]) of
{atomic,ok} ->
error_logger:error_msg("Creating new mnesia table for ~p~n", [?MODULE]),
ok;
{aborted, {already_exists, ?TABLE_NAME}} -> ok
end,
{ok, no_jsonrpc_state}.
%% @doc gen_server behaviour callback.
terminate(_Reason, _State) ->
%% FIXME: should we notify services here?
ok.
%% @doc gen_server behaviour callback.
code_change(_OldVsn, State, _Extra) ->
State.
%% @doc gen_server behaviour callback.
handle_call({lookup_service, Service}, _From, State) ->
case tx(lookup(service, Service)) of
none ->
{reply, not_found, State};
ServiceRec ->
{reply, ServiceRec, State}
end;
handle_call({register_service, Pid, ServiceDescription}, _From, State) ->
SD = ServiceDescription#service{handler = {pid, Pid}},
erlang:monitor(process, Pid),
tx(fun() ->
mnesia:write({?TABLE_NAME,{service_pid, Pid}, SD#service.name}),
mnesia:write({?TABLE_NAME,{service, SD#service.name}, SD})
end),
{reply, ok, State}.
%% @doc gen_server behaviour callback.
handle_cast(Request, State) ->
error_logger:error_msg("Unhandled cast in ~p: ~p", [?MODULE, Request]),
{noreply, State}.
%% @doc gen_server behaviour callback.
handle_info({'DOWN', _MonitorRef, process, DownPid, _Reason}, State) ->
case tx(lookup(service_pid, DownPid)) of
none ->
{noreply, State};
ServiceName ->
tx(fun() ->
mnesia:delete({?TABLE_NAME,{service_pid, DownPid}}),
mnesia:delete({?TABLE_NAME,{service, ServiceName}})
end),
{noreply, State}
end.
lookup(Type, Key) ->
fun() -> mnesia:read(?TABLE_NAME, {Type, Key}) end.
tx(Fun) ->
case mnesia:transaction(Fun) of
{atomic, []} -> none;
{atomic, ok} -> ok;
{atomic, abort} -> abort;
{atomic, [{?TABLE_NAME, _, X}]} -> X
end.