From f56479ffc22ca421d50ab951db0b91c34d6e3acd Mon Sep 17 00:00:00 2001 From: Magnus Klaar Date: Wed, 28 Dec 2011 18:01:21 +0100 Subject: Add cowboy_http_req:set_resp_body_fun/3 to rest --- src/cowboy_http_rest.erl | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) (limited to 'src/cowboy_http_rest.erl') diff --git a/src/cowboy_http_rest.erl b/src/cowboy_http_rest.erl index e825a98..a5333fe 100644 --- a/src/cowboy_http_rest.erl +++ b/src/cowboy_http_rest.erl @@ -748,7 +748,12 @@ set_resp_body(Req=#http_req{method=Method}, case call(Req5, State4, Fun) of {Body, Req6, HandlerState} -> State5 = State4#state{handler_state=HandlerState}, - {ok, Req7} = cowboy_http_req:set_resp_body(Body, Req6), + {ok, Req7} = case Body of + {stream, Len, Fun1} -> + cowboy_http_req:set_resp_body_fun(Len, Fun1, Req6); + _Contents -> + cowboy_http_req:set_resp_body(Body, Req6) + end, multiple_choices(Req7, State5) end; set_resp_body(Req, State) -> -- cgit v1.2.3