echo_protocol.erl 511 B

12345678910111213141516171819202122
  1. -module(echo_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. ok = ranch:accept_ack(Ref),
  10. loop(Socket, Transport).
  11. loop(Socket, Transport) ->
  12. case Transport:recv(Socket, 0, 5000) of
  13. {ok, Data} ->
  14. Transport:send(Socket, Data),
  15. loop(Socket, Transport);
  16. _ ->
  17. ok = Transport:close(Socket)
  18. end.