From cee8c6625bd7a09db88daf5e9767cced66fb5c15 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Lo=C3=AFc=20Hoguin?= Date: Fri, 26 Apr 2019 20:30:51 +0200 Subject: Add upgrade/ws tuples to gun:await/2,3,4 --- src/gun.erl | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) (limited to 'src/gun.erl') diff --git a/src/gun.erl b/src/gun.erl index b1f493b..7376a33 100644 --- a/src/gun.erl +++ b/src/gun.erl @@ -494,6 +494,8 @@ connect(ServerPid, Destination, Headers, ReqOpts) -> | {data, fin | nofin, binary()} | {trailers, resp_headers()} | {push, reference(), binary(), binary(), resp_headers()} + | {upgrade, [binary()], resp_headers()} + | {ws, ws_frame()} %% @todo Excluding ping/pong, for now. | {error, {stream_error | connection_error | down, any()} | timeout}. -spec await(pid(), reference()) -> await_result(). @@ -512,7 +514,6 @@ await(ServerPid, StreamRef, Timeout) -> demonitor(MRef, [flush]), Res. -%% @todo Add gun_upgrade and gun_ws? -spec await(pid(), reference(), timeout(), reference()) -> await_result(). await(ServerPid, StreamRef, Timeout, MRef) -> receive @@ -526,6 +527,10 @@ await(ServerPid, StreamRef, Timeout, MRef) -> {trailers, Trailers}; {gun_push, ServerPid, StreamRef, NewStreamRef, Method, URI, Headers} -> {push, NewStreamRef, Method, URI, Headers}; + {gun_upgrade, ServerPid, StreamRef, Protocols, Headers} -> + {upgrade, Protocols, Headers}; + {gun_ws, ServerPid, StreamRef, Frame} -> + {ws, Frame}; {gun_error, ServerPid, StreamRef, Reason} -> {error, {stream_error, Reason}}; {gun_error, ServerPid, Reason} -> -- cgit v1.2.3