.net 5 使用WebSocket

前台简单构造：

window.location.search.replace("?","") 获取页面传值zs或者ls

<html>
<head>
    <script type="text/javascript">
           const webSochet= new WebSocket('ws://localhost:47471/ws');

           var origin=window.location.search.replace("?","");

           webSochet.onopen=function(res)
           {
               webSochet.send(origin);
           }

           webSochet.onmessage=function(res)
           {
               console.log(res.data);
           }

           function add(){
               var msg=document.getElementById("msg").value;
               var to=document.getElementById("to").value;
               webSochet.send(`${msg}***${to}`);
           }
    </script>
</head>
<body>
    <textarea id="msg">

    </textarea>
    发给：<select id="to">
            <option>请选择</option>
            <option>zs</option>
            <option>ls</option>
    </select>
    <button onclick="add()">发送数据</button>
</body>
</html>

首先在中间件中使用 app.UseWebSockets();

app.UseAuthorization();
 
app.UseWebSockets();
 
app.UseEndpoints(endpoints =>
{
    endpoints.MapControllers();
});       

如何使用全局变量：在configService中注入

private readonly Dictionary<string, WebSocket> dic;
 
       //readonly static Dictionary<string, WebSocket> dic = new Dictionary<string, WebSocket>();
 
       public WeatherForecastController(Dictionary<string, WebSocket> dic)
       {
           this.dic = dic;
       }

后台代码演示

public class WeatherForecastController : ControllerBase
   {
       readonly static Dictionary<string, WebSocket> dic = new Dictionary<string, WebSocket>();
 
       [HttpGet("/ws")]
       public async Task Get()
       {
           if (HttpContext.WebSockets.IsWebSocketRequest)
           {
               //接受websocket客户端连接
               var webSocket = await HttpContext.WebSockets.AcceptWebSocketAsync();
               await Echo(webSocket, dic);
           }
           else
           {
               //不是websocket客户端请求
               HttpContext.Response.StatusCode = 400;
           }
       }
       /// <summary>
       /// 对客户端的处理，接受消息，发送消息，关闭连接
       /// </summary>
       /// <param name="webSocket"></param>
       /// <returns></returns>
       private async Task Echo(WebSocket webSocket, Dictionary<string, WebSocket> dic)
       {
           var buffer = new byte[1024 * 4];
           var result = await webSocket.ReceiveAsync(new ArraySegment<byte>(buffer), CancellationToken.None);
           var user = Encoding.UTF8.GetString(buffer, 0, buffer.Length);
           dic.Add(user, webSocket);
           while (!result.CloseStatus.HasValue)
           {
               var message = Encoding.UTF8.GetString(buffer, 0, buffer.Length);
               if (message.Contains("***"))
               {
                   var msgArr = message.Split("***");
                   var msgDate = msgArr[0];
                   var msgTo = msgArr[1];
                   if (dic.ContainsKey(msgTo))
                   {
                       var webSocketGit = dic[msgTo];
                       //向客户端发送消息
                       await webSocketGit.SendAsync(new ArraySegment<byte>(buffer, 0, buffer.Length), result.MessageType, result.EndOfMessage, CancellationToken.None);
                   }
                   else
                   {
                       throw new Exception("用户不存在");
                   }
               }
               else
               {
                   //向客户端发送消息
                   await webSocket.SendAsync(new ArraySegment<byte>(buffer, 0, buffer.Length), result.MessageType, result.EndOfMessage, CancellationToken.None);
               }
               //继续接受客户端消息
               result = await webSocket.ReceiveAsync(new ArraySegment<byte>(buffer), CancellationToken.None);
           }
           //关闭释放与客户端连接
           await webSocket.CloseAsync(result.CloseStatus.Value, result.CloseStatusDescription, CancellationToken.None);
       }
   }