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

admin

服务器端代码编写

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

9 Z/ ^3 d0 O9 R- s

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

3.Handler1.ashx代码如下：

1. using System;
   , N% @$ x" y; E1 Z! y
2. using System.Collections.Generic;
   ! S$ r: w, U# x( m0 `
3. using System.Linq;
   , [! h  U: g5 b
4. using System.Net.WebSockets;
   
5. using System.Text;
   
6. using System.Threading;
   
7. using System.Threading.Tasks;
   % m" E- B; v8 R) K1 p7 B- q+ ^
8. using System.Web;
   & r; m# N6 b" m0 U5 {; d* V
9. using System.Web.WebSockets;
   
10. 
    " c  v# m1 ^! W3 e9 c
11. 
    
12. namespace WebApplicationWebsocketHandler
    
13. {
    0 A# x' |- A5 }# Q. ^" n7 H
14.     /// <summary>
    ; ^) h$ m% Q. Q0 b/ ?
15.     /// 离线消息
    
16.     /// </summary>
    / j; [8 C4 S) W; u
17.     public class MessageInfo
    
18.     {
    
19.         public MessageInfo(DateTime _MsgTime, ArraySegment<byte> _MsgContent)
    9 ^% R3 L( T2 |' S
20.         {
    
21.             MsgTime = _MsgTime;
    / \7 |; O; X; ]4 }3 E
22.             MsgContent = _MsgContent;
    / N4 g* f- s  r( g
23.         }
    
24.         public DateTime MsgTime { get; set; }
    
25.         public ArraySegment<byte> MsgContent { get; set; }
    
26.     }
    
27. 
    ! ^' V& s1 O4 [: e: j' p$ M* c
28. 
    
29. 
    ' b/ A# |) ]$ }- X& i; P
30. 
    3 q, v* A. v0 o- k8 S
31.     /// <summary>
    ( S. H: x% y4 ?& x) k! O
32.     /// Handler1 的摘要说明
    
33.     /// </summary>
    
34.     public class Handler1 : IHttpHandler
    
35.     {
    
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>>();//离线消息池
    ' E- C3 `8 q+ @6 L$ ]  v
38.         public void ProcessRequest(HttpContext context)
    
39.         {
    1 B- R5 k) @3 j4 |6 e  u
40.             //context.Response.ContentType = "text/plain";
    . L3 H( C3 |* |1 J, Y; }# @
41.             //context.Response.Write("Hello World");
    8 K; {5 ]4 X7 F. |& t) ?
42.             if (context.IsWebSocketRequest)
    " W0 B3 q: ^+ W- ]; a
43.             {
    - ]% L' B3 ~/ y3 d
44.                 context.AcceptWebSocketRequest(ProcessChat);
    - E( n: ?. G8 ]9 L, R
45.             }
    0 ~/ c/ a8 W' W0 J3 T. \* v8 g
46.         }
    ( M* P. U: W0 v  t
47. 
    
48.         private async Task ProcessChat(AspNetWebSocketContext context)
    ! E9 \& E6 q! o; t1 a9 T; H
49.         {
    5 C$ D& K1 {9 U; h+ Y
50.             WebSocket socket = context.WebSocket;
    + }, L, Z& _" Q. |
51.             string user = context.QueryString["user"].ToString();
    , p% \4 ?6 b( c, d
52. 
    
53.             try
    
54.             {
    
55.                 #region 用户添加连接池
    
56.                 //第一次open时，添加到连接池中
    
57.                 if (!CONNECT_POOL.ContainsKey(user))
    
58.                     CONNECT_POOL.Add(user, socket);//不存在，添加
    
59.                 else
    " }- ]+ q) c; Z
60.                     if (socket != CONNECT_POOL[user])//当前对象不一致，更新
    / |+ k+ m7 N& Q" M" p6 `4 Q! [$ I  C
61.                         CONNECT_POOL[user] = socket;
    
62.                 #endregion
    9 ?  T( |, j4 ]% u/ `8 B; s: `
63. 
    * h+ e, I1 K0 g( t* S" N: Z2 e
64.                 #region 离线消息处理
    
65.                 if (MESSAGE_POOL.ContainsKey(user))
    ; {& h3 m4 v7 t3 a! e6 D
66.                 {
    ; `3 r& I0 ]8 x
67.                     List<MessageInfo> msgs = MESSAGE_POOL[user];
    9 P5 G% a7 u) X) w; o( t+ O
68.                     foreach (MessageInfo item in msgs)
    
69.                     {
    
70.                         await socket.SendAsync(item.MsgContent, WebSocketMessageType.Text, true, CancellationToken.None);
    : R3 B: P" F. K+ A, s  F/ |
71.                     }
    - {8 p: E- h5 p7 D
72.                     MESSAGE_POOL.Remove(user);//移除离线消息
    - ^  z9 i1 e/ {1 n
73.                 }
    5 I% @( f, [" Q/ K
74.                 #endregion
    
75. 
    3 C3 n1 f% s8 O. s  f) n2 u
76.                 string descUser = string.Empty;//目的用户
    
77.                 while (true)
    # j" E: f6 g2 B2 l% A) S
78.                 {
    
79.                     if (socket.State == WebSocketState.Open)
    $ a- j  ?4 D% j2 n2 z0 l
80.                     {
    
81.                         ArraySegment<byte> buffer = new ArraySegment<byte>(new byte[2048]);
    
82.                         WebSocketReceiveResult result = await socket.ReceiveAsync(buffer, CancellationToken.None);
    9 t8 U  y& g/ d/ }
83. 
    
84.                         #region 消息处理（字符截取、消息转发）
    
85.                         try
    9 M* E! j1 i! j1 q5 J4 d) T
86.                         {
    
87.                             #region 关闭Socket处理，删除连接池
    0 q0 E1 K# c4 y/ n) w
88.                             if (socket.State != WebSocketState.Open)//连接关闭
    " }" c- x) P$ h2 x* X6 l* q$ r  w
89.                             {
    2 P7 h, R/ F6 h2 \. C5 U5 D
90.                                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);//删除连接池
    
91.                                 break;
    * X2 [5 ~7 V6 ~; m
92.                             }
    . ]+ g" e3 ~6 h- T. R% h
93.                             #endregion
    
94. 
    
95.                             string userMsg = Encoding.UTF8.GetString(buffer.Array, 0, result.Count);//发送过来的消息
    
96.                             string[] msgList = userMsg.Split('|');
    , ?, d8 ^) q: A- X* Y( f
97.                             if (msgList.Length == 2)
    " i: N+ n6 O1 Q+ f* S4 p9 _+ J+ W1 L
98.                             {
    
99.                                 if (msgList[0].Trim().Length > 0)
    - ?7 K% q  A4 V' a( N
100.                                     descUser = msgList[0].Trim();//记录消息目的用户
     ; S6 P* L7 g: ~& C0 ]
101.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(msgList[1]));
     $ E3 y  ^7 ^9 T/ }8 O$ d4 i0 O
102.                             }
     
103.                             else
     
104.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(userMsg));
     9 y' y7 n% u8 ]3 y
105. 
     " ~: t' X  [1 a& k
106.                             if (CONNECT_POOL.ContainsKey(descUser))//判断客户端是否在线
     
107.                             {
     % o  w" `+ Y2 d' K0 f
108.                                 WebSocket destSocket = CONNECT_POOL[descUser];//目的客户端
     
109.                                 if (destSocket != null && destSocket.State == WebSocketState.Open)
     9 S+ O0 @' V6 K! @
110.                                     await destSocket.SendAsync(buffer, WebSocketMessageType.Text, true, CancellationToken.None);
     
111.                             }
     - x1 {" W1 X+ H. r
112.                             else
     % \7 ^" ?3 X: h2 k6 |
113.                             {
     
114.                                 Task.Run(() =>
     
115.                                 {
     ) n8 C# J0 g5 U* y) {  c# t0 x. |5 N
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.                                 });
     
120.                             }
     8 C3 }6 Y. K# i4 J
121.                         }
     
122.                         catch (Exception exs)
     9 \+ V- F. l* [% f& O
123.                         {
     ( \, g, i9 c2 R+ o) h( N) l
124.                             //消息转发异常处理，本次消息忽略 继续监听接下来的消息
     
125.                         }
     ) r/ P. h+ ]( O+ {# i5 P2 w
126.                         #endregion
     
127.                     }
     
128.                     else
     
129.                     {
     
130.                         break;
     
131.                     }
     
132.                 }//while end
     + B, R: M$ J1 C4 O
133.             }
     
134.             catch (Exception ex)
     9 _* M; G  A0 r6 m* S# I& y# i
135.             {
     
136.                 //整体异常处理
     8 [. v! o. }- f# X6 C
137.                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);
     : ]4 e- h9 e- N0 _/ A
138.             }
     1 p9 B7 v, b2 C0 ]
139.         }
     
140. 
     
141. 
     ( i+ n# D9 X( `5 ]
142.         public bool IsReusable
     6 Q+ N3 a) o+ v, ]- ^
143.         {
     ; |8 s+ B6 a; J
144.             get
     
145.             {
     
146.                 return false;
     / q& K2 h# o. d/ G( R
147.             }
     ) u7 t# V' S$ a; j( o
148.         }
     . H7 h0 a2 Y( {0 N( `; w9 \
149.     }
     * T. b2 }" L8 w) s* T2 o% A" u( U
150. }

复制代码

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

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

( ^  k: A2 C& X6 r9 I/ I7 B