83 lines
2.5 KiB
C#
83 lines
2.5 KiB
C#
using NadekoBot.Services.Impl;
|
|
using NLog;
|
|
using System;
|
|
using System.Diagnostics;
|
|
using System.Linq;
|
|
using System.Threading.Tasks;
|
|
using NadekoBot.Common.ShardCom;
|
|
|
|
namespace NadekoBot.Services
|
|
{
|
|
public class ShardsCoordinator
|
|
{
|
|
private readonly BotCredentials _creds;
|
|
private readonly Process[] _shardProcesses;
|
|
public ShardComMessage[] Statuses { get; }
|
|
public int GuildCount => Statuses.ToArray()
|
|
.Where(x => x != null)
|
|
.Sum(x => x.Guilds);
|
|
|
|
private readonly Logger _log;
|
|
private readonly ShardComServer _comServer;
|
|
private readonly int _port;
|
|
private readonly int _curProcessId;
|
|
|
|
public ShardsCoordinator(int port)
|
|
{
|
|
LogSetup.SetupLogger();
|
|
_creds = new BotCredentials();
|
|
_shardProcesses = new Process[_creds.TotalShards];
|
|
Statuses = new ShardComMessage[_creds.TotalShards];
|
|
_log = LogManager.GetCurrentClassLogger();
|
|
_port = port;
|
|
|
|
_comServer = new ShardComServer(port);
|
|
_comServer.Start();
|
|
|
|
_comServer.OnDataReceived += _comServer_OnDataReceived;
|
|
|
|
_curProcessId = Process.GetCurrentProcess().Id;
|
|
}
|
|
|
|
private Task _comServer_OnDataReceived(ShardComMessage msg)
|
|
{
|
|
Statuses[msg.ShardId] = msg;
|
|
if (msg.ConnectionState == Discord.ConnectionState.Disconnected || msg.ConnectionState == Discord.ConnectionState.Disconnecting)
|
|
_log.Error("!!! SHARD {0} IS IN {1} STATE", msg.ShardId, msg.ConnectionState.ToString());
|
|
return Task.CompletedTask;
|
|
}
|
|
|
|
public async Task RunAsync()
|
|
{
|
|
for (int i = 1; i < _creds.TotalShards; i++)
|
|
{
|
|
var p = Process.Start(new ProcessStartInfo()
|
|
{
|
|
FileName = _creds.ShardRunCommand,
|
|
Arguments = string.Format(_creds.ShardRunArguments, i, _curProcessId, _port)
|
|
});
|
|
await Task.Delay(5000);
|
|
}
|
|
}
|
|
|
|
public async Task RunAndBlockAsync()
|
|
{
|
|
try
|
|
{
|
|
await RunAsync().ConfigureAwait(false);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_log.Error(ex);
|
|
}
|
|
|
|
await Task.Delay(-1);
|
|
foreach (var p in _shardProcesses)
|
|
{
|
|
try { p.Kill(); } catch { }
|
|
try { p.Dispose(); } catch { }
|
|
}
|
|
}
|
|
}
|
|
}
|