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

admin

服务器端代码编写

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

! S6 l9 q3 z% n

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

3.Handler1.ashx代码如下：

1. using System;
   
2. using System.Collections.Generic;
   
3. using System.Linq;
   
4. using System.Net.WebSockets;
   
5. using System.Text;
   
6. using System.Threading;
   
7. using System.Threading.Tasks;
   
8. using System.Web;
   2 F0 o2 n2 F" q6 `
9. using System.Web.WebSockets;
   # {1 ]8 g- W* B: q) v! W8 t, N2 f7 |
10. 
    
11. 
    
12. namespace WebApplicationWebsocketHandler
    
13. {
    
14.     /// <summary>
    
15.     /// 离线消息
    9 I! j+ R. p9 O+ ^$ ~& n. v! q
16.     /// </summary>
    
17.     public class MessageInfo
    
18.     {
    3 C" C$ y5 l- ^: Y
19.         public MessageInfo(DateTime _MsgTime, ArraySegment<byte> _MsgContent)
    
20.         {
    : l2 x* O" n6 G& Q* N
21.             MsgTime = _MsgTime;
    
22.             MsgContent = _MsgContent;
    5 @3 _+ T! k7 i9 l) K# j
23.         }
    
24.         public DateTime MsgTime { get; set; }
    ; {4 Q1 `+ e) t& D  N
25.         public ArraySegment<byte> MsgContent { get; set; }
    / b# x7 R% B0 z! l: h; ]3 R
26.     }
    ! U* w& a5 A8 `. t
27. 
    
28. 
    8 r- C9 Z' O; E- t% W2 l
29. 
    
30. 
    
31.     /// <summary>
    & T: t. F* B9 j
32.     /// Handler1 的摘要说明
    7 D7 z% m: q5 ^) X( t
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>>();//离线消息池
    
38.         public void ProcessRequest(HttpContext context)
    5 p% M9 t/ q: y1 E
39.         {
    
40.             //context.Response.ContentType = "text/plain";
    - J6 h; E$ @6 h& T' y, V
41.             //context.Response.Write("Hello World");
    / p$ g1 ~7 F! A
42.             if (context.IsWebSocketRequest)
    
43.             {
    ! {: |! A6 M5 w1 J/ H" X
44.                 context.AcceptWebSocketRequest(ProcessChat);
    
45.             }
    5 A, D5 c5 Y) N! H+ M( l! C! M% \
46.         }
    
47. 
    
48.         private async Task ProcessChat(AspNetWebSocketContext context)
    
49.         {
    8 T* Y& V4 V/ B3 Z7 W. I) s. Y1 C( V
50.             WebSocket socket = context.WebSocket;
    , |- D! {  k2 k9 b9 j5 m$ z5 j9 Q
51.             string user = context.QueryString["user"].ToString();
      t4 ~/ D/ q( J6 g9 ?5 ?' n
52. 
    
53.             try
    ) A1 J8 W0 [2 A; q# x1 K% R
54.             {
    4 k1 W1 m: U" G9 e
55.                 #region 用户添加连接池
    
56.                 //第一次open时，添加到连接池中
    
57.                 if (!CONNECT_POOL.ContainsKey(user))
    - Y9 c. q( x+ U3 v' W( v, g
58.                     CONNECT_POOL.Add(user, socket);//不存在，添加
    ( Y$ z0 C- J$ {: s4 T5 z& @
59.                 else
    7 L8 w5 t' w3 c0 [
60.                     if (socket != CONNECT_POOL[user])//当前对象不一致，更新
    ' n" E  q% Q3 D8 B1 X0 P
61.                         CONNECT_POOL[user] = socket;
    
62.                 #endregion
    
63. 
    ; N$ i5 o0 D5 ?% C6 l4 E7 g, |
64.                 #region 离线消息处理
    
65.                 if (MESSAGE_POOL.ContainsKey(user))
    
66.                 {
    
67.                     List<MessageInfo> msgs = MESSAGE_POOL[user];
    
68.                     foreach (MessageInfo item in msgs)
    9 t" U8 v5 g; P0 g) e% Q
69.                     {
    ; H) v& m' b/ C$ R$ v( Z
70.                         await socket.SendAsync(item.MsgContent, WebSocketMessageType.Text, true, CancellationToken.None);
    ; I  L7 l. L9 X( W4 t2 O
71.                     }
    
72.                     MESSAGE_POOL.Remove(user);//移除离线消息
    
73.                 }
    % i- a8 t8 z9 p7 c9 h
74.                 #endregion
    
75. 
    9 ?/ F% ^6 o5 w# ^9 E# f
76.                 string descUser = string.Empty;//目的用户
    
77.                 while (true)
    ) ^* [, N! ?9 o1 g  Z8 C3 J' @
78.                 {
    
79.                     if (socket.State == WebSocketState.Open)
    
80.                     {
    : C0 T" S* S0 d$ t/ N" \- s
81.                         ArraySegment<byte> buffer = new ArraySegment<byte>(new byte[2048]);
    5 ^# X8 {6 `3 E1 x1 j! i
82.                         WebSocketReceiveResult result = await socket.ReceiveAsync(buffer, CancellationToken.None);
    
83. 
    / m" e8 _# _5 q
84.                         #region 消息处理（字符截取、消息转发）
    $ h& E6 p! H: v: g4 Z" A7 f
85.                         try
    & l6 T7 u* X6 n4 z/ y5 Q, Y0 h. C
86.                         {
    
87.                             #region 关闭Socket处理，删除连接池
    : K8 D8 x! h% A) V6 P6 T
88.                             if (socket.State != WebSocketState.Open)//连接关闭
    
89.                             {
    ! f/ K  M0 q0 C. ?' r" |
90.                                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);//删除连接池
    : }3 {! {0 `. X1 u3 c! V
91.                                 break;
      I! k" ~' f4 ~3 b# X
92.                             }
    
93.                             #endregion
    : p) Z% i; L5 |* ~: x
94. 
    
95.                             string userMsg = Encoding.UTF8.GetString(buffer.Array, 0, result.Count);//发送过来的消息
    5 W) ]6 C2 ^; L# o
96.                             string[] msgList = userMsg.Split('|');
    ' W  p. T. N2 S5 t& ^/ a, K/ P9 E
97.                             if (msgList.Length == 2)
    
98.                             {
    % @5 L3 N, e2 ^) @; H9 K
99.                                 if (msgList[0].Trim().Length > 0)
    , q4 S) w! m3 _0 [. q" D
100.                                     descUser = msgList[0].Trim();//记录消息目的用户
     
101.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(msgList[1]));
     . G; v  n: g% k; Q
102.                             }
     
103.                             else
     + X: c# X. s* @  y0 a7 [& ]/ E
104.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(userMsg));
     
105. 
     : i7 G) p3 j3 W. q5 O1 l
106.                             if (CONNECT_POOL.ContainsKey(descUser))//判断客户端是否在线
     
107.                             {
     
108.                                 WebSocket destSocket = CONNECT_POOL[descUser];//目的客户端
     / n2 u1 K5 z4 _5 k, t
109.                                 if (destSocket != null && destSocket.State == WebSocketState.Open)
     , H0 W8 L7 M7 z5 h7 q" t/ C
110.                                     await destSocket.SendAsync(buffer, WebSocketMessageType.Text, true, CancellationToken.None);
     - p( K* _* Q$ W) Z# a7 T3 ]+ K
111.                             }
     
112.                             else
     $ y/ A# V' U$ u' p
113.                             {
     
114.                                 Task.Run(() =>
     
115.                                 {
     0 k3 U! c/ b  O7 _+ a- ~
116.                                     if (!MESSAGE_POOL.ContainsKey(descUser))//将用户添加至离线消息池中
     
117.                                         MESSAGE_POOL.Add(descUser, new List<MessageInfo>());
     
118.                                     MESSAGE_POOL[descUser].Add(new MessageInfo(DateTime.Now, buffer));//添加离线消息
     + ~: I; x' r- `4 ~5 ]
119.                                 });
     
120.                             }
     # j: X- J2 m5 t
121.                         }
     
122.                         catch (Exception exs)
     
123.                         {
     ) v+ _! S5 |% S; H/ h" G% X# \
124.                             //消息转发异常处理，本次消息忽略 继续监听接下来的消息
       ^8 G' a4 J( t* M3 q
125.                         }
     ! u8 Q' L+ q; E8 p2 O& v
126.                         #endregion
     0 x+ c* _2 {/ w4 X$ G' |, C8 q! S1 [
127.                     }
     
128.                     else
     8 V) J7 z2 {# u: i1 A
129.                     {
     
130.                         break;
     
131.                     }
     
132.                 }//while end
     ( _* p0 E3 |/ G. }1 H
133.             }
     7 R: \* x# p0 z( D
134.             catch (Exception ex)
     
135.             {
     
136.                 //整体异常处理
     
137.                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);
     
138.             }
     8 e: I% O; s7 B! k& W) X' R
139.         }
     ) W6 {+ Y" O( Z! g; |3 V; p
140. 
     
141. 
     
142.         public bool IsReusable
     4 t0 ]( ^0 g( d" ?9 U/ o
143.         {
     
144.             get
     ( ]) @0 p. A4 c( H4 }- Z
145.             {
     & ?1 V# b* w- O4 [5 A7 J6 F: h
146.                 return false;
     0 _; A) k2 N3 n7 k9 o
147.             }
     # ?! |4 I4 {  U/ K( h& L
148.         }
     2 |  O1 b$ D/ L2 m- Q
149.     }
     
150. }

复制代码

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

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