add pid/port monitoring

pouriya · Jun 12, 2017 · 1a0a412 · 1a0a412
1 parent 742e6df
commit 1a0a412
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/src/lb.erl b/src/lb.erl
@@ -885,8 +885,10 @@ reply(Name, Dbg, {Pid, Tag}=From, Msg) ->
 
 
 do_subscribe(#?STATE{objects = []}=State, Object) ->
+    do_monitor(Object),
     {State#?STATE{objects = [Object], object = Object}, ok};
 do_subscribe(#?STATE{objects = Objects}=State, Object) ->
+    do_monitor(Object),
     {State#?STATE{objects = [Object|Objects]}, ok}.
 
 
@@ -895,6 +897,17 @@ do_subscribe(#?STATE{objects = Objects}=State, Object) ->
 
 
 
+do_monitor(Object) when erlang:is_pid(Object) ->
+    erlang:monitor(process, Object);
+do_monitor(Object) when erlang:is_port(Object) ->
+    erlang:monitor(port, Object).
+
+
+
+
+
+
+
 do_unsubscribe(#?STATE{objects = [Object]}=State, Object) ->
     {State#?STATE{objects = [], object = undefined}, ok};
 do_unsubscribe(#?STATE{objects = Objects, object =Object}=State, Object) ->