forked from krestenkrab/hanoidb
-
Notifications
You must be signed in to change notification settings - Fork 4
/
plain_rpc.erl
68 lines (56 loc) · 1.95 KB
/
plain_rpc.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
%% ----------------------------------------------------------------------------
%%
%% plain_rpc: RPC module to accompany plain_fsm
%%
%% Copyright 2011-2012 (c) Trifork A/S. All Rights Reserved.
%% http://trifork.com/ info@trifork.com
%%
%% This file is provided to you under the Apache License, Version 2.0 (the
%% "License"); you may not use this file except in compliance with the License.
%% You may obtain a copy of the License at
%%
%% http://www.apache.org/licenses/LICENSE-2.0
%%
%% Unless required by applicable law or agreed to in writing, software
%% distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
%% WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
%% License for the specific language governing permissions and limitations
%% under the License.
%%
%% ----------------------------------------------------------------------------
-module(plain_rpc).
-author('Kresten Krab Thorup <krab@trifork.com>').
-export([send_call/2, receive_reply/1, send_reply/2, call/2, call/3, cast/2]).
-include("include/plain_rpc.hrl").
send_call(PID, Request) ->
Ref = erlang:monitor(process, PID),
PID ! ?CALL({self(), Ref}, Request),
Ref.
cast(PID, Msg) ->
PID ! ?CAST(self(), Msg).
receive_reply(MRef) ->
receive
?REPLY(MRef, Reply) ->
erlang:demonitor(MRef, [flush]),
Reply;
{'DOWN', MRef, _, _, Reason} ->
exit(Reason)
end.
send_reply({PID,Ref}, Reply) ->
erlang:send(PID, ?REPLY(Ref, Reply)),
ok.
call(PID,Request) ->
call(PID, Request, infinity).
call(PID,Request,Timeout) ->
MRef = erlang:monitor(process, PID),
PID ! ?CALL({self(), MRef}, Request),
receive
?REPLY(MRef, Reply) ->
erlang:demonitor(MRef, [flush]),
Reply;
{'DOWN', MRef, _, _, Reason} ->
exit(Reason)
after Timeout ->
erlang:demonitor(MRef, [flush]),
exit({rpc_timeout, Request})
end.