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

admin

服务器端代码编写

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

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

3.Handler1.ashx代码如下：

1. using System;
   
2. using System.Collections.Generic;
     m! R8 ~* c& E! l. [& ]$ u/ d; O
3. using System.Linq;
   
4. using System.Net.WebSockets;
   
5. using System.Text;
   & R2 K( K/ i6 }* e/ @
6. using System.Threading;
   
7. using System.Threading.Tasks;
   
8. using System.Web;
   ' L3 a" S$ z/ D$ b6 ~
9. using System.Web.WebSockets;
   
10. 
    
11. 
    ) Z; z) U9 W3 B- @3 M7 V
12. namespace WebApplicationWebsocketHandler
    : ]: h5 ]6 k6 [: Z! A' Q5 S2 E
13. {
    , h2 Y$ s' y/ D3 v
14.     /// <summary>
    . w/ e( Z, ?7 m+ d: [0 {; O( A+ K4 N
15.     /// 离线消息
    
16.     /// </summary>
    4 c8 A, V2 w7 ~: ?$ I& d. y
17.     public class MessageInfo
    
18.     {
    9 Q5 I, q, B4 R  R% I$ i/ y: m
19.         public MessageInfo(DateTime _MsgTime, ArraySegment<byte> _MsgContent)
    1 p! y: F' B$ a0 x9 I; S
20.         {
    
21.             MsgTime = _MsgTime;
    
22.             MsgContent = _MsgContent;
    
23.         }
    , g1 x. `: c# o2 H
24.         public DateTime MsgTime { get; set; }
    
25.         public ArraySegment<byte> MsgContent { get; set; }
    
26.     }
    
27. 
    
28. 
    
29. 
    7 X8 W4 |5 J. }2 t- L$ H- S
30. 
    $ {; @* I+ e* D9 M! M/ @! b; {' x
31.     /// <summary>
    % Q1 f, |! Y; G( Q' Z  M1 C6 y
32.     /// Handler1 的摘要说明
    . ], a! W, l! p8 p" s
33.     /// </summary>
    2 m" h" q6 G  ~0 e$ W6 a' U
34.     public class Handler1 : IHttpHandler
    
35.     {
    ) i3 Q# b4 E8 ^# j, ?6 ^, Q
36.         private static Dictionary<string, WebSocket> CONNECT_POOL = new Dictionary<string, WebSocket>();//用户连接池
    1 _  g5 O! U& `$ ?
37.         private static Dictionary<string, List<MessageInfo>> MESSAGE_POOL = new Dictionary<string, List<MessageInfo>>();//离线消息池
    
38.         public void ProcessRequest(HttpContext context)
    . Y) x5 Y* V+ p  T2 Y
39.         {
    
40.             //context.Response.ContentType = "text/plain";
    
41.             //context.Response.Write("Hello World");
    
42.             if (context.IsWebSocketRequest)
    
43.             {
    
44.                 context.AcceptWebSocketRequest(ProcessChat);
    
45.             }
    
46.         }
    
47. 
    ( M& \, q( v5 ]0 D
48.         private async Task ProcessChat(AspNetWebSocketContext context)
    + n" O* I% ]( Z; `: v* \$ c: d% S
49.         {
    
50.             WebSocket socket = context.WebSocket;
    
51.             string user = context.QueryString["user"].ToString();
    ; t) f, w, c3 i8 m) s% F2 i
52. 
    
53.             try
    8 i/ v8 D; [, Q$ J; |
54.             {
    
55.                 #region 用户添加连接池
    7 K1 ~! n  Y! Y3 |* j- @/ |
56.                 //第一次open时，添加到连接池中
    $ u* @) ?% o; e2 Z) ?
57.                 if (!CONNECT_POOL.ContainsKey(user))
    0 b. l+ E$ d( _
58.                     CONNECT_POOL.Add(user, socket);//不存在，添加
    
59.                 else
    4 h: c' H: |1 d$ B
60.                     if (socket != CONNECT_POOL[user])//当前对象不一致，更新
    4 p: }% R$ j4 u
61.                         CONNECT_POOL[user] = socket;
    0 W' v) L& `* ~# b; |- V# i
62.                 #endregion
    ( z6 s2 l. L( L  S9 z
63. 
    
64.                 #region 离线消息处理
    + I/ Z: U! Z* Q, b7 ~1 r
65.                 if (MESSAGE_POOL.ContainsKey(user))
    0 ~2 s- L5 C  l1 P/ h
66.                 {
    * Y% {" H/ K! M0 ]& j5 {6 Z- b
67.                     List<MessageInfo> msgs = MESSAGE_POOL[user];
      ?$ B! T; [7 X$ t
68.                     foreach (MessageInfo item in msgs)
    " R: g0 A7 ]1 Y5 X5 Q' Q
69.                     {
    ' d! U. F1 r0 F8 g
70.                         await socket.SendAsync(item.MsgContent, WebSocketMessageType.Text, true, CancellationToken.None);
    
71.                     }
    9 ^0 X* s$ B" Z) ^. s
72.                     MESSAGE_POOL.Remove(user);//移除离线消息
    
73.                 }
    
74.                 #endregion
    2 O* V9 E, r/ Q0 z& R7 J
75. 
    % }  c3 U, A* `& I
76.                 string descUser = string.Empty;//目的用户
    
77.                 while (true)
    ' D3 s/ r1 |, y  v2 J
78.                 {
    
79.                     if (socket.State == WebSocketState.Open)
    1 Q! V( s$ ?5 {: a* H" a0 h
80.                     {
    8 c: x) b" R$ Q) k, d! `. i8 S: e
81.                         ArraySegment<byte> buffer = new ArraySegment<byte>(new byte[2048]);
    ; z9 w- u* @8 s0 D
82.                         WebSocketReceiveResult result = await socket.ReceiveAsync(buffer, CancellationToken.None);
    " z, I1 y8 t6 C3 @
83. 
    
84.                         #region 消息处理（字符截取、消息转发）
    
85.                         try
    ; G8 |0 B, L. C# k) o' {
86.                         {
    
87.                             #region 关闭Socket处理，删除连接池
    4 ?; n  f  h# j" q9 U* t7 ^  Z
88.                             if (socket.State != WebSocketState.Open)//连接关闭
    
89.                             {
    
90.                                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);//删除连接池
    
91.                                 break;
    ' V9 T; _3 C) E9 K' P
92.                             }
    ' V" t4 a# P7 c2 ~+ ]  Q
93.                             #endregion
    
94. 
    " e. D, H7 G& ~
95.                             string userMsg = Encoding.UTF8.GetString(buffer.Array, 0, result.Count);//发送过来的消息
    
96.                             string[] msgList = userMsg.Split('|');
    
97.                             if (msgList.Length == 2)
    
98.                             {
    
99.                                 if (msgList[0].Trim().Length > 0)
    
100.                                     descUser = msgList[0].Trim();//记录消息目的用户
     
101.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(msgList[1]));
     
102.                             }
     
103.                             else
     
104.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(userMsg));
     3 `9 n# u" I" X% _) ^
105. 
     ! C/ v$ `7 H" D- v9 n
106.                             if (CONNECT_POOL.ContainsKey(descUser))//判断客户端是否在线
     $ w  r2 A$ p; D9 _
107.                             {
     
108.                                 WebSocket destSocket = CONNECT_POOL[descUser];//目的客户端
     
109.                                 if (destSocket != null && destSocket.State == WebSocketState.Open)
     ' f, p, Y) |+ h
110.                                     await destSocket.SendAsync(buffer, WebSocketMessageType.Text, true, CancellationToken.None);
     . Y8 B* ]1 Y# F/ _& e. `+ h& k- T( W2 g
111.                             }
     9 N+ s& @. d" ?; V/ |2 {
112.                             else
     
113.                             {
     
114.                                 Task.Run(() =>
     
115.                                 {
     7 [$ \* ^& R: y: u
116.                                     if (!MESSAGE_POOL.ContainsKey(descUser))//将用户添加至离线消息池中
     2 U1 l2 g5 y7 {# |1 m, k/ Q* X
117.                                         MESSAGE_POOL.Add(descUser, new List<MessageInfo>());
     
118.                                     MESSAGE_POOL[descUser].Add(new MessageInfo(DateTime.Now, buffer));//添加离线消息
     ! H& ~4 U  n. d4 d: D* X
119.                                 });
     
120.                             }
     
121.                         }
     
122.                         catch (Exception exs)
     ' s) e7 P# u& ]5 n. Y& R, I
123.                         {
     
124.                             //消息转发异常处理，本次消息忽略 继续监听接下来的消息
     * Y- h+ |5 |. T3 Y: z0 ~- ^1 R- O
125.                         }
     
126.                         #endregion
     ! r" E4 r& k& `9 R- @; k( v$ D9 \
127.                     }
     1 a, l1 N# f0 U  M" C% m+ u, G/ p/ e* l
128.                     else
     
129.                     {
     
130.                         break;
     
131.                     }
     % K  b% b( x& x4 S
132.                 }//while end
     ) D; b' J7 \7 m
133.             }
     - y0 u! E( z7 Q) p, W& e
134.             catch (Exception ex)
     
135.             {
     
136.                 //整体异常处理
     
137.                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);
     
138.             }
     % E% U) }) T- h8 X8 v
139.         }
     . X4 Y+ c8 ^- |* [" _, `  }
140. 
       o9 E/ N2 S+ h+ p" `
141. 
     
142.         public bool IsReusable
     . z8 f1 Z3 H# \
143.         {
     
144.             get
     
145.             {
     
146.                 return false;
     - g3 ?  a: H/ n5 v
147.             }
     
148.         }
     ! f4 A; u& `6 X: Q# G. Y' N
149.     }
     
150. }

复制代码

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

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

. Z4 n- _. \, T# A