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

Add Message migration script

Browse source
This commit is contained in:
runebaas 2018-09-05 21:15:13 +02:00
parent a8c91c0372
commit fce6be56e2
No known key found for this signature in database
GPG key ID: 2677AF508D0300D6
5 changed files with 132 additions and 36 deletions
Download patch file Download diff file Expand all files Collapse all files

73
Geekbot.net/Database/MessageMigration.cs Normal file
View file

@ -0,0 +1,73 @@
using System;
using System.Linq;
using System.Threading.Tasks;
using Geekbot.net.Database.Models;
using Geekbot.net.Lib.AlmostRedis;
using Geekbot.net.Lib.Extensions;
using Geekbot.net.Lib.Logger;
namespace Geekbot.net.Database
{
public class MessageMigration
{
private readonly DatabaseContext _database;
private readonly IAlmostRedis _redis;
private readonly IGeekbotLogger _logger;
public MessageMigration(DatabaseContext database, IAlmostRedis redis, IGeekbotLogger logger)
{
_database = database;
_redis = redis;
_logger = logger;
}
public async Task Migrate()
{
_logger.Warning(LogSource.Migration, "Starting message migration");
try
{
var messageKeys = _redis.GetAllKeys().Where(e => e.ToString().EndsWith("Messages"));
foreach (var keyName in messageKeys)
{
try
{
var guildId = ulong.Parse(keyName.ToString().Split(':').FirstOrDefault());
var guildUsers = _redis.Db.HashGetAll(keyName);
foreach (var user in guildUsers)
{
try
{
var userId = ulong.Parse(user.Name);
if (userId != 0)
{
var userMessages = int.Parse(user.Value);
_database.Messages.Add(new MessagesModel
{
UserId = userId.AsLong(),
GuildId = guildId.AsLong(),
MessageCount = userMessages
});
}
}
catch (Exception e)
{
_logger.Error(LogSource.Migration, $"Failed to add record for a user in {guildId}", e);
}
}
await _database.SaveChangesAsync();
}
catch (Exception e)
{
_logger.Error(LogSource.Migration, "Failed to determinate guild", e);
}
}
_logger.Warning(LogSource.Migration, "Successfully finished message migration");
}
catch (Exception e)
{
_logger.Error(LogSource.Migration, "Message migration failed", e);
}
}
}
}