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
37
38
39
40
41
42
43
44
45
|
%% Feel free to use, reuse and abuse the code in this file.
-module(http_echo_body).
-behaviour(cowboy_http_handler).
-export([init/3, handle/2, terminate/3]).
init({_, http}, Req, _) ->
{ok, Req, undefined}.
handle(Req, State) ->
true = cowboy_req:has_body(Req),
{ok, Req3} = case cowboy_req:body(1000000, Req) of
{error, chunked} -> handle_chunked(Req);
{error, badlength} -> handle_badlength(Req);
{ok, Body, Req2} -> handle_body(Req2, Body)
end,
{ok, Req3, State}.
handle_chunked(Req) ->
{ok, Data, Req2} = read_body(Req, <<>>, 1000000),
{ok, Req3} = cowboy_req:reply(200, [], Data, Req2),
{ok, Req3}.
handle_badlength(Req) ->
{ok, Req2} = cowboy_req:reply(413, [], <<"Request entity too large">>, Req),
{ok, Req2}.
handle_body(Req, Body) ->
{Size, Req2} = cowboy_req:body_length(Req),
Size = byte_size(Body),
{ok, Req3} = cowboy_req:reply(200, [], Body, Req2),
{ok, Req3}.
terminate(_, _, _) ->
ok.
% Read chunked request content
read_body(Req, Acc, BodyLengthRemaining) ->
case cowboy_req:stream_body(Req) of
{ok, Data, Req2} ->
BodyLengthRem = BodyLengthRemaining - byte_size(Data),
read_body(Req2, << Acc/binary, Data/binary >>, BodyLengthRem);
{done, Req2} ->
{ok, Acc, Req2}
end.
|
