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

admin

服务器端代码编写

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

; |/ i7 ?- m( f+ P- R- E

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

3.Handler1.ashx代码如下：

1. using System;
   3 n3 G4 V5 D9 s! z: k
2. using System.Collections.Generic;
   - R* w) M% w, e
3. using System.Linq;
   ' m0 t/ ~+ C4 k& U
4. using System.Net.WebSockets;
   
5. using System.Text;
   
6. using System.Threading;
   ( @% ^, q1 t" u' g% u
7. using System.Threading.Tasks;
   
8. using System.Web;
   
9. using System.Web.WebSockets;
   1 ?  W0 i, W: `  ?- p' _
10. 
    ( b0 d! O  \6 a6 t2 B- [; s7 b
11. 
    
12. namespace WebApplicationWebsocketHandler
    
13. {
    
14.     /// <summary>
    
15.     /// 离线消息
    . k  ]2 a& D6 g  e( c
16.     /// </summary>
    0 R7 T- v3 I4 M, d0 U: p  C8 k
17.     public class MessageInfo
    % q$ d$ }7 m$ p, |. t
18.     {
    , D: `2 D" I. ~2 W* m8 O2 i, T
19.         public MessageInfo(DateTime _MsgTime, ArraySegment<byte> _MsgContent)
    
20.         {
    9 T  X4 V5 q! x9 o9 ]! e$ P
21.             MsgTime = _MsgTime;
    
22.             MsgContent = _MsgContent;
    
23.         }
    ) u. V) w, F& u  Z& z8 W" F
24.         public DateTime MsgTime { get; set; }
    
25.         public ArraySegment<byte> MsgContent { get; set; }
    3 n3 u1 K$ y- f, t# z) q3 J
26.     }
    
27. 
    6 q! P8 z5 s7 j; L
28. 
    : J+ r( ^) [0 U
29. 
    % d7 a) C9 C. j9 X
30. 
    
31.     /// <summary>
    
32.     /// Handler1 的摘要说明
    
33.     /// </summary>
    
34.     public class Handler1 : IHttpHandler
    
35.     {
    & g; G% r# l" D$ ^0 K* e
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)
    ' Z8 o, r! O* s5 r. r. u. A& B
39.         {
    
40.             //context.Response.ContentType = "text/plain";
    % B7 Y! I, Z) S: s. {9 {
41.             //context.Response.Write("Hello World");
    & V  v$ X; S% u( s4 _; g; u, F. S0 b
42.             if (context.IsWebSocketRequest)
    ) r$ y/ `5 x, C$ M
43.             {
    9 t2 h2 U- p9 x5 ^4 Y/ y5 n9 o& A3 q. O) C
44.                 context.AcceptWebSocketRequest(ProcessChat);
    " _0 Z) b+ z6 C4 F" B9 O
45.             }
    3 N2 e$ Z* P' u
46.         }
    
47. 
    6 U) h& K. l8 k( s
48.         private async Task ProcessChat(AspNetWebSocketContext context)
    
49.         {
    ; p( U) R3 y+ }5 H' z" p; Z  y( t
50.             WebSocket socket = context.WebSocket;
    
51.             string user = context.QueryString["user"].ToString();
    
52. 
    ' s7 ^7 }  G: q% ^# ~' x4 i
53.             try
    8 ?8 V# j. O. Z+ ?# O) ^: E
54.             {
    + \: K2 d$ B. q6 ^6 a0 e
55.                 #region 用户添加连接池
    + X; h) V  e# ?3 a0 A$ Z. ^  ?3 g
56.                 //第一次open时，添加到连接池中
    
57.                 if (!CONNECT_POOL.ContainsKey(user))
    . q* _8 R- Z' K2 x5 n  B3 }
58.                     CONNECT_POOL.Add(user, socket);//不存在，添加
    # l& f1 m! J0 z, h8 v5 Y  R1 T" T% y
59.                 else
    
60.                     if (socket != CONNECT_POOL[user])//当前对象不一致，更新
    
61.                         CONNECT_POOL[user] = socket;
    
62.                 #endregion
    
63. 
    ) d! J# K7 g3 m* H7 w, c, |
64.                 #region 离线消息处理
    
65.                 if (MESSAGE_POOL.ContainsKey(user))
    
66.                 {
    
67.                     List<MessageInfo> msgs = MESSAGE_POOL[user];
    4 A1 t0 L9 R/ |( F; N
68.                     foreach (MessageInfo item in msgs)
    1 E+ r- \4 x4 T1 _
69.                     {
    
70.                         await socket.SendAsync(item.MsgContent, WebSocketMessageType.Text, true, CancellationToken.None);
    
71.                     }
    
72.                     MESSAGE_POOL.Remove(user);//移除离线消息
    ! i; c1 q4 w7 {: Q0 b
73.                 }
    
74.                 #endregion
    - G$ M7 G5 a* h
75. 
    ; g* Z6 f# [( d7 \3 @- h$ d) I; v+ l
76.                 string descUser = string.Empty;//目的用户
    
77.                 while (true)
    
78.                 {
    2 P2 R7 H+ }( K2 V; e+ G" P; [
79.                     if (socket.State == WebSocketState.Open)
    5 C- A' Z# {0 Y
80.                     {
    
81.                         ArraySegment<byte> buffer = new ArraySegment<byte>(new byte[2048]);
    : u$ j, E( B$ y$ L
82.                         WebSocketReceiveResult result = await socket.ReceiveAsync(buffer, CancellationToken.None);
    ( f. s, e+ U2 g% }
83. 
    % n3 r4 p" U2 o9 H: ]% A+ i
84.                         #region 消息处理（字符截取、消息转发）
    2 _5 Q5 \% T1 x& _9 @( X# x9 M
85.                         try
    # u1 i1 k4 ]* v
86.                         {
    
87.                             #region 关闭Socket处理，删除连接池
    
88.                             if (socket.State != WebSocketState.Open)//连接关闭
    2 ?3 x: h0 n" {( d8 r  ]/ d
89.                             {
    
90.                                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);//删除连接池
    8 P+ \, w1 q* T
91.                                 break;
    9 C" E, I& R, k- x3 N: x9 `/ J0 ?$ _
92.                             }
    
93.                             #endregion
    
94. 
    
95.                             string userMsg = Encoding.UTF8.GetString(buffer.Array, 0, result.Count);//发送过来的消息
    5 e5 Y, p3 b3 E) B
96.                             string[] msgList = userMsg.Split('|');
    
97.                             if (msgList.Length == 2)
    
98.                             {
    / ~1 J: b3 F: `
99.                                 if (msgList[0].Trim().Length > 0)
    
100.                                     descUser = msgList[0].Trim();//记录消息目的用户
     
101.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(msgList[1]));
     6 W* \, O  E' W. Q  f; h
102.                             }
     
103.                             else
     
104.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(userMsg));
     
105. 
     
106.                             if (CONNECT_POOL.ContainsKey(descUser))//判断客户端是否在线
     9 c# }" U4 y, i, e6 _6 I4 l8 A
107.                             {
     
108.                                 WebSocket destSocket = CONNECT_POOL[descUser];//目的客户端
     5 T0 }# x: q4 g3 N" N- ?
109.                                 if (destSocket != null && destSocket.State == WebSocketState.Open)
     ) l6 x; \. s2 I/ b8 ^$ g- L( g6 c9 r
110.                                     await destSocket.SendAsync(buffer, WebSocketMessageType.Text, true, CancellationToken.None);
     ) a- a) L% f' j% j$ O2 i
111.                             }
     - ~' q8 ^- R% e# q
112.                             else
     * V$ {$ s( H. @4 J* e  M6 g
113.                             {
     
114.                                 Task.Run(() =>
     
115.                                 {
     
116.                                     if (!MESSAGE_POOL.ContainsKey(descUser))//将用户添加至离线消息池中
     7 x# h. w9 q' x
117.                                         MESSAGE_POOL.Add(descUser, new List<MessageInfo>());
     # B6 g+ g: y; m+ _
118.                                     MESSAGE_POOL[descUser].Add(new MessageInfo(DateTime.Now, buffer));//添加离线消息
     
119.                                 });
     
120.                             }
     , j9 q9 _& Y, p# [' T
121.                         }
     , H7 b/ G8 `6 w% Z9 S. g4 ~
122.                         catch (Exception exs)
     & i$ ?4 d+ B& h
123.                         {
     
124.                             //消息转发异常处理，本次消息忽略 继续监听接下来的消息
     . M4 I* p9 p7 L% M8 a
125.                         }
     , Q% q, A, H" B
126.                         #endregion
     , Y7 k; C, l$ R* x+ {
127.                     }
     1 D, {$ w! N( s" T  b
128.                     else
     
129.                     {
     
130.                         break;
     - Z% \1 N: c  {) h4 [. ?- J
131.                     }
     ' `6 V3 H- @5 Y
132.                 }//while end
     ! ^# I. l( s+ r# Y2 m$ {, e( x8 @
133.             }
     
134.             catch (Exception ex)
     
135.             {
     " v' X# m' o6 K
136.                 //整体异常处理
     
137.                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);
     
138.             }
     & M1 o% ]1 U/ O/ e/ L3 w/ e
139.         }
     : g! v% h$ ]8 b% ~! ?
140. 
     
141. 
     
142.         public bool IsReusable
     
143.         {
     
144.             get
     
145.             {
       `  ~" f, m4 Z4 R+ |1 U2 t+ k
146.                 return false;
     
147.             }
     " K# s! W% R  ~) U2 Z2 U* N5 U
148.         }
     ( D# p5 u: `. s4 e2 ]. }5 q5 p9 \
149.     }
     , {! `3 a! v" b, u8 ?4 C
150. }

复制代码

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

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