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

admin

服务器端代码编写

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

: Q/ f' H1 ]4 I2 W

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

3.Handler1.ashx代码如下：

1. using System;
   / w: s( D; N: D& `% H. z
2. using System.Collections.Generic;
   
3. using System.Linq;
   6 f) Y( y" z, q3 w# q
4. using System.Net.WebSockets;
   
5. using System.Text;
   # C7 r" Q! m, E8 p! i+ j, I
6. using System.Threading;
   ' Z. o0 g/ I/ E4 D& \
7. using System.Threading.Tasks;
   
8. using System.Web;
   
9. using System.Web.WebSockets;
   
10. 
    $ ]* u$ \/ t9 `1 \8 u3 W" I
11. 
    
12. namespace WebApplicationWebsocketHandler
      X9 ]; Q: e8 Q0 t( S8 D! r/ Z5 ~& `
13. {
    
14.     /// <summary>
    * M$ a. W  O: K% c: |! F) [$ f( D% P
15.     /// 离线消息
    
16.     /// </summary>
    7 T1 x6 s$ g# z: H! r" w+ m
17.     public class MessageInfo
    ; z5 W2 y  t! g  e4 U1 ~5 {
18.     {
    * P: m+ |' q' u6 O- }* W) \. \
19.         public MessageInfo(DateTime _MsgTime, ArraySegment<byte> _MsgContent)
    $ o5 b  a# c7 G% J/ \
20.         {
    
21.             MsgTime = _MsgTime;
    : J$ |; B" l: K6 [$ G$ W4 k9 C
22.             MsgContent = _MsgContent;
    
23.         }
    
24.         public DateTime MsgTime { get; set; }
    5 U( Z3 [5 G5 J6 b
25.         public ArraySegment<byte> MsgContent { get; set; }
    
26.     }
    , \0 k& T/ b& U) v! s
27. 
    
28. 
    5 U/ E$ s3 ~  l* Z1 U% R
29. 
    0 I- u* C( O& K
30. 
    0 d! J2 h4 v, P0 F( I2 {
31.     /// <summary>
    
32.     /// Handler1 的摘要说明
    4 h3 h! y5 v& {3 T$ |5 M0 x- Q! ~
33.     /// </summary>
    
34.     public class Handler1 : IHttpHandler
    
35.     {
    
36.         private static Dictionary<string, WebSocket> CONNECT_POOL = new Dictionary<string, WebSocket>();//用户连接池
    5 Z9 e1 e4 A: G1 \- R
37.         private static Dictionary<string, List<MessageInfo>> MESSAGE_POOL = new Dictionary<string, List<MessageInfo>>();//离线消息池
    - Q: u& R4 L; ~- O5 @5 Y
38.         public void ProcessRequest(HttpContext context)
    
39.         {
    
40.             //context.Response.ContentType = "text/plain";
    
41.             //context.Response.Write("Hello World");
    
42.             if (context.IsWebSocketRequest)
    
43.             {
    ( D; c* {3 A% t0 _7 H2 l+ f- |
44.                 context.AcceptWebSocketRequest(ProcessChat);
    / c. L+ t5 P$ Y8 ?
45.             }
    
46.         }
    ( H  C9 o  u- Z& [1 v5 ]
47. 
    4 o9 Q8 d# h/ e
48.         private async Task ProcessChat(AspNetWebSocketContext context)
    
49.         {
    : c9 C1 E- m) j; ?- X$ E" H- n# E
50.             WebSocket socket = context.WebSocket;
    
51.             string user = context.QueryString["user"].ToString();
    * I2 P6 [1 Y1 ]. Z0 ?+ d, u4 c. j
52. 
    
53.             try
    
54.             {
    
55.                 #region 用户添加连接池
    8 D! R' s) A- x  ^- h4 T
56.                 //第一次open时，添加到连接池中
    * P% @, N# v1 f! @  d( x( @# Y$ L" P
57.                 if (!CONNECT_POOL.ContainsKey(user))
    7 l( R  o- @2 n  J2 R
58.                     CONNECT_POOL.Add(user, socket);//不存在，添加
    
59.                 else
    6 F- [3 `" R  L* ~3 Q$ L' f
60.                     if (socket != CONNECT_POOL[user])//当前对象不一致，更新
    2 T* z; t2 g- r$ Z! i1 G) Y: g
61.                         CONNECT_POOL[user] = socket;
    / `% C  O0 E( Z2 B8 h% a$ G  i, z
62.                 #endregion
    
63. 
    - Y) F) [: _' {6 _. d! s
64.                 #region 离线消息处理
    % ]$ m2 D; C% x/ D. Y) z
65.                 if (MESSAGE_POOL.ContainsKey(user))
    
66.                 {
      D1 E+ p/ @6 B4 ^
67.                     List<MessageInfo> msgs = MESSAGE_POOL[user];
    
68.                     foreach (MessageInfo item in msgs)
    
69.                     {
    
70.                         await socket.SendAsync(item.MsgContent, WebSocketMessageType.Text, true, CancellationToken.None);
    
71.                     }
    
72.                     MESSAGE_POOL.Remove(user);//移除离线消息
    ( b1 {! ?. Z5 U/ d& S7 k- b
73.                 }
    
74.                 #endregion
    
75. 
    
76.                 string descUser = string.Empty;//目的用户
    
77.                 while (true)
    3 D+ c4 n2 f& o6 \
78.                 {
    
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);
    # v6 J- O  Z& d
83. 
    2 [" ]( k4 w0 n+ I9 d3 C
84.                         #region 消息处理（字符截取、消息转发）
    
85.                         try
    
86.                         {
    
87.                             #region 关闭Socket处理，删除连接池
    , F+ s; N* b& \( U
88.                             if (socket.State != WebSocketState.Open)//连接关闭
    
89.                             {
      \% D1 A6 W2 B9 k) U1 t
90.                                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);//删除连接池
    " D' c# v0 [. |/ _) [! x% C
91.                                 break;
    
92.                             }
    
93.                             #endregion
    & s4 b" G) f  L3 v( m
94. 
    + @/ z; H2 Y- ]3 l1 o: `: i( k
95.                             string userMsg = Encoding.UTF8.GetString(buffer.Array, 0, result.Count);//发送过来的消息
    2 }% {; Q" j: |+ B2 Y7 V( j
96.                             string[] msgList = userMsg.Split('|');
    # w- R7 M: o+ c* @0 A% t( v
97.                             if (msgList.Length == 2)
    
98.                             {
    + k$ H( v% A$ q) y" H
99.                                 if (msgList[0].Trim().Length > 0)
    " X* n# L6 `0 `0 t  p+ h- L
100.                                     descUser = msgList[0].Trim();//记录消息目的用户
     
101.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(msgList[1]));
     + o) b3 b( S. n: c0 ^( q
102.                             }
     
103.                             else
     
104.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(userMsg));
     
105. 
     4 i2 l) |0 ^2 F. S7 \
106.                             if (CONNECT_POOL.ContainsKey(descUser))//判断客户端是否在线
     
107.                             {
     
108.                                 WebSocket destSocket = CONNECT_POOL[descUser];//目的客户端
     
109.                                 if (destSocket != null && destSocket.State == WebSocketState.Open)
     & M: F2 J6 ]  D$ T# g+ R$ i, \% B
110.                                     await destSocket.SendAsync(buffer, WebSocketMessageType.Text, true, CancellationToken.None);
     
111.                             }
     
112.                             else
     5 V5 L9 t5 U5 `
113.                             {
     * I3 R+ i5 B. S6 n8 h6 x
114.                                 Task.Run(() =>
     5 E- [: H! I/ ?5 ~; s
115.                                 {
     
116.                                     if (!MESSAGE_POOL.ContainsKey(descUser))//将用户添加至离线消息池中
     ! `) R- c" @3 q0 S/ r
117.                                         MESSAGE_POOL.Add(descUser, new List<MessageInfo>());
     
118.                                     MESSAGE_POOL[descUser].Add(new MessageInfo(DateTime.Now, buffer));//添加离线消息
     
119.                                 });
     6 C# N; t1 M* `" J8 f0 d
120.                             }
     
121.                         }
     ( ~+ f$ b4 e& n, D, C- u6 k3 E( H! t
122.                         catch (Exception exs)
     
123.                         {
     
124.                             //消息转发异常处理，本次消息忽略 继续监听接下来的消息
     . X% ?! |5 m) G2 V% r7 F
125.                         }
     
126.                         #endregion
     + T  n4 }, G1 D( q$ |
127.                     }
     
128.                     else
     
129.                     {
     : l  Y  h3 b3 n$ q
130.                         break;
     
131.                     }
     
132.                 }//while end
     
133.             }
     
134.             catch (Exception ex)
     
135.             {
     
136.                 //整体异常处理
     
137.                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);
     ) ~( H4 y+ ?- O! e; z8 k0 ~
138.             }
     2 s7 y9 n8 T$ Y3 X
139.         }
     
140. 
     4 ?' J. c3 |4 `  E: u0 }6 y
141. 
     ; I2 V8 j% i- M4 d5 L7 C
142.         public bool IsReusable
     2 j7 x' e- L/ p4 L. k# Y* K
143.         {
     - \3 Q3 i" c; ]9 x) I2 ~
144.             get
     
145.             {
       F- T8 R% p- b. r" e
146.                 return false;
     
147.             }
     
148.         }
     
149.     }
     
150. }

复制代码

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

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

  f: w$ p& u6 O, Z# z