mirror of
https://github.com/wabbajack-tools/wabbajack.git
synced 2024-08-30 18:42:17 +00:00
101 lines
3.0 KiB
C#
101 lines
3.0 KiB
C#
using System;
|
|
using System.Collections.Immutable;
|
|
using System.Data;
|
|
using System.Data.SQLite;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using Wabbajack.Common;
|
|
using Wabbajack.DTOs.Streams;
|
|
using Wabbajack.DTOs.Vfs;
|
|
using Wabbajack.Hashing.xxHash64;
|
|
using Wabbajack.Paths;
|
|
using Wabbajack.Paths.IO;
|
|
using Wabbajack.VFS.Interfaces;
|
|
|
|
namespace Wabbajack.VFS;
|
|
|
|
public class VFSDiskCache : IVfsCache
|
|
{
|
|
private readonly SQLiteConnection _conn;
|
|
private readonly string _connectionString;
|
|
private readonly AbsolutePath _path;
|
|
|
|
public VFSDiskCache(AbsolutePath path)
|
|
{
|
|
_path = path;
|
|
|
|
if (!_path.Parent.DirectoryExists())
|
|
_path.Parent.CreateDirectory();
|
|
|
|
_connectionString = string.Intern($"URI=file:{path};Pooling=True;Max Pool Size=100; Journal Mode=Memory;");
|
|
_conn = new SQLiteConnection(_connectionString);
|
|
_conn.Open();
|
|
|
|
using var cmd = new SQLiteCommand(_conn);
|
|
cmd.CommandText = @"CREATE TABLE IF NOT EXISTS VFSCache (
|
|
Hash BIGINT PRIMARY KEY,
|
|
Contents BLOB)
|
|
WITHOUT ROWID";
|
|
cmd.ExecuteNonQuery();
|
|
}
|
|
|
|
public async Task<IndexedVirtualFile?> Get(Hash hash, CancellationToken token)
|
|
{
|
|
if (hash == default)
|
|
throw new ArgumentException("Cannot cache default hashes");
|
|
|
|
await using var cmd = new SQLiteCommand(_conn);
|
|
cmd.CommandText = @"SELECT Contents FROM VFSCache WHERE Hash = @hash";
|
|
cmd.Parameters.AddWithValue("@hash", (long) hash);
|
|
|
|
await using var rdr = cmd.ExecuteReader();
|
|
while (rdr.Read())
|
|
{
|
|
var data = IndexedVirtualFileExtensions.Read(rdr.GetStream(0));
|
|
return data;
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
public async Task Put(IndexedVirtualFile ivf, CancellationToken token)
|
|
{
|
|
await using var ms = new MemoryStream();
|
|
// Top level path gets renamed when read, we don't want the absolute path
|
|
// here else the reader will blow up when it tries to convert the value
|
|
ivf.Name = (RelativePath) "not/applicable";
|
|
ivf.Write(ms);
|
|
ms.Position = 0;
|
|
await InsertIntoVFSCache(ivf.Hash, ms);
|
|
}
|
|
|
|
private async Task InsertIntoVFSCache(Hash hash, MemoryStream data)
|
|
{
|
|
await using var cmd = new SQLiteCommand(_conn);
|
|
cmd.CommandText = @"INSERT INTO VFSCache (Hash, Contents) VALUES (@hash, @contents)";
|
|
cmd.Parameters.AddWithValue("@hash", (long) hash);
|
|
var val = new SQLiteParameter("@contents", DbType.Binary) {Value = data.ToArray()};
|
|
cmd.Parameters.Add(val);
|
|
try
|
|
{
|
|
await cmd.ExecuteNonQueryAsync();
|
|
}
|
|
catch (SQLiteException ex)
|
|
{
|
|
if (ex.Message.StartsWith("constraint failed"))
|
|
return;
|
|
throw;
|
|
}
|
|
}
|
|
|
|
public void VacuumDatabase()
|
|
{
|
|
using var cmd = new SQLiteCommand(_conn);
|
|
cmd.CommandText = @"VACUUM";
|
|
cmd.PrepareAsync();
|
|
|
|
cmd.ExecuteNonQuery();
|
|
}
|
|
} |