aboutsummaryrefslogtreecommitdiffstats
path: root/test/handlers/switch_handler_h.erl
blob: f1580d121b815c0ea552baf50c026c687a350ec2 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
-module(switch_handler_h).

-export([init/2]).
-export([content_types_provided/2]).
-export([provide/2]).
-export([info/3]).

init(Req, State) ->
	{cowboy_rest, Req, State}.

content_types_provided(Req, State) ->
	{[{<<"text/plain">>, provide}], Req, State}.

provide(Req0, run) ->
	Req = cowboy_req:stream_reply(200, Req0),
	send_after(0),
	{{switch_handler, cowboy_loop}, Req, 0};
provide(Req0, hibernate) ->
	Req = cowboy_req:stream_reply(200, Req0),
	send_after(0),
	{{switch_handler, cowboy_loop, hibernate}, Req, 0}.

send_after(N) ->
	erlang:send_after(100, self(), {stream, msg(N)}).

msg(0) -> <<"Hello\n">>;
msg(1) -> <<"streamed\n">>;
msg(2) -> <<"world!\n">>;
msg(3) -> stop.

info({stream, stop}, Req, State) ->
	{stop, Req, State};
info({stream, What}, Req, State) ->
	cowboy_req:stream_body(What, nofin, Req),
	send_after(State + 1),
	{ok, Req, State + 1}.