WebSocketForWebAPI:https://gitee.com/lycz/WebSocketForWebAPI
服務端發送消息,控制器代碼
private readonly ClientWebSocket webSocket = new ClientWebSocket(); private readonly CancellationToken _cancellation = new CancellationToken(); [HttpPost] public async Task SendMsg(string msg) { await webSocket.ConnectAsync(new Uri("ws://localhost:56486/api/msg"), _cancellation); var sendBytes = Encoding.UTF8.GetBytes(msg);//發送的數據 var bsend = new ArraySegment<byte>(sendBytes); await webSocket.SendAsync(bsend, WebSocketMessageType.Binary, true, _cancellation); await webSocket.CloseAsync(WebSocketCloseStatus.NormalClosure, "1", _cancellation); webSocket.Dispose(); }
WebSocketForWebAPI代碼
后台:MsgApiController.cs
[RoutePrefix("api/msg")] public class MsgApiController : ApiController { private static List<WebSocket> _sockets = new List<WebSocket>(); [Route] [HttpGet] public HttpResponseMessage Connect() { HttpContext.Current.AcceptWebSocketRequest(ProcessRequest); //在服務器端接受Web Socket請求,傳入的函數作為Web Socket的處理函數,待Web Socket建立后該函數會被調用,在該函數中可以對Web Socket進行消息收發 return Request.CreateResponse(HttpStatusCode.SwitchingProtocols); //構造同意切換至Web Socket的Response. } public async Task ProcessRequest(AspNetWebSocketContext context) { var socket = context.WebSocket;//傳入的context中有當前的web socket對象 _sockets.Add(socket);//此處將web socket對象加入一個靜態列表中 //進入一個無限循環,當web socket close是循環結束 while (true) { var buffer = new ArraySegment<byte>(new byte[1024]); var receivedResult = await socket.ReceiveAsync(buffer, CancellationToken.None);//對web socket進行異步接收數據 if (receivedResult.MessageType == WebSocketMessageType.Close) { await socket.CloseAsync(WebSocketCloseStatus.Empty, string.Empty, CancellationToken.None);//如果client發起close請求,對client進行ack _sockets.Remove(socket); break; } if (socket.State == System.Net.WebSockets.WebSocketState.Open) { string recvMsg = Encoding.UTF8.GetString(buffer.Array, 0, receivedResult.Count); var recvBytes = Encoding.UTF8.GetBytes(recvMsg); var sendBuffer = new ArraySegment<byte>(buffer.Array); foreach (var innerSocket in _sockets)//當接收到文本消息時,對當前服務器上所有web socket連接進行廣播 { if (innerSocket != socket) { await innerSocket.SendAsync(sendBuffer, WebSocketMessageType.Text, true, CancellationToken.None); } } } } } }
前端:
Msg.cshtml
@{ Layout = null; } <!DOCTYPE html> <html> <head> <script src="http://libs.baidu.com/jquery/2.0.0/jquery.js"></script> <script> var webSocket; function sendSocketMsg() { var msg = $("#msg").val(); webSocket.send(msg); showMsg(msg, "blue"); } function openSocket() { if (webSocket != null && typeof (webSocket) != "undefined") { closeSocket(); } webSocket = new WebSocket("ws://" + location.hostname + ":" + location.port + "/api/msg"); webSocket.onopen = function () { showMsg("連接建立"); } webSocket.onerror = function () { showMsg("發生異常"); } webSocket.onmessage = function (event) { showMsg(event.data, "yellow"); } webSocket.onclose = function () { showMsg("連接關閉"); } } function closeSocket() { if (webSocket != null && typeof (webSocket) != "undefined") { webSocket.close(); } } function showMsg(msg, type) { if (type === null || typeof (type) === "undefined") type = "gray"; $("#show").append("<span class='" + type + "'>" + msg + "</span><br>"); } </script> <style> #show { border: 1px solid #ff6a00; padding: 10px; } .blue { border: 1px solid #0094ff; } .yellow { margin-left: 300px; border: 1px solid #b6ff00; } .gray { font-size: 10px; border: 1px solid #a9a9a9; } </style> </head> <body> <button onclick="openSocket()">建立連接</button> <button onclick="closeSocket()">斷開連接</button> <hr /> <input type="text" id="msg" /> <button onclick="sendSocketMsg()">發送信息</button> <hr /> <div id="show"></div> </body> </html>
