From 34307a584149abbf0b2e5b33beb2fca4c585b0d1 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Lo=C3=AFc=20Hoguin?= Date: Sun, 3 Jun 2018 20:31:09 +0200 Subject: Change messages to gun_upgrade and gun_ws with stream reference --- src/gun.erl | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) (limited to 'src/gun.erl') diff --git a/src/gun.erl b/src/gun.erl index cad1da1..1224c82 100644 --- a/src/gun.erl +++ b/src/gun.erl @@ -381,6 +381,7 @@ await(ServerPid, StreamRef, Timeout) -> demonitor(MRef, [flush]), Res. +%% @todo Add gun_upgrade and gun_ws? await(ServerPid, StreamRef, Timeout, MRef) -> receive {gun_inform, ServerPid, StreamRef, Status, Headers} -> @@ -494,9 +495,9 @@ flush_pid(ServerPid) -> flush_pid(ServerPid); {gun_error, ServerPid, _} -> flush_pid(ServerPid); - {gun_ws_upgrade, ServerPid, _, _} -> + {gun_upgrade, ServerPid, _, _, _} -> flush_pid(ServerPid); - {gun_ws, ServerPid, _} -> + {gun_ws, ServerPid, _, _} -> flush_pid(ServerPid); {'DOWN', _, process, ServerPid, _} -> flush_pid(ServerPid) @@ -517,6 +518,10 @@ flush_ref(StreamRef) -> {gun_push, _, StreamRef, _, _, _, _, _} -> flush_ref(StreamRef); {gun_error, _, StreamRef, _} -> + flush_ref(StreamRef); + {gun_upgrade, _, StreamRef, _, _} -> + flush_ref(StreamRef); + {gun_ws, _, StreamRef, _} -> flush_ref(StreamRef) after 0 -> ok -- cgit v1.2.3