ToxicMushroom/Dalamud
1
0
Fork 0
mirror of https://github.com/goatcorp/Dalamud.git synced 2025-12-14 12:44:16 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

feat: handle db load failures gracefully

Browse source
This commit is contained in:
goat 2023-09-29 18:37:45 +02:00
parent c1fd08cc93
commit 416bb42f5b
No known key found for this signature in database
GPG key ID: 49E2AA8C6A76498B
1 changed files with 43 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

47
Dalamud/Storage/ReliableFileStorage.cs
View file

@ -27,7 +27,7 @@ public class ReliableFileStorage : IServiceType, IDisposable
{ {
private static readonly ModuleLog Log = new("VFS"); private static readonly ModuleLog Log = new("VFS");
private SQLiteConnection db; private SQLiteConnection? db;
/// <summary> /// <summary>
/// Initializes a new instance of the <see cref="ReliableFileStorage"/> class. /// Initializes a new instance of the <see cref="ReliableFileStorage"/> class.
@ -38,8 +38,27 @@ public class ReliableFileStorage : IServiceType, IDisposable
var databasePath = Path.Combine(vfsDbPath, "dalamudVfs.db"); var databasePath = Path.Combine(vfsDbPath, "dalamudVfs.db");
Log.Verbose("Initializing VFS database at {Path}", databasePath); Log.Verbose("Initializing VFS database at {Path}", databasePath);
this.db = new SQLiteConnection(databasePath, SQLiteOpenFlags.ReadWrite | SQLiteOpenFlags.Create | SQLiteOpenFlags.FullMutex);
this.db.CreateTable<DbFile>(); try
{
this.SetupDb(vfsDbPath);
}
catch (Exception ex)
{
Log.Error(ex, "Failed to load VFS database, starting fresh");
try
{
if (File.Exists(vfsDbPath))
File.Delete(vfsDbPath);
}
catch (Exception)
{
// ignored
}
this.SetupDb(vfsDbPath);
}
} }
/// <summary> /// <summary>
@ -56,6 +75,9 @@ public class ReliableFileStorage : IServiceType, IDisposable
if (File.Exists(path)) if (File.Exists(path))
return true; return true;
if (this.db == null)
return false;
// If the file doesn't actually exist on the FS, but it does in the DB, we can say YES and read operations will read from the DB instead // If the file doesn't actually exist on the FS, but it does in the DB, we can say YES and read operations will read from the DB instead
var normalizedPath = NormalizePath(path); var normalizedPath = NormalizePath(path);
@ -94,6 +116,12 @@ public class ReliableFileStorage : IServiceType, IDisposable
public void WriteAllBytes(string path, byte[] bytes, Guid containerId = default) public void WriteAllBytes(string path, byte[] bytes, Guid containerId = default)
{ {
ArgumentException.ThrowIfNullOrEmpty(path); ArgumentException.ThrowIfNullOrEmpty(path);
if (this.db == null)
{
Util.WriteAllBytesSafe(path, bytes);
return;
}
this.db.RunInTransaction(() => this.db.RunInTransaction(() =>
{ {
@ -227,6 +255,10 @@ public class ReliableFileStorage : IServiceType, IDisposable
if (forceBackup) if (forceBackup)
{ {
// If the db failed to load, act as if the file does not exist
if (this.db == null)
throw new FileNotFoundException("Backup database was not available");
var normalizedPath = NormalizePath(path); var normalizedPath = NormalizePath(path);
var file = this.db.Table<DbFile>().FirstOrDefault(f => f.Path == normalizedPath && f.ContainerId == containerId); var file = this.db.Table<DbFile>().FirstOrDefault(f => f.Path == normalizedPath && f.ContainerId == containerId);
if (file == null) if (file == null)
@ -253,7 +285,7 @@ public class ReliableFileStorage : IServiceType, IDisposable
/// <inheritdoc/> /// <inheritdoc/>
public void Dispose() public void Dispose()
{ {
this.db.Dispose(); this.db?.Dispose();
} }
/// <summary> /// <summary>
@ -269,6 +301,13 @@ public class ReliableFileStorage : IServiceType, IDisposable
return path; return path;
} }
private void SetupDb(string path)
{
this.db = new SQLiteConnection(path,
SQLiteOpenFlags.ReadWrite | SQLiteOpenFlags.Create | SQLiteOpenFlags.FullMutex);
this.db.CreateTable<DbFile>();
}
private class DbFile private class DbFile
{ {