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

admin

服务器端代码编写

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

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

3.Handler1.ashx代码如下：

1. using System;
   ' v9 l- I/ ~( x! _
2. using System.Collections.Generic;
   
3. using System.Linq;
   6 b8 a0 _% i4 F: B
4. using System.Net.WebSockets;
   5 _: V, C7 N# y4 f: |9 A
5. using System.Text;
   
6. using System.Threading;
   7 J' Y, G8 W0 }7 ~) V8 c3 W  B
7. using System.Threading.Tasks;
   1 X3 R9 W* B; Y
8. using System.Web;
   
9. using System.Web.WebSockets;
   0 @4 I1 W  d  b3 w& a
10. 
    1 R+ w; h  g$ x2 w  w- K
11. 
    
12. namespace WebApplicationWebsocketHandler
    
13. {
    
14.     /// <summary>
    # f5 j* `( @, D% O* {" d& E+ T2 Y
15.     /// 离线消息
    # p+ r" S* U) m) Y
16.     /// </summary>
    / R/ t% s# c% e& |8 J6 R* l4 y7 e
17.     public class MessageInfo
    . q  m; K! Z+ B3 O5 {
18.     {
    
19.         public MessageInfo(DateTime _MsgTime, ArraySegment<byte> _MsgContent)
    2 g- C- [! A' o0 x' u
20.         {
    
21.             MsgTime = _MsgTime;
    4 n! Y/ A* G/ K6 j2 y+ e5 u
22.             MsgContent = _MsgContent;
    6 O# g  @6 c% _, G$ L# Y7 V
23.         }
    
24.         public DateTime MsgTime { get; set; }
    
25.         public ArraySegment<byte> MsgContent { get; set; }
    
26.     }
    6 A7 f% Z) m$ u- U
27. 
    
28. 
    
29. 
    
30. 
    
31.     /// <summary>
    
32.     /// Handler1 的摘要说明
    
33.     /// </summary>
    5 B: O' \- O9 T: S+ y! R
34.     public class Handler1 : IHttpHandler
    1 a# y- |* k3 L0 C" e- S7 e/ _4 u
35.     {
    " M( k7 H, e; |
36.         private static Dictionary<string, WebSocket> CONNECT_POOL = new Dictionary<string, WebSocket>();//用户连接池
    ( `9 P- T& K1 Y7 C
37.         private static Dictionary<string, List<MessageInfo>> MESSAGE_POOL = new Dictionary<string, List<MessageInfo>>();//离线消息池
    
38.         public void ProcessRequest(HttpContext context)
    4 C# Y& N0 [% h' r5 t6 w% M, H
39.         {
    
40.             //context.Response.ContentType = "text/plain";
    
41.             //context.Response.Write("Hello World");
    
42.             if (context.IsWebSocketRequest)
    1 K( r6 \+ A# a) R- G( B; H: A' y% x- G
43.             {
    
44.                 context.AcceptWebSocketRequest(ProcessChat);
    
45.             }
    
46.         }
      p# M& ~  o$ r/ X0 X
47. 
    
48.         private async Task ProcessChat(AspNetWebSocketContext context)
    : E5 H/ [, e" N0 s1 a% @/ I
49.         {
    1 I+ D5 V& R- z; I! \# A
50.             WebSocket socket = context.WebSocket;
    
51.             string user = context.QueryString["user"].ToString();
    & V  P$ C6 C+ K2 j% O9 r% ?9 G- L- G, ]8 j
52. 
    
53.             try
    * r2 D# l/ l6 N. x; j
54.             {
    + i: }$ ~# p4 @; ]8 G
55.                 #region 用户添加连接池
    
56.                 //第一次open时，添加到连接池中
    
57.                 if (!CONNECT_POOL.ContainsKey(user))
    $ |5 W& _' r! U# U7 q; u# o
58.                     CONNECT_POOL.Add(user, socket);//不存在，添加
    # \. Y% r% @- f
59.                 else
    ) R6 J1 M, n9 o0 c0 m6 x0 M; q- S
60.                     if (socket != CONNECT_POOL[user])//当前对象不一致，更新
    * M" J2 ?4 V' {, s! B1 s3 ^
61.                         CONNECT_POOL[user] = socket;
    
62.                 #endregion
    
63. 
    0 A2 N2 n& o9 s4 B
64.                 #region 离线消息处理
    * A% P8 B2 Y" j" z+ W; ^
65.                 if (MESSAGE_POOL.ContainsKey(user))
    
66.                 {
    
67.                     List<MessageInfo> msgs = MESSAGE_POOL[user];
    ; S9 L2 H8 _! w5 @
68.                     foreach (MessageInfo item in msgs)
    
69.                     {
    3 Z9 F  ^' B4 G* T
70.                         await socket.SendAsync(item.MsgContent, WebSocketMessageType.Text, true, CancellationToken.None);
    + [) j( h% O5 h! ~: d/ c! j* t& R
71.                     }
    
72.                     MESSAGE_POOL.Remove(user);//移除离线消息
    $ i4 J3 D9 v1 ^. n$ S/ v/ f$ H
73.                 }
    8 X0 [0 K4 C8 n! n2 ^
74.                 #endregion
    7 Q* D& B& V0 _3 _
75. 
    
76.                 string descUser = string.Empty;//目的用户
    
77.                 while (true)
    % y  q/ o- _. M+ C$ e7 L5 U
78.                 {
    2 @; p$ D3 y1 K: U
79.                     if (socket.State == WebSocketState.Open)
    ) [$ A  W, B) _
80.                     {
    
81.                         ArraySegment<byte> buffer = new ArraySegment<byte>(new byte[2048]);
    
82.                         WebSocketReceiveResult result = await socket.ReceiveAsync(buffer, CancellationToken.None);
    % ]; W( L9 {$ v
83. 
    / m3 D( X9 Q7 J4 N3 P+ d2 O5 S0 J: e
84.                         #region 消息处理（字符截取、消息转发）
    3 R1 D' \2 S+ C% @/ [" e, d
85.                         try
    
86.                         {
    
87.                             #region 关闭Socket处理，删除连接池
    
88.                             if (socket.State != WebSocketState.Open)//连接关闭
    6 e, j! K3 ~6 P
89.                             {
      J# ]; `1 D) D1 D
90.                                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);//删除连接池
    8 N+ }* W4 _! p: W
91.                                 break;
    0 [$ H6 d' R) K8 E+ o4 y; @+ d6 ^0 q
92.                             }
    ! e8 _) R% l/ o9 m4 \1 x  W
93.                             #endregion
    & ~# ~8 D' }/ g# `% ?
94. 
    
95.                             string userMsg = Encoding.UTF8.GetString(buffer.Array, 0, result.Count);//发送过来的消息
    ! q2 C1 t+ w, I- V' {" X
96.                             string[] msgList = userMsg.Split('|');
    4 ?9 N! Y# e8 m: S
97.                             if (msgList.Length == 2)
    
98.                             {
      A9 h& K1 Q# x) ]  a- ]
99.                                 if (msgList[0].Trim().Length > 0)
    " E6 Y* f0 f( x, t  a+ j8 H/ C* D0 G$ V
100.                                     descUser = msgList[0].Trim();//记录消息目的用户
     2 }4 m$ v5 S' d: K
101.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(msgList[1]));
     - L6 D" b; ^. p# A3 D
102.                             }
     
103.                             else
     : a: s- {' s/ O3 b  L
104.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(userMsg));
     . o; ~, ?0 M& e/ ~- e% B
105. 
     
106.                             if (CONNECT_POOL.ContainsKey(descUser))//判断客户端是否在线
     
107.                             {
     
108.                                 WebSocket destSocket = CONNECT_POOL[descUser];//目的客户端
     9 W" q& [1 |; V- h. X( L1 i# L) c& \
109.                                 if (destSocket != null && destSocket.State == WebSocketState.Open)
     0 ~: p, k. a' G/ B6 ~
110.                                     await destSocket.SendAsync(buffer, WebSocketMessageType.Text, true, CancellationToken.None);
     : ?4 I  Z# i' {% L) i
111.                             }
     
112.                             else
     
113.                             {
     * P1 b. e' M$ Q1 Q9 q
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));//添加离线消息
     
119.                                 });
     5 y, e) T: N8 z
120.                             }
       z. G  j, m2 N+ K7 {
121.                         }
     
122.                         catch (Exception exs)
     + d0 P* B; k4 o9 n
123.                         {
     , ^; [2 C' G: {
124.                             //消息转发异常处理，本次消息忽略 继续监听接下来的消息
     ! T! H# E" x: a1 y) `0 w) s
125.                         }
     1 z3 Y# O" n  U8 S
126.                         #endregion
     : d) D7 v2 E! g% }# T' k& j
127.                     }
     
128.                     else
     5 @2 D% D( f2 w
129.                     {
     
130.                         break;
     % ]8 v4 ]  I" ~# ]# r4 A
131.                     }
     0 `& f( p! v; v3 ^8 d
132.                 }//while end
     2 D! T. I$ _6 s8 e& V, p
133.             }
     
134.             catch (Exception ex)
     
135.             {
     
136.                 //整体异常处理
     
137.                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);
     - s, M, u. x* p' [
138.             }
     
139.         }
     + m1 A. h4 Q0 C% K- e& `
140. 
     
141. 
     
142.         public bool IsReusable
     ) |3 F2 B$ K; |
143.         {
     , L/ p! q; ~+ A, e* j( I! f" U) C6 t
144.             get
     
145.             {
     9 R# D2 E+ m& M2 t: w# O
146.                 return false;
     
147.             }
     
148.         }
     
149.     }
     / B8 J( F' e4 n
150. }

复制代码

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

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