Expose ContextFactory to allow for connection flushing

This commit is contained in:
Dean Herbert 2020-05-06 18:31:17 +09:00
parent 4ac5ed71f4
commit 754afb9c0b
1 changed files with 12 additions and 12 deletions

View File

@ -121,7 +121,7 @@ public OsuGameBase()
protected override IReadOnlyDependencyContainer CreateChildDependencies(IReadOnlyDependencyContainer parent) =>
dependencies = new DependencyContainer(base.CreateChildDependencies(parent));
private DatabaseContextFactory contextFactory;
protected DatabaseContextFactory ContextFactory;
protected override UserInputManager CreateUserInputManager() => new OsuUserInputManager();
@ -130,7 +130,7 @@ private void load()
{
Resources.AddStore(new DllResourceStore(OsuResources.ResourceAssembly));
dependencies.Cache(contextFactory = new DatabaseContextFactory(Storage));
dependencies.Cache(ContextFactory = new DatabaseContextFactory(Storage));
dependencies.CacheAs(Storage);
@ -161,7 +161,7 @@ private void load()
runMigrations();
dependencies.Cache(SkinManager = new SkinManager(Storage, contextFactory, Host, Audio, new NamespacedResourceStore<byte[]>(Resources, "Skins/Legacy")));
dependencies.Cache(SkinManager = new SkinManager(Storage, ContextFactory, Host, Audio, new NamespacedResourceStore<byte[]>(Resources, "Skins/Legacy")));
dependencies.CacheAs<ISkinSource>(SkinManager);
if (API == null) API = new APIAccess(LocalConfig);
@ -170,12 +170,12 @@ private void load()
var defaultBeatmap = new DummyWorkingBeatmap(Audio, Textures);
dependencies.Cache(RulesetStore = new RulesetStore(contextFactory, Storage));
dependencies.Cache(FileStore = new FileStore(contextFactory, Storage));
dependencies.Cache(RulesetStore = new RulesetStore(ContextFactory, Storage));
dependencies.Cache(FileStore = new FileStore(ContextFactory, Storage));
// ordering is important here to ensure foreign keys rules are not broken in ModelStore.Cleanup()
dependencies.Cache(ScoreManager = new ScoreManager(RulesetStore, () => BeatmapManager, Storage, API, contextFactory, Host));
dependencies.Cache(BeatmapManager = new BeatmapManager(Storage, contextFactory, RulesetStore, API, Audio, Host, defaultBeatmap));
dependencies.Cache(ScoreManager = new ScoreManager(RulesetStore, () => BeatmapManager, Storage, API, ContextFactory, Host));
dependencies.Cache(BeatmapManager = new BeatmapManager(Storage, ContextFactory, RulesetStore, API, Audio, Host, defaultBeatmap));
// this should likely be moved to ArchiveModelManager when another case appers where it is necessary
// to have inter-dependent model managers. this could be obtained with an IHasForeign<T> interface to
@ -189,8 +189,8 @@ List<ScoreInfo> getBeatmapScores(BeatmapSetInfo set)
BeatmapManager.ItemRemoved += i => ScoreManager.Delete(getBeatmapScores(i), true);
BeatmapManager.ItemAdded += i => ScoreManager.Undelete(getBeatmapScores(i), true);
dependencies.Cache(KeyBindingStore = new KeyBindingStore(contextFactory, RulesetStore));
dependencies.Cache(SettingsStore = new SettingsStore(contextFactory));
dependencies.Cache(KeyBindingStore = new KeyBindingStore(ContextFactory, RulesetStore));
dependencies.Cache(SettingsStore = new SettingsStore(ContextFactory));
dependencies.Cache(RulesetConfigCache = new RulesetConfigCache(SettingsStore));
dependencies.Cache(new SessionStatics());
dependencies.Cache(new OsuColour());
@ -279,7 +279,7 @@ private void runMigrations()
{
try
{
using (var db = contextFactory.GetForWrite(false))
using (var db = ContextFactory.GetForWrite(false))
db.Context.Migrate();
}
catch (Exception e)
@ -288,12 +288,12 @@ private void runMigrations()
// if we failed, let's delete the database and start fresh.
// todo: we probably want a better (non-destructive) migrations/recovery process at a later point than this.
contextFactory.ResetDatabase();
ContextFactory.ResetDatabase();
Logger.Log("Database purged successfully.", LoggingTarget.Database);
// only run once more, then hard bail.
using (var db = contextFactory.GetForWrite(false))
using (var db = ContextFactory.GetForWrite(false))
db.Context.Migrate();
}
}