mirror of
https://github.com/chylex/Discord-History-Tracker.git
synced 2025-06-09 11:52:09 +03:00
Refresh message statistics after tracking messages in a background thread
This commit is contained in:
parent
fd09ac496e
commit
6ce0ef7d55
@ -36,10 +36,13 @@ 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();
|
||||||
|
|
||||||
@ -51,6 +54,7 @@ namespace DHT.Server.Database.Sqlite {
|
|||||||
}
|
}
|
||||||
|
|
||||||
public void Dispose() {
|
public void Dispose() {
|
||||||
|
messageStatisticsThread.Dispose();
|
||||||
pool.Dispose();
|
pool.Dispose();
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -299,7 +303,7 @@ namespace DHT.Server.Database.Sqlite {
|
|||||||
}
|
}
|
||||||
|
|
||||||
tx.Commit();
|
tx.Commit();
|
||||||
UpdateMessageStatistics(conn);
|
messageStatisticsThread.RequestUpdate();
|
||||||
}
|
}
|
||||||
|
|
||||||
public int CountMessages(MessageFilter? filter = null) {
|
public int CountMessages(MessageFilter? filter = null) {
|
||||||
|
54
app/Server/Database/Sqlite/SqliteMessageStatisticsThread.cs
Normal file
54
app/Server/Database/Sqlite/SqliteMessageStatisticsThread.cs
Normal file
@ -0,0 +1,54 @@
|
|||||||
|
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();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
Loading…
x
Reference in New Issue
Block a user