Implement migration for TournamentStorage

This commit is contained in:
Shivam 2020-06-08 03:03:57 +02:00
parent 9a20ffa8a3
commit ba5a747ac9
1 changed files with 70 additions and 5 deletions

View File

@ -1,10 +1,14 @@
// Copyright (c) ppy Pty Ltd <contact@ppy.sh>. Licensed under the MIT Licence.
// See the LICENCE file in the repository root for full licence text.
using System;
using System.Linq;
using System.Threading;
using osu.Framework.IO.Stores;
using osu.Framework.Logging;
using osu.Framework.Platform;
using osu.Game.IO;
using System.IO;
using osu.Game.Tournament.Configuration;
namespace osu.Game.Tournament
@ -32,16 +36,77 @@ public TournamentStorage(GameHost host)
this.host = host;
storageConfig = new TournamentStorageManager(host.Storage);
var customTournamentPath = storageConfig.Get<string>(StorageConfig.CurrentTournament);
var currentTournament = storageConfig.Get<string>(StorageConfig.CurrentTournament);
if (!string.IsNullOrEmpty(customTournamentPath))
if (!string.IsNullOrEmpty(currentTournament))
{
ChangeTargetStorage(UnderlyingStorage.GetStorageForDirectory("tournaments/" + customTournamentPath));
} else {
ChangeTargetStorage(UnderlyingStorage.GetStorageForDirectory("tournaments/default"));
ChangeTargetStorage(UnderlyingStorage.GetStorageForDirectory("tournaments" + Path.DirectorySeparatorChar + currentTournament));
}
else
{
// Migrating old storage format to the new one.
Migrate();
Logger.Log("Migrating files from old storage to new.");
}
VideoStorage = new TournamentVideoStorage(this);
Logger.Log("Using tournament storage: " + GetFullPath(string.Empty));
}
private void Migrate()
{
var defaultPath = "tournaments/default";
var source = new DirectoryInfo(GetFullPath("tournament"));
var destination = new DirectoryInfo(GetFullPath(defaultPath));
Directory.CreateDirectory(destination.FullName);
if (host.Storage.Exists("bracket.json"))
{
Logger.Log("Migrating bracket to default tournament storage.");
var bracketFile = new System.IO.FileInfo(GetFullPath(string.Empty) + Path.DirectorySeparatorChar + GetFiles(string.Empty, "bracket.json").First());
attemptOperation(() => bracketFile.CopyTo(Path.Combine(destination.FullName, bracketFile.Name), true));
}
Logger.Log("Migrating other assets to default tournament storage.");
copyRecursive(source, destination);
ChangeTargetStorage(UnderlyingStorage.GetStorageForDirectory(defaultPath));
storageConfig.Set<string>(StorageConfig.CurrentTournament, defaultPath);
storageConfig.Save();
}
private void copyRecursive(DirectoryInfo source, DirectoryInfo destination, bool topLevelExcludes = true)
{
// based off example code https://docs.microsoft.com/en-us/dotnet/api/system.io.directoryinfo
foreach (System.IO.FileInfo fi in source.GetFiles())
{
attemptOperation(() => fi.CopyTo(Path.Combine(destination.FullName, fi.Name), true));
}
foreach (DirectoryInfo dir in source.GetDirectories())
{
copyRecursive(dir, destination.CreateSubdirectory(dir.Name), false);
}
}
private void attemptOperation(Action action, int attempts = 10)
{
while (true)
{
try
{
action();
return;
}
catch (Exception)
{
if (attempts-- == 0)
throw;
}
Thread.Sleep(250);
}
}
}
}