From 29c4508884f4e70dc23f486da64aee491009b124 Mon Sep 17 00:00:00 2001 From: Tristan Sloughter Date: Fri, 1 Feb 2013 10:40:00 -0600 Subject: add guards on set_resp_body_fun for streamlen and streamfun --- src/cowboy_req.erl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src') diff --git a/src/cowboy_req.erl b/src/cowboy_req.erl index 0745192..b3bc375 100644 --- a/src/cowboy_req.erl +++ b/src/cowboy_req.erl @@ -847,7 +847,7 @@ set_resp_body(Body, Req) -> %% Cowboy will add a "Transfer-Encoding: identity" header to the %% response. -spec set_resp_body_fun(resp_body_fun(), Req) -> Req when Req::req(). -set_resp_body_fun(StreamFun, Req) -> +set_resp_body_fun(StreamFun, Req) when is_function(StreamFun) -> Req#http_req{resp_body=StreamFun}. %% @doc Add a body function to the response. @@ -863,7 +863,7 @@ set_resp_body_fun(StreamFun, Req) -> %% fewer bytes than declared the behaviour is undefined. -spec set_resp_body_fun(non_neg_integer(), resp_body_fun(), Req) -> Req when Req::req(). -set_resp_body_fun(StreamLen, StreamFun, Req) -> +set_resp_body_fun(StreamLen, StreamFun, Req) when is_integer(StreamLen), is_function(StreamFun) -> Req#http_req{resp_body={StreamLen, StreamFun}}. %% @doc Return whether the given header has been set for the response. -- cgit v1.2.3