1
0
mirror of https://github.com/ppy/osu.git synced 2024-12-15 01:43:20 +08:00

Add automatic transaction handling to realm helper methods

This commit is contained in:
Dean Herbert 2022-07-07 17:32:48 +09:00
parent d4c539687e
commit 5197d0fa9e
2 changed files with 50 additions and 4 deletions

View File

@ -59,6 +59,25 @@ namespace osu.Game.Tests.Database
});
}
[Test]
public void TestNestedWriteCalls()
{
RunTestWithRealm((realm, _) =>
{
var beatmap = new BeatmapInfo(CreateRuleset(), new BeatmapDifficulty(), new BeatmapMetadata());
var liveBeatmap = beatmap.ToLive(realm);
realm.Run(r =>
r.Write(_ =>
r.Write(_ =>
r.Add(beatmap)))
);
Assert.IsFalse(liveBeatmap.PerformRead(l => l.Hidden));
});
}
[Test]
public void TestAccessAfterAttach()
{

View File

@ -8,18 +8,45 @@ namespace osu.Game.Database
{
public static class RealmExtensions
{
/// <summary>
/// Perform a write operation against the provided realm instance.
/// </summary>
/// <remarks>
/// This will automatically start a transaction if not already in one.
/// </remarks>
/// <param name="realm">The realm to operate on.</param>
/// <param name="function">The write operation to run.</param>
public static void Write(this Realm realm, Action<Realm> function)
{
using var transaction = realm.BeginWrite();
Transaction? transaction = null;
if (!realm.IsInTransaction)
transaction = realm.BeginWrite();
function(realm);
transaction.Commit();
transaction?.Commit();
}
/// <summary>
/// Perform a write operation against the provided realm instance.
/// </summary>
/// <remarks>
/// This will automatically start a transaction if not already in one.
/// </remarks>
/// <param name="realm">The realm to operate on.</param>
/// <param name="function">The write operation to run.</param>
public static T Write<T>(this Realm realm, Func<Realm, T> function)
{
using var transaction = realm.BeginWrite();
Transaction? transaction = null;
if (!realm.IsInTransaction)
transaction = realm.BeginWrite();
var result = function(realm);
transaction.Commit();
transaction?.Commit();
return result;
}