OpenCodeBox/fishbait
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Added regions to joining load balancer

Browse source
This commit is contained in:
Derek S 2021-04-09 02:45:08 -05:00
parent 8b6919abd3
commit 3b17246764
1 changed files with 17 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
LoadBalancerProject-DONT-IMPORT-INTO-UNITY/LRM_LoadBalancer/Endpoint/Endpoint.cs
View file

@ -107,6 +107,11 @@ namespace LightReflectiveMirror.LoadBalancing
// collection being modified while iterating.
var servers = Program.instance.availableRelayServers.ToList();
var low = lowest;
var lowestTotal = lowest;
var region = context.Request.Headers["x-Region"];
// If we cant parse it (due to not being included or not being valid) then just join any region.
int regionID = int.TryParse(region, out regionID) ? regionID : 0;
if (servers.Count == 0)
{
@ -114,14 +119,25 @@ namespace LightReflectiveMirror.LoadBalancing
return;
}
bool foundServerInRegion = false;
for (int i = 0; i < servers.Count; i++)
{
if (servers[i].Value.connectedClients < low.Value.connectedClients)
{
if ((int)servers[i].Key.serverRegion == regionID)
{
low = servers[i];
foundServerInRegion = true;
}
lowestTotal = servers[i];
}
}
// If the region didnt have a single node, then just give him the lowest node we looped over.
if (!foundServerInRegion)
low = lowestTotal;
// respond with the server ip
// if the string is still dummy then theres no servers
await context.Response.SendResponseAsync(low.Key.address != "Dummy" ? JsonConvert.SerializeObject(low.Key) : HttpStatusCode.InternalServerError);