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

admin

服务器端代码编写

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

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

3.Handler1.ashx代码如下：

1. using System;
   
2. using System.Collections.Generic;
   
3. using System.Linq;
   . w0 `9 ~  Y3 a
4. using System.Net.WebSockets;
   
5. using System.Text;
   + m; H9 u1 K$ c! U" T6 y
6. using System.Threading;
   
7. using System.Threading.Tasks;
   ! W/ y7 K) A% o5 R2 A& c( i& {# k6 ^
8. using System.Web;
   ) m0 X- ]2 ?$ d' ~0 E8 {
9. using System.Web.WebSockets;
   
10. 
    4 L" L- ]7 g! S
11. 
    
12. namespace WebApplicationWebsocketHandler
    / W  x+ `" c0 W1 z
13. {
    
14.     /// <summary>
      h/ a! c6 b) w6 Y* g" o) j
15.     /// 离线消息
    
16.     /// </summary>
    
17.     public class MessageInfo
    # z' M  s1 e. E# f& P2 a* v- A  `) z
18.     {
      v, R" U+ d0 |+ t% H
19.         public MessageInfo(DateTime _MsgTime, ArraySegment<byte> _MsgContent)
    
20.         {
      s$ _9 |) [) V# Z9 H8 p$ R# l
21.             MsgTime = _MsgTime;
    " ]: `, T* }2 u- Q/ A; Z9 `. H
22.             MsgContent = _MsgContent;
    
23.         }
    ) E9 E  D* n6 k0 S  i( N
24.         public DateTime MsgTime { get; set; }
    
25.         public ArraySegment<byte> MsgContent { get; set; }
    
26.     }
    6 T. w5 j2 Q  C6 l, u
27. 
      Q: m9 P. e; j1 x' q
28. 
    
29. 
    
30. 
    " [  p2 y& q5 Q- _3 o4 n. r
31.     /// <summary>
    & \8 X+ A- n' H% K$ S. ^
32.     /// Handler1 的摘要说明
    " A3 h7 Q- A5 x, P. q' C9 ?
33.     /// </summary>
    1 u5 v9 A0 ^: S: V6 Y$ y& X
34.     public class Handler1 : IHttpHandler
    ( v1 o2 \) b9 ^0 E, |6 D4 ~; `
35.     {
    3 b: e7 i. z% l) F- U! C
36.         private static Dictionary<string, WebSocket> CONNECT_POOL = new Dictionary<string, WebSocket>();//用户连接池
    # k5 h) ?3 _& B
37.         private static Dictionary<string, List<MessageInfo>> MESSAGE_POOL = new Dictionary<string, List<MessageInfo>>();//离线消息池
    
38.         public void ProcessRequest(HttpContext context)
    
39.         {
    
40.             //context.Response.ContentType = "text/plain";
    
41.             //context.Response.Write("Hello World");
    " \1 D+ {/ R' z, I, d; @5 v
42.             if (context.IsWebSocketRequest)
    , ?: m: h* D& O  y/ C
43.             {
    # G( C7 u- P' H( d' S4 C
44.                 context.AcceptWebSocketRequest(ProcessChat);
    
45.             }
    3 W, T. k6 y  e- V3 V; _; G: g
46.         }
    
47. 
    . X- Z! ~# r/ w7 [' G5 m
48.         private async Task ProcessChat(AspNetWebSocketContext context)
    
49.         {
    
50.             WebSocket socket = context.WebSocket;
    $ b# Q/ w& u9 G* z
51.             string user = context.QueryString["user"].ToString();
    , F2 d' E/ X6 o7 \6 X
52. 
    7 u! u" O! \# `
53.             try
    . j  V6 J2 y: r  z
54.             {
    ) d2 f' Q8 g+ A# W6 D# @
55.                 #region 用户添加连接池
    
56.                 //第一次open时，添加到连接池中
    # H3 }, [8 w" J/ N. K$ G: l
57.                 if (!CONNECT_POOL.ContainsKey(user))
    
58.                     CONNECT_POOL.Add(user, socket);//不存在，添加
    # J6 _+ \& |. }2 m
59.                 else
    
60.                     if (socket != CONNECT_POOL[user])//当前对象不一致，更新
    
61.                         CONNECT_POOL[user] = socket;
      O) V5 U9 z1 r) L" A: @
62.                 #endregion
    . w/ Y1 Q7 x& P- e
63. 
      r! W1 U2 E: L4 Q! ]0 I" d
64.                 #region 离线消息处理
    
65.                 if (MESSAGE_POOL.ContainsKey(user))
    
66.                 {
    
67.                     List<MessageInfo> msgs = MESSAGE_POOL[user];
    ( R% V+ h/ X2 x0 J3 r' b
68.                     foreach (MessageInfo item in msgs)
    0 n# _- X; [$ ]
69.                     {
    ) r; r! s, W( L  {+ F1 N+ I' ?
70.                         await socket.SendAsync(item.MsgContent, WebSocketMessageType.Text, true, CancellationToken.None);
    " h0 \6 l/ b9 R; D4 s0 t- ]0 _
71.                     }
    7 A  s& i4 g; q0 O; h# s" m- \
72.                     MESSAGE_POOL.Remove(user);//移除离线消息
    
73.                 }
    
74.                 #endregion
    
75. 
    3 |5 a/ ?2 n) @  |* N# p
76.                 string descUser = string.Empty;//目的用户
    
77.                 while (true)
    
78.                 {
    
79.                     if (socket.State == WebSocketState.Open)
    7 B0 B: f% [, d0 T3 X
80.                     {
    
81.                         ArraySegment<byte> buffer = new ArraySegment<byte>(new byte[2048]);
    
82.                         WebSocketReceiveResult result = await socket.ReceiveAsync(buffer, CancellationToken.None);
    
83. 
    
84.                         #region 消息处理（字符截取、消息转发）
    / ~7 e- z8 q' L( ~+ j) X2 F
85.                         try
    
86.                         {
    
87.                             #region 关闭Socket处理，删除连接池
    
88.                             if (socket.State != WebSocketState.Open)//连接关闭
    
89.                             {
    0 G7 @3 P& E3 w& H' m
90.                                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);//删除连接池
    
91.                                 break;
    ' R$ t- N  V8 M5 k5 ~9 q  H
92.                             }
    
93.                             #endregion
    % x# q- a8 y( ^( p. O
94. 
    
95.                             string userMsg = Encoding.UTF8.GetString(buffer.Array, 0, result.Count);//发送过来的消息
    
96.                             string[] msgList = userMsg.Split('|');
    % k! M' I' h: c
97.                             if (msgList.Length == 2)
    ! v6 N: y+ M  u& S  D
98.                             {
    ! q' ?. J  h8 i5 s
99.                                 if (msgList[0].Trim().Length > 0)
    
100.                                     descUser = msgList[0].Trim();//记录消息目的用户
     # C4 p& ?; L) o. B# P
101.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(msgList[1]));
     $ w3 W5 Z! T- J6 f- q( Q
102.                             }
     
103.                             else
     
104.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(userMsg));
     5 i# e7 s( q6 J7 `0 N* h8 I
105. 
     : W! ^( \6 J2 |" |; Q, \
106.                             if (CONNECT_POOL.ContainsKey(descUser))//判断客户端是否在线
     
107.                             {
     
108.                                 WebSocket destSocket = CONNECT_POOL[descUser];//目的客户端
     + L0 ~5 d0 g; V" r3 O
109.                                 if (destSocket != null && destSocket.State == WebSocketState.Open)
     
110.                                     await destSocket.SendAsync(buffer, WebSocketMessageType.Text, true, CancellationToken.None);
     
111.                             }
     9 Q; g1 W5 B2 p+ k; C
112.                             else
     
113.                             {
     
114.                                 Task.Run(() =>
     
115.                                 {
     
116.                                     if (!MESSAGE_POOL.ContainsKey(descUser))//将用户添加至离线消息池中
     & i. I7 K, n/ @; x1 S2 V! [
117.                                         MESSAGE_POOL.Add(descUser, new List<MessageInfo>());
     
118.                                     MESSAGE_POOL[descUser].Add(new MessageInfo(DateTime.Now, buffer));//添加离线消息
     
119.                                 });
     
120.                             }
     
121.                         }
     7 d4 C( e' O% W
122.                         catch (Exception exs)
     
123.                         {
     
124.                             //消息转发异常处理，本次消息忽略 继续监听接下来的消息
     - _8 k( j- u# g' Y4 \# {$ |
125.                         }
     
126.                         #endregion
     
127.                     }
     ( f  f8 j5 k9 W! x' P
128.                     else
     # x* p  Q- i  H- e8 w2 `$ Q) f
129.                     {
     , `! U7 G5 a: b8 D$ B$ V' a
130.                         break;
     4 {9 C$ K# J  S' N
131.                     }
     
132.                 }//while end
     + C& `% \' O6 E; J! V6 y. z9 j$ X6 j
133.             }
     1 d# b' P6 z% {1 T. }; k* j
134.             catch (Exception ex)
     " [- @, l2 t" r
135.             {
     
136.                 //整体异常处理
     
137.                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);
     . c% \2 W; Q/ q9 ]! _- ~# D# v
138.             }
     
139.         }
     
140. 
     9 w' k* y8 K" e' u
141. 
     . K$ T6 s  D4 `$ y
142.         public bool IsReusable
     
143.         {
     ( P/ ^' a2 y& a6 x' M+ C4 n
144.             get
     9 j* M, S) @% b5 O) }. ^
145.             {
     0 Q" H0 [8 S7 ~( m8 X2 Y' ^
146.                 return false;
     . T5 p0 D, o' v5 p7 S: z
147.             }
     
148.         }
     ' O) j0 N/ m7 _- |  r! c1 J
149.     }
     
150. }

复制代码

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

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

% T) o, R3 P% ]  P& x