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

Dedicated Logger class

Browse source
This commit is contained in:
cxxpxr 2021-04-06 18:53:27 -04:00
parent e9cc15011b
commit 41166cb82d
3 changed files with 53 additions and 28 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -1,4 +1,5 @@
using Grapevine;
using LightReflectiveMirror.Debug;
using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Logging;
@ -44,15 +45,12 @@ namespace LightReflectiveMirror.LoadBalancing
string publicIP = req.Headers["PIP"];
string address = context.Request.RemoteEndPoint.Address.ToString();
if(Program.showDebugLogs)
Console.WriteLine("Received auth req [" + receivedAuthKey + "] == [" + Program.conf.AuthKey + "]");
Logger.WriteLogMessage("Received auth req [" + receivedAuthKey + "] == [" + Program.conf.AuthKey + "]");
// if server is authenticated
if (receivedAuthKey != null && address != null && endpointPort != null && gamePort != null && receivedAuthKey == Program.conf.AuthKey)
{
if(Program.showDebugLogs)
Console.WriteLine($"Server accepted: {address}:{gamePort}");
Logger.WriteLogMessage($"Server accepted: {address}:{gamePort}");
try
{

37
LoadBalancerProject-DONT-IMPORT-INTO-UNITY/LRM_LoadBalancer/Logger.cs Normal file
View file

@ -0,0 +1,37 @@
using System;
namespace LightReflectiveMirror.Debug
{
public static class Logger
{
private static LogConfiguration conf;
public static void ConfigureLogger(LogConfiguration config) => conf = config;
public static void WriteLogMessage(string message, ConsoleColor color = ConsoleColor.White, bool oneLine = false)
{
if(!conf.sendLogs) { return; }
Console.ForegroundColor = color;
if (oneLine)
Console.Write(message);
else
Console.WriteLine(message);
}
public static void ForceLogMessage(string message, ConsoleColor color = ConsoleColor.White, bool oneLine = false)
{
Console.ForegroundColor = color;
if (oneLine)
Console.Write(message);
else
Console.WriteLine(message);
}
public struct LogConfiguration
{
public bool sendLogs;
}
}
}

36
LoadBalancerProject-DONT-IMPORT-INTO-UNITY/LRM_LoadBalancer/Program.cs
View file

@ -1,3 +1,4 @@
using LightReflectiveMirror.Debug;
using Newtonsoft.Json;
using System;
using System.Collections.Generic;
@ -30,26 +31,29 @@ namespace LightReflectiveMirror.LoadBalancing
public async Task MainAsync()
{
WriteTitle();
instance = this;
startupTime = DateTime.Now;
if (!File.Exists(CONFIG_PATH))
{
File.WriteAllText(CONFIG_PATH, JsonConvert.SerializeObject(new Config(), Formatting.Indented));
WriteLogMessage("A config.json file was generated. Please configure it to the proper settings and re-run!", ConsoleColor.Yellow);
Logger.ForceLogMessage("A config.json file was generated. Please configure it to the proper settings and re-run!", ConsoleColor.Yellow);
Console.ReadKey();
Environment.Exit(0);
}
else
{
conf = JsonConvert.DeserializeObject<Config>(File.ReadAllText(CONFIG_PATH));
Logger.ConfigureLogger(new Logger.LogConfiguration { sendLogs = conf.ShowDebugLogs });
_pingDelay = conf.ConnectedServerPingRate;
showDebugLogs = conf.ShowDebugLogs;
if (new EndpointServer().Start(conf.EndpointPort))
WriteLogMessage("Endpoint server started successfully", ConsoleColor.Green);
Logger.ForceLogMessage("Endpoint server started successfully", ConsoleColor.Green);
else
WriteLogMessage("Endpoint server started unsuccessfully", ConsoleColor.Red);
Logger.ForceLogMessage("Endpoint server started unsuccessfully", ConsoleColor.Red);
}
var pingThread = new Thread(new ThreadStart(PingServers));
@ -66,7 +70,7 @@ namespace LightReflectiveMirror.LoadBalancing
if (availableRelayServers.ContainsKey(relayAddr))
{
WriteLogMessage($"LRM Node {serverIP}:{port} tried to register while already registered!");
Logger.ForceLogMessage($"LRM Node {serverIP}:{port} tried to register while already registered!");
return;
}
@ -115,8 +119,7 @@ namespace LightReflectiveMirror.LoadBalancing
{
while (true)
{
if (showDebugLogs)
WriteLogMessage("Pinging " + availableRelayServers.Count + " available relays");
Logger.WriteLogMessage("Pinging " + availableRelayServers.Count + " available relays");
// Create a new list so we can modify the collection in our loop.
var keys = new List<RelayAddress>(availableRelayServers.Keys);
@ -132,8 +135,7 @@ namespace LightReflectiveMirror.LoadBalancing
var serverStats = wc.DownloadString(url);
var deserializedData = JsonConvert.DeserializeObject<RelayServerInfo>(serverStats);
if (showDebugLogs)
WriteLogMessage("Server " + keys[i].Address + " still exists, keeping in collection.");
Logger.ForceLogMessage("Server " + keys[i].Address + " still exists, keeping in collection.");
if (availableRelayServers.ContainsKey(keys[i]))
availableRelayServers[keys[i]] = deserializedData;
@ -143,9 +145,7 @@ namespace LightReflectiveMirror.LoadBalancing
catch (Exception ex)
{
// server doesnt exist anymore probably
// do more shit here
if (showDebugLogs)
WriteLogMessage("Server " + keys[i] + " does not exist anymore, removing", ConsoleColor.Red);
Logger.WriteLogMessage("Server " + keys[i] + " does not exist anymore, removing", ConsoleColor.Red);
availableRelayServers.Remove(keys[i]);
}
}
@ -182,19 +182,9 @@ namespace LightReflectiveMirror.LoadBalancing
"\nHarambe Memorial Initializing... OK" +
"\nBananas Initializing... OK\n";
WriteLogMessage(t, ConsoleColor.Green);
WriteLogMessage(load, ConsoleColor.Cyan);
Logger.ForceLogMessage(t, ConsoleColor.Green);
Logger.ForceLogMessage(load, ConsoleColor.Cyan);
}
static void WriteLogMessage(string message, ConsoleColor color = ConsoleColor.White, bool oneLine = false)
{
Console.ForegroundColor = color;
if (oneLine)
Console.Write(message);
else
Console.WriteLine(message);
}
}
}