Merge pull request #1 from lrascao/fix/remove-async-ref

Remove async ref when exit signaled
rebar · Mar 12, 2016 · ffef7d0 · ffef7d0
2 parents 3409aa7 + 3b7fd49
commit ffef7d0
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/retest_sh.erl b/src/retest_sh.erl
@@ -83,7 +83,7 @@ run(Cmd, Opts) ->
 stop(Ref) ->
     #sh { pid = Pid, port = Port } = erlang:get(Ref),
     _ = os:cmd(?FMT("kill ~s", [Pid])),
-    exit_loop(Port).
+    exit_loop(Ref, Port).
 
 stop_all() ->
     _ = [ {ok, _} = stop(Ref) || {Ref, Sh} <- erlang:get(),
@@ -115,13 +115,14 @@ read_pid(Port) ->
             {error, {stopped, Rc}}
     end.
 
-exit_loop(Port) ->
+exit_loop(Ref, Port) ->
     receive
         {Port, {data, {_, Line}}} ->
             ?DEBUG("~p: ~s\n", [erlang:get(retest_module), Line]),
-            exit_loop(Port);
+            exit_loop(Ref, Port);
 
         {Port, {exit_status, Rc}} ->
+            erlang:erase(Ref),
             {ok, Rc}
     end.