mirror of
https://github.com/ninenines/cowboy.git
synced 2025-07-14 20:30:23 +00:00
5f122d9fa6
This commit prevents erlang messages from keeping a websocket connection alive. Previously, the timer was canceled upon any activity. Now, the timeout is only canceled when actual data is sent from the client. The handler_loop_timeout/1 function is called from websocket_data/4 instead of handler_before_loop/4. It is also called after every successful reply in handler_call/4.
33 lines
924 B
Erlang
33 lines
924 B
Erlang
%% Feel free to use, reuse and abuse the code in this file.
|
|
|
|
-module(ws_timeout_cancel_handler).
|
|
-behaviour(cowboy_http_handler).
|
|
-behaviour(cowboy_websocket_handler).
|
|
-export([init/3, handle/2, terminate/2]).
|
|
-export([websocket_init/3, websocket_handle/3,
|
|
websocket_info/3, websocket_terminate/3]).
|
|
|
|
init(_Any, _Req, _Opts) ->
|
|
{upgrade, protocol, cowboy_websocket}.
|
|
|
|
handle(_Req, _State) ->
|
|
exit(badarg).
|
|
|
|
terminate(_Req, _State) ->
|
|
exit(badarg).
|
|
|
|
websocket_init(_TransportName, Req, _Opts) ->
|
|
erlang:start_timer(500, self(), should_not_cancel_timer),
|
|
{ok, Req, undefined, 1000}.
|
|
|
|
websocket_handle({text, Data}, Req, State) ->
|
|
{reply, {text, Data}, Req, State};
|
|
websocket_handle({binary, Data}, Req, State) ->
|
|
{reply, {binary, Data}, Req, State}.
|
|
|
|
websocket_info(_Info, Req, State) ->
|
|
erlang:start_timer(500, self(), should_not_cancel_timer),
|
|
{ok, Req, State}.
|
|
|
|
websocket_terminate(_Reason, _Req, _State) ->
|
|
ok.
|