Monitor owner and terminate when it dies

1 files changed, 9 insertions, 2 deletions

diff --git a/src/gun.erl b/src/gun.erl
index 4304299..16397c9 100644
--- a/src/gun.erl
+++ b/src/gun.erl
@@ -99,6 +99,7 @@
 -record(state, {
 	parent :: pid(),
 	owner :: pid(),
+	owner_ref :: reference(),
 	host :: inet:hostname(),
 	port :: inet:port_number(),
 	opts :: opts(),
@@ -487,7 +488,9 @@ init(Parent, Owner, Host, Port, Opts) ->
 		tcp -> ranch_tcp;
 		ssl -> ranch_ssl
 	end,
-	connect(#state{parent=Parent, owner=Owner, host=Host, port=Port, opts=Opts, transport=Transport}, Retry).
+	OwnerRef = monitor(process, Owner),
+	connect(#state{parent=Parent, owner=Owner, owner_ref=OwnerRef,
+		host=Host, port=Port, opts=Opts, transport=Transport}, Retry).
 
 default_transport(443) -> ssl;
 default_transport(_) -> tcp.
@@ -576,7 +579,7 @@ before_loop(State=#state{opts=Opts, protocol=Protocol}) ->
 	KeepaliveRef = erlang:send_after(Keepalive, self(), keepalive),
 	loop(State#state{keepalive_ref=KeepaliveRef}).
 
-loop(State=#state{parent=Parent, owner=Owner, host=Host, port=Port, opts=Opts,
+loop(State=#state{parent=Parent, owner=Owner, owner_ref=OwnerRef, host=Host, port=Port, opts=Opts,
 		socket=Socket, transport=Transport, protocol=Protocol, protocol_state=ProtoState}) ->
 	{OK, Closed, Error} = Transport:messages(),
 	Transport:setopts(Socket, [{active, once}]),
@@ -636,6 +639,10 @@ loop(State=#state{parent=Parent, owner=Owner, host=Host, port=Port, opts=Opts,
 		{shutdown, Owner} ->
 			%% @todo Protocol:shutdown?
 			ok;
+		{'DOWN', OwnerRef, process, Owner, Reason} ->
+			Protocol:close(ProtoState),
+			Transport:close(Socket),
+			error({owner_gone, Reason});
 		{system, From, Request} ->
 			sys:handle_system_msg(Request, From, Parent, ?MODULE, [],
 				{loop, State});