daan/geekbot
1
0
Fork 0
Code Issues 2 Pull requests Projects Releases Packages Wiki Activity

Update Migration Script

Browse source
This commit is contained in:
runebaas 2018-08-25 23:26:46 +02:00
parent 8a62f94e01
commit e18a512a9b
No known key found for this signature in database
GPG key ID: 2677AF508D0300D6
2 changed files with 8 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
Geekbot.net/Database/RedisMigration.cs
View file

@ -33,7 +33,7 @@ namespace Geekbot.net.Database
_logger.Information(LogSource.Migration, "Starting migration process"); _logger.Information(LogSource.Migration, "Starting migration process");
var keys = _redis.GetAllKeys().Where(e => e.ToString().EndsWith("Messages")); var keys = _redis.GetAllKeys().Where(e => e.ToString().EndsWith("Messages"));
var guilds = new List<SocketGuild>(); var allGuilds = new List<SocketGuild>();
foreach (var key in keys) foreach (var key in keys)
{ {
@ -41,7 +41,7 @@ namespace Geekbot.net.Database
{ {
var g = _client.GetGuild(ulong.Parse(key.ToString().Split(':').First())); var g = _client.GetGuild(ulong.Parse(key.ToString().Split(':').First()));
Console.WriteLine(g.Name); Console.WriteLine(g.Name);
guilds.Add(g); allGuilds.Add(g);
} }
catch (Exception e) catch (Exception e)
{ {
@ -49,16 +49,12 @@ namespace Geekbot.net.Database
} }
} }
_logger.Information(LogSource.Migration, $"Found {guilds.Count} guilds in redis"); _logger.Information(LogSource.Migration, $"Found {allGuilds.Count} guilds in redis");
var guilds = allGuilds.FindAll(e => e.MemberCount < 10000);
foreach (var guild in guilds) foreach (var guild in guilds)
{ {
if (guild.MemberCount > 10000)
{
_logger.Information(LogSource.Migration, $"Skipping {guild.Name} because server size is to big ({guild.MemberCount})");
break;
}
_logger.Information(LogSource.Migration, $"Start Migration for {guild.Name}"); _logger.Information(LogSource.Migration, $"Start Migration for {guild.Name}");
#region Quotes #region Quotes
/** /**
@ -315,7 +311,7 @@ namespace Geekbot.net.Database
*/ */
try try
{ {
var data = guild.Users; var data = guild.Users.ToList().FindAll(e => !e.IsBot);
foreach (var user in data) foreach (var user in data)
{ {
try try
@ -325,6 +321,7 @@ namespace Geekbot.net.Database
await Task.Delay(100); await Task.Delay(100);
if (user.Username == null) break; if (user.Username == null) break;
} }
var namesSerialized = _redis.Db.HashGet($"User:{user.Id}", "UsedNames").ToString(); var namesSerialized = _redis.Db.HashGet($"User:{user.Id}", "UsedNames").ToString();
var names = namesSerialized != null var names = namesSerialized != null
? Utf8Json.JsonSerializer.Deserialize<string[]>(namesSerialized) ? Utf8Json.JsonSerializer.Deserialize<string[]>(namesSerialized)

3
Geekbot.net/Handlers.cs
View file

@ -183,8 +183,7 @@ namespace Geekbot.net
try try
{ {
var guildSocketData = ((IGuildChannel) channel).Guild; var guildSocketData = ((IGuildChannel) channel).Guild;
var guild = _database.GuildSettings.FirstOrDefault(g => var guild = _database.GuildSettings.FirstOrDefault(g => g.GuildId.Equals(guildSocketData.Id.AsLong()));
g.GuildId.Equals(guildSocketData.Id.AsLong()));
if (guild?.ShowDelete ?? false) if (guild?.ShowDelete ?? false)
{ {
var modChannelSocket = (ISocketMessageChannel) await _client.GetChannelAsync(guild.ModChannel.AsUlong()); var modChannelSocket = (ISocketMessageChannel) await _client.GetChannelAsync(guild.ModChannel.AsUlong());