mirror of
https://github.com/ninenines/cowboy.git
synced 2025-07-14 20:30:23 +00:00
647e95aed1
This should have been done a *long* time ago, back when I initially
added Websocket support. This is the first part of two in improving
loop handler support with regards to socket closure.
Reason may include: {normal, shutdown} for the most normal shutdown,
{normal, timeout} for a loop handler timeout shutdown, or {error, _}
if an error occured.
28 lines
891 B
Erlang
28 lines
891 B
Erlang
%% Feel free to use, reuse and abuse the code in this file.
|
|
|
|
-module(http_handler_stream_body).
|
|
-behaviour(cowboy_http_handler).
|
|
-export([init/3, handle/2, terminate/3]).
|
|
|
|
-record(state, {headers, body, reply}).
|
|
|
|
init({_Transport, http}, Req, Opts) ->
|
|
Headers = proplists:get_value(headers, Opts, []),
|
|
Body = proplists:get_value(body, Opts, "http_handler_stream_body"),
|
|
Reply = proplists:get_value(reply, Opts),
|
|
{ok, Req, #state{headers=Headers, body=Body, reply=Reply}}.
|
|
|
|
handle(Req, State=#state{headers=_Headers, body=Body, reply=Reply}) ->
|
|
SFun = fun(Socket, Transport) -> Transport:send(Socket, Body) end,
|
|
Req2 = case Reply of
|
|
set_resp ->
|
|
SLen = iolist_size(Body),
|
|
cowboy_req:set_resp_body_fun(SLen, SFun, Req);
|
|
set_resp_close ->
|
|
cowboy_req:set_resp_body_fun(SFun, Req)
|
|
end,
|
|
{ok, Req3} = cowboy_req:reply(200, Req2),
|
|
{ok, Req3, State}.
|
|
|
|
terminate(_, _, _) ->
|
|
ok.
|