mirror of
https://github.com/ninenines/cowboy.git
synced 2025-07-14 12:20:24 +00:00
81de580aee
This can be used to limit the maximum frame size before some authentication or other validation is completed.
31 lines
1.1 KiB
Erlang
31 lines
1.1 KiB
Erlang
%% This module sets options based on the frame received.
|
|
|
|
-module(ws_set_options_commands_h).
|
|
-behavior(cowboy_websocket).
|
|
|
|
-export([init/2]).
|
|
-export([websocket_handle/2]).
|
|
-export([websocket_info/2]).
|
|
|
|
init(Req, RunOrHibernate) ->
|
|
{cowboy_websocket, Req, RunOrHibernate,
|
|
#{idle_timeout => infinity}}.
|
|
|
|
%% Set the idle_timeout option dynamically.
|
|
websocket_handle({text, <<"idle_timeout_short">>}, State=run) ->
|
|
{[{set_options, #{idle_timeout => 500}}], State};
|
|
websocket_handle({text, <<"idle_timeout_short">>}, State=hibernate) ->
|
|
{[{set_options, #{idle_timeout => 500}}], State, hibernate};
|
|
%% Set the max_frame_size option dynamically.
|
|
websocket_handle({text, <<"max_frame_size_small">>}, State=run) ->
|
|
{[{set_options, #{max_frame_size => 1000}}], State};
|
|
websocket_handle({text, <<"max_frame_size_small">>}, State=hibernate) ->
|
|
{[{set_options, #{max_frame_size => 1000}}], State, hibernate};
|
|
%% We just echo binary frames.
|
|
websocket_handle(Frame={binary, _}, State=run) ->
|
|
{[Frame], State};
|
|
websocket_handle(Frame={binary, _}, State=hibernate) ->
|
|
{[Frame], State, hibernate}.
|
|
|
|
websocket_info(_Info, State) ->
|
|
{[], State}.
|