64 lines
2 KiB
C#
64 lines
2 KiB
C#
using Microsoft.Extensions.Logging;
|
|
using SharpChat.Configuration;
|
|
using SharpChat.Messages;
|
|
using System.Data.SQLite;
|
|
using ZLogger;
|
|
using NativeSQLiteConnection = System.Data.SQLite.SQLiteConnection;
|
|
|
|
namespace SharpChat.SQLite;
|
|
|
|
public class SQLiteStorage(ILogger logger, string connString) : Storage, IDisposable {
|
|
public const string MEMORY = "file::memory:?cache=shared";
|
|
public const string DEFAULT = "sharpchat.db";
|
|
|
|
public SQLiteConnection Connection { get; } = new SQLiteConnection(new NativeSQLiteConnection(connString).OpenAndReturn());
|
|
|
|
public MessageStorage CreateMessageStorage() {
|
|
return new SQLiteMessageStorage(logger, Connection);
|
|
}
|
|
|
|
public async Task UpgradeStorage() {
|
|
logger.ZLogInformation($"Upgrading storage...");
|
|
await new SQLiteMigrations(logger, Connection).RunMigrations();
|
|
}
|
|
|
|
public static string BuildConnectionString(Config config) {
|
|
return BuildConnectionString(
|
|
config.ReadValue("path", DEFAULT)!,
|
|
config.ReadValue("pass")
|
|
);
|
|
}
|
|
|
|
public static string BuildConnectionString(string path, string? password) {
|
|
return new SQLiteConnectionStringBuilder {
|
|
DataSource = string.IsNullOrWhiteSpace(path) ? MEMORY : path,
|
|
DateTimeFormat = SQLiteDateFormats.ISO8601,
|
|
DateTimeKind = DateTimeKind.Utc,
|
|
FailIfMissing = false,
|
|
ForeignKeys = true,
|
|
JournalMode = SQLiteJournalModeEnum.Wal,
|
|
LegacyFormat = false,
|
|
Password = string.IsNullOrWhiteSpace(password) ? null : password,
|
|
ReadOnly = false,
|
|
UseUTF16Encoding = false,
|
|
}.ToString();
|
|
}
|
|
|
|
private bool disposed = false;
|
|
|
|
~SQLiteStorage() {
|
|
DoDispose();
|
|
}
|
|
|
|
public void Dispose() {
|
|
DoDispose();
|
|
GC.SuppressFinalize(this);
|
|
}
|
|
|
|
private void DoDispose() {
|
|
if(disposed)
|
|
return;
|
|
disposed = true;
|
|
Connection.Dispose();
|
|
}
|
|
}
|