1234567891011121314151617181920212223 |
- -module(trap_exit_protocol).
- -behaviour(ranch_protocol).
- -export([start_link/3]).
- -export([init/3]).
- start_link(Ref, Transport, Opts) ->
- Pid = spawn_link(?MODULE, init, [Ref, Transport, Opts]),
- {ok, Pid}.
- init(Ref, Transport, _Opts = []) ->
- process_flag(trap_exit, true),
- {ok, Socket} = ranch:handshake(Ref),
- loop(Socket, Transport).
- loop(Socket, Transport) ->
- case Transport:recv(Socket, 0, infinity) of
- {ok, Data} ->
- Transport:send(Socket, Data),
- loop(Socket, Transport);
- _ ->
- ok = Transport:close(Socket)
- end.
|