mirror of
https://github.com/ninenines/cowboy.git
synced 2025-07-14 20:30:23 +00:00
Add and document the new multipart code
The old undocumented API is removed entirely. While a documentation exists for the new API, it will not be considered set in stone until further testing has been performed, and a file upload example has been added. The new API should be a little more efficient than the old API, especially with smaller messages.
This commit is contained in:
parent
1f5342f3b8
commit
917cf99e10
10 changed files with 299 additions and 387 deletions
|
@ -1,4 +1,4 @@
|
|||
%% Copyright (c) 2011-2013, Loïc Hoguin <essen@ninenines.eu>
|
||||
%% Copyright (c) 2011-2014, Loïc Hoguin <essen@ninenines.eu>
|
||||
%% Copyright (c) 2011, Anthony Ramine <nox@dev-extend.eu>
|
||||
%%
|
||||
%% Permission to use, copy, modify, and/or distribute this software for any
|
||||
|
@ -49,6 +49,7 @@
|
|||
-export([keepalive_nl/1]).
|
||||
-export([keepalive_stream_loop/1]).
|
||||
-export([multipart/1]).
|
||||
-export([multipart_large/1]).
|
||||
-export([nc_rand/1]).
|
||||
-export([nc_zero/1]).
|
||||
-export([onrequest/1]).
|
||||
|
@ -135,6 +136,7 @@ groups() ->
|
|||
keepalive_nl,
|
||||
keepalive_stream_loop,
|
||||
multipart,
|
||||
multipart_large,
|
||||
nc_rand,
|
||||
nc_zero,
|
||||
pipeline,
|
||||
|
@ -391,6 +393,7 @@ init_dispatch(Config) ->
|
|||
{"/static_specify_file/[...]", cowboy_static,
|
||||
{file, ?config(static_dir, Config) ++ "/style.css"}},
|
||||
{"/multipart", http_multipart, []},
|
||||
{"/multipart/large", http_multipart_stream, []},
|
||||
{"/echo/body", http_echo_body, []},
|
||||
{"/echo/body_qs", http_body_qs, []},
|
||||
{"/param_all", rest_param_all, []},
|
||||
|
@ -755,8 +758,8 @@ multipart(Config) ->
|
|||
"This is a preamble."
|
||||
"\r\n--OHai\r\nX-Name:answer\r\n\r\n42"
|
||||
"\r\n--OHai\r\nServer:Cowboy\r\n\r\nIt rocks!\r\n"
|
||||
"\r\n--OHai--"
|
||||
"This is an epiloque."
|
||||
"\r\n--OHai--\r\n"
|
||||
"This is an epilogue."
|
||||
>>,
|
||||
{ok, Client2} = cowboy_client:request(<<"POST">>,
|
||||
build_url("/multipart", Config),
|
||||
|
@ -770,6 +773,21 @@ multipart(Config) ->
|
|||
{[{<<"server">>, <<"Cowboy">>}], <<"It rocks!\r\n">>}
|
||||
].
|
||||
|
||||
multipart_large(Config) ->
|
||||
Client = ?config(client, Config),
|
||||
Boundary = "----------",
|
||||
Big = << 0:9000000/unit:8 >>,
|
||||
Bigger = << 0:9999999/unit:8 >>,
|
||||
Body = ["--", Boundary, "\r\ncontent-length: 9000000\r\n\r\n", Big, "\r\n",
|
||||
"--", Boundary, "\r\ncontent-length: 9999999\r\n\r\n", Bigger, "\r\n",
|
||||
"--", Boundary, "--\r\n"],
|
||||
{ok, Client2} = cowboy_client:request(<<"POST">>,
|
||||
build_url("/multipart/large", Config),
|
||||
[{<<"content-type">>, ["multipart/x-large; boundary=", Boundary]}],
|
||||
Body, Client),
|
||||
{ok, 200, _, _} = cowboy_client:response(Client2),
|
||||
ok.
|
||||
|
||||
nc_reqs(Config, Input) ->
|
||||
Cat = os:find_executable("cat"),
|
||||
Nc = os:find_executable("nc"),
|
||||
|
|
|
@ -8,22 +8,18 @@ init({_Transport, http}, Req, []) ->
|
|||
{ok, Req, {}}.
|
||||
|
||||
handle(Req, State) ->
|
||||
{Result, Req2} = acc_multipart(Req),
|
||||
{Result, Req2} = acc_multipart(Req, []),
|
||||
{ok, Req3} = cowboy_req:reply(200, [], term_to_binary(Result), Req2),
|
||||
{ok, Req3, State}.
|
||||
|
||||
terminate(_, _, _) ->
|
||||
ok.
|
||||
|
||||
acc_multipart(Req) ->
|
||||
acc_multipart(cowboy_req:multipart_data(Req), []).
|
||||
|
||||
acc_multipart({headers, Headers, Req}, Acc) ->
|
||||
acc_multipart(cowboy_req:multipart_data(Req), [{Headers, []}|Acc]);
|
||||
acc_multipart({body, Data, Req}, [{Headers, BodyAcc}|Acc]) ->
|
||||
acc_multipart(cowboy_req:multipart_data(Req), [{Headers, [Data|BodyAcc]}|Acc]);
|
||||
acc_multipart({end_of_part, Req}, [{Headers, BodyAcc}|Acc]) ->
|
||||
acc_multipart(cowboy_req:multipart_data(Req),
|
||||
[{Headers, list_to_binary(lists:reverse(BodyAcc))}|Acc]);
|
||||
acc_multipart({eof, Req}, Acc) ->
|
||||
{lists:reverse(Acc), Req}.
|
||||
acc_multipart(Req, Acc) ->
|
||||
case cowboy_req:part(Req) of
|
||||
{ok, Headers, Req2} ->
|
||||
{ok, Body, Req3} = cowboy_req:part_body(Req2),
|
||||
acc_multipart(Req3, [{Headers, Body}|Acc]);
|
||||
{done, Req2} ->
|
||||
{lists:reverse(Acc), Req2}
|
||||
end.
|
||||
|
|
34
test/http_SUITE_data/http_multipart_stream.erl
Normal file
34
test/http_SUITE_data/http_multipart_stream.erl
Normal file
|
@ -0,0 +1,34 @@
|
|||
%% Feel free to use, reuse and abuse the code in this file.
|
||||
|
||||
-module(http_multipart_stream).
|
||||
-behaviour(cowboy_http_handler).
|
||||
-export([init/3, handle/2, terminate/3]).
|
||||
|
||||
init(_, Req, []) ->
|
||||
{ok, Req, undefined}.
|
||||
|
||||
handle(Req, State) ->
|
||||
Req2 = multipart(Req),
|
||||
{ok, Req3} = cowboy_req:reply(200, Req2),
|
||||
{ok, Req3, State}.
|
||||
|
||||
terminate(_, _, _) ->
|
||||
ok.
|
||||
|
||||
multipart(Req) ->
|
||||
case cowboy_req:part(Req) of
|
||||
{ok, [{<<"content-length">>, BinLength}], Req2} ->
|
||||
Length = list_to_integer(binary_to_list(BinLength)),
|
||||
{Length, Req3} = stream_body(Req2, 0),
|
||||
multipart(Req3);
|
||||
{done, Req2} ->
|
||||
Req2
|
||||
end.
|
||||
|
||||
stream_body(Req, N) ->
|
||||
case cowboy_req:part_body(Req) of
|
||||
{ok, Data, Req2} ->
|
||||
{N + byte_size(Data), Req2};
|
||||
{more, Data, Req2} ->
|
||||
stream_body(Req2, N + byte_size(Data))
|
||||
end.
|
Loading…
Add table
Add a link
Reference in a new issue