在Asp.net core中实现websocket通信

在ASP.NET Core中实现WebSocket通信的完整攻略如下：

步骤一：创建ASP.NET Core Web应用程序

首先，我们需要创建一个ASP.NET Core Web应用程序。可以使用Visual Studio或者命令行工具创建一个新的ASP.NET Core Web应用程序。

步骤二：添加WebSocket中间件

在ASP.NET Core中，可以使用WebSocket中间件来处理WebSocket通信。要添加WebSocket中间件，请在Startup.cs文件中的Configure方法中添加以下代码：

app.UseWebSockets();

步骤三：处理WebSocket请求

要处理WebSocket请求，请创建一个WebSocket处理程序。可以使用ASP.NET Core中的WebSocketHandler类来创建WebSocket处理程序。以下是一个示例WebSocket处理程序：

public class MyWebSocketHandler : WebSocketHandler
{
    public MyWebSocketHandler(WebSocketConnectionManager webSocketConnectionManager) : base(webSocketConnectionManager)
    {
    }

    public override async Task ReceiveAsync(WebSocket socket, WebSocketReceiveResult result, byte[] buffer)
    {
        string message = Encoding.UTF8.GetString(buffer, 0, result.Count);
        await SendMessageToAllAsync(message);
    }
}

在上面的代码中，我们创建了一个名为MyWebSocketHandler的WebSocket处理程序，并覆盖了ReceiveAsync方法来处理接收到的消息。在这个示例中，我们将接收到的消息发送给所有连接的WebSocket客户端。

步骤四：管理WebSocket连接

要管理WebSocket连接，请创建一个WebSocket连接管理器。可以使用以下代码创建WebSocket连接管理器：

public class WebSocketConnectionManager
{
    private readonly ConcurrentDictionary<string, WebSocket> _sockets = new ConcurrentDictionary<string, WebSocket>();

    public WebSocket GetSocketById(string id)
    {
        return _sockets.TryGetValue(id, out WebSocket socket) ? socket : null;
    }

    public ConcurrentDictionary<string, WebSocket> GetAllSockets()
    {
        return _sockets;
    }

    public string AddSocket(WebSocket socket)
    {
        string socketId = Guid.NewGuid().ToString();
        _sockets.TryAdd(socketId, socket);
        return socketId;
    }

    public async Task RemoveSocketAsync(string id)
    {
        WebSocket socket = GetSocketById(id);
        if (socket != null)
        {
            await socket.CloseAsync(WebSocketCloseStatus.NormalClosure, "Connection closed", CancellationToken.None);
            _sockets.TryRemove(id, out WebSocket removedSocket);
        }
    }
}

在上面的代码中，我们创建了一个名为WebSocketConnectionManager的WebSocket连接管理器，并使用ConcurrentDictionary来存储WebSocket连接。我们还提供了一些方法来添加、删除和获取WebSocket连接。

步骤五：将WebSocket处理程序添加到DI容器

要将WebSocket处理程序添加到DI容器中，请在Startup.cs文件中的ConfigureServices方法中添加以下代码：

services.AddSingleton<WebSocketConnectionManager>();
services.AddWebSocketManager();
services.AddTransient<MyWebSocketHandler>();

在上面的代码中，我们将WebSocket连接管理器和WebSocket处理程序添加到DI容器中。

步骤六：创建WebSocket控制器

要创建WebSocket控制器，请创建一个继承自Controller类的控制器，并使用以下代码添加WebSocket处理程序：

public class WebSocketController : Controller
{
    private readonly WebSocketConnectionManager _webSocketConnectionManager;
    private readonly MyWebSocketHandler _webSocketHandler;

    public WebSocketController(WebSocketConnectionManager webSocketConnectionManager, MyWebSocketHandler webSocketHandler)
    {
        _webSocketConnectionManager = webSocketConnectionManager;
        _webSocketHandler = webSocketHandler;
    }

    [HttpGet("/ws")]
    public async Task<IActionResult> Get()
    {
        if (HttpContext.WebSockets.IsWebSocketRequest)
        {
            WebSocket webSocket = await HttpContext.WebSockets.AcceptWebSocketAsync();
            string socketId = _webSocketConnectionManager.AddSocket(webSocket);
            await _webSocketHandler.OnConnectedAsync(webSocket);
            await ReceiveAsync(webSocket, async (result, buffer) =>
            {
                if (result.MessageType == WebSocketMessageType.Text)
                {
                    await _webSocketHandler.ReceiveAsync(webSocket, result, buffer);
                }
                else if (result.MessageType == WebSocketMessageType.Close)
                {
                    await _webSocketHandler.OnDisconnectedAsync(webSocket);
                }
            });
            await _webSocketConnectionManager.RemoveSocketAsync(socketId);
        }

        return new BadRequestResult();
    }

    private async Task ReceiveAsync(WebSocket socket, Action<WebSocketReceiveResult, byte[]> handleMessage)
    {
        byte[] buffer = new byte[1024 * 4];
        while (socket.State == WebSocketState.Open)
        {
            WebSocketReceiveResult result = await socket.ReceiveAsync(new ArraySegment<byte>(buffer), CancellationToken.None);
            handleMessage(result, buffer);
        }
    }
}

在上面的代码中，我们创建了一个名为WebSocketController的控制器，并使用HttpGet特性将其映射到/ws路由。在Get方法中，我们检查请求是否为WebSocket请求，并将WebSocket连接添加到WebSocket连接管理器中。然后，我们使用ReceiveAsync方法来处理接收到的消息，并在WebSocket连接关闭时将其从WebSocket连接管理器中删除。

示例一：使用JavaScript创建WebSocket客户端

以下是一个使用JavaScript创建WebSocket客户端的示例：

var socket = new WebSocket("ws://localhost:5000/ws");

socket.onopen = function (event) {
    console.log("WebSocket connected");
};

socket.onmessage = function (event) {
    console.log("Received message: " + event.data);
};

socket.onclose = function (event) {
    console.log("WebSocket disconnected");
};

socket.send("Hello, WebSocket!");

在上面的代码中，我们创建了一个名为socket的WebSocket客户端，并使用onopen、onmessage和onclose事件处理程序来处理WebSocket连接的状态和接收到的消息。我们还使用send方法向WebSocket服务器发送消息。

示例二：使用C#创建WebSocket客户端

以下是一个使用C#创建WebSocket客户端的示例：

using (ClientWebSocket clientWebSocket = new ClientWebSocket())
{
    await clientWebSocket.ConnectAsync(new Uri("ws://localhost:5000/ws"), CancellationToken.None);

    byte[] buffer = new byte[1024];
    while (true)
    {
        WebSocketReceiveResult result = await clientWebSocket.ReceiveAsync(new ArraySegment<byte>(buffer), CancellationToken.None);
        string message = Encoding.UTF8.GetString(buffer, 0, result.Count);
        Console.WriteLine("Received message: " + message);
    }
}

在上面的代码中，我们使用ClientWebSocket类创建了一个名为clientWebSocket的WebSocket客户端，并使用ConnectAsync方法连接到WebSocket服务器。然后，我们使用ReceiveAsync方法来处理接收到的消息。