Client Web Socket implementation (#46)

2026-07-24 12:06:27 +00:00 · 2023-05-14 20:46:38 +00:00
parent a25f2e5acb
commit de08c0e3b0
13 changed files with 234 additions and 107 deletions
@@ -0,0 +1,88 @@
+using Microsoft.Extensions.Logging;
+using System.Extensions;
+using System.Threading;
+using System.Threading.Tasks;
+
+namespace System.Net.WebSockets;
+
+// TODO: Cannot properly test because ClientWebSocket is sealed
+public sealed class ClientWebSocket<TScope> : IClientWebSocket<TScope>, IDisposable
+{
+    private readonly ILogger<TScope>? logger;
+    private readonly ClientWebSocket internalWebSocket = new();
+
+    public ClientWebSocketOptions Options => this.internalWebSocket.Options;
+
+    public WebSocketCloseStatus? CloseStatus => this.internalWebSocket.CloseStatus;
+
+    public string CloseStatusDescription => this.internalWebSocket.CloseStatusDescription;
+
+    public string SubProtocol => this.internalWebSocket.SubProtocol;
+
+    public WebSocketState State => this.internalWebSocket.State;
+
+    public ClientWebSocket()
+    {
+    }
+
+    public ClientWebSocket(ILogger<TScope> logger)
+    {
+        this.logger = logger.ThrowIfNull(nameof(logger));
+    }
+
+    public ClientWebSocket(ClientWebSocket clientWebSocket, ILogger<TScope> logger)
+    {
+        this.internalWebSocket = clientWebSocket.ThrowIfNull(nameof(clientWebSocket));
+        this.logger = logger.ThrowIfNull(nameof(logger));
+    }
+
+    public void Dispose()
+    {
+        this.internalWebSocket?.Dispose();
+    }
+
+    public void Abort()
+    {
+        this.internalWebSocket.Abort();
+    }
+
+    public Task CloseAsync(WebSocketCloseStatus closeStatus, string statusDescription, CancellationToken cancellationToken)
+    {
+        var scopedLogger = this.logger?.CreateScopedLogger(nameof(CloseAsync), string.Empty);
+        scopedLogger?.LogInformation($"Closing websocket. Status [{closeStatus}]. Status Description [{statusDescription}]");
+        return this.internalWebSocket.CloseAsync(closeStatus, statusDescription, cancellationToken);
+    }
+
+    public Task CloseOutputAsync(WebSocketCloseStatus closeStatus, string statusDescription, CancellationToken cancellationToken)
+    {
+        var scopedLogger = this.logger?.CreateScopedLogger(nameof(CloseOutputAsync), string.Empty);
+        scopedLogger?.LogInformation($"Closing output. Status [{closeStatus}]. Status Description [{statusDescription}]");
+        return this.internalWebSocket.CloseOutputAsync(closeStatus, statusDescription, cancellationToken);
+    }
+
+    public Task ConnectAsync(Uri uri, CancellationToken cancellationToken)
+    {
+        var scopedLogger = this.logger?.CreateScopedLogger(nameof(ConnectAsync), string.Empty);
+        scopedLogger?.LogInformation($"Connecting to {uri}");
+        return this.ConnectAsync(uri, cancellationToken);
+    }
+
+    public async Task<WebSocketReceiveResult> ReceiveAsync(ArraySegment<byte> buffer, CancellationToken cancellationToken)
+    {
+        var scopedLogger = this.logger?.CreateScopedLogger(nameof(ConnectAsync), string.Empty);
+        scopedLogger?.LogInformation($"Attempting to receive bytes");
+        var result = await this.ReceiveAsync(buffer, cancellationToken);
+        scopedLogger?.LogInformation($"Received message [{result.MessageType}]");
+        scopedLogger?.LogDebug($"Type: {result.MessageType}\nCount: {result.Count}\nEndOfMessage: {result.EndOfMessage}\nCloseStatus: {result.CloseStatus}\nCloseStatusDescription: {result.CloseStatusDescription}");
+
+        return result;
+    }
+
+    public Task SendAsync(ArraySegment<byte> buffer, WebSocketMessageType messageType, bool endOfMessage, CancellationToken cancellationToken)
+    {
+        var scopedLogger = this.logger?.CreateScopedLogger(nameof(SendAsync), string.Empty);
+        scopedLogger?.LogInformation($"Sending bytes");
+        scopedLogger?.LogDebug($"Type: {messageType}\nCount: {buffer.Count}\nEndOfMessage: {endOfMessage}");
+        return this.SendAsync(buffer, messageType, endOfMessage, cancellationToken);
+    }
+}
@@ -0,0 +1,31 @@
+using System.Threading.Tasks;
+using System.Threading;
+
+namespace System.Net.WebSockets;
+
+public interface IClientWebSocket<TScope>
+{
+    WebSocketCloseStatus? CloseStatus { get; }
+    
+    string CloseStatusDescription { get; }
+    
+    ClientWebSocketOptions Options { get; }
+    
+    WebSocketState State { get; }
+    
+    string SubProtocol { get; }
+
+    void Abort();
+
+    Task CloseAsync(WebSocketCloseStatus closeStatus, string statusDescription, CancellationToken cancellationToken);
+
+    Task CloseOutputAsync(WebSocketCloseStatus closeStatus, string statusDescription, CancellationToken cancellationToken);
+
+    Task ConnectAsync(Uri uri, CancellationToken cancellationToken);
+
+    void Dispose();
+
+    Task<WebSocketReceiveResult> ReceiveAsync(ArraySegment<byte> buffer, CancellationToken cancellationToken);
+
+    Task SendAsync(ArraySegment<byte> buffer, WebSocketMessageType messageType, bool endOfMessage, CancellationToken cancellationToken);
+}