在现代Web开发中，实时通信已经成为许多应用不可或缺的一部分。WebSocket作为一种支持全双工通信的协议，被广泛应用于需要实时数据交互的场景，如在线聊天、实时通知系统等。ASP.NET Core提供了强大的支持来实现WebSocket通信。本文将介绍在ASP.NET Core中实现WebSocket通信的两种主要方式。

一、通过自定义Middleware实现WebSocket

1. 创建WebSocket Middleware

首先，需要创建一个自定义的WebSocket Middleware来处理WebSocket请求。这个Middleware将检查传入的请求是否为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), WebSocketMessageType.Text, true, CancellationToken.None);
            }
            else if (result.MessageType == WebSocketMessageType.Close)
            {
                await webSocket.CloseAsync(result.CloseStatus.Value, result.CloseStatusDescription, CancellationToken.None);
                break;
            }
        }
    }
}

2. 注册WebSocket Middleware

在Startup.cs文件的Configure方法中注册上面创建的WebSocket Middleware。

public void Configure(IApplicationBuilder app, IWebHostEnvironment env)
{
    // ... 其他配置

    app.UseWebSockets(); // 必须先启用WebSocket支持
    app.UseMiddleware<WebSocketMiddleware>("/ws"); // 将Middleware映射到/ws路径

    app.UseRouting();
    app.UseEndpoints(endpoints =>
    {
        endpoints.MapControllers();
        // 其他路由配置
    });
}

二、通过SignalR实现WebSocket

虽然SignalR不是WebSocket的直接实现，但它内部封装了WebSocket，并提供了更高级别的抽象，使得开发者可以更容易地实现实时通信功能。

1. 安装SignalR包

首先，在项目中安装ASP.NET Core SignalR的NuGet包。

dotnet add package Microsoft.AspNetCore.SignalR

2. 创建Hub类

Hub类是SignalR的核心，它定义了服务器可以调用的方法以及客户端可以调用的方法。

public class ChatHub : Hub
{
    public async Task SendMessage(string user, string message)
    {
        await Clients.All.SendAsync("ReceiveMessage", user, message);
    }
}

3. 配置SignalR

在Startup.cs中配置SignalR路由。

public void Configure(IApplicationBuilder app, IWebHostEnvironment env)
{
    // ... 其他配置

    app.UseRouting();

    app.UseEndpoints(endpoints =>
    {
        endpoints.MapHub<ChatHub>("/chatHub"); // 将Hub映射到/chatHub路径
    });
}

4. 使用SignalR客户端连接Hub

在客户端（浏览器或移动应用），使用SignalR客户端库连接到Hub，并监听消息。

const connection = new signalR.HubConnectionBuilder()
    .withUrl("/chatHub")
    .build();

connection.on("ReceiveMessage", (user, message) => {
    console.log(`${user}: ${message}`);
});

connection.start().catch(err => console.error(err.toString()));

// 发送消息到服务器
connection.invoke("SendMessage", "Alice", "Hello everyone!");

三、总结

ASP.NET Core提供了两种主要方式来实现WebSocket通信：通过自定义Middleware直接操作WebSocket连接，以及使用SignalR来封装WebSocket并提供更高级别的抽象。开发者可以根据项目需求和个人偏好选择合适的方式。自定义Middleware提供了更多的控制力，而SignalR则简化了实时通信的实现过程。