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

admin

服务器端代码编写

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

* e" a( c7 c' n/ ?. q

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

3.Handler1.ashx代码如下：

1. using System;
   
2. using System.Collections.Generic;
   4 C: D# i( k; F9 f
3. using System.Linq;
   ( s) C' k0 H/ B. q" K
4. using System.Net.WebSockets;
   & `7 u0 q, }# M0 V: z  J0 [& C' [. S+ h
5. using System.Text;
   
6. using System.Threading;
   
7. using System.Threading.Tasks;
   
8. using System.Web;
   8 h$ i9 _. \* Z6 D7 K+ @) @+ G
9. using System.Web.WebSockets;
     O* c* t& o. m
10. 
    
11. 
    $ q2 F% m7 w4 n' `% x' d0 h; ]
12. namespace WebApplicationWebsocketHandler
    2 t& u2 f5 S. n' f. O
13. {
    2 I$ T$ l; U1 E+ ?
14.     /// <summary>
    , S- C2 \1 V  ?- d: V( r# o
15.     /// 离线消息
    . m- A: B% \& U3 c5 s; D
16.     /// </summary>
    
17.     public class MessageInfo
    
18.     {
    
19.         public MessageInfo(DateTime _MsgTime, ArraySegment<byte> _MsgContent)
    
20.         {
    
21.             MsgTime = _MsgTime;
    . z1 z( B5 ]& R: o5 Q: c" N
22.             MsgContent = _MsgContent;
    
23.         }
    
24.         public DateTime MsgTime { get; set; }
    
25.         public ArraySegment<byte> MsgContent { get; set; }
    + F- J. T0 v7 B3 ~* I" a: j& y; n; L* y
26.     }
    
27. 
    + x4 ?  Z" M# w+ ?( I6 n
28. 
    
29. 
    - p+ _2 u1 Z3 Z& s
30. 
    ( _3 y* Q( w" D4 Y, T: C# O0 c
31.     /// <summary>
    
32.     /// Handler1 的摘要说明
    
33.     /// </summary>
    
34.     public class Handler1 : IHttpHandler
    
35.     {
    0 H/ |+ j9 d+ {+ d$ W- W7 B
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>>();//离线消息池
    3 x( c" L# t1 c6 h, {
38.         public void ProcessRequest(HttpContext context)
    , i% t7 R9 a- V" t7 n
39.         {
    - P( ^, b3 Y  ?& H9 C
40.             //context.Response.ContentType = "text/plain";
    
41.             //context.Response.Write("Hello World");
    8 j1 r/ o6 B' L! n; x( W. K
42.             if (context.IsWebSocketRequest)
    , X8 Q0 X1 c, J
43.             {
    
44.                 context.AcceptWebSocketRequest(ProcessChat);
    
45.             }
    $ m; w9 _, r7 J( G
46.         }
    
47. 
    4 Z$ C; `2 r; h. ?8 [5 {
48.         private async Task ProcessChat(AspNetWebSocketContext context)
    
49.         {
    
50.             WebSocket socket = context.WebSocket;
    
51.             string user = context.QueryString["user"].ToString();
    6 X5 N/ Q: B  P1 Y/ x, [# w. e! L
52. 
    2 P7 c0 R* q; g; F# A, S
53.             try
    ( o. t) n8 k5 o
54.             {
    + b' p7 f/ w) Z1 @. a- t/ T7 r
55.                 #region 用户添加连接池
    
56.                 //第一次open时，添加到连接池中
    
57.                 if (!CONNECT_POOL.ContainsKey(user))
    ' b+ X8 r' a+ K
58.                     CONNECT_POOL.Add(user, socket);//不存在，添加
    ; K& K, j% t. S2 ~. {9 c% d
59.                 else
    
60.                     if (socket != CONNECT_POOL[user])//当前对象不一致，更新
    
61.                         CONNECT_POOL[user] = socket;
    
62.                 #endregion
    
63. 
    
64.                 #region 离线消息处理
    
65.                 if (MESSAGE_POOL.ContainsKey(user))
    ; b4 A* f# v* b, X5 R- v% z, ?
66.                 {
    
67.                     List<MessageInfo> msgs = MESSAGE_POOL[user];
    2 u- p! H* t; h
68.                     foreach (MessageInfo item in msgs)
    , C1 x- u8 f7 o5 P
69.                     {
      ^3 \( b; y! h- g  |7 c' E# t4 @
70.                         await socket.SendAsync(item.MsgContent, WebSocketMessageType.Text, true, CancellationToken.None);
    
71.                     }
    ; A9 Y3 V$ U4 W2 A
72.                     MESSAGE_POOL.Remove(user);//移除离线消息
    , [8 n9 d; b; j, w! |6 `5 X: u- Z
73.                 }
    / h8 G- h' }6 K7 N* {' z1 V
74.                 #endregion
    % |, t: U2 B& U' x
75. 
    
76.                 string descUser = string.Empty;//目的用户
    
77.                 while (true)
    
78.                 {
    % W( Y7 e$ Z* l& T! c$ i
79.                     if (socket.State == WebSocketState.Open)
    , V- k* s0 Z+ ~3 |
80.                     {
    
81.                         ArraySegment<byte> buffer = new ArraySegment<byte>(new byte[2048]);
    
82.                         WebSocketReceiveResult result = await socket.ReceiveAsync(buffer, CancellationToken.None);
    , C- Y  B1 N2 T3 ?2 P, ^0 C
83. 
    
84.                         #region 消息处理（字符截取、消息转发）
    
85.                         try
    " R, i6 m+ \7 h7 G' E/ l
86.                         {
    9 |- _  v1 P# a3 q
87.                             #region 关闭Socket处理，删除连接池
    ( l" R% f0 W# H1 a- |% {% `
88.                             if (socket.State != WebSocketState.Open)//连接关闭
    
89.                             {
    
90.                                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);//删除连接池
    ; P8 O# W5 a, T/ L& m" G3 R; @
91.                                 break;
    
92.                             }
    
93.                             #endregion
    
94. 
    
95.                             string userMsg = Encoding.UTF8.GetString(buffer.Array, 0, result.Count);//发送过来的消息
    ) X- c1 I; m8 D3 [7 X* H1 n+ W
96.                             string[] msgList = userMsg.Split('|');
    
97.                             if (msgList.Length == 2)
    * M; v2 Q1 b1 `# n+ h- b
98.                             {
    
99.                                 if (msgList[0].Trim().Length > 0)
    / w7 z/ [, ^1 p, |, Y7 l
100.                                     descUser = msgList[0].Trim();//记录消息目的用户
     6 b" t  v1 n$ g8 a9 ^; b
101.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(msgList[1]));
     ; a0 f7 _( m7 A* Y1 e
102.                             }
     ; S* b1 x: ]1 Z4 J, c
103.                             else
     1 W$ \6 E+ n0 x4 e! V
104.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(userMsg));
     ; O( A7 y/ ?( {! o
105. 
     
106.                             if (CONNECT_POOL.ContainsKey(descUser))//判断客户端是否在线
     
107.                             {
     
108.                                 WebSocket destSocket = CONNECT_POOL[descUser];//目的客户端
     5 t* a* }. I7 G1 d9 |- X& a7 p
109.                                 if (destSocket != null && destSocket.State == WebSocketState.Open)
     
110.                                     await destSocket.SendAsync(buffer, WebSocketMessageType.Text, true, CancellationToken.None);
     
111.                             }
     
112.                             else
     
113.                             {
     3 O" C  S9 m9 A6 E7 f) ~9 K
114.                                 Task.Run(() =>
     
115.                                 {
     # }4 h1 r3 k; ]7 |  q5 x
116.                                     if (!MESSAGE_POOL.ContainsKey(descUser))//将用户添加至离线消息池中
     4 i; Y( o) n. z
117.                                         MESSAGE_POOL.Add(descUser, new List<MessageInfo>());
     
118.                                     MESSAGE_POOL[descUser].Add(new MessageInfo(DateTime.Now, buffer));//添加离线消息
     
119.                                 });
     5 V9 [' R$ V# |/ G* v9 r
120.                             }
     
121.                         }
     
122.                         catch (Exception exs)
     
123.                         {
     + n5 T1 y" Z$ L2 o
124.                             //消息转发异常处理，本次消息忽略 继续监听接下来的消息
       [0 u4 S0 d% R0 c0 `
125.                         }
     & w8 u! W9 u# ]0 i5 X" n
126.                         #endregion
     
127.                     }
       e, g8 U/ Z* C3 z( G  y  ?) b# O: k$ \
128.                     else
     ; W8 f6 C- b% u  N
129.                     {
     - t1 _* T' D- p
130.                         break;
     
131.                     }
     
132.                 }//while end
     % D! D5 ?5 L: u. ?' W7 x
133.             }
     
134.             catch (Exception ex)
     # b5 U9 v4 \/ F- J8 }
135.             {
     
136.                 //整体异常处理
     0 J$ G% I1 o7 J  P! O5 B' a
137.                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);
     . C% G! A' ~( J7 ^
138.             }
     0 \$ ?8 M& k- ~, U2 ~3 z- n
139.         }
     
140. 
     : u; N8 z! L+ E( U
141. 
     
142.         public bool IsReusable
     1 l6 N8 C# q) w6 |. z9 z5 w
143.         {
     ) [: U+ S5 m: X5 Y
144.             get
     
145.             {
     0 f: S- V0 g8 Y% a
146.                 return false;
     
147.             }
     
148.         }
     
149.     }
     
150. }

复制代码

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

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