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

admin

服务器端代码编写

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

. R' g$ T1 ^, u7 u) W5 \

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

3.Handler1.ashx代码如下：

1. using System;
   
2. using System.Collections.Generic;
   4 K' f. U% m# L+ i) j% l
3. using System.Linq;
   9 h2 ^' l; n& C  N+ m1 T7 H- V
4. using System.Net.WebSockets;
   + d- t+ Z" i4 ~3 r% b
5. using System.Text;
   ) U$ I' ^( Z3 Q# W4 q9 ^
6. using System.Threading;
   
7. using System.Threading.Tasks;
   1 m: x/ Q9 ^+ V& @+ t; @* y$ E
8. using System.Web;
   6 x  t# I  k  y! v, A! V5 g
9. using System.Web.WebSockets;
   
10. 
    
11. 
    " G+ P5 l/ Y1 R0 A
12. namespace WebApplicationWebsocketHandler
    . x) N4 T# j1 M% r  @3 S& x
13. {
    / S# {2 D) Q( V, [& }6 A7 U" x
14.     /// <summary>
    3 A1 b+ i- b$ o) P
15.     /// 离线消息
    
16.     /// </summary>
    
17.     public class MessageInfo
    3 J) U& N, L# I, f- y! n
18.     {
    
19.         public MessageInfo(DateTime _MsgTime, ArraySegment<byte> _MsgContent)
    
20.         {
    4 W+ n/ i: C  ~/ ?2 S. d. Y
21.             MsgTime = _MsgTime;
    
22.             MsgContent = _MsgContent;
    $ I; U2 l$ o2 y2 F: n
23.         }
    
24.         public DateTime MsgTime { get; set; }
    5 ]+ q* @; p" l1 o8 Q4 ]
25.         public ArraySegment<byte> MsgContent { get; set; }
    
26.     }
    # W# V3 L8 j! N" w4 @  w. L
27. 
    ; @: a# A8 |7 W2 _* D' _+ k: ]! Q
28. 
    
29. 
    
30. 
    
31.     /// <summary>
    : Z" J- Z  Y; _+ r9 Z) h
32.     /// Handler1 的摘要说明
    
33.     /// </summary>
    # l5 _2 _2 N5 \, G
34.     public class Handler1 : IHttpHandler
    0 s' J/ @: J* y8 u
35.     {
    
36.         private static Dictionary<string, WebSocket> CONNECT_POOL = new Dictionary<string, WebSocket>();//用户连接池
    , F" z( Q) v# J1 E% C1 i
37.         private static Dictionary<string, List<MessageInfo>> MESSAGE_POOL = new Dictionary<string, List<MessageInfo>>();//离线消息池
    
38.         public void ProcessRequest(HttpContext context)
    
39.         {
    ! y$ x7 p9 T2 X: q+ Z
40.             //context.Response.ContentType = "text/plain";
    
41.             //context.Response.Write("Hello World");
    5 X0 @  Q1 k( _# _
42.             if (context.IsWebSocketRequest)
    1 F/ U8 p+ L0 V; ]+ H. y, n3 A, e
43.             {
    
44.                 context.AcceptWebSocketRequest(ProcessChat);
    1 v8 P# w9 T' x8 y1 w2 ~$ ]
45.             }
    6 q  t3 B$ R/ X0 R
46.         }
    * C! M! M% T4 ~8 A3 v! D$ d
47. 
    5 X' u1 {. n, \9 C" g% X# s
48.         private async Task ProcessChat(AspNetWebSocketContext context)
    % N) w( }: T( u1 ]+ v! f1 i$ \9 f
49.         {
    0 E( U; j7 A$ H* N0 _( G4 Z' `
50.             WebSocket socket = context.WebSocket;
    9 e5 U1 @% W' Z: L0 z# y
51.             string user = context.QueryString["user"].ToString();
    
52. 
    
53.             try
    
54.             {
    
55.                 #region 用户添加连接池
    
56.                 //第一次open时，添加到连接池中
    / t6 N5 A9 e1 S: ~. s
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;
    
62.                 #endregion
    
63. 
    ' G1 O& ]! J/ P
64.                 #region 离线消息处理
    8 o9 l9 J- f. W" e
65.                 if (MESSAGE_POOL.ContainsKey(user))
    & p! {/ O, d5 N! U
66.                 {
    
67.                     List<MessageInfo> msgs = MESSAGE_POOL[user];
    
68.                     foreach (MessageInfo item in msgs)
    
69.                     {
    ' ?; v% U8 L, ]+ X' [* {
70.                         await socket.SendAsync(item.MsgContent, WebSocketMessageType.Text, true, CancellationToken.None);
    ; p. ?: _- Q6 f; v9 T$ g( X& g( D& u
71.                     }
    6 {8 t+ L9 |% C: x
72.                     MESSAGE_POOL.Remove(user);//移除离线消息
    
73.                 }
    
74.                 #endregion
    
75. 
    4 x0 h: I8 g" Q- T: i
76.                 string descUser = string.Empty;//目的用户
    
77.                 while (true)
    " P  O- e4 P( ]5 g$ m
78.                 {
    
79.                     if (socket.State == WebSocketState.Open)
    # p# O, s. ^; f$ W4 }
80.                     {
    
81.                         ArraySegment<byte> buffer = new ArraySegment<byte>(new byte[2048]);
    
82.                         WebSocketReceiveResult result = await socket.ReceiveAsync(buffer, CancellationToken.None);
    
83. 
    
84.                         #region 消息处理（字符截取、消息转发）
    
85.                         try
    % t. c  X: |; L& L* t
86.                         {
    
87.                             #region 关闭Socket处理，删除连接池
    , `1 A- k1 G- ^% [
88.                             if (socket.State != WebSocketState.Open)//连接关闭
    % Q: \- T8 e6 u( o1 Q# \8 E1 z
89.                             {
    
90.                                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);//删除连接池
    
91.                                 break;
    
92.                             }
    
93.                             #endregion
    1 K: Q7 x- o/ v& w% C7 V6 L2 b
94. 
    1 P/ b, n5 `( W" e
95.                             string userMsg = Encoding.UTF8.GetString(buffer.Array, 0, result.Count);//发送过来的消息
    2 ~2 f! _+ Z7 D* \
96.                             string[] msgList = userMsg.Split('|');
    3 P9 D& J$ U2 y# E: ~8 w; b/ B
97.                             if (msgList.Length == 2)
    1 R0 d8 W1 ]/ V
98.                             {
    
99.                                 if (msgList[0].Trim().Length > 0)
    , P1 U8 @+ O! l) O% m! b5 w8 }( z# }3 }
100.                                     descUser = msgList[0].Trim();//记录消息目的用户
     8 @% N" ~' {9 e- f! }3 @: M
101.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(msgList[1]));
     4 Y) ~  Q1 ?* X7 [7 ~
102.                             }
     
103.                             else
     
104.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(userMsg));
     
105. 
     
106.                             if (CONNECT_POOL.ContainsKey(descUser))//判断客户端是否在线
     
107.                             {
     . n) x0 T, I$ B3 {
108.                                 WebSocket destSocket = CONNECT_POOL[descUser];//目的客户端
     
109.                                 if (destSocket != null && destSocket.State == WebSocketState.Open)
     
110.                                     await destSocket.SendAsync(buffer, WebSocketMessageType.Text, true, CancellationToken.None);
     2 q, d1 {. L1 K8 S2 f/ E
111.                             }
     
112.                             else
     
113.                             {
     / i5 Z7 E/ D+ N( o
114.                                 Task.Run(() =>
     
115.                                 {
     
116.                                     if (!MESSAGE_POOL.ContainsKey(descUser))//将用户添加至离线消息池中
     
117.                                         MESSAGE_POOL.Add(descUser, new List<MessageInfo>());
     
118.                                     MESSAGE_POOL[descUser].Add(new MessageInfo(DateTime.Now, buffer));//添加离线消息
     8 B6 q& x+ L6 c
119.                                 });
     , p' k; B/ O3 j; i
120.                             }
     
121.                         }
     
122.                         catch (Exception exs)
     7 l+ F- |2 z# ]$ \+ t/ K- e& m
123.                         {
     + R) V! ]# u6 \. e1 p1 t# d% C9 c' d
124.                             //消息转发异常处理，本次消息忽略 继续监听接下来的消息
     . k- p8 d' e, c( n
125.                         }
     . O( N7 k+ \9 S# f- K
126.                         #endregion
     
127.                     }
     
128.                     else
     
129.                     {
     
130.                         break;
     
131.                     }
     
132.                 }//while end
     & r. t; G( o  n" Y
133.             }
     5 V( ]. _- c- c2 V' h4 k
134.             catch (Exception ex)
     $ p, f( l% \/ S
135.             {
     
136.                 //整体异常处理
     
137.                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);
       z! ^! x- k! F  n
138.             }
     
139.         }
     : M& i6 J8 g( ]# _9 H
140. 
     
141. 
     
142.         public bool IsReusable
     * H  x: ~* ^  ~
143.         {
     ! \/ d$ ?( B- D1 w
144.             get
     
145.             {
     
146.                 return false;
     
147.             }
     " Q$ m) W3 q3 l* o$ }# b7 X: H2 w
148.         }
     
149.     }
     
150. }

复制代码

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

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

) ]% X( M$ n' B0 V- E( n6 c