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

admin

服务器端代码编写

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

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

3.Handler1.ashx代码如下：

1. using System;
   
2. using System.Collections.Generic;
   $ P( B6 N6 c& n
3. using System.Linq;
   $ a6 P0 S, W& j8 ]6 _
4. using System.Net.WebSockets;
   
5. using System.Text;
   ) U: b/ Y- R" ~$ t6 h
6. using System.Threading;
   2 C2 r0 b" M4 u( k
7. using System.Threading.Tasks;
   
8. using System.Web;
   - |- v+ \, K& t
9. using System.Web.WebSockets;
   
10. 
    
11. 
    # ^: x, D/ \/ E$ J6 a' S9 a
12. namespace WebApplicationWebsocketHandler
    
13. {
    
14.     /// <summary>
    
15.     /// 离线消息
    ) h9 M. Q+ l' H2 K
16.     /// </summary>
    
17.     public class MessageInfo
    
18.     {
    7 I! V( D2 a2 \
19.         public MessageInfo(DateTime _MsgTime, ArraySegment<byte> _MsgContent)
    
20.         {
    
21.             MsgTime = _MsgTime;
    
22.             MsgContent = _MsgContent;
    1 |% z. N, F& U0 j
23.         }
    
24.         public DateTime MsgTime { get; set; }
    & g% m+ l% c, T! M& y
25.         public ArraySegment<byte> MsgContent { get; set; }
    
26.     }
    * Y' H! L( j4 M6 a! i% I* _
27. 
    
28. 
    $ W2 u# g. \- X0 n
29. 
    4 [! Y6 u' A- d$ `$ D" ?. X
30. 
    
31.     /// <summary>
    ; r; z: V& [9 e. S
32.     /// Handler1 的摘要说明
    6 R& t2 C9 f8 f( \* d
33.     /// </summary>
    " W( U/ Y. F3 R* y7 W
34.     public class Handler1 : IHttpHandler
    
35.     {
    
36.         private static Dictionary<string, WebSocket> CONNECT_POOL = new Dictionary<string, WebSocket>();//用户连接池
    
37.         private static Dictionary<string, List<MessageInfo>> MESSAGE_POOL = new Dictionary<string, List<MessageInfo>>();//离线消息池
    + d% G- ~; @& V  |, T# N
38.         public void ProcessRequest(HttpContext context)
    
39.         {
    6 O( }: G8 G7 y/ l/ @: W. u8 t
40.             //context.Response.ContentType = "text/plain";
    
41.             //context.Response.Write("Hello World");
    
42.             if (context.IsWebSocketRequest)
    1 O7 `+ r- d" D& X, `
43.             {
    3 Z$ Q5 W' o& p9 [- u) h' a. l3 U
44.                 context.AcceptWebSocketRequest(ProcessChat);
    5 c5 O+ K, U- f( r" ?+ ~: {' v
45.             }
    % v' n0 E+ j, ?0 v
46.         }
    
47. 
    ! K4 h1 A6 i' s) z+ ?& {! O
48.         private async Task ProcessChat(AspNetWebSocketContext context)
    
49.         {
    
50.             WebSocket socket = context.WebSocket;
    ! n* O1 I5 U* \9 P
51.             string user = context.QueryString["user"].ToString();
    ) V8 M+ Q: X% P" V% A; ~/ E
52. 
    
53.             try
    
54.             {
      v' }% @3 J" e0 R4 Z( k, b# E
55.                 #region 用户添加连接池
    ( w0 F  `6 T5 S. d- r2 v5 c1 I
56.                 //第一次open时，添加到连接池中
    7 Z) ~9 t$ N' s: O' B. ?
57.                 if (!CONNECT_POOL.ContainsKey(user))
    8 u/ k5 F7 e2 S" Z6 o% ~
58.                     CONNECT_POOL.Add(user, socket);//不存在，添加
    * z: l' o: R. j& ]$ Y& _
59.                 else
    5 f( t7 J3 d9 N0 N% k# c# g
60.                     if (socket != CONNECT_POOL[user])//当前对象不一致，更新
    ) K% r3 A4 j6 M8 W/ e: o
61.                         CONNECT_POOL[user] = socket;
    ! _/ B5 E  W5 G/ b- n% ]
62.                 #endregion
    
63. 
    
64.                 #region 离线消息处理
    
65.                 if (MESSAGE_POOL.ContainsKey(user))
    3 }/ Q6 s0 Z4 Y8 F5 B
66.                 {
    3 l2 Q. U7 k9 p  ?6 Y" X
67.                     List<MessageInfo> msgs = MESSAGE_POOL[user];
    7 x% q) p, R2 L8 [- I) u
68.                     foreach (MessageInfo item in msgs)
    
69.                     {
    
70.                         await socket.SendAsync(item.MsgContent, WebSocketMessageType.Text, true, CancellationToken.None);
    
71.                     }
    
72.                     MESSAGE_POOL.Remove(user);//移除离线消息
    
73.                 }
    ) ]5 Z& o( q- E7 i( v# ]* \! C
74.                 #endregion
    + N  h: d6 p' o% r" g6 ?* `
75. 
    * L# A$ p& v4 n5 C; ]  L
76.                 string descUser = string.Empty;//目的用户
    
77.                 while (true)
    5 @4 w& }7 b% E& O, ]8 _! y2 c
78.                 {
    
79.                     if (socket.State == WebSocketState.Open)
    % v1 V+ G( }& F; a2 Z2 l
80.                     {
    . A3 g- B5 M5 J; D3 U7 E2 c( v
81.                         ArraySegment<byte> buffer = new ArraySegment<byte>(new byte[2048]);
    
82.                         WebSocketReceiveResult result = await socket.ReceiveAsync(buffer, CancellationToken.None);
    - V. U; n" [) H/ G% I/ D5 u5 v
83. 
    
84.                         #region 消息处理（字符截取、消息转发）
    
85.                         try
    
86.                         {
    
87.                             #region 关闭Socket处理，删除连接池
    
88.                             if (socket.State != WebSocketState.Open)//连接关闭
    ( V: w) `3 \9 Y, q7 U3 c+ z
89.                             {
    
90.                                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);//删除连接池
    
91.                                 break;
    " Z' [  W" S2 N8 \3 ]
92.                             }
    * H$ B' ?$ m1 n
93.                             #endregion
    
94. 
    
95.                             string userMsg = Encoding.UTF8.GetString(buffer.Array, 0, result.Count);//发送过来的消息
    
96.                             string[] msgList = userMsg.Split('|');
    + ~& q8 [! W' B  U1 E; A  @  g0 f
97.                             if (msgList.Length == 2)
    
98.                             {
    " u; n  M6 ~7 j$ S7 l, p, d/ f! Z
99.                                 if (msgList[0].Trim().Length > 0)
    
100.                                     descUser = msgList[0].Trim();//记录消息目的用户
     + G: ]+ x3 N, d- n+ w3 ]0 k
101.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(msgList[1]));
     
102.                             }
     
103.                             else
     : _7 q! F$ U5 ]% x- c
104.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(userMsg));
     
105. 
     
106.                             if (CONNECT_POOL.ContainsKey(descUser))//判断客户端是否在线
     " j+ V5 E! R6 L) D
107.                             {
     
108.                                 WebSocket destSocket = CONNECT_POOL[descUser];//目的客户端
     
109.                                 if (destSocket != null && destSocket.State == WebSocketState.Open)
     : @( c1 F) {& Y, e& i
110.                                     await destSocket.SendAsync(buffer, WebSocketMessageType.Text, true, CancellationToken.None);
     5 m* Z8 [4 z2 `$ v
111.                             }
     - W4 q, J$ f4 \0 A
112.                             else
     . V3 {3 n1 T" N( g2 c+ O
113.                             {
     3 s- I8 x6 q7 n; _: E! N5 I. h$ v
114.                                 Task.Run(() =>
     6 T3 u4 H$ V" @# Q; D1 R" [# K
115.                                 {
     9 O( O( F) w& J8 x
116.                                     if (!MESSAGE_POOL.ContainsKey(descUser))//将用户添加至离线消息池中
     + i& M7 O, Z! a3 f- `
117.                                         MESSAGE_POOL.Add(descUser, new List<MessageInfo>());
     
118.                                     MESSAGE_POOL[descUser].Add(new MessageInfo(DateTime.Now, buffer));//添加离线消息
     % P3 N5 X6 C  ?! V
119.                                 });
     # J& i. f2 c0 j( J+ _; H( f; d9 C& w
120.                             }
     " N: n+ k  Y" }& g- Z# g0 M9 q- D
121.                         }
     
122.                         catch (Exception exs)
     
123.                         {
     . x$ h" c1 A2 f! q
124.                             //消息转发异常处理，本次消息忽略 继续监听接下来的消息
     
125.                         }
     , G/ ^$ l; p! I& a& W* W& p
126.                         #endregion
     
127.                     }
     6 m/ I0 y( i& v
128.                     else
     0 e& O% b# }+ [
129.                     {
     
130.                         break;
     # E# r; R! q; S2 ]. Z
131.                     }
     
132.                 }//while end
     
133.             }
     1 K$ w' R/ K& I3 l5 p
134.             catch (Exception ex)
     , A; |, r* q1 p& n+ r
135.             {
     0 O5 \' u$ |5 Z3 k6 @. y6 a
136.                 //整体异常处理
     8 }& F2 l3 c0 c4 i2 _8 q
137.                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);
     ' L9 [3 o$ r, @: U* p) m- D
138.             }
     7 @/ m( N: m" |( S# j
139.         }
     
140. 
     1 Z0 S0 D) _/ p7 ^, F+ R! Y( d
141. 
     
142.         public bool IsReusable
     / I; b! Q7 o# q/ w
143.         {
     $ g- q/ w/ e1 v
144.             get
     ) t! G! T/ J" j  `
145.             {
     
146.                 return false;
     
147.             }
     
148.         }
     
149.     }
     
150. }

复制代码

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

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