Asp.Net Core应用程序WebSocket(客户端/服务器)
创始人
2024-09-16 02:00:26
0

以下是一个使用Asp.Net Core的WebSocket的简单示例:

首先,在Startup.cs文件中,添加以下代码以启用WebSocket:

public void ConfigureServices(IServiceCollection services)
{
    services.AddWebSocketManager();
    // 添加其他服务配置
}

public void Configure(IApplicationBuilder app, IWebHostEnvironment env)
{
    app.UseWebSockets();
    app.UseWebSocketServer();
    // 添加其他中间件配置
}

然后,创建一个WebSocket中间件WebSocketServerMiddleware.cs:

public class WebSocketServerMiddleware
{
    private readonly RequestDelegate _next;
    private readonly IWebSocketManager _webSocketManager;

    public WebSocketServerMiddleware(RequestDelegate next, IWebSocketManager webSocketManager)
    {
        _next = next;
        _webSocketManager = webSocketManager;
    }

    public async Task Invoke(HttpContext context)
    {
        if (context.WebSockets.IsWebSocketRequest)
        {
            WebSocket webSocket = await context.WebSockets.AcceptWebSocketAsync();

            // 处理WebSocket连接
            var socketId = _webSocketManager.AddSocket(webSocket);
            await _webSocketManager.HandleConnectionAsync(socketId);

            await Receive(webSocket, async (result, buffer) =>
            {
                if (result.MessageType == WebSocketMessageType.Text)
                {
                    await _webSocketManager.HandleMessageAsync(socketId, result, buffer);
                    return;
                }

                if (result.MessageType == WebSocketMessageType.Close)
                {
                    await _webSocketManager.HandleDisconnectedAsync(socketId);
                    return;
                }
            });
        }
        else
        {
            await _next(context);
        }
    }

    private async Task Receive(WebSocket webSocket, Action handleMessage)
    {
        var buffer = new byte[1024 * 4];

        while (webSocket.State == WebSocketState.Open)
        {
            var result = await webSocket.ReceiveAsync(buffer, CancellationToken.None);
            handleMessage(result, buffer);
        }
    }
}

接下来,创建一个WebSocket管理器WebSocketManager.cs:

public interface IWebSocketManager
{
    string AddSocket(WebSocket socket);
    Task HandleConnectionAsync(string socketId);
    Task HandleMessageAsync(string socketId, WebSocketReceiveResult result, byte[] buffer);
    Task HandleDisconnectedAsync(string socketId);
}

public class WebSocketManager : IWebSocketManager
{
    private ConcurrentDictionary _sockets = new ConcurrentDictionary();

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

    public async Task HandleConnectionAsync(string socketId)
    {
        // 处理新的WebSocket连接
        // 可以向客户端发送初始消息
        await SendMessageAsync(socketId, "Connected");
    }

    public async Task HandleMessageAsync(string socketId, WebSocketReceiveResult result, byte[] buffer)
    {
        // 处理接收到的消息
        var message = Encoding.UTF8.GetString(buffer, 0, result.Count);
        await SendMessageAsync(socketId, $"Received: {message}");
    }

    public async Task HandleDisconnectedAsync(string socketId)
    {
        // 处理WebSocket断开连接
        _sockets.TryRemove(socketId, out var socket);
        await socket.CloseAsync(WebSocketCloseStatus.NormalClosure, "Closed by the server", CancellationToken.None);
    }

    private async Task SendMessageAsync(string socketId, string message)
    {
        if (_sockets.TryGetValue(socketId, out var socket))
        {
            var buffer = Encoding.UTF8.GetBytes(message);
            await socket.SendAsync(buffer, WebSocketMessageType.Text, true, CancellationToken.None);
        }
    }
}

最后,在Startup.cs文件中添加WebSocket服务的扩展方法:

public static class WebSocketExtensions
{
    public static IServiceCollection AddWebSocketManager(this IServiceCollection services)
    {
        services.AddSingleton();
        return services;
    }

    public static IApplicationBuilder UseWebSocketServer(this IApplicationBuilder app)
    {
        app.UseMiddleware();
        return app;
    }
}

现在,你可以在控制器或其他地方注入IWebSocketManager,并使用它进行WebSocket通信。例如,在控制器中:

[ApiController]
[Route("api/[controller]")]
public class WebSocketController : ControllerBase
{
    private readonly IWebSocketManager _webSocketManager;

    public WebSocketController(IWebSocketManager webSocketManager)
    {
        _webSocketManager = webSocketManager;
    }

    [HttpGet]
    public async Task Get()
    {
        // 向所有连接的客户端发送消息

相关内容

热门资讯

记者发布!pokemmo修改器... 记者发布!pokemmo修改器手机版,wepoker私人辅助器,总是真的是有挂(有挂助手)1、操作简...
实测分享!德普之星透视辅助软件... 实测分享!德普之星透视辅助软件是真的吗,德州hhpoker脚本,真是是有挂(有挂分享)1、下载好脚本...
揭秘真相!wepoker辅助器... 揭秘真相!wepoker辅助器官方,we poker辅助器,本来真的有挂(有挂总结)1、wepoke...
科技通报!wepoker怎么增... 科技通报!wepoker怎么增加运气,HH平台挂,一贯真的有挂(有挂细节)亲,关键说明,透视脚本安卓...
技术分享!hhpoker有没有... 技术分享!hhpoker有没有辅助,hhpoker脚本,真是真的有挂(有挂方式)1、很好的工具软件,...
一分钟揭秘!pokemmo手机... 一分钟揭秘!pokemmo手机版修改器,大菠萝辅助器,原来真的是有挂(有挂总结)1、让任何用户在无需...
9分钟了解!wepoker透视... 9分钟了解!wepoker透视苹果系统,wepoker辅助透视软件,果然存在有挂(有挂教学)1、we...
免费测试版!werplan辅助... 免费测试版!werplan辅助软件,德普之星透视辅助软件,确实是有挂(真的有挂)辅助器是一种具有地方...
玩家实测!wepoker有辅助... 玩家实测!wepoker有辅助插件吗,wepoker透视脚本是什么,切实真的是有挂(有挂总结)1、打...
科普分享!wpk可以作弊吗,h... 科普分享!wpk可以作弊吗,hhpoker德州有挂吗,原来是有挂(有人有挂)1、进入到是否有挂之后,...