mattburchett/NadekoBot
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fixes for global nadeko

Browse Source
This commit is contained in:
Master Kwoth 2017-10-29 04:46:42 +01:00
parent 8ceb6f5c46
commit d91b7632f6
2 changed files with 22 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
NadekoBot.Core/Common/ShardCom/ShardComMessage.cs
View File

@ -5,9 +5,19 @@ namespace NadekoBot.Common.ShardCom
{
public class ShardComMessage
{
public int ShardId { get; set; }
public ConnectionState ConnectionState { get; set; }
public int Guilds { get; set; }
public DateTime Time { get; set; }
public ShardComMessage Clone() =>
new ShardComMessage
{
ShardId = ShardId,
ConnectionState = ConnectionState,
Guilds = Guilds,
Time = Time,
};
}
}

18
NadekoBot.Core/Services/ShardsCoordinator.cs
View File

@ -96,13 +96,18 @@ namespace NadekoBot.Core.Services
#else
_shardStartQueue.Enqueue(i);
#endif
//set the shard's initial state in redis cache
_defaultShardState.ShardId = i;
//set the shard's initial state in redis cache
var msg = _defaultShardState.Clone();
msg.ShardId = i;
//this is to avoid the shard coordinator thinking that
//the shard is unresponsive while startup up
_defaultShardState.Time = DateTime.UtcNow + TimeSpan.FromSeconds(45 * (i + 1));
var delay = 45;
#if GLOBAL_NADEKO
delay = 180;
#endif
msg.Time = DateTime.UtcNow + TimeSpan.FromSeconds(delay * (i + 1));
db.ListRightPush(_key + "_shardstats",
JsonConvert.SerializeObject(_defaultShardState),
JsonConvert.SerializeObject(msg),
flags: CommandFlags.FireAndForget);
}
@ -141,10 +146,11 @@ namespace NadekoBot.Core.Services
private void OnStop(int shardId)
{
var db = _redis.GetDatabase();
_defaultShardState.ShardId = shardId;
var msg = _defaultShardState.Clone();
msg.ShardId = shardId;
db.ListSetByIndex(_key + "_shardstats",
shardId,
JsonConvert.SerializeObject(_defaultShardState),
JsonConvert.SerializeObject(msg),
CommandFlags.FireAndForget);
var p = _shardProcesses[shardId];
_shardProcesses[shardId] = null;