diff options
Diffstat (limited to 'src/gun_raw.erl')
-rw-r--r-- | src/gun_raw.erl | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/src/gun_raw.erl b/src/gun_raw.erl index bef2c1c..369353e 100644 --- a/src/gun_raw.erl +++ b/src/gun_raw.erl @@ -26,6 +26,7 @@ %% @todo down -record(raw_state, { + ref :: reference() | undefined, reply_to :: pid(), socket :: inet:socket() | ssl:sslsocket(), transport :: module() @@ -39,12 +40,13 @@ name() -> raw. opts_name() -> raw_opts. has_keepalive() -> false. -init(ReplyTo, Socket, Transport, _Opts) -> - {connected_data_only, #raw_state{reply_to=ReplyTo, socket=Socket, transport=Transport}}. +init(ReplyTo, Socket, Transport, Opts) -> + StreamRef = maps:get(stream_ref, Opts, undefined), + {connected_data_only, #raw_state{ref=StreamRef, reply_to=ReplyTo, socket=Socket, transport=Transport}}. -handle(Data, State=#raw_state{reply_to=ReplyTo}, _, EvHandlerState) -> +handle(Data, State=#raw_state{ref=StreamRef, reply_to=ReplyTo}, _, EvHandlerState) -> %% When we take over the entire connection there is no stream reference. - ReplyTo ! {gun_data, self(), undefined, nofin, Data}, + ReplyTo ! {gun_data, self(), StreamRef, nofin, Data}, {{state, State}, EvHandlerState}. %% We can always close immediately. |