温馨提示×

c# websocket服务器搭建的方法是什么

小亿
110
2024-01-08 21:34:28
栏目: 云计算

在C#中搭建WebSocket服务器的方法是通过使用WebSocket类和相关的命名空间来实现。

以下是一个简单的例子:

  1. 首先,确保你的项目引用了System.Net.WebSockets命名空间。

  2. 创建一个WebSocket服务器端的类,并在其中初始化WebSocket对象。例如:

using System;
using System.Net;
using System.Net.WebSockets;
using System.Threading;
using System.Threading.Tasks;

public class MyWebSocketServer
{
    private HttpListener _httpListener;
    private CancellationTokenSource _cts;

    public MyWebSocketServer(string url)
    {
        _httpListener = new HttpListener();
        _httpListener.Prefixes.Add(url);
        _cts = new CancellationTokenSource();
    }

    public async Task Start()
    {
        _httpListener.Start();
        Console.WriteLine("WebSocket server started.");

        while (!_cts.Token.IsCancellationRequested)
        {
            HttpListenerContext context = await _httpListener.GetContextAsync();
            if (context.Request.IsWebSocketRequest)
            {
                ProcessWebSocketRequest(context);
            }
            else
            {
                // Handle non-WebSocket request
                context.Response.StatusCode = 400;
                context.Response.Close();
            }
        }
    }

    private async void ProcessWebSocketRequest(HttpListenerContext context)
    {
        HttpListenerWebSocketContext webSocketContext = null;

        try
        {
            webSocketContext = await context.AcceptWebSocketAsync(subProtocol: null);
            Console.WriteLine("WebSocket connection established.");

            await HandleWebSocketConnection(webSocketContext.WebSocket);
        }
        catch (Exception ex)
        {
            Console.WriteLine("WebSocket connection failed: " + ex.Message);
            if (webSocketContext != null)
            {
                webSocketContext.WebSocket.CloseAsync(WebSocketCloseStatus.InternalServerError, "Internal Server Error", CancellationToken.None);
            }
        }
    }

    private async Task HandleWebSocketConnection(WebSocket webSocket)
    {
        byte[] buffer = new byte[1024];
        WebSocketReceiveResult result = await webSocket.ReceiveAsync(new ArraySegment<byte>(buffer), CancellationToken.None);

        while (!result.CloseStatus.HasValue)
        {
            // Handle received data
            string receivedMessage = System.Text.Encoding.UTF8.GetString(buffer, 0, result.Count);
            Console.WriteLine("Received: " + receivedMessage);

            // Echo the received message back to the client
            await webSocket.SendAsync(new ArraySegment<byte>(buffer, 0, result.Count), WebSocketMessageType.Text, true, CancellationToken.None);

            // Get next message
            result = await webSocket.ReceiveAsync(new ArraySegment<byte>(buffer), CancellationToken.None);
        }

        await webSocket.CloseAsync(result.CloseStatus.Value, result.CloseStatusDescription, CancellationToken.None);
        Console.WriteLine("WebSocket connection closed.");
    }

    public void Stop()
    {
        _cts.Cancel();
        _httpListener.Stop();
        _httpListener.Close();
    }
}
  1. 在程序的入口点创建一个MyWebSocketServer实例并启动服务器。例如:
class Program
{
    static async Task Main(string[] args)
    {
        string url = "http://localhost:8080/";
        MyWebSocketServer server = new MyWebSocketServer(url);
        await server.Start();

        Console.WriteLine("Press any key to stop the server.");
        Console.ReadKey();

        server.Stop();
    }
}

以上代码创建了一个简单的WebSocket服务器,监听指定的URL,并处理接收到的WebSocket连接和消息。你可以根据自己的需求进行修改和扩展。

0