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

admin

服务器端代码编写

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

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

3.Handler1.ashx代码如下：

1. using System;
   ) o0 n. G! O; a: k
2. using System.Collections.Generic;
   
3. using System.Linq;
   
4. using System.Net.WebSockets;
   6 a" y0 w3 u0 r8 i! v7 x6 e% d- S
5. using System.Text;
   
6. using System.Threading;
   8 U: R9 P1 k2 Q! y
7. using System.Threading.Tasks;
   . ^2 o% o- e1 h8 G
8. using System.Web;
   
9. using System.Web.WebSockets;
   
10. 
    
11. 
    
12. namespace WebApplicationWebsocketHandler
    $ n9 d' S) x' U. g' `( ]
13. {
    
14.     /// <summary>
    , |) W0 s) f2 Y' s  O( Z3 c9 m# a3 H
15.     /// 离线消息
    
16.     /// </summary>
    6 U: M4 u1 O! T, ]! y% p2 l6 k
17.     public class MessageInfo
    ) c( X( m- Y( k' _' W/ q! p
18.     {
    * B6 F. Z, e# x; s
19.         public MessageInfo(DateTime _MsgTime, ArraySegment<byte> _MsgContent)
    ( Q: q. v1 Q) Y/ ^/ I! B" q
20.         {
    3 `1 t7 K8 t3 ~: X0 E9 c- u  h
21.             MsgTime = _MsgTime;
    
22.             MsgContent = _MsgContent;
    
23.         }
    6 r7 J1 E. p# c+ V! ~
24.         public DateTime MsgTime { get; set; }
    # @6 Q+ L2 C( X- S( M
25.         public ArraySegment<byte> MsgContent { get; set; }
    ) Q+ g1 Q; C0 ]7 Y2 J% N$ V
26.     }
    
27. 
    ' a) M0 h+ _+ P5 G4 j/ M1 ~0 l
28. 
    
29. 
    ) V' c+ ^# Y/ V( K+ k, W
30. 
    
31.     /// <summary>
    1 p( f3 q& g# H
32.     /// Handler1 的摘要说明
    
33.     /// </summary>
    ! @0 U. {$ `! }1 b$ w2 u/ i. e
34.     public class Handler1 : IHttpHandler
    
35.     {
    : Z. l0 O0 R; q8 D( y8 O& [3 W  ~  z
36.         private static Dictionary<string, WebSocket> CONNECT_POOL = new Dictionary<string, WebSocket>();//用户连接池
    9 v1 B" l: L5 b
37.         private static Dictionary<string, List<MessageInfo>> MESSAGE_POOL = new Dictionary<string, List<MessageInfo>>();//离线消息池
    
38.         public void ProcessRequest(HttpContext context)
    
39.         {
    - ?0 w% q% n% v0 E& D
40.             //context.Response.ContentType = "text/plain";
    $ l  g/ I0 i  b( V: U# ~) P$ ~
41.             //context.Response.Write("Hello World");
    - j* j! w) |1 Q) R$ A0 x" m
42.             if (context.IsWebSocketRequest)
    1 o9 o/ c  P$ C8 D( s
43.             {
    
44.                 context.AcceptWebSocketRequest(ProcessChat);
    
45.             }
    7 M% i% J2 V1 z
46.         }
    
47. 
    # h- o+ S" l" w0 }5 D& D
48.         private async Task ProcessChat(AspNetWebSocketContext context)
    
49.         {
    8 g5 e0 e7 t. p& P2 A
50.             WebSocket socket = context.WebSocket;
    
51.             string user = context.QueryString["user"].ToString();
    4 z, c) N  E8 r
52. 
    
53.             try
    - r6 w$ m' F0 t3 `8 N# z3 m
54.             {
    
55.                 #region 用户添加连接池
    % B9 X+ k* d- @7 ^/ b! @  l$ C
56.                 //第一次open时，添加到连接池中
    
57.                 if (!CONNECT_POOL.ContainsKey(user))
    
58.                     CONNECT_POOL.Add(user, socket);//不存在，添加
    - B: S  }/ Y4 A( u( d
59.                 else
    ( O& t; |& E! y1 ^
60.                     if (socket != CONNECT_POOL[user])//当前对象不一致，更新
    ; F4 h+ L4 b' [# i9 \( s
61.                         CONNECT_POOL[user] = socket;
    
62.                 #endregion
    
63. 
    
64.                 #region 离线消息处理
    7 b  Q# j2 F2 T8 e. |1 b
65.                 if (MESSAGE_POOL.ContainsKey(user))
    
66.                 {
      M6 [" l6 r7 N7 [
67.                     List<MessageInfo> msgs = MESSAGE_POOL[user];
    
68.                     foreach (MessageInfo item in msgs)
    $ d2 {* {/ k! n4 }1 C
69.                     {
    % a' P: u+ V6 E* F# H( t. ^( o% w
70.                         await socket.SendAsync(item.MsgContent, WebSocketMessageType.Text, true, CancellationToken.None);
    
71.                     }
    
72.                     MESSAGE_POOL.Remove(user);//移除离线消息
    ' f' H# r2 t1 L* @% X& W
73.                 }
    ' t: ?: a: N& o1 P* i
74.                 #endregion
    
75. 
    ' [0 B2 L6 i- T, p
76.                 string descUser = string.Empty;//目的用户
    
77.                 while (true)
    
78.                 {
    
79.                     if (socket.State == WebSocketState.Open)
    & V3 b+ ^# ~( f* ^8 h) ~( h3 k0 S
80.                     {
    
81.                         ArraySegment<byte> buffer = new ArraySegment<byte>(new byte[2048]);
    ) L# e; Z6 [* n# a% a
82.                         WebSocketReceiveResult result = await socket.ReceiveAsync(buffer, CancellationToken.None);
    % ]' B# X  E: X7 z, D& l6 x0 O& T
83. 
    ! I3 h9 e9 q  P6 x2 R& j1 H1 Z+ Z
84.                         #region 消息处理（字符截取、消息转发）
    
85.                         try
    
86.                         {
    ! h, [& Q! H$ }+ Z# e
87.                             #region 关闭Socket处理，删除连接池
    
88.                             if (socket.State != WebSocketState.Open)//连接关闭
    
89.                             {
    # V* X7 S. B* ]7 w
90.                                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);//删除连接池
    
91.                                 break;
    . m% o" u! ~$ f6 |
92.                             }
    
93.                             #endregion
    ! y4 _( ^$ T5 X
94. 
    
95.                             string userMsg = Encoding.UTF8.GetString(buffer.Array, 0, result.Count);//发送过来的消息
    
96.                             string[] msgList = userMsg.Split('|');
    . q7 a/ {! k& J2 M. f: G
97.                             if (msgList.Length == 2)
    
98.                             {
    / s1 W2 H9 k# Z
99.                                 if (msgList[0].Trim().Length > 0)
    
100.                                     descUser = msgList[0].Trim();//记录消息目的用户
     
101.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(msgList[1]));
       L9 M7 g* ]/ u# Y! W
102.                             }
     " v- g4 S3 t1 r6 r
103.                             else
     5 T5 o8 v2 u( F  I( X. ]% ?& J& ^
104.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(userMsg));
     ; G' _; k! Y! l; E! i
105. 
     9 C- M6 i5 Z& W) \  }8 Z- h( b
106.                             if (CONNECT_POOL.ContainsKey(descUser))//判断客户端是否在线
     
107.                             {
     
108.                                 WebSocket destSocket = CONNECT_POOL[descUser];//目的客户端
     7 R9 I: N: T, A2 `/ _0 A- I
109.                                 if (destSocket != null && destSocket.State == WebSocketState.Open)
     
110.                                     await destSocket.SendAsync(buffer, WebSocketMessageType.Text, true, CancellationToken.None);
     + p" Q1 z" w9 v- G8 I
111.                             }
     : T! \2 [6 V! A: a/ k* H( ]
112.                             else
     
113.                             {
     * X- t, W! h4 c7 Z( I3 [
114.                                 Task.Run(() =>
     
115.                                 {
     
116.                                     if (!MESSAGE_POOL.ContainsKey(descUser))//将用户添加至离线消息池中
     ( R4 b3 t/ g3 F# |
117.                                         MESSAGE_POOL.Add(descUser, new List<MessageInfo>());
     0 {- p: F+ r. E, o" ]
118.                                     MESSAGE_POOL[descUser].Add(new MessageInfo(DateTime.Now, buffer));//添加离线消息
     7 c* X7 X8 d- S- r8 |2 R
119.                                 });
     2 N; z4 t* j6 R& L" ?
120.                             }
     4 ]; w$ ^2 W$ v/ z
121.                         }
     + G5 }( ?' w1 z' e0 g2 }. E+ _' V. P
122.                         catch (Exception exs)
     
123.                         {
     $ r; G: V1 b9 g5 |+ D: c% c0 X6 |
124.                             //消息转发异常处理，本次消息忽略 继续监听接下来的消息
     
125.                         }
     
126.                         #endregion
     $ z* i0 G6 q( ~5 K7 a0 s. }
127.                     }
     
128.                     else
     7 R  ?8 ?6 N0 o  e0 Z& s  E" g; [
129.                     {
     ) F6 Y/ }: _  X* l, O  k3 x- o5 @
130.                         break;
     
131.                     }
     4 m/ U" G% [7 I1 p. [; G4 y6 \
132.                 }//while end
     9 w1 z3 U0 P9 N) _% X; Z
133.             }
     
134.             catch (Exception ex)
     . q2 }; \* J+ v" K
135.             {
     
136.                 //整体异常处理
     
137.                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);
     2 N: G* w% x  |) }! ^1 @
138.             }
     
139.         }
     / c7 X6 A" U5 C1 K# E! e& w" f
140. 
     + o+ Y+ i9 H( a6 N6 [3 O
141. 
     7 D$ N% b1 D, ~$ r, I- a
142.         public bool IsReusable
     ! y9 {, F% S! \) J# @: S: W
143.         {
     
144.             get
     
145.             {
     : s. {: D$ u. L
146.                 return false;
     7 W6 o. n- r( R4 Y3 J% s
147.             }
     
148.         }
     
149.     }
     
150. }

复制代码

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

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