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

admin

服务器端代码编写

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

" V, }6 }1 f5 L% O2 z+ }

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

3.Handler1.ashx代码如下：

1. using System;
   & F. J0 J1 n# f9 R0 Q
2. using System.Collections.Generic;
   
3. using System.Linq;
   
4. using System.Net.WebSockets;
   
5. using System.Text;
   / M9 {; q1 X* r. p% H2 ]. |
6. using System.Threading;
   
7. using System.Threading.Tasks;
   9 p" x9 |1 P6 _; l7 D
8. using System.Web;
   
9. using System.Web.WebSockets;
   " b1 N: P# C7 J; Q
10. 
    
11. 
    
12. namespace WebApplicationWebsocketHandler
    
13. {
    
14.     /// <summary>
    
15.     /// 离线消息
    
16.     /// </summary>
    ) R5 b& Q1 ~6 [) X, d' v
17.     public class MessageInfo
    
18.     {
    
19.         public MessageInfo(DateTime _MsgTime, ArraySegment<byte> _MsgContent)
    + R' k/ N) n7 K+ j$ V& Q" x
20.         {
    
21.             MsgTime = _MsgTime;
    , {- _. U, }4 X0 K, F3 o
22.             MsgContent = _MsgContent;
    , R+ b! N- b  G9 A0 N
23.         }
    
24.         public DateTime MsgTime { get; set; }
    6 a. ?5 |" X/ X& P
25.         public ArraySegment<byte> MsgContent { get; set; }
    
26.     }
    8 _- M" M& r) j; @2 E" }
27. 
    
28. 
    # ^$ x5 E! i! _, ]$ i
29. 
    
30. 
    
31.     /// <summary>
    3 t/ O, P$ U( l8 j
32.     /// Handler1 的摘要说明
    
33.     /// </summary>
    
34.     public class Handler1 : IHttpHandler
    
35.     {
    " I" {) a8 O  s* a( 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>>();//离线消息池
    
38.         public void ProcessRequest(HttpContext context)
    
39.         {
    
40.             //context.Response.ContentType = "text/plain";
    
41.             //context.Response.Write("Hello World");
    
42.             if (context.IsWebSocketRequest)
    
43.             {
    . Y. P: w& \5 R$ y9 f7 x% u2 `
44.                 context.AcceptWebSocketRequest(ProcessChat);
    
45.             }
    
46.         }
    
47. 
    ! R7 Z4 K7 K) i* O+ T5 M' y! r2 u
48.         private async Task ProcessChat(AspNetWebSocketContext context)
    
49.         {
    0 v9 Y" {& v8 H& X" L- t
50.             WebSocket socket = context.WebSocket;
    
51.             string user = context.QueryString["user"].ToString();
    
52. 
    
53.             try
    
54.             {
    9 b* U% N' O% b1 d
55.                 #region 用户添加连接池
    " h& G2 k9 h; ^" _
56.                 //第一次open时，添加到连接池中
    # }$ C( |! a1 m9 j+ Q4 o
57.                 if (!CONNECT_POOL.ContainsKey(user))
    4 G* W! A, I4 L+ C
58.                     CONNECT_POOL.Add(user, socket);//不存在，添加
    
59.                 else
    
60.                     if (socket != CONNECT_POOL[user])//当前对象不一致，更新
    
61.                         CONNECT_POOL[user] = socket;
    0 D) [1 u1 @$ s/ b
62.                 #endregion
    
63. 
    5 e: {- Z  p4 f0 I. z1 b6 l! ^4 X
64.                 #region 离线消息处理
    
65.                 if (MESSAGE_POOL.ContainsKey(user))
    
66.                 {
    + F# t/ X! N3 G$ f
67.                     List<MessageInfo> msgs = MESSAGE_POOL[user];
    
68.                     foreach (MessageInfo item in msgs)
    # J4 o6 M2 c- P
69.                     {
    % a7 l2 Z3 Q3 t2 ]# i" s# ~
70.                         await socket.SendAsync(item.MsgContent, WebSocketMessageType.Text, true, CancellationToken.None);
    8 Q* T/ x% F$ E* O! a
71.                     }
    ' f/ \1 w* X$ n# Z9 B* Q
72.                     MESSAGE_POOL.Remove(user);//移除离线消息
    
73.                 }
    
74.                 #endregion
    
75. 
    $ R/ r$ {! N- [; `* B, _5 X
76.                 string descUser = string.Empty;//目的用户
    
77.                 while (true)
    
78.                 {
    9 X3 q/ n& n7 }; ?' @6 x- T
79.                     if (socket.State == WebSocketState.Open)
    
80.                     {
    
81.                         ArraySegment<byte> buffer = new ArraySegment<byte>(new byte[2048]);
    0 b# k% m7 }) Q2 f( l$ }9 x. }
82.                         WebSocketReceiveResult result = await socket.ReceiveAsync(buffer, CancellationToken.None);
    / r  i+ c4 T2 D4 w( X, W
83. 
    0 T+ r, |& r. J) k5 Y, E
84.                         #region 消息处理（字符截取、消息转发）
    ) t2 p5 S* u  z% J0 O$ {) J8 M
85.                         try
    " Y# L6 L3 \, p' Q0 w2 ]5 y
86.                         {
    - \1 p. ~" K: c6 d) u! @2 S* o
87.                             #region 关闭Socket处理，删除连接池
    
88.                             if (socket.State != WebSocketState.Open)//连接关闭
    : m, u: Q4 [+ @$ B6 `0 u
89.                             {
    ; W) c3 g% k1 c5 m& F
90.                                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);//删除连接池
    ! r0 r3 X& x- C/ E1 z
91.                                 break;
    
92.                             }
    % V  N( z3 o  V# b( ]
93.                             #endregion
    
94. 
    % J4 a, n+ E+ V: A+ x
95.                             string userMsg = Encoding.UTF8.GetString(buffer.Array, 0, result.Count);//发送过来的消息
    
96.                             string[] msgList = userMsg.Split('|');
    : i7 a4 g6 m, d; E6 Q
97.                             if (msgList.Length == 2)
    $ L2 P1 h! n( X( q
98.                             {
    
99.                                 if (msgList[0].Trim().Length > 0)
    
100.                                     descUser = msgList[0].Trim();//记录消息目的用户
     
101.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(msgList[1]));
     $ \$ o! f! G7 V+ `3 r
102.                             }
     ' e+ d2 ~& q6 ?4 E
103.                             else
     1 Z& Y4 r1 T( a( o
104.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(userMsg));
     
105. 
     % x7 A4 E1 m; P7 B' d
106.                             if (CONNECT_POOL.ContainsKey(descUser))//判断客户端是否在线
     
107.                             {
     6 R* x3 M; o4 f6 M- j
108.                                 WebSocket destSocket = CONNECT_POOL[descUser];//目的客户端
     , e, m' _2 E+ R
109.                                 if (destSocket != null && destSocket.State == WebSocketState.Open)
     
110.                                     await destSocket.SendAsync(buffer, WebSocketMessageType.Text, true, CancellationToken.None);
     
111.                             }
     
112.                             else
     ' L- u$ j/ O9 m$ t" ]
113.                             {
     
114.                                 Task.Run(() =>
     0 \  |8 b2 F4 o& [! S0 e( i
115.                                 {
     
116.                                     if (!MESSAGE_POOL.ContainsKey(descUser))//将用户添加至离线消息池中
     ' I5 h$ _4 l  s; |& s
117.                                         MESSAGE_POOL.Add(descUser, new List<MessageInfo>());
     
118.                                     MESSAGE_POOL[descUser].Add(new MessageInfo(DateTime.Now, buffer));//添加离线消息
     ( u- s% n. H5 S" O/ @
119.                                 });
     
120.                             }
     
121.                         }
     . \/ c7 I! I9 H; I7 f# L+ d" e
122.                         catch (Exception exs)
     ! D5 e& u, [( C; c: s  n# ]
123.                         {
     
124.                             //消息转发异常处理，本次消息忽略 继续监听接下来的消息
     7 f- ]# @; u, d, i5 N! k
125.                         }
     . c# M& S" ?" f4 L
126.                         #endregion
     
127.                     }
     ( V3 V/ F9 F- }
128.                     else
     
129.                     {
     
130.                         break;
     # o( ]; O( t7 j0 _: T( a  y1 E. u
131.                     }
     
132.                 }//while end
     ! |/ H5 E% Y0 @, T/ E
133.             }
     & w* d3 c8 g  X: g# m- x# F6 Q* R/ h
134.             catch (Exception ex)
     / R9 d" J- p1 t1 T/ d  a
135.             {
     0 L2 I" M7 d, v
136.                 //整体异常处理
     
137.                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);
     ) g& L* D& G3 Q' A
138.             }
     
139.         }
     ) \( ~! E9 @5 d4 ^7 F( _$ c7 Q
140. 
     4 g  g0 r4 ]# b
141. 
     
142.         public bool IsReusable
     6 z/ I: c3 F+ Y9 D
143.         {
     
144.             get
     
145.             {
     
146.                 return false;
     
147.             }
     
148.         }
     8 d" k( r: _0 [+ N4 b
149.     }
     ( ?/ z/ K5 w' m! |5 u) S$ V
150. }

复制代码

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

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