| | | 1 | | // Copyright (c) ZeroC, Inc. |
| | | 2 | | |
| | | 3 | | using IceRpc.Internal; |
| | | 4 | | using IceRpc.Transports; |
| | | 5 | | using Microsoft.Extensions.Logging; |
| | | 6 | | using Microsoft.Extensions.Logging.Abstractions; |
| | | 7 | | using System.Net.Security; |
| | | 8 | | |
| | | 9 | | namespace IceRpc; |
| | | 10 | | |
| | | 11 | | /// <summary>Default implementation of <see cref="IClientProtocolConnectionFactory" />.</summary> |
| | | 12 | | public sealed class ClientProtocolConnectionFactory : IClientProtocolConnectionFactory |
| | | 13 | | { |
| | | 14 | | private readonly SslClientAuthenticationOptions? _clientAuthenticationOptions; |
| | | 15 | | private readonly ConnectionOptions _connectionOptions; |
| | | 16 | | private readonly IDuplexClientTransport _duplexClientTransport; |
| | | 17 | | private readonly DuplexConnectionOptions _duplexConnectionOptions; |
| | | 18 | | private readonly ILogger _logger; |
| | | 19 | | private readonly IMultiplexedClientTransport _multiplexedClientTransport; |
| | | 20 | | private readonly MultiplexedConnectionOptions _multiplexedConnectionOptions; |
| | | 21 | | |
| | | 22 | | /// <summary>Constructs a client protocol connection factory.</summary> |
| | | 23 | | /// <param name="connectionOptions">The connection options.</param> |
| | | 24 | | /// <param name="connectTimeout">The connect timeout.</param> |
| | | 25 | | /// <param name="clientAuthenticationOptions">The client authentication options.</param> |
| | | 26 | | /// <param name="duplexClientTransport">The duplex client transport. <see langword="null" /> is equivalent to <see |
| | | 27 | | /// cref="IDuplexClientTransport.Default" />.</param> |
| | | 28 | | /// <param name="multiplexedClientTransport">The multiplexed client transport. <see langword="null" /> is equivalent |
| | | 29 | | /// to <see cref="IMultiplexedClientTransport.Default" />.</param> |
| | | 30 | | /// <param name="logger">The logger. <see langword="null" /> is equivalent to <see cref="NullLogger.Instance" />. |
| | | 31 | | /// </param> |
| | 89 | 32 | | public ClientProtocolConnectionFactory( |
| | 89 | 33 | | ConnectionOptions connectionOptions, |
| | 89 | 34 | | TimeSpan connectTimeout, |
| | 89 | 35 | | SslClientAuthenticationOptions? clientAuthenticationOptions = null, |
| | 89 | 36 | | IDuplexClientTransport? duplexClientTransport = null, |
| | 89 | 37 | | IMultiplexedClientTransport? multiplexedClientTransport = null, |
| | 89 | 38 | | ILogger? logger = null) |
| | 89 | 39 | | { |
| | 89 | 40 | | _clientAuthenticationOptions = clientAuthenticationOptions; |
| | 89 | 41 | | _connectionOptions = connectionOptions; |
| | | 42 | | |
| | 89 | 43 | | _duplexClientTransport = duplexClientTransport ?? IDuplexClientTransport.Default; |
| | 89 | 44 | | _duplexConnectionOptions = new DuplexConnectionOptions |
| | 89 | 45 | | { |
| | 89 | 46 | | Pool = connectionOptions.Pool, |
| | 89 | 47 | | MinSegmentSize = connectionOptions.MinSegmentSize, |
| | 89 | 48 | | }; |
| | | 49 | | |
| | 89 | 50 | | _multiplexedClientTransport = multiplexedClientTransport ?? IMultiplexedClientTransport.Default; |
| | | 51 | | |
| | | 52 | | // If the dispatcher is null, we don't allow the peer to open streams for incoming requests. The only stream |
| | | 53 | | // which is accepted locally is the control stream created by the peer. |
| | 89 | 54 | | _multiplexedConnectionOptions = new MultiplexedConnectionOptions |
| | 89 | 55 | | { |
| | 89 | 56 | | HandshakeTimeout = connectTimeout, |
| | 89 | 57 | | |
| | 89 | 58 | | MaxBidirectionalStreams = connectionOptions.Dispatcher is null ? 0 : |
| | 89 | 59 | | connectionOptions.MaxIceRpcBidirectionalStreams, |
| | 89 | 60 | | |
| | 89 | 61 | | // Add an additional stream for the icerpc protocol control stream. |
| | 89 | 62 | | MaxUnidirectionalStreams = connectionOptions.Dispatcher is null ? 1 : |
| | 89 | 63 | | connectionOptions.MaxIceRpcUnidirectionalStreams + 1, |
| | 89 | 64 | | |
| | 89 | 65 | | Pool = connectionOptions.Pool, |
| | 89 | 66 | | MinSegmentSize = connectionOptions.MinSegmentSize, |
| | 89 | 67 | | }; |
| | | 68 | | |
| | 89 | 69 | | _logger = logger ?? NullLogger.Instance; |
| | 89 | 70 | | } |
| | | 71 | | |
| | | 72 | | /// <summary>Creates a protocol connection to the specified server address.</summary> |
| | | 73 | | /// <param name="serverAddress">The address of the server.</param> |
| | | 74 | | /// <returns>The new protocol connection.</returns> |
| | | 75 | | /// <remarks>The protocol connection returned by this factory method is not connected. The caller must call |
| | | 76 | | /// <see cref="IProtocolConnection.ConnectAsync" /> exactly once on this connection before calling |
| | | 77 | | /// <see cref="IInvoker.InvokeAsync" />.</remarks> |
| | | 78 | | public IProtocolConnection CreateConnection(ServerAddress serverAddress) |
| | 92 | 79 | | { |
| | 92 | 80 | | IProtocolConnection connection = |
| | 92 | 81 | | serverAddress.Protocol == Protocol.Ice ? |
| | 92 | 82 | | new IceProtocolConnection( |
| | 92 | 83 | | _duplexClientTransport.CreateConnection( |
| | 92 | 84 | | serverAddress, |
| | 92 | 85 | | _duplexConnectionOptions, |
| | 92 | 86 | | _clientAuthenticationOptions), |
| | 92 | 87 | | transportConnectionInformation: null, |
| | 92 | 88 | | _connectionOptions) : |
| | 92 | 89 | | new IceRpcProtocolConnection( |
| | 92 | 90 | | _multiplexedClientTransport.CreateConnection( |
| | 92 | 91 | | serverAddress, |
| | 92 | 92 | | _multiplexedConnectionOptions, |
| | 92 | 93 | | _clientAuthenticationOptions), |
| | 92 | 94 | | transportConnectionInformation: null, |
| | 92 | 95 | | _connectionOptions, |
| | 92 | 96 | | taskExceptionObserver: _logger == NullLogger.Instance ? null : |
| | 92 | 97 | | new LogTaskExceptionObserver(_logger)); |
| | | 98 | | |
| | 92 | 99 | | connection = new MetricsProtocolConnectionDecorator(connection, Metrics.ClientMetrics, connectStarted: false); |
| | | 100 | | |
| | 92 | 101 | | return _logger == NullLogger.Instance ? connection : |
| | 92 | 102 | | new LogProtocolConnectionDecorator(connection, serverAddress, remoteNetworkAddress: null, _logger); |
| | 92 | 103 | | } |
| | | 104 | | } |