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

admin

服务器端代码编写

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

  G7 Y5 E( Q, q7 l! r

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

3.Handler1.ashx代码如下：

1. using System;
   
2. using System.Collections.Generic;
   
3. using System.Linq;
   
4. using System.Net.WebSockets;
   5 Y& s8 _$ ~! B: a' v  y
5. using System.Text;
   5 H% H4 n3 ?, P* U
6. using System.Threading;
   
7. using System.Threading.Tasks;
   
8. using System.Web;
   
9. using System.Web.WebSockets;
   
10. 
    # l! {) G/ A2 q# X) X
11. 
    5 X1 s$ L& S5 F9 r3 l, ?0 l$ A
12. namespace WebApplicationWebsocketHandler
    8 n, S2 K: z; U0 ~) V
13. {
    
14.     /// <summary>
    7 T8 X) J: K4 p# l7 |/ G2 Z$ o
15.     /// 离线消息
    
16.     /// </summary>
    
17.     public class MessageInfo
    : T" A1 H1 ^9 {
18.     {
    $ X0 t1 r* u  l9 Q3 h4 Q4 e1 ^
19.         public MessageInfo(DateTime _MsgTime, ArraySegment<byte> _MsgContent)
    1 l7 |: y. C, Y7 Y1 b7 T
20.         {
    # m6 {. p/ }# v: s# b7 Y
21.             MsgTime = _MsgTime;
    0 G3 U5 h3 o3 T/ X3 o
22.             MsgContent = _MsgContent;
    * b: g( m# @2 E0 ]1 {
23.         }
    
24.         public DateTime MsgTime { get; set; }
    3 B; z, e$ n1 t5 `& y- [$ R# B
25.         public ArraySegment<byte> MsgContent { get; set; }
    
26.     }
    
27. 
    7 i; q, L% ?6 s. O' q# v$ _( b( e
28. 
    ) I7 u! B5 ~; y5 g4 h, w
29. 
    1 Z3 j( \# @+ R' x! g/ Y
30. 
    2 p0 T+ C8 y* u  f8 v
31.     /// <summary>
    
32.     /// Handler1 的摘要说明
    $ ]2 O9 r  o3 H7 p
33.     /// </summary>
    
34.     public class Handler1 : IHttpHandler
    $ S+ g" h$ Q! G
35.     {
    9 A% |- w8 G) Z
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>>();//离线消息池
    
38.         public void ProcessRequest(HttpContext context)
    
39.         {
    5 L! @( A) p5 n! ]
40.             //context.Response.ContentType = "text/plain";
    1 N, R) k# B* J1 H( a8 b
41.             //context.Response.Write("Hello World");
    
42.             if (context.IsWebSocketRequest)
    
43.             {
    $ D# O, u; M3 J, h3 g7 O
44.                 context.AcceptWebSocketRequest(ProcessChat);
    " w8 Q' b% ?" Z6 z& i) \
45.             }
    
46.         }
    $ a/ e6 f( a/ w
47. 
    
48.         private async Task ProcessChat(AspNetWebSocketContext context)
    1 C2 t1 p2 H9 {1 K5 U' @
49.         {
    
50.             WebSocket socket = context.WebSocket;
    3 X( l2 P& J7 y1 n9 M# a
51.             string user = context.QueryString["user"].ToString();
    1 |, v7 F+ G, V* i! i" @) O
52. 
    7 H  |! h: V' _7 m+ i
53.             try
    & `, v& E$ n4 B- f
54.             {
    $ M* Q- R/ i! f- L6 j! R- G) j4 ?
55.                 #region 用户添加连接池
    5 @! k8 A" T# f
56.                 //第一次open时，添加到连接池中
    
57.                 if (!CONNECT_POOL.ContainsKey(user))
    " c9 H4 j% d+ _4 b- d5 p
58.                     CONNECT_POOL.Add(user, socket);//不存在，添加
    
59.                 else
    / W6 d2 k. V2 c8 h4 ]5 k( e6 W
60.                     if (socket != CONNECT_POOL[user])//当前对象不一致，更新
    
61.                         CONNECT_POOL[user] = socket;
    
62.                 #endregion
    
63. 
    6 v- S$ }4 b% d  p8 f* J$ f
64.                 #region 离线消息处理
    0 g% t/ Q8 q: N
65.                 if (MESSAGE_POOL.ContainsKey(user))
    
66.                 {
    ) w/ K: C/ @# g6 t
67.                     List<MessageInfo> msgs = MESSAGE_POOL[user];
    
68.                     foreach (MessageInfo item in msgs)
    3 V2 X$ R7 C/ y" E+ f7 c
69.                     {
    
70.                         await socket.SendAsync(item.MsgContent, WebSocketMessageType.Text, true, CancellationToken.None);
    
71.                     }
    7 X: R# [- A; f! B, x* ?0 p2 ^
72.                     MESSAGE_POOL.Remove(user);//移除离线消息
    
73.                 }
    
74.                 #endregion
    
75. 
    ; S8 C! u: k9 c: A1 N* z0 }# m
76.                 string descUser = string.Empty;//目的用户
    
77.                 while (true)
    
78.                 {
    
79.                     if (socket.State == WebSocketState.Open)
    - X* ^: o, Z+ d  z/ w4 R2 U: f
80.                     {
    
81.                         ArraySegment<byte> buffer = new ArraySegment<byte>(new byte[2048]);
    
82.                         WebSocketReceiveResult result = await socket.ReceiveAsync(buffer, CancellationToken.None);
    
83. 
    # L  {# ^. ]8 i/ k( c5 D
84.                         #region 消息处理（字符截取、消息转发）
    8 j' |# r7 j/ e. X* h9 C1 d! M
85.                         try
    
86.                         {
    ( F! J/ {1 m& X" Z! p  y! K6 g
87.                             #region 关闭Socket处理，删除连接池
    4 s; d$ I9 t$ [: \+ d
88.                             if (socket.State != WebSocketState.Open)//连接关闭
    
89.                             {
    ( h7 ~7 t7 G$ h% C
90.                                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);//删除连接池
    
91.                                 break;
    
92.                             }
    
93.                             #endregion
    
94. 
    
95.                             string userMsg = Encoding.UTF8.GetString(buffer.Array, 0, result.Count);//发送过来的消息
    
96.                             string[] msgList = userMsg.Split('|');
    
97.                             if (msgList.Length == 2)
    * [. }+ U" k3 j1 N$ Q
98.                             {
    $ O: U" m, ~+ O; {, x5 s$ L
99.                                 if (msgList[0].Trim().Length > 0)
    
100.                                     descUser = msgList[0].Trim();//记录消息目的用户
     
101.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(msgList[1]));
     8 D" l; P! X) s' `
102.                             }
     4 R0 T: C  \5 h% ^9 V* F
103.                             else
     
104.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(userMsg));
     
105. 
     ( @* ?* B+ W6 @  g" S# v  E* N! t
106.                             if (CONNECT_POOL.ContainsKey(descUser))//判断客户端是否在线
     0 ^0 H. ~: w7 z1 I
107.                             {
     1 V, |# K# s7 Y( c2 K
108.                                 WebSocket destSocket = CONNECT_POOL[descUser];//目的客户端
     & t8 D1 v* B  [5 i' {. f4 S! Y4 G
109.                                 if (destSocket != null && destSocket.State == WebSocketState.Open)
     
110.                                     await destSocket.SendAsync(buffer, WebSocketMessageType.Text, true, CancellationToken.None);
     7 i8 J9 E& ?: {+ z( y
111.                             }
     % ^2 T6 F6 L, B* ?/ t- x6 f
112.                             else
     % A4 L7 ~9 Q3 e( w$ {
113.                             {
     2 ?& [+ v3 h6 T1 F8 q0 P( U6 p7 Y" J% `
114.                                 Task.Run(() =>
     
115.                                 {
     
116.                                     if (!MESSAGE_POOL.ContainsKey(descUser))//将用户添加至离线消息池中
     3 {) o: Y3 Y& E, d) g4 U
117.                                         MESSAGE_POOL.Add(descUser, new List<MessageInfo>());
       z2 t- H% v; l! h- `
118.                                     MESSAGE_POOL[descUser].Add(new MessageInfo(DateTime.Now, buffer));//添加离线消息
     
119.                                 });
     & @; p8 D& v7 `7 O
120.                             }
     
121.                         }
     
122.                         catch (Exception exs)
     ) {) ?5 y$ m% k) I- Q! `8 X
123.                         {
     , [& h$ _" d2 l! q! b
124.                             //消息转发异常处理，本次消息忽略 继续监听接下来的消息
     
125.                         }
     
126.                         #endregion
     
127.                     }
     : v& ?* j1 w; I  x" }) q
128.                     else
     7 s+ s  H3 y7 R# e5 ^
129.                     {
     
130.                         break;
     
131.                     }
     0 }' W8 n* e. f9 T; Y& X
132.                 }//while end
     
133.             }
     
134.             catch (Exception ex)
     
135.             {
     & B2 A: d; j' t! k- `/ \' W
136.                 //整体异常处理
     % @, D, \# U' H, h( X4 o
137.                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);
     $ r/ U7 L; K! G; f2 |( i" M
138.             }
     * w9 T9 y& c  c2 b
139.         }
     
140. 
     
141. 
     
142.         public bool IsReusable
     
143.         {
     
144.             get
     + _/ h- L5 c+ C; o$ E/ m
145.             {
     1 N- B/ K2 R& D# U& {# w7 }: T
146.                 return false;
     
147.             }
     
148.         }
     
149.     }
     & h* d# u" O/ N/ P8 S1 {
150. }

复制代码

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

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

6 k6 u- j& N* Z1 g4 t