ngx_dynamic_proxy_pass_module/rule.lua at master · marsqing/ngx_dynamic_proxy_pass_module · GitHub

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
--uid = get_cookie("UID");
upstream = "six@0";
--f=io.open("/Users/hupeng/log", "aw");
function choose_upstream()
	uid = get_ngx_http_variable("cookie_uid");

	--uid = "test";
	--f:write(string.format("%d %d %d\n", uid, 1, 1));
	ups = {get_upstream_list()};
	--for key, value in pairs(ups) do
	--	f:write(string.format("%s %s %d %d\n", key, value, 1, 1));
	--end
	if #ups == 0 then
		return;
	end
	ups_cnt = #ups;
	i = 2;
	bucket_cnt = 0;
	while i <= ups_cnt do
		bucket_cnt = bucket_cnt + ups[i]
		i = i + 2;
	end
	modus = uid % bucket_cnt;
	--modus = 0;
	--f:write(string.format("%d %d\n", modus, uid));
	i = 2;
	j = 1;
	bucket_search = ups[i];
	while j < ups_cnt do
		if modus < bucket_search then
			upstream = ups[j];
			break;
		end
		i = i + 2;
		bucket_search = bucket_search + ups[i];
		j = j + 2;
	end
end

--f:write(string.format("%s\n", upstream));
--f:write(string.format("%d %d %d\n", uid, bucket_cnt, modus));
--i = 1;
--while i <= #ups do
--	f:write(string.format("%s\n", ups[i]));
--	i = i + 1;
--end
--f:write(string.format("arg_name=%s\n", get_ngx_http_variable("arg_name")));
--f:close();
--upstream = "six@0";