h1or4dme8/osu-lazer
1
0
Fork 0
mirror of https://github.com/ppy/osu.git synced 2024-09-22 09:27:34 +08:00
Code Issues Releases Wiki Activity

Tidy up RealmContextFactory; remove delete/dispose method which wouldn't work due to threading

Browse Source
This commit is contained in:
Dean Herbert 2021-01-12 14:36:35 +09:00
parent 8cbad1dc1c
commit 0dca9c8c46
1 changed files with 20 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
osu.Game/Database/RealmContextFactory.cs
View File

@ -2,6 +2,7 @@
// See the LICENCE file in the repository root for full licence text.
using System.Threading;
using osu.Framework.Logging;
using osu.Framework.Platform;
using osu.Framework.Statistics;
using Realms;
@ -16,8 +17,11 @@ namespace osu.Game.Database
private const int schema_version = 5;
private ThreadLocal<Realm> threadContexts;
private readonly ThreadLocal<Realm> threadContexts;
/// <summary>
/// Lock object which is held for the duration of a write operation (via <see cref="GetForWrite"/>).
/// </summary>
private readonly object writeLock = new object();
private ThreadLocal<bool> refreshCompleted = new ThreadLocal<bool>();
@ -32,10 +36,11 @@ namespace osu.Game.Database
{
this.storage = storage;
recreateThreadContexts();
threadContexts = new ThreadLocal<Realm>(createContext, true);
using (CreateContext())
using (var realm = Get())
{
Logger.Log($"Opened realm {database_name} at version {realm.Config.SchemaVersion}");
// creating a context will ensure our schema is up-to-date and migrated.
}
}
@ -95,7 +100,7 @@ namespace osu.Game.Database
var context = threadContexts.Value;
if (context?.IsClosed != false)
threadContexts.Value = context = CreateContext();
threadContexts.Value = context = createContext();
contexts_open.Value = threadContexts.Values.Count;
@ -110,6 +115,17 @@ namespace osu.Game.Database
return context;
}
private Realm createContext()
{
contexts_created.Value++;
return Realm.GetInstance(new RealmConfiguration(storage.GetFullPath($"{database_name}.realm", true))
{
SchemaVersion = schema_version,
MigrationCallback = onMigration,
});
}
private void usageCompleted(RealmWriteUsage usage)
{
int usages = Interlocked.Decrement(ref currentWriteUsages);
@ -142,33 +158,5 @@ namespace osu.Game.Database
Monitor.Exit(writeLock);
}
}
private void recreateThreadContexts()
{
// Contexts for other threads are not disposed as they may be in use elsewhere. Instead, fresh contexts are exposed
// for other threads to use, and we rely on the finalizer inside OsuDbContext to handle their previous contexts
threadContexts?.Value.Dispose();
threadContexts = new ThreadLocal<Realm>(CreateContext, true);
}
protected virtual Realm CreateContext()
{
contexts_created.Value++;
return Realm.GetInstance(new RealmConfiguration(storage.GetFullPath($"{database_name}.realm", true))
{
SchemaVersion = schema_version,
MigrationCallback = onMigration,
});
}
public void ResetDatabase()
{
lock (writeLock)
{
recreateThreadContexts();
storage.DeleteDatabase(database_name);
}
}
}
}