aboutsummaryrefslogtreecommitdiff
path: root/erlbal.erl
blob: 417eb1ce774392e030b930c21651972ee7ec3eb5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
-module(erlbal).
-export([make_request/0, start_bal/0, kill_bal/0, start_server/1]).

start_server(ID) ->
	PID = spawn(fun() -> serverloop(ID) end),
	balancer ! {add_node, PID}.

serverloop(ID) ->
	receive
		{request, From} ->
			balancer ! {response, From, ID},
			serverloop(ID);
		die ->
			ok
	end.

start_bal() ->
	register(balancer, spawn(fun() -> balloop([], 1) end)).

kill_bal() ->
	balancer ! die.

balloop(Serverlist, Nextserver) ->
	receive
		{add_node, PID} ->
			balloop(Serverlist ++ [PID], Nextserver);
		{request, From} ->
			Serv = lists:nth(Nextserver, Serverlist),
			Serv ! {request, From},
			NS = Nextserver + 1,
			SLL = length(Serverlist),
			if
				NS > SLL ->
					balloop(Serverlist, 1);
				true ->
					balloop(Serverlist, Nextserver+1)
			end;
		{response, To, ID} ->
			To ! ID,
			balloop(Serverlist, Nextserver);
		die ->
			lists:foreach(fun(X) -> X ! die end, Serverlist)
	end.


make_request() ->
	balancer ! {request, self()},
	receive
		ID ->
			ID
	end.