Compare commits

..

No commits in common. "6ce0ef7d55d73b632bfef379e0969cd483cc58b0" and "3e891e19c364e3ea18c6236ee4a6896fdae02508" have entirely different histories.

4 changed files with 8 additions and 73 deletions

View File

@ -53,12 +53,9 @@ namespace DHT.Desktop.Main.Pages {
await progressDialog.ShowDialog(window); await progressDialog.ShowDialog(window);
} }
private const int BatchSize = 500; private const int BatchSize = 100;
private async Task GenerateRandomData(int channelCount, int userCount, int messageCount, IProgressCallback callback) { private async Task GenerateRandomData(int channelCount, int userCount, int messageCount, IProgressCallback callback) {
int batchCount = (messageCount + BatchSize - 1) / BatchSize;
await callback.Update("Adding messages in batches of " + BatchSize, 0, batchCount);
var rand = new Random(); var rand = new Random();
var server = new DHT.Server.Data.Server { var server = new DHT.Server.Data.Server {
Id = RandomId(rand), Id = RandomId(rand),
@ -90,6 +87,9 @@ namespace DHT.Desktop.Main.Pages {
db.AddChannel(channel); db.AddChannel(channel);
} }
int batchCount = (messageCount + BatchSize - 1) / BatchSize;
await callback.Update("Adding messages in batches of 100", 0, batchCount);
var now = DateTimeOffset.Now; var now = DateTimeOffset.Now;
int batchIndex = 0; int batchIndex = 0;
@ -119,8 +119,8 @@ namespace DHT.Desktop.Main.Pages {
db.AddMessages(messages); db.AddMessages(messages);
messageCount -= BatchSize; messageCount -= 100;
await callback.Update("Adding messages in batches of " + BatchSize, ++batchIndex, batchCount); await callback.Update("Adding messages in batches of 100", ++batchIndex, batchCount);
} }
} }

View File

@ -36,13 +36,10 @@ namespace DHT.Server.Database.Sqlite {
private readonly Log log; private readonly Log log;
private readonly SqliteConnectionPool pool; private readonly SqliteConnectionPool pool;
private readonly SqliteMessageStatisticsThread messageStatisticsThread;
private SqliteDatabaseFile(string path, SqliteConnectionPool pool) { private SqliteDatabaseFile(string path, SqliteConnectionPool pool) {
this.log = Log.ForType(typeof(SqliteDatabaseFile), System.IO.Path.GetFileName(path)); this.log = Log.ForType(typeof(SqliteDatabaseFile), System.IO.Path.GetFileName(path));
this.pool = pool; this.pool = pool;
this.messageStatisticsThread = new SqliteMessageStatisticsThread(pool, UpdateMessageStatistics);
this.Path = path; this.Path = path;
this.Statistics = new DatabaseStatistics(); this.Statistics = new DatabaseStatistics();
@ -54,7 +51,6 @@ namespace DHT.Server.Database.Sqlite {
} }
public void Dispose() { public void Dispose() {
messageStatisticsThread.Dispose();
pool.Dispose(); pool.Dispose();
} }
@ -303,7 +299,7 @@ namespace DHT.Server.Database.Sqlite {
} }
tx.Commit(); tx.Commit();
messageStatisticsThread.RequestUpdate(); UpdateMessageStatistics(conn);
} }
public int CountMessages(MessageFilter? filter = null) { public int CountMessages(MessageFilter? filter = null) {

View File

@ -1,54 +0,0 @@
using System;
using System.Threading;
using DHT.Server.Database.Sqlite.Utils;
namespace DHT.Server.Database.Sqlite {
sealed class SqliteMessageStatisticsThread : IDisposable {
private readonly SqliteConnectionPool pool;
private readonly Action<ISqliteConnection> action;
private readonly CancellationTokenSource cancellationTokenSource = new();
private readonly CancellationToken cancellationToken;
private readonly AutoResetEvent requestEvent = new (false);
public SqliteMessageStatisticsThread(SqliteConnectionPool pool, Action<ISqliteConnection> action) {
this.pool = pool;
this.action = action;
this.cancellationToken = cancellationTokenSource.Token;
var thread = new Thread(RunThread) {
Name = "DHT message statistics thread",
IsBackground = true
};
thread.Start();
}
public void Dispose() {
try {
cancellationTokenSource.Cancel();
} catch (ObjectDisposedException) {}
}
public void RequestUpdate() {
try {
requestEvent.Set();
} catch (ObjectDisposedException) {}
}
private void RunThread() {
try {
while (!cancellationToken.IsCancellationRequested) {
if (requestEvent.WaitOne(TimeSpan.FromMilliseconds(100))) {
using var conn = pool.Take();
action(conn);
}
}
} finally {
cancellationTokenSource.Dispose();
requestEvent.Dispose();
}
}
}
}

View File

@ -24,14 +24,7 @@ namespace DHT.Server.Database.Sqlite.Utils {
for (int i = 0; i < poolSize; i++) { for (int i = 0; i < poolSize; i++) {
var conn = new SqliteConnection(connectionString); var conn = new SqliteConnection(connectionString);
conn.Open(); conn.Open();
free.Add(new PooledConnection(this, conn));
var pooledConn = new PooledConnection(this, conn);
using (var cmd = pooledConn.Command("PRAGMA journal_mode=WAL")) {
cmd.ExecuteNonQuery();
}
free.Add(pooledConn);
} }
used = new List<PooledConnection>(poolSize); used = new List<PooledConnection>(poolSize);