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

admin

服务器端代码编写

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

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

3.Handler1.ashx代码如下：

1. using System;
   2 t/ B( m+ ]" e  I
2. using System.Collections.Generic;
   
3. using System.Linq;
   
4. using System.Net.WebSockets;
   
5. using System.Text;
   
6. using System.Threading;
   ( a" ?1 Z9 K0 ?/ U& W# n0 s3 g1 h0 ]
7. using System.Threading.Tasks;
   % G* Y- H" Z" J% [$ t6 j
8. using System.Web;
   
9. using System.Web.WebSockets;
   
10. 
    3 @: e( ~! f5 [! ~& i
11. 
    
12. namespace WebApplicationWebsocketHandler
    + r3 [9 w( T  q3 z% h1 a. m
13. {
    1 ?7 C0 U, G1 _# v* ~. ~% q
14.     /// <summary>
    ; z7 R- u3 _, I% ^  j% H: c
15.     /// 离线消息
    
16.     /// </summary>
    & g; B. p; A- Z6 D
17.     public class MessageInfo
    
18.     {
    & V3 e9 @9 f$ I) ^- U& \3 J0 _
19.         public MessageInfo(DateTime _MsgTime, ArraySegment<byte> _MsgContent)
    ! A7 V2 h3 Z. c. t, x
20.         {
    
21.             MsgTime = _MsgTime;
    $ r1 J3 n/ Q9 Y6 v/ L( h8 ~) B+ p0 l
22.             MsgContent = _MsgContent;
    
23.         }
    * N# u1 h7 v% K" v; J* W
24.         public DateTime MsgTime { get; set; }
    1 k3 K. [/ Z# K! a6 E
25.         public ArraySegment<byte> MsgContent { get; set; }
    
26.     }
    
27. 
    8 o. i! {( V: D( x
28. 
    ( t1 c: b: V9 ?# y. n
29. 
      z- i2 \5 U# c4 y
30. 
    
31.     /// <summary>
    ( U: a  n" ^, l( n& T
32.     /// Handler1 的摘要说明
    4 x, _8 b6 {7 S; U
33.     /// </summary>
    
34.     public class Handler1 : IHttpHandler
    / S: q  U3 J3 M# Y$ W& n
35.     {
    1 n' H! Y' u  ~/ u# @$ m: v
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>>();//离线消息池
    1 R4 D( K4 m# b+ ?# Y& Y: ]- x
38.         public void ProcessRequest(HttpContext context)
    
39.         {
    
40.             //context.Response.ContentType = "text/plain";
    
41.             //context.Response.Write("Hello World");
    : V3 v* j( a4 d- M1 Y6 ]1 n8 J' U. N
42.             if (context.IsWebSocketRequest)
    
43.             {
    
44.                 context.AcceptWebSocketRequest(ProcessChat);
    
45.             }
    
46.         }
    ) I& N: I3 v7 P$ ?5 k
47. 
    * S1 j* ~: x: e5 o/ D
48.         private async Task ProcessChat(AspNetWebSocketContext context)
    6 m6 C$ `% N( c+ r; ~5 i6 H9 g
49.         {
    
50.             WebSocket socket = context.WebSocket;
    
51.             string user = context.QueryString["user"].ToString();
    
52. 
    $ S+ ~9 @. k+ [2 a
53.             try
    
54.             {
    
55.                 #region 用户添加连接池
    2 @0 |' i: t+ D' z1 c
56.                 //第一次open时，添加到连接池中
    ( j) Y( O. U/ v* B1 x
57.                 if (!CONNECT_POOL.ContainsKey(user))
    
58.                     CONNECT_POOL.Add(user, socket);//不存在，添加
    7 w( \, a. b- R# b
59.                 else
    
60.                     if (socket != CONNECT_POOL[user])//当前对象不一致，更新
    
61.                         CONNECT_POOL[user] = socket;
    
62.                 #endregion
    
63. 
    
64.                 #region 离线消息处理
    
65.                 if (MESSAGE_POOL.ContainsKey(user))
    - H2 U% m, d  ~# I' P0 D
66.                 {
    
67.                     List<MessageInfo> msgs = MESSAGE_POOL[user];
    ( `* J( `, k3 j
68.                     foreach (MessageInfo item in msgs)
    1 y5 C5 j; t" ~' w% a8 M/ l+ X  y
69.                     {
    
70.                         await socket.SendAsync(item.MsgContent, WebSocketMessageType.Text, true, CancellationToken.None);
    
71.                     }
    2 D+ o( n; X: w, i
72.                     MESSAGE_POOL.Remove(user);//移除离线消息
    
73.                 }
    
74.                 #endregion
    1 s+ }% Q" y7 o* l3 p6 t
75. 
    2 o# F, g, U1 P& ~* n7 V$ K
76.                 string descUser = string.Empty;//目的用户
    
77.                 while (true)
    3 X4 }* A8 v+ G5 I+ b
78.                 {
      P; n' S8 j7 Q  D8 p
79.                     if (socket.State == WebSocketState.Open)
    
80.                     {
    
81.                         ArraySegment<byte> buffer = new ArraySegment<byte>(new byte[2048]);
    . Y5 v& H1 e8 f! J4 P
82.                         WebSocketReceiveResult result = await socket.ReceiveAsync(buffer, CancellationToken.None);
    % v# f+ Z6 P. t
83. 
    
84.                         #region 消息处理（字符截取、消息转发）
    ! C4 o0 F6 W/ ]/ @- b: r
85.                         try
    
86.                         {
    0 @0 c# Z% n6 [* u
87.                             #region 关闭Socket处理，删除连接池
    
88.                             if (socket.State != WebSocketState.Open)//连接关闭
    
89.                             {
    3 q+ y* ?3 R% r) I0 `" ~
90.                                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);//删除连接池
    - j- j# D9 T8 p8 [6 B/ I
91.                                 break;
    
92.                             }
    . J" ?& W/ |; ~$ e6 U% j: }
93.                             #endregion
    % G. w- |; V, L- W% Z+ N
94. 
    0 g( p4 N  ?$ \, d0 S
95.                             string userMsg = Encoding.UTF8.GetString(buffer.Array, 0, result.Count);//发送过来的消息
    % s  d! y+ i5 c2 p) g; R5 J
96.                             string[] msgList = userMsg.Split('|');
    * X& w4 h3 Y) t- ^* v4 I+ [
97.                             if (msgList.Length == 2)
    9 n8 `+ d) j' x. ]
98.                             {
    
99.                                 if (msgList[0].Trim().Length > 0)
    
100.                                     descUser = msgList[0].Trim();//记录消息目的用户
     
101.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(msgList[1]));
     
102.                             }
     2 G3 Q  X3 j, x, ?7 Y
103.                             else
     
104.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(userMsg));
     
105. 
     
106.                             if (CONNECT_POOL.ContainsKey(descUser))//判断客户端是否在线
     , l; U. ^# m, \) G  h4 N
107.                             {
     1 |/ W% k  T% f8 N4 i  y
108.                                 WebSocket destSocket = CONNECT_POOL[descUser];//目的客户端
     : S/ r1 t. v, y3 y* C9 l" `, ?
109.                                 if (destSocket != null && destSocket.State == WebSocketState.Open)
     
110.                                     await destSocket.SendAsync(buffer, WebSocketMessageType.Text, true, CancellationToken.None);
     : D9 c1 f$ D; T5 a8 E
111.                             }
     
112.                             else
     
113.                             {
     + x$ O% M2 W2 Q, B% x3 X# u% z
114.                                 Task.Run(() =>
       v8 j9 X- n: e! g
115.                                 {
     
116.                                     if (!MESSAGE_POOL.ContainsKey(descUser))//将用户添加至离线消息池中
     ) ~3 Z4 t: B$ C2 G' i6 @0 |* D- S
117.                                         MESSAGE_POOL.Add(descUser, new List<MessageInfo>());
     
118.                                     MESSAGE_POOL[descUser].Add(new MessageInfo(DateTime.Now, buffer));//添加离线消息
     - g0 K2 g6 P* p$ c) l  U: N5 W
119.                                 });
     2 f3 U- P, h$ V. i
120.                             }
     * p: L% Z: R9 ?% ]. o' |4 l
121.                         }
     
122.                         catch (Exception exs)
     
123.                         {
     1 L- I$ X  Y  O# y2 x: N; }, ?
124.                             //消息转发异常处理，本次消息忽略 继续监听接下来的消息
     
125.                         }
     + A# M% |/ O: @6 ?) y! k
126.                         #endregion
       i! \  @  Q7 E8 R8 ^" s* i
127.                     }
     
128.                     else
     * X2 W; k8 B( T- y
129.                     {
     
130.                         break;
     
131.                     }
     8 @, `% n8 E; n
132.                 }//while end
     # R& D3 z+ U5 d# Y' K$ f- i0 o
133.             }
     & z2 F5 q: H$ I) ~1 ]/ ?
134.             catch (Exception ex)
     3 R8 y) i, f  r
135.             {
     ( M  f3 N1 k' r% s' G/ a
136.                 //整体异常处理
     
137.                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);
     * C" O1 ^' I( B$ O1 ~2 z
138.             }
     * N" c4 a. \# g0 Q, g
139.         }
     
140. 
     , V4 k8 X7 w. S7 S' m
141. 
     3 c6 j% Y5 Z; j
142.         public bool IsReusable
     * s4 m9 g7 p: \1 u# k" _' Q6 M
143.         {
     
144.             get
     . g0 y' o5 |6 O; h
145.             {
     . i  V4 Z$ U6 i9 i& h6 F8 |
146.                 return false;
     ; u3 M0 k6 G: J! x; H
147.             }
     5 k" W9 ~% V7 F% p2 m' ^) X
148.         }
     
149.     }
     + _" ?- {4 ?; ]
150. }

复制代码

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

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