| | | 1 | | // Copyright (c) ZeroC, Inc. |
| | | 2 | | |
| | | 3 | | using IceRpc.Transports; |
| | | 4 | | |
| | | 5 | | namespace IceRpc.Internal; |
| | | 6 | | |
| | | 7 | | /// <summary>Provides a log decorator for protocol connections.</summary> |
| | | 8 | | internal class MetricsProtocolConnectionDecorator : IProtocolConnection |
| | | 9 | | { |
| | | 10 | | private readonly bool _connectStarted; |
| | | 11 | | private readonly IProtocolConnection _decoratee; |
| | | 12 | | private bool _isConnected; |
| | | 13 | | private bool _isShutdown; |
| | | 14 | | private readonly Metrics _metrics; |
| | | 15 | | |
| | | 16 | | public async Task<(TransportConnectionInformation ConnectionInformation, Task ShutdownRequested)> ConnectAsync( |
| | | 17 | | CancellationToken cancellationToken) |
| | 158 | 18 | | { |
| | 158 | 19 | | if (!_connectStarted) |
| | 91 | 20 | | { |
| | 91 | 21 | | _metrics.ConnectStart(); |
| | 91 | 22 | | } |
| | | 23 | | try |
| | 158 | 24 | | { |
| | 158 | 25 | | (TransportConnectionInformation connectionInformation, Task shutdownRequested) = |
| | 158 | 26 | | await _decoratee.ConnectAsync(cancellationToken).ConfigureAwait(false); |
| | 132 | 27 | | _metrics.ConnectSuccess(); |
| | 132 | 28 | | _isConnected = true; |
| | 132 | 29 | | return (connectionInformation, shutdownRequested); |
| | | 30 | | } |
| | 26 | 31 | | catch |
| | 26 | 32 | | { |
| | 26 | 33 | | _metrics.ConnectionFailure(); |
| | 26 | 34 | | throw; |
| | | 35 | | } |
| | | 36 | | finally |
| | 158 | 37 | | { |
| | 158 | 38 | | _metrics.ConnectStop(); |
| | 158 | 39 | | } |
| | 132 | 40 | | } |
| | | 41 | | |
| | | 42 | | public async ValueTask DisposeAsync() |
| | 158 | 43 | | { |
| | 158 | 44 | | await _decoratee.DisposeAsync().ConfigureAwait(false); |
| | 158 | 45 | | if (_isConnected) |
| | 132 | 46 | | { |
| | 132 | 47 | | if (!_isShutdown) |
| | 82 | 48 | | { |
| | | 49 | | // shutdown failed or didn't call shutdown at all. |
| | 82 | 50 | | _metrics.ConnectionFailure(); |
| | 82 | 51 | | } |
| | 132 | 52 | | _metrics.ConnectionDisconnected(); |
| | 132 | 53 | | } |
| | 158 | 54 | | } |
| | | 55 | | |
| | | 56 | | public Task<IncomingResponse> InvokeAsync(OutgoingRequest request, CancellationToken cancellationToken) => |
| | 130 | 57 | | _decoratee.InvokeAsync(request, cancellationToken); |
| | | 58 | | |
| | | 59 | | public async Task ShutdownAsync(CancellationToken cancellationToken) |
| | 62 | 60 | | { |
| | 62 | 61 | | await _decoratee.ShutdownAsync(cancellationToken).ConfigureAwait(false); |
| | 50 | 62 | | _isShutdown = true; |
| | 50 | 63 | | } |
| | | 64 | | |
| | | 65 | | /// <summary>A protocol connection decorator to log connection metrics.</summary> |
| | | 66 | | /// <param name="decoratee">The protocol connection decoratee.</param> |
| | | 67 | | /// <param name="metrics">The metrics object used to log the metrics.</param> |
| | | 68 | | /// <param name="connectStarted">Whether or not the ConnectStart events has be already logged. For server connection |
| | | 69 | | /// the event is logged from a separate <see cref="Server.IConnector"/> decorator.</param> |
| | 158 | 70 | | internal MetricsProtocolConnectionDecorator( |
| | 158 | 71 | | IProtocolConnection decoratee, |
| | 158 | 72 | | Metrics metrics, |
| | 158 | 73 | | bool connectStarted) |
| | 158 | 74 | | { |
| | 158 | 75 | | _connectStarted = connectStarted; |
| | 158 | 76 | | _decoratee = decoratee; |
| | 158 | 77 | | _metrics = metrics; |
| | 158 | 78 | | } |
| | | 79 | | } |