Browse Source

Add the private get_resp_state/1 function used by cowboy_protocol

Loïc Hoguin 12 years ago
parent
commit
e55ec7d43d
2 changed files with 7 additions and 1 deletions
  1. 1 1
      src/cowboy_protocol.erl
  2. 6 0
      src/cowboy_req.erl

+ 1 - 1
src/cowboy_protocol.erl

@@ -233,7 +233,7 @@ onrequest(Req, State=#state{onrequest=undefined}) ->
 	dispatch(Req, State);
 onrequest(Req, State=#state{onrequest=OnRequest}) ->
 	Req2 = OnRequest(Req),
-	case Req2#http_req.resp_state of
+	case cowboy_req:get_resp_state(Req2) of
 		waiting -> dispatch(Req2, State);
 		_ -> next_request(Req2, State, ok)
 	end.

+ 6 - 0
src/cowboy_req.erl

@@ -108,6 +108,7 @@
 -export([add_header/3]).
 -export([set_buffer/2]).
 -export([set_bindings/4]).
+-export([get_resp_state/1]).
 
 %% Misc API.
 -export([compact/1]).
@@ -949,6 +950,11 @@ set_bindings(HostInfo, PathInfo, Bindings, Req) ->
 	Req#http_req{host_info=HostInfo, path_info=PathInfo,
 		bindings=Bindings}.
 
+%% @private
+-spec get_resp_state(req()) -> locked | waiting | chunks | done.
+get_resp_state(#http_req{resp_state=RespState}) ->
+	RespState.
+
 %% Misc API.
 
 %% @doc Compact the request data by removing all non-system information.