C#实现WebSocket服务器：（03）消息收发的封装

Anlige

已于 2024-10-12 23:52:40 修改

阅读量6.7k

点赞数 11

分类专栏： WebSocket HTTP服务器文章标签： c# websocket http

于 2021-09-11 15:40:36 首次发布

本文链接：https://blog.youkuaiyun.com/moasp/article/details/120237666

版权

HTTP服务器同时被 2 个专栏收录

15 篇文章

订阅专栏

WebSocket

6 篇文章

订阅专栏

本文介绍了如何将WebSocket服务器的逻辑封装到一个名为Messager的抽象类中，包括客户端连接、断开、消息接收和发送等事件处理。通过创建自定义的Messager子类，可以方便地处理不同业务逻辑。此外，还展示了如何使用这个封装好的类进行测试，例如处理文本消息、二进制消息以及Close和Ping帧的交互。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

前文我们实现了WebSocket消息的读取和发送：C#实现WebSocket服务器：（02）消息帧分析和代码实现
但是我们所有的逻辑都是写在OnWebSocket方法的，并不方便扩展，现在我们封装一个Messager抽象类，封装消息的读取和发送。

Messager类封装

我们将在类中暴露一些必要的方法。
下面我们用消息这个词，而不是帧，因为一条消息可能是由多个帧(例如Text帧+N个Continuation帧)组成的。

方法	说明
Accept	在OnWebSocket方法中，调用具体Messager的Accept方法，开始接收客户端的消息
OnConnected	客户端新连接
OnDisconnected	客户端断开连接
OnNewFrame	服务器收到新帧，主要用于调试了展示用，尽量不要在这里处理业务逻辑
OnText	服务器收到文本消息
OnBinary	服务器收到二进制消息
Close	关闭WebSocket，服务端会向客户端发送Close帧，服务端收到客户端的Close响应后，真正关闭连接
Send	发送消息给客户端，实现了多个重载，用来发送不同类型的消息
Ping	发送Ping帧给客户端，客户端会响应一个Pong帧给服务器
其他方法和帧处理我们都不暴露出来了，程序内部会自动处理。
例如：服务器收到Ping帧会自动响应Pong帧给客户端、Close帧也会自动处理。
Github上有具体的代码实现：https://github.com/sometiny/http/blob/main/src/WebSocket/Messager.cs
我们就不拷贝过来了，占篇幅。

Messager类的使用

Messager类为抽象类，不能直接实例化，需要我们实现自己的的Messager类，继承Messager抽象类即可。
下面实现了一个Messager用来作测试用。
这里也可以找到源码：https://github.com/sometiny/http/blob/main/demo/WebSocketMessager.cs

/// <summary>
/// 实现一个Messager类
/// </summary>
public class MyMessager : Messager
{
    private EndPoint _remoteEndPoint = null;
    public MyMessager(Stream stream) : base(stream) {

        //获取客户端的连接信息
        if(stream is BufferedNetworkStream networkStream)
        {
            _remoteEndPoint = networkStream.BaseSocket.RemoteEndPoint ;
        }
    }

    /// <summary>
    /// 客户端新连接
    /// </summary>
    protected override void OnConnected()
    {
        Console.WriteLine($"{DateTime.Now:HH:mm:ss} > 新客户端连接：{_remoteEndPoint}");
    }

    /// <summary>
    /// 客户端断开连接
    /// </summary>
    protected override void OnDisconnected()
    {
        Console.WriteLine($"{DateTime.Now:HH:mm:ss} > 连接断开：{_remoteEndPoint}");
    }

    /// <summary>
    /// 接收到新的帧，仅展示下
    /// </summary>
    /// <param name="frame"></param>
    protected override void OnNewFrame(Frame frame)
    {
        var color = Console.ForegroundColor;
        Console.ForegroundColor = ConsoleColor.DarkYellow;
        Console.WriteLine($"{DateTime.Now:HH:mm:ss} > 接收到新帧；帧类型：{frame.OpCode}，结束帧：{frame.Fin}，携带掩码：{frame.Mask}，长度：{frame.PayloadLength}");
        Console.ForegroundColor = color;
    }

    /// <summary>
    /// 收到Text消息时的实现
    /// 里面定义两个特使的消息：close和ping，用来测试服务器主动发送Close和Ping帧。
    /// </summary>
    /// <param name="payload">完整消息</param>
    protected override void OnText(string payload)
    {
        Console.WriteLine($"{DateTime.Now:HH:mm:ss} > 文本数据：{payload}");
        if(payload == "close")
        {
            Send($"服务器接收到close指令，关闭连接。");
            Close();
            return;
        }
        if (payload == "ping")
        {
            Send($"服务器接收到ping指令，发送ping。");
            Ping();
            return;
        }
        Send($"服务器接收到文本数据：{payload}");
    }

    /// <summary>
    /// 接收到二进制消息
    /// 二进制消息是通过流来读取的，不像Text，直接一股脑读取全部消息。
    /// </summary>
    /// <param name="inputStream">输入流</param>
    protected override void OnBinary(Stream inputStream)
    {
        //为了测试，我们把二进制消息读取到字节数组。
        byte[] payload = StreamUtils.ReadAllBytes(inputStream);
        Console.WriteLine($"{DateTime.Now:HH:mm:ss} > 二进制数据，长度：{payload.Length}");
        Send($"服务器接收到二进制数据，长度：{payload.Length}");
    }
}

测试

我们使用上面实现的MyMessager进行测试。
实例化一个服务器，使用MyMessager来接收和发送消息。

public class HttpServer : HttpServerBase
{
    public HttpServer() : base()
    {
        //设置根目录
        WebRoot = Path.GetFullPath(Path.Combine(AppDomain.CurrentDomain.BaseDirectory, "web"));
    }
    protected override void OnWebSocket(HttpRequest request, Stream stream)
    {
        //使用自己实现的Messager类处理业务。
        //可以根据请求的Url不同去实例化不同的Messager，处理不同的业务逻辑。
        new MyMessager(stream).Accept();
    }
}