-
Notifications
You must be signed in to change notification settings - Fork 0
/
ThreadedLogger.cs
103 lines (89 loc) · 2.99 KB
/
ThreadedLogger.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
// originally from https://stackoverflow.com/a/1187568/53236
namespace twot
{
using System;
using System.Collections.Generic;
using System.IO;
using System.Text;
using System.Threading;
internal class ThreadedLogger : IDisposable
{
private readonly Queue<Action> queue = new Queue<Action>();
private readonly ManualResetEvent hasNewItems = new ManualResetEvent(false);
private readonly ManualResetEvent terminate = new ManualResetEvent(false);
private readonly ManualResetEvent waiting = new ManualResetEvent(false);
private readonly Thread? loggingThread;
private readonly FileStream? fileStream;
private readonly bool enabled;
private readonly byte[] newLine = Encoding.UTF8.GetBytes(Environment.NewLine);
public ThreadedLogger(string filename, bool enabled)
{
this.enabled = enabled;
if (this.enabled)
{
this.fileStream = new FileStream(filename, FileMode.Append, FileAccess.Write, FileShare.Read);
this.loggingThread = new Thread(new ThreadStart(this.ProcessQueue));
this.loggingThread.IsBackground = true;
this.loggingThread.Start();
}
}
public void LogMessage(string row)
{
if (!this.enabled)
{
return;
}
lock (this.queue)
{
this.queue.Enqueue(() => this.WriteLogMessage(row));
}
this.hasNewItems.Set();
}
public void Flush()
{
this.waiting.WaitOne();
}
public void Dispose()
{
this.terminate.Set();
this.terminate.Dispose();
this.waiting.Dispose();
this.hasNewItems.Dispose();
this.loggingThread?.Join();
this.fileStream?.Dispose();
GC.SuppressFinalize(this);
}
private void ProcessQueue()
{
while (true)
{
this.waiting.Set();
var i = ManualResetEvent.WaitAny(new WaitHandle[] { this.hasNewItems, this.terminate });
if (i == 1)
{
return;
}
this.hasNewItems.Reset();
this.waiting.Reset();
Queue<Action> queueCopy;
lock (this.queue)
{
queueCopy = new Queue<Action>(this.queue);
this.queue.Clear();
}
foreach (var log in queueCopy)
{
log();
}
}
}
private void WriteLogMessage(string row)
{
this.fileStream!.Write(Encoding.UTF8.GetBytes(row));
if (!row.EndsWith(Environment.NewLine, StringComparison.InvariantCultureIgnoreCase))
{
this.fileStream.Write(this.newLine);
}
}
}
}