第一節：.Net版基於WebSocket的聊天室樣例

時間 2019-11-11

標籤一節基於 websocket 聊天室樣例简体版

原文原文鏈接

一. 說在前面的話html

　　該篇文章爲實時通信系列的第一節，基於WebSocket編寫了一個簡易版聊天樣例，主要做用是爲引出後面SignalR系列的用法及其強大方便之處，經過這個樣例與後續的SignalR對比，能夠體會到用原生的WebSocket編程的不靈活以及繁瑣之處。（PS：下面那個案例編寫+調試用了大約2個小時的時間）前端

　特別強調：該案例只是一個簡易版的Demo，實現了一些基本的功能，起到一個基本入門的做用，在一些細節方面的限制、驗證、以及關鍵信息的存放位置是有必定缺陷的，在實際項目中，須要根據實際狀況進行調整。jquery

　後面打算以一個真實的微信項目爲例【二人奪寶】以下圖，介紹裏面須要用到的實時通信技術，即SignalR在裏面起到的做用，你會發現若是業務複雜，若是採用原生的WebSocket進行編程，「你能累死」，或者說有些功能很難實現。web

　最後以一個真實的App混合開發項目【紅猴代派】以下圖，介紹用SignalR替代推送技術的實現。編程

　ps：鑑於保密協議，以上提到的兩個項目名稱並不是最終上線的真實名稱。瀏覽器

二. WebSocket簡介安全

　　WebSocket能夠經過JS與服務器端創建實時的雙向通訊，它主要包括：2個核心方法、4個監聽函數、1個核心屬性。服務器

2個核心方法：微信

　　① send() 向遠程服務器發送信息websocket

　　② close() 關閉與遠程服務器的鏈接

4個監聽函數：

　　① onopen 當網絡鏈接創建成功時觸發該事件
　　② onerror 當網絡發生錯誤時觸發該事件
　　③ onclose 當網絡鏈接被關閉時觸發該事件
　　④ onmessage 當websocket接收到服務器發來的消息的時觸發該事件

1個核心屬性：readyState，它有四種狀態

　　① WebSocket.OPEN：表示與服務器已經創建好鏈接

　　② WebSocket.CLOSED：表示與服務器鏈接已經斷開

　　③ WebSocket.CONNECTING：表示正在嘗試與服務創建鏈接

　　④ WebSocket.CLOSING：表示正在關閉與服務器鏈接

服務器端主要用到兩個方法：

　　① SendAsync() 發送信息給客戶端

　　② CloseAsync() 關閉該Socket鏈接

三. 聊天室實現的功能

　　1. 經過輸入登陸標記，進行創建鏈接，會提示當前全部在線的人(包括本身)，登陸成功。（Ps：同一個標記不能重複登陸）

　　2. 點擊關閉鏈接，斷開當前鏈接，提示除了本身之外的全部人，「某某離開了」。

　　3. 羣發功能：給全部在線的人（本身除外）的發送消息。

　　4. 單發功能：點對點發送消息。（須要輸入內容和接收人的標識）

四. 代碼分享

　　代碼主要分爲3部分組成：前端html頁面、服務器端ashx通常處理程序、封裝的消息發送幫助類。

前端頁面代碼：

  1 @{
  2     Layout = null;
  3 }
  4 
  5 <!DOCTYPE html>
  6 
  7 <html>
  8 <head>
  9     <meta name="viewport" content="width=device-width" />
 10     <title>Index</title>
 11     <script src="~/Scripts/jquery-3.3.1.min.js"></script>
 12     <script>
 13 
 14         //封裝WebSocket的相關方法
 15         (function (w) {
 16             //聲明全局對象
 17             var ws; //websocket對象
 18             var url = "ws://localhost:9008/webSocketHandler.ashx";   //鏈接地址
 19             var mainUitls = {
 20                 //1.初始化基本事件
 21                 init: function () {
 22                     this.initClick();
 23                 },
 24                 //2.創建通信事件
 25                 initConnect: function () {
 26                     var newUrl = url + "?userKey=" + $("#j_userKey").val();
 27                     ws = new WebSocket(newUrl);
 28                     //2.1 網絡創建鏈接的時候觸發該事件
 29                     ws.onopen = function () {
 30                         $('#j_notice').html('已經鏈接');
 31                     }
 32                     //2.2 接受服務器發來消息時觸發事件
 33                     ws.onmessage = function (evt) {
 34                         $("#j_Msg").append("<li>" + evt.data + "</li>");
 35                     }
 36                     //2.3 網絡錯誤時候觸發
 37                     ws.onerror = function (evt) {
 38                         $('#j_notice').html(JSON.stringify(evt));
 39                     }
 40                     //2.4 當鏈接關閉的時候觸發該事件
 41                     ws.onclose = function () {
 42                         //這裏根據實際場景來編寫 (好比斷線重連機制)
 43                         //該版本只是簡單的給個提示
 44                         $('#j_notice').html("鏈接斷開");
 45                     }
 46                 },
 47                 //3. 各類點擊事件
 48                 initClick: function () {
 49                     //3.1 創建鏈接
 50                     $("#j_connect").on("click", function () {
 51                         if ($("#j_userKey").val()=="") {
 52                             $('#j_notice').html("請輸入用戶標記");
 53                             return;
 54                         }
 55                         mainUitls.initConnect();
 56                     });
 57                     //3.2 關閉鏈接
 58                     $("#j_close").on("click", function () {
 59                         ws.close();
 60                     });
 61                     //3.3 羣發發送消息
 62                     $("#j_send").on("click", function () {
 63                         //表示與服務器已經創建好鏈接
 64                         if (ws.readyState == WebSocket.OPEN) {
 65                             ws.send($('#j_content').val());
 66                         }
 67                          //表示與服務器鏈接已經斷開
 68                         else if (ws.readyState == WebSocket.CLOSED) {
 69                             $('#j_notice').html('與服務器鏈接已經斷開');
 70                         }
 71                         //表示正在嘗試與服務創建鏈接
 72                         else if (ws.readyState == WebSocket.CONNECTING) {
 73                             $('#j_notice').html('正在嘗試與服務創建鏈接');
 74                         }
 75                         //正在關閉與服務器鏈接
 76                         else if (ws.readyState == WebSocket.CLOSING) {
 77                             $('#j_notice').html('正在關閉與服務器鏈接');
 78                         }
 79 
 80                     });
 81                     //3.4 單發消息
 82                     $("#j_send2").on("click", function () {
 83                         var msg = $('#j_receiveUserKey').val();
 84                         if (msg=="") {
 85                             $('#j_notice').html('請輸入接收人的標識');
 86                             return;
 87                         }
 88                         //下面對內容進行拼接
 89                        // "$--$--**" + msg + "$--$--**" + "發送的內容";
 90 
 91                         var finalMsg = "$--$--**" + msg + "$--$--**" + $('#j_content').val();
 92 
 93                         //表示與服務器已經創建好鏈接
 94                         if (ws.readyState == WebSocket.OPEN) {
 95                             ws.send(finalMsg);
 96                         }
 97                         //表示與服務器鏈接已經斷開
 98                         else if (ws.readyState == WebSocket.CLOSED) {
 99                             $('#j_notice').html('與服務器鏈接已經斷開');
100                         }
101                         //表示正在嘗試與服務創建鏈接
102                         else if (ws.readyState == WebSocket.CONNECTING) {
103                             $('#j_notice').html('正在嘗試與服務創建鏈接');
104                         }
105                         //正在關閉與服務器鏈接
106                         else if (ws.readyState == WebSocket.CLOSING) {
107                             $('#j_notice').html('正在關閉與服務器鏈接');
108                         }
109 
110                     });
111                 }
112 
113             };
114 
115             w.mainUitls = mainUitls;
116         })(window)
117 
118         $(function () {
119             mainUitls.init();
120         });
121     </script>
122 </head>
123 <body>
124     <div>
125         <div><span>提示：</span><span id="j_notice"></span></div>
126         <div style="margin-top:20px">
127             <input type="text" name="name" value="" placeholder="請輸入登陸標識" id="j_userKey" />
128             <button id="j_connect">創建鏈接</button>
129             <button id="j_close">關閉鏈接</button>
130         </div>
131         <div style="margin-top:20px">
132             <input type="text" value="" placeholder="請輸入發送內容" id="j_content" />
133             <button id="j_send">羣發</button>
134         </div>
135 
136         <div style="margin-top:20px">
137             <input type="text" value="" placeholder="請輸入接收人標識" id="j_receiveUserKey" />
138             <button id="j_send2">單發</button>
139         </div>
140         <div>
141             <ul id="j_Msg"></ul>
142         </div>
143     </div>
144 </body>
145 </html>

View Code

通常處理程序代碼：

  1 using System;
  2 using System.Collections.Concurrent;
  3 using System.Collections.Generic;
  4 using System.Linq;
  5 using System.Net.WebSockets;
  6 using System.Text;
  7 using System.Threading;
  8 using System.Threading.Tasks;
  9 using System.Web;
 10 using System.Web.SessionState;
 11 using System.Web.WebSockets;
 12 
 13 namespace WebSocketDemo
 14 {
 15     /// <summary>
 16     /// webSocketHander
 17     /// </summary>
 18     public class webSocketHandler : IHttpHandler, IRequiresSessionState
 19     {
 20         //用戶登記標識
 21         private string userKey = "";
 22         public void ProcessRequest(HttpContext context)
 23         {
 24             if (context.IsWebSocketRequest)
 25             {
 26                 this.userKey = context.Request.QueryString["userKey"];
 27                 context.AcceptWebSocketRequest(ProcessChat);
 28             }
 29             else
 30             {
 31                 context.Response.Write("不是WebSocket請求");
 32             }
 33         }
 34         private async Task ProcessChat(AspNetWebSocketContext context)
 35         {
 36             WebSocket socket = context.WebSocket;
 37             CancellationToken cancellationToken = new CancellationToken();
 38             bool isExits = WebManager.AddUser(userKey, socket);
 39             if (isExits == false)
 40             {
 41                 //表示該用戶有在線
 42                 await WebManager.SendToMySelf(cancellationToken, $"用戶【{this.userKey}】 已在線", this.userKey);
 43             }
 44             else
 45             {
 46                 //表示登陸成功
 47                 //某人登錄後，給羣裏其餘人發送提示信息(本人除外)
 48                 await WebManager.SendLoginSucesssNotice(cancellationToken, $"用戶【{this.userKey}】 進入聊天室,當前時間：{DateTime.Now.ToString("yyyy-MM-dd HH:mm:ss")}");
 49                 while (socket.State == WebSocketState.Open)
 50                 {
 51                     ArraySegment<byte> buffer = new ArraySegment<byte>(new byte[2048]);
 52                     //接受指令
 53                     WebSocketReceiveResult result = await socket.ReceiveAsync(buffer, cancellationToken);
 54                     //表示是關閉指令
 55                     if (result.MessageType == WebSocketMessageType.Close)
 56                     {
 57                         //移除該用戶對應的socket對象
 58                         WebManager.RemoveUser(userKey);
 59                         //發送離開提醒
 60                         await WebManager.SendOutNotice(cancellationToken, $"用戶【{this.userKey}】 離開聊天室,當前時間：{DateTime.Now.ToString("yyyy-MM-dd HH:mm:ss")}");
 61                         await socket.CloseAsync(WebSocketCloseStatus.NormalClosure, String.Empty, cancellationToken);
 62                     }
 63                     //獲取是發送消息指令
 64                     else
 65                     {
 66                         string userMsg = Encoding.UTF8.GetString(buffer.Array, 0, result.Count);
 67                         //這裏要區分是單發仍是羣發(經過與前端在內容開頭作一個標記，來區分是單發仍是羣發)
 68                         if (userMsg.Length > 8 && userMsg.Substring(0, 8) == "$--$--**")
 69                         {
 70                             //表示是單發，截取內容和接受者的標記
 71                             var array = userMsg.Split(new string[] { "$--$--**" }, StringSplitOptions.None);
 72                             var receiveNotice = array[1];
 73                             string content = $"用戶【{this.userKey}】 發來消息：{array[2]},當前時間：{DateTime.Now.ToString("yyyy-MM-dd HH:mm:ss")}";
 74                             WebManager.SendSingleMessage(cancellationToken, content, receiveNotice);
 75                         }
 76                         else
 77                         {
 78                             //表示羣發信息 
 79                             string content = $"用戶【{this.userKey}】 羣發消息：{userMsg},當前時間：{DateTime.Now.ToString("yyyy-MM-dd HH:mm:ss")}";
 80                             WebManager.SendAllMessage(cancellationToken, content, userKey);
 81                         }
 82                     }
 83                 }
 84             }
 85 
 86         }
 87 
 88         /// <summary>
 89         ///  true 若是 System.Web.IHttpHandler 實例是可重複使用; 不然爲 false。
 90         /// </summary>
 91         public bool IsReusable
 92         {
 93             get
 94             {
 95                 return false;
 96             }
 97         }
 98     }
 99 
100 
101 }

View Code

幫助類代碼：

  1 using System;
  2 using System.Collections.Concurrent;
  3 using System.Collections.Generic;
  4 using System.Linq;
  5 using System.Net.WebSockets;
  6 using System.Text;
  7 using System.Threading;
  8 using System.Threading.Tasks;
  9 using System.Web;
 10 using System.Web.SessionState;
 11 
 12 
 13 namespace WebSocketDemo
 14 {
 15     /// <summary>
 16     /// webSocket幫助類
 17     /// </summary>
 18     public class WebManager
 19     {
 20         /// <summary>
 21         /// 這採用ConcurrentDictionary字典，是線程安全的，不須要加鎖
 22         /// </summary>
 23         private static ConcurrentDictionary<string, WebSocket> _UserDictionary = new ConcurrentDictionary<string, WebSocket>();
 24 
 25         #region 01-增長用戶
 26         /// <summary>
 27         /// 增長用戶
 28         /// </summary>
 29         /// <param name="userKey"></param>
 30         /// <param name="socket"></param>
 31         public static bool AddUser(string userKey, WebSocket socket)
 32         {
 33 
 34             bool flag = _UserDictionary.Select(d => d.Key).ToList().Contains(userKey);
 35             if (flag == false)
 36             {
 37                 _UserDictionary[userKey] = socket;
 38                 return true;
 39             }
 40             else
 41             {
 42                 //表示該用戶在線
 43                 return false;
 44 
 45             }
 46         }
 47         #endregion
 48 
 49         #region 02-移除用戶
 50         /// <summary>
 51         /// 移除用戶
 52         /// </summary>
 53         /// <param name="userKey"></param>
 54         public static void RemoveUser(string userKey)
 55         {
 56             WebSocket socket = null;
 57             _UserDictionary.TryRemove(userKey, out socket);
 58         }
 59         #endregion
 60 
 61         #region 03-登陸提醒
 62 
 63         /// <summary>
 64         /// 登陸提醒（包括本身）
 65         /// </summary>
 66         /// <param name="cancellationToken"></param>
 67         /// <param name="content"></param>
 68         /// <returns></returns>
 69         public static async Task SendLoginSucesssNotice(CancellationToken cancellationToken, string content)
 70         {
 71             ArraySegment<byte> buffer = new ArraySegment<byte>(new byte[2048]);
 72             buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(content));
 73             //登陸提醒（包括本身）
 74             foreach (var socket in _UserDictionary.Select(d => d.Value))
 75             {
 76                 await socket.SendAsync(buffer, WebSocketMessageType.Text, true, cancellationToken);
 77             }
 78         }
 79 
 80         #endregion
 81 
 82         #region 04-離開提醒
 83 
 84         /// <summary>
 85         /// 離開提醒
 86         /// </summary>
 87         /// <param name="cancellationToken"></param>
 88         /// <param name="content"></param>
 89         /// <returns></returns>
 90         public static async Task SendOutNotice(CancellationToken cancellationToken, string content)
 91         {
 92             ArraySegment<byte> buffer = new ArraySegment<byte>(new byte[2048]);
 93             buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(content));
 94             //離開提醒
 95             foreach (var socket in _UserDictionary.Select(d => d.Value))
 96             {
 97                 await socket.SendAsync(buffer, WebSocketMessageType.Text, true, cancellationToken);
 98             }
 99         }
100 
101         #endregion
102 
103         #region 05-羣發消息，不包括本身
104         /// <summary>
105         /// 羣發消息，不包括本身
106         /// </summary>
107         /// <param name="cancellationToken"></param>
108         /// <param name="content"></param>
109         /// <param name="myUserKey">當前用戶標記</param>
110         /// <returns></returns>
111         public static void SendAllMessage(CancellationToken cancellationToken, string content, string myUserKey)
112         {
113             try
114             {
115                 ArraySegment<byte> buffer = new ArraySegment<byte>(new byte[2048]);
116                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(content));
117                 //羣發消息，但不包括本身
118                 foreach (var item in _UserDictionary)
119                 {
120                     if (item.Key.ToString() != myUserKey)
121                     {
122                         item.Value.SendAsync(buffer, WebSocketMessageType.Text, true, cancellationToken);
123                     }
124                 }
125             }
126             catch (Exception ex)
127             {
128                 var msg = ex.Message;
129             }
130         }
131         #endregion
132 
133         #region 06-單發消息
134         /// <summary>
135         /// 單發消息
136         /// </summary>
137         /// <param name="cancellationToken"></param>
138         /// <param name="content"></param>
139         /// <param name="receiveKey">接收者的標識</param>
140         /// <returns></returns>
141         public static void SendSingleMessage(CancellationToken cancellationToken, string content, string receiveKey)
142         {
143             try
144             {
145                 ArraySegment<byte> buffer = new ArraySegment<byte>(new byte[2048]);
146                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(content));
147                 //單發消息
148                 foreach (var item in _UserDictionary)
149                 {
150                     if (item.Key.ToString() == receiveKey)
151                     {
152                         item.Value.SendAsync(buffer, WebSocketMessageType.Text, true, cancellationToken);
153                     }
154                 }
155             }
156             catch (Exception ex)
157             {
158                 var msg = ex.Message;
159             }
160         }
161         #endregion
162 
163         #region 07-給本身發送消息
164 
165         /// <summary>
166         /// 給本身發送消息
167         /// </summary>
168         /// <param name="cancellationToken"></param>
169         /// <param name="content"></param>
170         /// <param name="userKey">當前標記</param>
171         /// <returns></returns>
172         public static async Task SendToMySelf(CancellationToken cancellationToken, string content,string userKey)
173         {
174             ArraySegment<byte> buffer = new ArraySegment<byte>(new byte[2048]);
175             buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(content));
176             //離開提醒
177             foreach (var item in _UserDictionary)
178             {
179                 if (item.Key.ToString() == userKey)
180                 {
181                   await  item.Value.SendAsync(buffer, WebSocketMessageType.Text, true, cancellationToken);
182                 }
183             }
184         }
185         #endregion
186 
187     }
188 }