trap_exit_protocol.erl 552 B

1234567891011121314151617181920212223
  1. -module(trap_exit_protocol).
  2. -behaviour(ranch_protocol).
  3. -export([start_link/4]).
  4. -export([init/4]).
  5. start_link(Ref, Socket, Transport, Opts) ->
  6. Pid = spawn_link(?MODULE, init, [Ref, Socket, Transport, Opts]),
  7. {ok, Pid}.
  8. init(Ref, Socket, Transport, _Opts = []) ->
  9. process_flag(trap_exit, true),
  10. ok = ranch:accept_ack(Ref),
  11. loop(Socket, Transport).
  12. loop(Socket, Transport) ->
  13. case Transport:recv(Socket, 0, infinity) of
  14. {ok, Data} ->
  15. Transport:send(Socket, Data),
  16. loop(Socket, Transport);
  17. _ ->
  18. ok = Transport:close(Socket)
  19. end.