官方学习资料:WebSocket服务器。
本文开发环境:Win10 + VS2019 + .NetCore 3.1 + SuperSocket.WebSocket.Server 2.0.0-beta.8。
Gitee:SuperSocketV2Sample。
WebSocket服务器使用WebSocketHostBuilder类创建,可使用链式函数设置消息回调处理、WebSocket参数配置等。如果要添加自定义命令,那么需要把WebSocketPackage转换为StringPackageInfo,Demo中的StringPackageConverter就是实现这个转换的类。
需要注意的是,命令类继承自IAsyncCommand<WebSocketSession, StringPackageInfo>,和之前的基类IAsyncCommand<StringPackageInfo>有所不同。命令类的实现请参考Gitee项目源码。
此外,如果使用UseCommand注册了命令,无论UseCommand回调方法中是否注册命令,UseWebSocketMessageHandler方法都不会被调用。
1、StringPackageConverter
using System; using System.Linq; using SuperSocket.Command; using SuperSocket.ProtoBase; using SuperSocket.WebSocket; namespace SuperSocketV2Sample.WebSocketServer.Server { /// <summary> /// WebSocketPackage转换器 /// </summary> public class StringPackageConverter : IPackageMapper<WebSocketPackage, StringPackageInfo> { public StringPackageInfo Map(WebSocketPackage package) { var packInfo = new StringPackageInfo(); var array = package.Message.Split(‘ ‘, 3, StringSplitOptions.RemoveEmptyEntries); packInfo.Key = array[0]; packInfo.Parameters = array.Skip(1).ToArray(); packInfo.Body = string.Join(‘ ‘, packInfo.Parameters); return packInfo; } } }
2、Program
using System; using System.Collections.Generic; using System.Threading.Tasks; using Microsoft.Extensions.Configuration; using Microsoft.Extensions.Hosting; using Microsoft.Extensions.Logging; using SuperSocket.Command; using SuperSocket.ProtoBase; using SuperSocket.WebSocket.Server; using SuperSocketV2Sample.WebSocketServer.Commands; using SuperSocketV2Sample.WebSocketServer.Server; namespace SuperSocketV2Sample.WebSocketServer { class Program { static async Task Main() { var host = WebSocketHostBuilder.Create() //注册WebSocket消息处理器(使用UseCommand注册命令之后该处理器不起作用) .UseWebSocketMessageHandler(async (session, package) => { Console.WriteLine($@"{DateTime.Now:yyyy-MM-dd HH:mm:ss fff} Receive message: {package.Message}."); //Send message back var message = $@"{DateTime.Now:yyyy-MM-dd HH:mm:ss fff} Hello from WebSocket Server: {package.Message}."; await session.SendAsync(message); }) .UseCommand<StringPackageInfo, StringPackageConverter>(commandOptions => { //注册命令 commandOptions.AddCommand<AddCommand>(); commandOptions.AddCommand<DivCommand>(); commandOptions.AddCommand<MulCommand>(); commandOptions.AddCommand<SubCommand>(); commandOptions.AddCommand<EchoCommand>(); }) .ConfigureAppConfiguration((hostCtx, configApp) => { configApp.AddInMemoryCollection(new Dictionary<string, string> { {"serverOptions:name", "TestServer"}, {"serverOptions:listeners:0:ip", "Any"}, {"serverOptions:listeners:0:port", "4040"} }); }) .ConfigureLogging((hostCtx, loggingBuilder) => { //添加控制台输出 loggingBuilder.AddConsole(); }) .Build(); await host.RunAsync(); } } }