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
|
// Copyright (c) ZeroC, Inc. All rights reserved.
using System;
using System.Collections.Generic;
using System.Threading.Channels;
using System.Threading.Tasks;
namespace ZeroC.Ice
{
internal sealed class LoggerAdminLogger : ILogger, IAsyncDisposable
{
public string Prefix => LocalLogger.Prefix;
internal ILogger LocalLogger { get; }
private static long Now() => DateTimeOffset.UtcNow.ToUnixTimeMilliseconds() * 1000;
private readonly Channel<(List<LogForwarder>, LogMessage)> _channel;
private readonly LoggerAdmin _loggerAdmin;
public ILogger CloneWithPrefix(string prefix) => LocalLogger.CloneWithPrefix(prefix);
public async ValueTask DisposeAsync()
{
if (LocalLogger is FileLogger fileLogger)
{
fileLogger.Dispose();
}
_channel.Writer.Complete();
await _loggerAdmin.DisposeAsync().ConfigureAwait(false);
}
public void Error(string message)
{
var logMessage = new LogMessage(LogMessageType.ErrorMessage, Now(), "", message);
LocalLogger.Error(message);
Log(logMessage);
}
public IAsyncLoggerAdmin GetFacet() => _loggerAdmin;
public void Print(string message)
{
var logMessage = new LogMessage(LogMessageType.PrintMessage, Now(), "", message);
LocalLogger.Print(message);
Log(logMessage);
}
public void Trace(string category, string message)
{
var logMessage = new LogMessage(LogMessageType.TraceMessage, Now(), category, message);
LocalLogger.Trace(category, message);
Log(logMessage);
}
public void Warning(string message)
{
var logMessage = new LogMessage(LogMessageType.WarningMessage, Now(), "", message);
LocalLogger.Warning(message);
Log(logMessage);
}
internal LoggerAdminLogger(Communicator communicator, ILogger localLogger)
{
LocalLogger = (localLogger as LoggerAdminLogger)?.LocalLogger ?? localLogger;
_loggerAdmin = new LoggerAdmin(communicator, this);
// Create an unbounded channel to ensure the messages are sent from a separate thread. We don't allow
// synchronous continuations to ensure that writes on the channel are never processed by the writer
// thread.
_channel = Channel.CreateUnbounded<(List<LogForwarder>, LogMessage)>(new UnboundedChannelOptions
{
AllowSynchronousContinuations = false,
SingleReader = true,
SingleWriter = false
});
Task.Run(async () =>
{
// The enumeration completes when the channel writer Complete method is called.
await foreach ((List<LogForwarder> forwarders, LogMessage logMessage) in _channel.Reader.ReadAllAsync())
{
foreach (LogForwarder forwarder in forwarders)
{
forwarder.Queue("log", LocalLogger, proxy => proxy.LogAsync(logMessage));
}
}
});
}
internal void Log(LogMessage logMessage)
{
List<LogForwarder>? logForwarderList = _loggerAdmin.Log(logMessage);
if (logForwarderList != null)
{
_channel.Writer.TryWrite((logForwarderList, logMessage));
}
}
}
}
|
