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

admin

服务器端代码编写

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

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

3.Handler1.ashx代码如下：

1. using System;
   . G0 c: {4 V( y' d# x# l3 a
2. using System.Collections.Generic;
   : t& }# {9 h0 |" z
3. using System.Linq;
   * @& R9 l0 R7 M2 R' b$ ~$ l; h
4. using System.Net.WebSockets;
   
5. using System.Text;
   
6. using System.Threading;
   1 H6 m& i  }9 l! ]6 o. L
7. using System.Threading.Tasks;
   " a) A6 M3 K0 Y) k
8. using System.Web;
   - J- ~- ~3 l4 d9 c6 c- `
9. using System.Web.WebSockets;
   
10. 
    7 P+ ~  t# u) U6 t$ U5 c$ E
11. 
    , {$ i( f% e, r6 z9 s6 U7 A
12. namespace WebApplicationWebsocketHandler
    
13. {
    7 _/ K4 _8 i" G# L
14.     /// <summary>
    ; p5 p5 s- [% U" c( V7 ?& J
15.     /// 离线消息
    
16.     /// </summary>
    # s- a! o  m3 e6 v7 h
17.     public class MessageInfo
    5 F# I4 i4 i' h2 R. s
18.     {
    
19.         public MessageInfo(DateTime _MsgTime, ArraySegment<byte> _MsgContent)
    , p/ @$ L+ {9 P7 g- E
20.         {
    4 K- @1 ~: t; u
21.             MsgTime = _MsgTime;
    
22.             MsgContent = _MsgContent;
    
23.         }
    
24.         public DateTime MsgTime { get; set; }
    
25.         public ArraySegment<byte> MsgContent { get; set; }
    ; H8 a7 i: F4 h. K- i3 B
26.     }
    
27. 
    : E8 r. [) z5 Q; E# k$ h( D/ A
28. 
    
29. 
    
30. 
    
31.     /// <summary>
    
32.     /// Handler1 的摘要说明
    6 _; g& ?' t7 t/ }* l" \# n: _
33.     /// </summary>
    
34.     public class Handler1 : IHttpHandler
    5 ^4 b8 T7 U: e$ P" u8 }
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>>();//离线消息池
    ( V* G: z' @1 Z' t$ [
38.         public void ProcessRequest(HttpContext context)
    $ e/ t' P+ q1 q0 W
39.         {
    
40.             //context.Response.ContentType = "text/plain";
    ; [/ S! V% H2 ?+ E. n4 \% y1 C' s. r
41.             //context.Response.Write("Hello World");
    . i% I8 [" r- a
42.             if (context.IsWebSocketRequest)
    
43.             {
    . Z0 y( |4 W" x
44.                 context.AcceptWebSocketRequest(ProcessChat);
    
45.             }
    ! \( ?3 Y5 I5 |! @  b3 C" h
46.         }
    
47. 
    
48.         private async Task ProcessChat(AspNetWebSocketContext context)
    3 K8 B- e; R* x. s. f
49.         {
    1 i, o9 [9 W5 d4 G
50.             WebSocket socket = context.WebSocket;
    
51.             string user = context.QueryString["user"].ToString();
    
52. 
    
53.             try
    1 h% _( _! d7 n3 a4 h) v
54.             {
    
55.                 #region 用户添加连接池
    
56.                 //第一次open时，添加到连接池中
    
57.                 if (!CONNECT_POOL.ContainsKey(user))
    
58.                     CONNECT_POOL.Add(user, socket);//不存在，添加
    
59.                 else
    ) i# L3 P! Q0 Y  A* e& [9 G- \& m
60.                     if (socket != CONNECT_POOL[user])//当前对象不一致，更新
    
61.                         CONNECT_POOL[user] = socket;
    
62.                 #endregion
    ; V6 D& ^" c( S  g5 Q
63. 
    
64.                 #region 离线消息处理
    
65.                 if (MESSAGE_POOL.ContainsKey(user))
    9 S! ]6 {4 {  |  ~; k0 k
66.                 {
      q8 S5 W' T. ^
67.                     List<MessageInfo> msgs = MESSAGE_POOL[user];
    ! E/ }- G% b- a) e+ j
68.                     foreach (MessageInfo item in msgs)
    
69.                     {
    ' @' V- G9 Z: {# b2 R' `6 b' i
70.                         await socket.SendAsync(item.MsgContent, WebSocketMessageType.Text, true, CancellationToken.None);
    
71.                     }
    3 h- H& f; Z3 }
72.                     MESSAGE_POOL.Remove(user);//移除离线消息
    : k! o- [$ }; b. _' ~
73.                 }
    ; h% W8 T# X# l4 V' @4 U$ |
74.                 #endregion
    
75. 
    7 ^$ H% j$ @* Y4 m
76.                 string descUser = string.Empty;//目的用户
    
77.                 while (true)
    
78.                 {
    ! y- F& d! f, n: ?. R7 C
79.                     if (socket.State == WebSocketState.Open)
    
80.                     {
    
81.                         ArraySegment<byte> buffer = new ArraySegment<byte>(new byte[2048]);
    
82.                         WebSocketReceiveResult result = await socket.ReceiveAsync(buffer, CancellationToken.None);
    ) A% t, v7 T( O( k/ V* A$ L1 V
83. 
    
84.                         #region 消息处理（字符截取、消息转发）
    . l( O* l# z, P, X8 z! o
85.                         try
    
86.                         {
      Y1 t2 M  \9 b4 |7 j" m" \& \+ l
87.                             #region 关闭Socket处理，删除连接池
    6 e' a9 m+ ~* S
88.                             if (socket.State != WebSocketState.Open)//连接关闭
    
89.                             {
    
90.                                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);//删除连接池
    
91.                                 break;
    ' N9 O; g# h8 L  W+ f; S, @! ~
92.                             }
    , k6 T( W3 ~; w8 B
93.                             #endregion
    5 J' M6 l& v$ }( d  v! Z4 q4 r% n
94. 
    0 d3 A4 `/ O9 D
95.                             string userMsg = Encoding.UTF8.GetString(buffer.Array, 0, result.Count);//发送过来的消息
    5 r' X, x% }  j* z3 h* O9 W
96.                             string[] msgList = userMsg.Split('|');
      [8 k! y* Y/ u" }5 n3 w1 ?! Y
97.                             if (msgList.Length == 2)
    
98.                             {
    
99.                                 if (msgList[0].Trim().Length > 0)
    ) k9 B7 x* ?) a, \" o
100.                                     descUser = msgList[0].Trim();//记录消息目的用户
     8 A; {; L, C' C8 I6 z) O
101.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(msgList[1]));
     * s" k; Y" d2 r
102.                             }
     
103.                             else
     $ ?6 C# C" M( y& P$ I. v
104.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(userMsg));
     ' v, `, A- M, z9 P6 l
105. 
     7 ~' t5 [( Z' p+ ^  w4 W" Q+ M
106.                             if (CONNECT_POOL.ContainsKey(descUser))//判断客户端是否在线
     
107.                             {
     : l  M" ~" m) F; p, e+ ]
108.                                 WebSocket destSocket = CONNECT_POOL[descUser];//目的客户端
     
109.                                 if (destSocket != null && destSocket.State == WebSocketState.Open)
     ( P0 F" K. Y% w* C0 ~1 ^
110.                                     await destSocket.SendAsync(buffer, WebSocketMessageType.Text, true, CancellationToken.None);
     
111.                             }
     
112.                             else
     
113.                             {
     
114.                                 Task.Run(() =>
     + N" q5 X7 m- |( U
115.                                 {
     
116.                                     if (!MESSAGE_POOL.ContainsKey(descUser))//将用户添加至离线消息池中
     % u$ Y- k% o- x* Q6 Y1 X0 l
117.                                         MESSAGE_POOL.Add(descUser, new List<MessageInfo>());
     
118.                                     MESSAGE_POOL[descUser].Add(new MessageInfo(DateTime.Now, buffer));//添加离线消息
     $ a& Z/ a$ k8 }
119.                                 });
     * \% I* |4 S3 R& |! c: H; a
120.                             }
     ! d- F8 U6 |$ J. R( c# q
121.                         }
     - {, q) f6 T8 z
122.                         catch (Exception exs)
     
123.                         {
     
124.                             //消息转发异常处理，本次消息忽略 继续监听接下来的消息
     
125.                         }
     # d$ n# u5 p5 u, |
126.                         #endregion
     
127.                     }
     6 j( f; t- d, i4 g. J
128.                     else
     
129.                     {
     ) |5 f$ A" E: y6 I/ A! s, V
130.                         break;
     
131.                     }
     
132.                 }//while end
     1 d+ x3 Q9 d* m4 v: H& G# \
133.             }
     
134.             catch (Exception ex)
     
135.             {
     
136.                 //整体异常处理
       p. y3 T3 G+ R* @2 K' Z
137.                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);
     5 _5 j$ ?1 S% y% U$ @: V4 O
138.             }
     
139.         }
     
140. 
       y0 @0 K: y4 a, _
141. 
     
142.         public bool IsReusable
     ) N/ z  F+ v, x
143.         {
     ' {& F2 z3 v/ W* B3 d- Q) m( k
144.             get
     0 T0 }5 _4 g6 t# H
145.             {
     
146.                 return false;
     - o: J& t+ v0 s9 W# X, N
147.             }
     
148.         }
     
149.     }
     ; |2 T8 y7 Z, h' e
150. }

复制代码

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

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