wip

1 files changed, 42 insertions, 0 deletions

diff --git a/learn-you-some-erlang/my_server.erl b/learn-you-some-erlang/my_server.erl
new file mode 100644
index 0000000..ed83a7a
--- /dev/null
+++ b/learn-you-some-erlang/my_server.erl
@@ -0,0 +1,42 @@
+-module(my_server).
+-export([start/2, start_link/2, call/2, cast/2, reply/2]).
+
+%%% Public API
+start(Module, InitialState) ->
+    spawn(fun() -> init(Module, InitialState) end).
+
+start_link(Module, InitialState) ->
+    spawn_link(fun() -> init(Module, InitialState) end).
+
+call(Pid, Msg) ->
+    Ref = erlang:monitor(process, Pid),
+    Pid ! {sync, self(), Ref, Msg},
+    receive
+        {Ref, Reply} ->
+            erlang:demonitor(Ref, [flush]),
+            Reply;
+        {'DOWN', Ref, process, Pid, Reason} ->
+            erlang:error(Reason)
+    after 5000 ->
+        erlang:error(timeout)
+    end.
+
+cast(Pid, Msg) ->
+    Pid ! {async, Msg},
+    ok.
+
+reply({Pid, Ref}, Reply) ->
+    Pid ! {Ref, Reply}.
+
+%%% Private stuff
+init(Module, InitialState) ->
+    loop(Module, Module:init(InitialState)).
+
+loop(Module, State) ->
+    receive
+        {async, Msg} ->
+             loop(Module, Module:handle_cast(Msg, State));
+        {sync, Pid, Ref, Msg} ->
+             loop(Module, Module:handle_call(Msg, {Pid, Ref}, State))
+    end.
+