From 6d2ccec261ead5f3f3d62aa4d42735930f44e5c8 Mon Sep 17 00:00:00 2001
From: Nathan Lasseter <nathan.je.lasseter@googlemail.com>
Date: Thu, 27 Jan 2011 21:48:19 +0000
Subject: Added del_node and list_nodes

---
 erlbal.erl | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/erlbal.erl b/erlbal.erl
index f0bca7e..6dfefad 100644
--- a/erlbal.erl
+++ b/erlbal.erl
@@ -26,6 +26,11 @@ bal_loop(Serverlist, Nextserver) ->
 	receive
 		{add_node, PID} ->
 			bal_loop(Serverlist ++ [PID], Nextserver);
+		{del_node, PID} ->
+			bal_loop(Serverlist -- [PID], Nextserver);
+		{list_nodes, From} ->
+			From ! Serverlist,
+			bal_loop(Serverlist, Nextserver);
 		{request, From, ARGS} ->
 			Serv = lists:nth(Nextserver, Serverlist),
 			Serv ! {request, From, ARGS},
@@ -48,6 +53,6 @@ bal_loop(Serverlist, Nextserver) ->
 make_request(Balancer, ARGS) ->
 	Balancer ! {request, self(), ARGS},
 	receive
-		ID ->
-			ID
+		Ret ->
+			Ret
 	end.
-- 
cgit v1.2.1