在IIS上搭建WebSocket服务器（二）

admin

服务器端代码编写

1.新建一个ASP.net Web MVC5项目

2.新建一个“一般处理程序”

3.Handler1.ashx代码如下：

1. using System;
   
2. using System.Collections.Generic;
   
3. using System.Linq;
   / t; |  c+ M) ]
4. using System.Net.WebSockets;
   
5. using System.Text;
   - L. o8 P3 n$ X% R  H6 d7 v: f
6. using System.Threading;
   
7. using System.Threading.Tasks;
   
8. using System.Web;
   
9. using System.Web.WebSockets;
   % m* V& d0 ?9 ?2 o+ U0 \5 z; O
10. 
    
11. 
    ' Y4 X; ]5 H( d6 x# U) a
12. namespace WebApplicationWebsocketHandler
    " _. C8 G% i5 @
13. {
    2 _" e) [4 [3 H
14.     /// <summary>
    . j% m+ U. g3 i, u! K! O& a2 g
15.     /// 离线消息
    % c' V# l; o9 Y8 j" Y4 J( o* D* H+ B
16.     /// </summary>
    ) N: x: T! c3 B& Z; i" j9 Q" W
17.     public class MessageInfo
    
18.     {
    
19.         public MessageInfo(DateTime _MsgTime, ArraySegment<byte> _MsgContent)
    
20.         {
    
21.             MsgTime = _MsgTime;
      N* a9 w& r- {- z
22.             MsgContent = _MsgContent;
    
23.         }
    
24.         public DateTime MsgTime { get; set; }
    
25.         public ArraySegment<byte> MsgContent { get; set; }
    
26.     }
    # K: h8 v1 ?* K2 n9 C6 V
27. 
    
28. 
    # M; Q* K% O6 |& R! W
29. 
    
30. 
    
31.     /// <summary>
    
32.     /// Handler1 的摘要说明
    ) k3 Z5 C8 Y7 h0 Z
33.     /// </summary>
    
34.     public class Handler1 : IHttpHandler
    3 Z/ U) K; B& ?: Q! `( _
35.     {
    - ?. q8 _. i! X7 g
36.         private static Dictionary<string, WebSocket> CONNECT_POOL = new Dictionary<string, WebSocket>();//用户连接池
    ! H2 }( V, U* T& @
37.         private static Dictionary<string, List<MessageInfo>> MESSAGE_POOL = new Dictionary<string, List<MessageInfo>>();//离线消息池
    
38.         public void ProcessRequest(HttpContext context)
    - E! v5 u' U3 a, [
39.         {
    
40.             //context.Response.ContentType = "text/plain";
    
41.             //context.Response.Write("Hello World");
    2 t% b" i) j" Q- b
42.             if (context.IsWebSocketRequest)
    - r* v  @5 L& X) x# w
43.             {
    % `( v' d! n" Q9 R& f/ s
44.                 context.AcceptWebSocketRequest(ProcessChat);
      t7 u3 K( O% i  X# j
45.             }
    
46.         }
    * Q( U( A3 w: ?* ]7 J
47. 
    0 `& B3 g3 J1 h. i& B% Q
48.         private async Task ProcessChat(AspNetWebSocketContext context)
    
49.         {
    
50.             WebSocket socket = context.WebSocket;
    : d/ @3 Y- f+ p, c9 y" {, i8 Q
51.             string user = context.QueryString["user"].ToString();
    * @' M' B% h  A* Z. p5 B
52. 
    5 b9 R0 B7 X/ H2 G9 Q  j
53.             try
    ! U. X! Y4 }4 J2 T
54.             {
    
55.                 #region 用户添加连接池
    
56.                 //第一次open时，添加到连接池中
    7 G* R* D% ]3 B; z+ Z% _
57.                 if (!CONNECT_POOL.ContainsKey(user))
    
58.                     CONNECT_POOL.Add(user, socket);//不存在，添加
    
59.                 else
    
60.                     if (socket != CONNECT_POOL[user])//当前对象不一致，更新
    
61.                         CONNECT_POOL[user] = socket;
    1 m( x5 p8 Y: z1 t" W6 S" g; s# h
62.                 #endregion
    4 i: Y- e# ~0 l% N- W) t
63. 
    
64.                 #region 离线消息处理
    : P8 n" z2 I; _% Q8 c
65.                 if (MESSAGE_POOL.ContainsKey(user))
    2 a7 @4 S+ Q+ H7 S/ x
66.                 {
    , ~6 _. V# L9 Z
67.                     List<MessageInfo> msgs = MESSAGE_POOL[user];
    
68.                     foreach (MessageInfo item in msgs)
    ! j" m* p$ K2 L
69.                     {
    
70.                         await socket.SendAsync(item.MsgContent, WebSocketMessageType.Text, true, CancellationToken.None);
    
71.                     }
    : B4 A* \, i0 G1 g% y
72.                     MESSAGE_POOL.Remove(user);//移除离线消息
    
73.                 }
    
74.                 #endregion
    + s! y% k6 P. W" x  Z" j! t7 D
75. 
    
76.                 string descUser = string.Empty;//目的用户
    - W' a: l6 i$ e$ L: S! g
77.                 while (true)
    : m" |/ S7 s6 O2 N/ C4 \
78.                 {
    
79.                     if (socket.State == WebSocketState.Open)
    + C& S  N5 z: ~) d; l
80.                     {
    
81.                         ArraySegment<byte> buffer = new ArraySegment<byte>(new byte[2048]);
    
82.                         WebSocketReceiveResult result = await socket.ReceiveAsync(buffer, CancellationToken.None);
    
83. 
    6 j9 q: U9 f8 b  F; X! b
84.                         #region 消息处理（字符截取、消息转发）
    . W# `6 g9 A2 z0 w
85.                         try
    
86.                         {
    
87.                             #region 关闭Socket处理，删除连接池
    7 [* X' x1 X3 A5 ~6 X! U) u. M
88.                             if (socket.State != WebSocketState.Open)//连接关闭
    0 {) p4 K5 X  Q! w
89.                             {
    . T! I3 Y" j  _# ]  ^
90.                                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);//删除连接池
    
91.                                 break;
    
92.                             }
    
93.                             #endregion
    , T1 x/ c  l3 w* i  d, u6 j0 L
94. 
    
95.                             string userMsg = Encoding.UTF8.GetString(buffer.Array, 0, result.Count);//发送过来的消息
    . t& z! o, L2 T5 x
96.                             string[] msgList = userMsg.Split('|');
    
97.                             if (msgList.Length == 2)
    7 A( g- y" k8 H9 N8 ?3 f
98.                             {
    
99.                                 if (msgList[0].Trim().Length > 0)
    " D8 S. c( \/ O) ?" s3 E; J
100.                                     descUser = msgList[0].Trim();//记录消息目的用户
     
101.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(msgList[1]));
     $ O# i. h9 [( O" l  o( [
102.                             }
     
103.                             else
     
104.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(userMsg));
     
105. 
     
106.                             if (CONNECT_POOL.ContainsKey(descUser))//判断客户端是否在线
     . w/ Y5 c) J+ O
107.                             {
     0 w* Q4 y! i2 O' X6 i. I
108.                                 WebSocket destSocket = CONNECT_POOL[descUser];//目的客户端
     : `8 z5 Y0 \: M, r
109.                                 if (destSocket != null && destSocket.State == WebSocketState.Open)
     ) W! b2 i7 s8 X* S
110.                                     await destSocket.SendAsync(buffer, WebSocketMessageType.Text, true, CancellationToken.None);
     $ T# U3 E5 `' ?& a$ l  S
111.                             }
     ' B9 C. W* n' ?) Y4 n0 @
112.                             else
     ; ]' S$ C# t4 W3 V. j9 u4 B% f
113.                             {
     5 }' u3 P5 p2 O2 Q& Y2 g, H8 v
114.                                 Task.Run(() =>
     , a  J$ b0 c1 C, i" Z/ [
115.                                 {
     5 p. |0 p! t# l
116.                                     if (!MESSAGE_POOL.ContainsKey(descUser))//将用户添加至离线消息池中
     1 C. \, b' }- O2 ^( N  |, m
117.                                         MESSAGE_POOL.Add(descUser, new List<MessageInfo>());
     
118.                                     MESSAGE_POOL[descUser].Add(new MessageInfo(DateTime.Now, buffer));//添加离线消息
     
119.                                 });
     8 S2 D2 @" r# }2 G+ f1 c/ L
120.                             }
     ; O) [1 h1 M! m; v; X
121.                         }
     
122.                         catch (Exception exs)
     
123.                         {
     
124.                             //消息转发异常处理，本次消息忽略 继续监听接下来的消息
     
125.                         }
     0 e* L$ U3 _1 N. c" \! y
126.                         #endregion
     & A- g+ j+ k0 Y6 F, O$ U4 T
127.                     }
     0 W; X1 P9 T4 @/ d; N6 }
128.                     else
     % R% q3 i9 G" c1 |0 _
129.                     {
     
130.                         break;
     
131.                     }
     , P) H& A3 L! R* `1 H) h
132.                 }//while end
     
133.             }
     2 v' z4 m& ]9 q3 Q
134.             catch (Exception ex)
     
135.             {
     
136.                 //整体异常处理
     " c2 N) F7 \4 K3 K5 U
137.                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);
     
138.             }
     5 k9 o: f& G7 t5 |2 q
139.         }
       Z' K5 m0 e$ Y; T9 t( @! W
140. 
     
141. 
     ) X  t/ U, H. F( ]* R$ L& I7 @
142.         public bool IsReusable
     4 `% ?* a  ?7 N0 B. v9 F; I$ ?
143.         {
     
144.             get
     
145.             {
     
146.                 return false;
     
147.             }
     
148.         }
     3 x2 b* I2 H% g) g0 L+ K
149.     }
     
150. }

复制代码

4.运行看是否报错，若没错将我们的服务器网站发布到IIS（支持WebSocket的IIS上，win7的系统时不可以的）上

点击“生成”->"发布........",以文件系统的方式发布，目标位置为我们创建的IIS网站对应的物理路径

% S: G$ A4 Y: u" T( T, h