|
17 | 17 | error/2, |
18 | 18 | ctx/1, |
19 | 19 | ctx/2, |
20 | | - end_stream/1, |
21 | 20 | handle_streams/2, |
22 | 21 | handle_call/2, |
23 | 22 | handle_info/2]). |
24 | 23 |
|
25 | 24 | -export([init/3, |
26 | | - on_receive_request_headers/2, |
| 25 | + on_receive_headers/2, |
27 | 26 | on_send_push_promise/2, |
28 | | - on_receive_request_data/2, |
29 | | - on_request_end_stream/1]). |
30 | | - |
31 | | -%% state getters and setters |
32 | | --export([stream_handler_state/1, |
33 | | - stream_handler_state/2, |
34 | | - stream_req_headers/1 |
35 | | -]). |
| 27 | + on_receive_data/2, |
| 28 | + on_end_stream/1]). |
36 | 29 |
|
37 | 30 | %% state getters and setters |
38 | 31 | -export([stream_handler_state/1, |
@@ -111,7 +104,8 @@ init(ConnPid, StreamId, [Socket, ServicesTable, AuthFun, UnaryInterceptor, |
111 | 104 | stats_handler=StatsHandler}, |
112 | 105 | {ok, State}. |
113 | 106 |
|
114 | | -on_receive_request_headers(Headers, State=#state{ctx=_Ctx}) -> |
| 107 | +on_receive_headers(Headers, State=#state{ctx=_Ctx}) -> |
| 108 | + %% proplists:get_value(<<":method">>, Headers) =:= <<"POST">>, |
115 | 109 | Metadata = grpcbox_utils:headers_to_metadata(Headers), |
116 | 110 | Ctx = case parse_options(<<"grpc-timeout">>, Headers) of |
117 | 111 | infinity -> |
@@ -273,9 +267,9 @@ ctx_with_stream(Ctx, Stream) -> |
273 | 267 | from_ctx(Ctx) -> |
274 | 268 | ctx:get(Ctx, ctx_stream_key). |
275 | 269 |
|
276 | | -on_receive_request_data(_, State=#state{method=undefined}) -> |
| 270 | +on_receive_data(_, State=#state{method=undefined}) -> |
277 | 271 | {ok, State}; |
278 | | -on_receive_request_data(Bin, State=#state{request_encoding=Encoding, |
| 272 | +on_receive_data(Bin, State=#state{request_encoding=Encoding, |
279 | 273 | buffer=Buffer}) -> |
280 | 274 | try |
281 | 275 | {NewBuffer, Messages} = grpcbox_frame:split(<<Buffer/binary, Bin/binary>>, Encoding), |
@@ -362,9 +356,9 @@ on_end_stream_(State = #state{method=#method{input={_Input, true}, |
362 | 356 | on_end_stream_(#state{method=#method{input={_Input, false}, |
363 | 357 | output={_Output, true}}}) -> |
364 | 358 | ok; |
365 | | -on_request_end_stream_(State=#state{method=#method{output={_Output, false}}}) -> |
| 359 | +on_end_stream_(State=#state{method=#method{output={_Output, false}}}) -> |
366 | 360 | end_stream(State); |
367 | | -on_request_end_stream_(State) -> |
| 361 | +on_end_stream_(State) -> |
368 | 362 | end_stream(State). |
369 | 363 |
|
370 | 364 | %% Internal |
|
0 commit comments