如何使用WebSockets跟踪已建立的连接

本文关键字：建立 连接 跟踪 何使用 WebSockets | 更新日期: 2023-09-27 18:26:01

我正在尝试为跨平台设备提供实时聊天服务。问题是System.Net.WebSockets命名空间不允许我直接跟踪已建立的连接。我可以获取当前连接的sessionID，但我怎么能说对特定客户端执行以下操作await socket.SendAsync(buffer, WebSocketMessageType.Text, CancellationToken.None)呢？

如果我能使用Microsoft.WebSockets，我就有可能创建一个WebSocketCollection()并做一些类似client.Send(message)的事情，但我不能通过它发送ArraySegment<byte[]>。我还认为这更适合AJAX客户端和网站等。

我现在有以下代码片段：

public class WSHandler : IHttpHandler
{
    event NewConnectionEventHandler NewConnection;
    public void ProcessRequest(HttpContext context)
    {
        if (context.IsWebSocketRequest)
        {
            context.AcceptWebSocketRequest(ProcessWSChat);
        }
    }
    public bool IsReusable { get { return false; } }
    private async Task ProcessWSChat(AspNetWebSocketContext context)
    {
        WebSocket socket = context.WebSocket;
        int myHash = socket.GetHashCode();
        while (true)
        {
        ArraySegment<byte> buffer = new ArraySegment<byte>(new byte[1024]);
        WebSocketReceiveResult result = await socket.ReceiveAsync(
            buffer, CancellationToken.None);
            if (socket.State == WebSocketState.Open)
            {
                string userMessage = Encoding.ASCII.GetString(
                    buffer.Array, 0, result.Count);
                userMessage = "You sent: " + userMessage + " at " +
                    DateTime.Now.ToLongTimeString() + " from ip " +
                    context.UserHostAddress.ToString();
                buffer = new ArraySegment<byte>(
                    Encoding.ASCII.GetBytes(userMessage));
                await socket.SendAsync(
                    buffer, WebSocketMessageType.Text, true, CancellationToken.None);
            }
            else
            {
                break;
            }
        }
    }
}

如何扩展我的项目、保存会话/连接并调用特定的用户连接以便向其发送消息？

如何使用WebSockets跟踪已建立的连接

通常，您不会做这样的事情。WebSocket应该只是另一层或子系统的端点。例如，事件驱动的体系结构。在连接时，您应该启动处理程序，从连接的套接字（如cookie、URL或其他）收集数据，并通知其他用户，具有该cookie/URL的用户已连接到该套接字中。

关于您的代码：

  • 不要在循环中声明读取缓冲区，您可以重用它
  • WebSocket总是以UTF8而不是ASCII的形式发送文本数据

这将是在一个小项目中跟踪用户的一种方式，但我建议您将WebSocket插入另一个消息基础设施，如MassTransit:

public class WSHandler : IHttpHandler
{
    public bool IsReusable { get { return false; } }
    public void ProcessRequest(HttpContext context)
    {
        if (context.IsWebSocketRequest)
        {
            context.AcceptWebSocketRequest(ProcessWSChat);
        }
    }
    static readonly ConcurrentDictionary<IPrincipal, WebSocket> _users = new ConcurrentDictionary<IPrincipal, WebSocket>();
    private async Task ProcessWSChat(AspNetWebSocketContext context)
    {
        WebSocket socket = context.WebSocket;
        ArraySegment<byte> buffer = new ArraySegment<byte>(new byte[4096]);
        //Identify user by cookie or whatever and create a user Object
        var cookie = context.Cookies["mycookie"];
        var url = context.RequestUri;
        IPrincipal myUser = GetUser(cookie, url);
        // Or uses the user that came from the ASP.NET authentication.
        myUser = context.User;
        _users.AddOrUpdate(myUser, socket, (p, w) => socket);
        while (socket.State == WebSocketState.Open)
        {
            WebSocketReceiveResult result = await socket.ReceiveAsync(buffer, CancellationToken.None)
                                                        .ConfigureAwait(false);
            String userMessage = Encoding.UTF8.GetString(buffer.Array, 0, result.Count);
            userMessage = "You sent: " + userMessage + " at " +
                DateTime.Now.ToLongTimeString() + " from ip " + context.UserHostAddress.ToString();
            var sendbuffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(userMessage));
            await socket.SendAsync(sendbuffer , WebSocketMessageType.Text, true, CancellationToken.None)
                        .ConfigureAwait(false);
        }
        // when the connection ends, try to remove the user
        WebSocket ows;
        if (_users.TryRemove(myUser, out ows))
        {
            if (ows != socket)
            {
                // whops! user reconnected too fast and you are removing
                // the new connection, put it back
                _users.AddOrUpdate(myUser, ows, (p, w) => ows);
            }
        }
    }
    private IPrincipal GetUser(HttpCookie cookie, Uri url)
    {
        throw new NotImplementedException();
    }
}

我想重新发明轮子会很愚蠢，所以我最好使用SignalR。

感谢@Jonesy让我了解这个Microsoft库！

编辑：

SignalR在非"IIS Express"模式下无法正常工作。"本地IIS"导致整个项目失败。

编辑：

SignalR现在工作。重写规则使得无法获得正确的路径。

相关文章：