active_echo_protocol.erl 663 B

1234567891011121314151617181920212223242526
  1. -module(active_echo_protocol).
  2. -behaviour(ranch_protocol).
  3. -export([start_link/4]).
  4. -export([init/4]).
  5. start_link(ListenerPid, Socket, Transport, Opts) ->
  6. Pid = spawn_link(?MODULE, init, [ListenerPid, Socket, Transport, Opts]),
  7. {ok, Pid}.
  8. init(ListenerPid, Socket, Transport, _Opts = []) ->
  9. ok = ranch:accept_ack(ListenerPid),
  10. loop(Socket, Transport).
  11. loop(Socket, Transport) ->
  12. {OK, Closed, Error} = Transport:messages(),
  13. Transport:setopts(Socket, [{active, once}]),
  14. receive
  15. {OK, Socket, Data} ->
  16. Transport:send(Socket, Data),
  17. loop(Socket, Transport);
  18. {Closed, Socket} ->
  19. ok;
  20. {Error, Socket, _} ->
  21. ok = Transport:close(Socket)
  22. end.