h1or4dme8/osu-lazer
1
0
Fork 0
mirror of https://github.com/ppy/osu.git synced 2024-11-11 15:07:44 +08:00
Code Issues Releases Wiki Activity

Detach beatmap sets asynchronously

Browse Source
This commit is contained in:
Dean Herbert 2024-08-28 17:46:36 +09:00
parent b1f653899c
commit c2c1dccf2d
No known key found for this signature in database
1 changed files with 45 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

57
osu.Game/Database/DetachedBeatmapStore.cs
View File

@ -4,6 +4,7 @@
using System;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
using osu.Framework.Allocation;
using osu.Framework.Bindables;
using osu.Framework.Graphics;
@ -30,9 +31,10 @@ namespace osu.Game.Database
}
[BackgroundDependencyLoader]
private void load()
private void load(CancellationToken cancellationToken)
{
realmSubscription = realm.RegisterForNotifications(getBeatmapSets, beatmapSetsChanged);
realmSubscription = realm.RegisterForNotifications(r => r.All<BeatmapSetInfo>().Where(s => !s.DeletePending && !s.Protected), beatmapSetsChanged);
loaded.Wait(cancellationToken);
}
private void beatmapSetsChanged(IRealmCollection<BeatmapSetInfo> sender, ChangeSet? changes)
@ -50,23 +52,56 @@ namespace osu.Game.Database
return;
}
detachedBeatmapSets.Clear();
detachedBeatmapSets.AddRange(sender.Detach());
// Detaching beatmaps takes some time, so let's make sure it doesn't run on the update thread.
var frozenSets = sender.Freeze();
Task.Factory.StartNew(() =>
{
realm.Run(_ =>
{
var detached = frozenSets.Detach();
detachedBeatmapSets.Clear();
detachedBeatmapSets.AddRange(detached);
loaded.Set();
});
}, TaskCreationOptions.LongRunning);
loaded.Set();
return;
}
foreach (int i in changes.DeletedIndices.OrderDescending())
detachedBeatmapSets.RemoveAt(i);
removeAt(i);
foreach (int i in changes.InsertedIndices)
{
detachedBeatmapSets.Insert(i, sender[i].Detach());
}
insert(sender[i].Detach(), i);
foreach (int i in changes.NewModifiedIndices)
detachedBeatmapSets.ReplaceRange(i, 1, new[] { sender[i].Detach() });
replaceRange(sender[i].Detach(), i);
}
private void replaceRange(BeatmapSetInfo set, int i)
{
if (loaded.IsSet)
detachedBeatmapSets.ReplaceRange(i, 1, new[] { set });
else
Schedule(() => { detachedBeatmapSets.ReplaceRange(i, 1, new[] { set }); });
}
private void insert(BeatmapSetInfo set, int i)
{
if (loaded.IsSet)
detachedBeatmapSets.Insert(i, set);
else
Schedule(() => { detachedBeatmapSets.Insert(i, set); });
}
private void removeAt(int i)
{
if (loaded.IsSet)
detachedBeatmapSets.RemoveAt(i);
else
Schedule(() => { detachedBeatmapSets.RemoveAt(i); });
}
protected override void Dispose(bool isDisposing)
@ -75,7 +110,5 @@ namespace osu.Game.Database
loaded.Set();
realmSubscription?.Dispose();
}
private IQueryable<BeatmapSetInfo> getBeatmapSets(Realm realm) => realm.All<BeatmapSetInfo>().Where(s => !s.DeletePending && !s.Protected);
}
}