asp.net core WebSocket的使用

ASP.NET Core中使用WebSocket可以实现双向通信,以下是ASP.NET Core WebSocket的使用步骤:

创建WebSocketMiddleware

创建一个自定义的WebSocketMiddleware来处理WebSocket请求,代码如下所示:

public class WebSocketMiddleware
{
    private readonly RequestDelegate _next;
    public WebSocketMiddleware(RequestDelegate next)
    {
        _next = next;
    }
    public async Task Invoke(HttpContext context)
    {
        if (context.WebSockets.IsWebSocketRequest)
        {
            WebSocket webSocket = await context.WebSockets.AcceptWebSocketAsync();
            await EchoAsync(context, webSocket);
        }
        else
        {
            await _next(context);
        }
    }
    private async Task EchoAsync(HttpContext context, WebSocket webSocket)
    {
        var buffer = new byte[1024 * 4];
        while (webSocket.State == WebSocketState.Open)
        {
            var result = await webSocket.ReceiveAsync(new ArraySegment<byte>(buffer), CancellationToken.None);
            if (result.MessageType == WebSocketMessageType.Text)
            {
                var incomingMessage = Encoding.UTF8.GetString(buffer, 0, result.Count);
                var outgoingMessage = "You sent " + incomingMessage + " at " + DateTime.Now.ToLongTimeString();
                var outgoingBuffer = Encoding.UTF8.GetBytes(outgoingMessage);
                await webSocket.SendAsync(new ArraySegment<byte>(outgoingBuffer, 0, outgoingBuffer.Length), result.MessageType, result.EndOfMessage, CancellationToken.None);
            }
            else if (result.MessageType == WebSocketMessageType.Close)
            {
                await webSocket.CloseAsync(result.CloseStatus.Value, result.CloseStatusDescription, CancellationToken.None);
            }
        }
    }
}

注册WebSocketMiddleware

在Startup.cs文件的Configure方法中注册WebSocketMiddleware,代码如下所示:

public void Configure(IApplicationBuilder app, IWebHostEnvironment env)
{
    app.UseRouting();
    app.UseEndpoints(endpoints =>
    {
        endpoints.MapGet("/", async context =>
        {
            await context.Response.WriteAsync("Hello World!");
        });
        endpoints.MapMiddleware<WebSocketMiddleware>("/ws");
    });
}

连接WebSocket

使用JavaScript代码连接WebSocket,代码如下所示:

var ws = new WebSocket("ws://" + location.host + "/ws");
ws.onopen = function () {
    console.log("WebSocket connection opened.");
};
ws.onmessage = function (event) {
    console.log("Received message: " + event.data);
};
ws.onclose = function () {
    console.log("WebSocket connection closed.");
};
ws.send("Hello, WebSocket!");

以上是ASP.NET Core中使用WebSocket的简单示例,通过WebSocket可以实现双向通信,可以在实时通讯、在线游戏等领域中得到广泛应用。

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
好的,下面是一个简单的使用 .NET 6 的 WebSocket 实现聊天室功能的代码示例: 服务端代码: ``` csharp using Microsoft.AspNetCore.Builder; using Microsoft.AspNetCore.Http; using Microsoft.Extensions.Logging; using System.Collections.Concurrent; using System.Net.WebSockets; using System.Text; using System.Threading; using System.Threading.Tasks; public class ChatRoom { private readonly ConcurrentDictionary<string, WebSocket> _sockets = new ConcurrentDictionary<string, WebSocket>(); private readonly ILogger<ChatRoom> _logger; public ChatRoom(ILogger<ChatRoom> logger) { _logger = logger; } public async Task HandleWebSocketAsync(HttpContext context) { if (!context.WebSockets.IsWebSocketRequest) { context.Response.StatusCode = 400; return; } WebSocket webSocket = await context.WebSockets.AcceptWebSocketAsync(); string socketId = Guid.NewGuid().ToString(); _sockets.TryAdd(socketId, webSocket); _logger.LogInformation($"WebSocket {socketId} connected"); try { await HandleIncomingMessageAsync(webSocket, socketId); } finally { _sockets.TryRemove(socketId, out _); await webSocket.CloseAsync(WebSocketCloseStatus.NormalClosure, "WebSocket closed", CancellationToken.None); _logger.LogInformation($"WebSocket {socketId} disconnected"); } } private async Task HandleIncomingMessageAsync(WebSocket webSocket, string socketId) { byte[] buffer = new byte[1024 * 4]; while (webSocket.State == WebSocketState.Open) { WebSocketReceiveResult result = await webSocket.ReceiveAsync(new ArraySegment<byte>(buffer), CancellationToken.None); if (result.MessageType == WebSocketMessageType.Close) { break; } string message = Encoding.UTF8.GetString(buffer, 0, result.Count); _logger.LogInformation($"WebSocket {socketId} received message: {message}"); await BroadcastMessageAsync(message, socketId); } } private async Task BroadcastMessageAsync(string message, string excludeSocketId) { foreach (KeyValuePair<string, WebSocket> pair in _sockets) { if (pair.Key != excludeSocketId && pair.Value.State == WebSocketState.Open) { await pair.Value.SendAsync(new ArraySegment<byte>(Encoding.UTF8.GetBytes(message)), WebSocketMessageType.Text, true, CancellationToken.None); } } } } public static class ChatRoomExtensions { public static IApplicationBuilder UseChatRoom(this IApplicationBuilder app) { return app.Use(async (context, next) => { if (context.Request.Path == "/chat") { ChatRoom chatRoom = context.RequestServices.GetService(typeof(ChatRoom)) as ChatRoom; await chatRoom.HandleWebSocketAsync(context); } else { await next(); } }); } } ``` 客户端代码: ``` csharp using System; using System.Net.WebSockets; using System.Text; using System.Threading; using System.Threading.Tasks; public class ChatClient : IDisposable { private readonly ClientWebSocket _webSocket = new ClientWebSocket(); public async Task ConnectAsync(Uri uri) { await _webSocket.ConnectAsync(uri, CancellationToken.None); Console.WriteLine("Connected to WebSocket server"); Task receiveTask = ReceiveAsync(); while (_webSocket.State == WebSocketState.Open) { string message = Console.ReadLine(); byte[] buffer = Encoding.UTF8.GetBytes(message); await _webSocket.SendAsync(new ArraySegment<byte>(buffer), WebSocketMessageType.Text, true, CancellationToken.None); } await receiveTask; } private async Task ReceiveAsync() { byte[] buffer = new byte[1024 * 4]; while (_webSocket.State == WebSocketState.Open) { WebSocketReceiveResult result = await _webSocket.ReceiveAsync(new ArraySegment<byte>(buffer), CancellationToken.None); if (result.MessageType == WebSocketMessageType.Close) { break; } string message = Encoding.UTF8.GetString(buffer, 0, result.Count); Console.WriteLine(message); } } public async Task DisconnectAsync() { await _webSocket.CloseAsync(WebSocketCloseStatus.NormalClosure, "WebSocket closed", CancellationToken.None); Console.WriteLine("Disconnected from WebSocket server"); } public void Dispose() { _webSocket.Dispose(); } } ``` 使用方法: 在 ASP.NET Core 应用程序的 `Startup.cs` 文件中调用 `UseChatRoom()` 扩展方法,将 `/chat` 路径映射到 `ChatRoom` 类: ``` csharp app.UseChatRoom(); ``` 在客户端代码中创建 `ChatClient` 实例,然后调用 `ConnectAsync()` 方法连接到 WebSocket 服务器: ``` csharp ChatClient client = new ChatClient(); await client.ConnectAsync(new Uri("ws://localhost:5000/chat")); ``` 当用户在控制台中输入文本时,调用 `SendAsync()` 方法将文本发送到服务器: ``` csharp string message = Console.ReadLine(); await client.SendAsync(message); ``` 当用户想要断开连接时,调用 `DisconnectAsync()` 方法断开连接: ``` csharp await client.DisconnectAsync(); ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值