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

admin

服务器端代码编写

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

1 c) w0 z- \3 a4 |: l  H- N

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

3.Handler1.ashx代码如下：

1. using System;
   
2. using System.Collections.Generic;
   ) A% s0 e  l: q$ C6 o
3. using System.Linq;
   ! F+ H" C5 j1 I: r/ O
4. using System.Net.WebSockets;
   ; S. l# s) d) M# ?" O/ T' I2 ]; r
5. using System.Text;
   
6. using System.Threading;
   & N. }1 Z# K! Q1 g8 ^
7. using System.Threading.Tasks;
   
8. using System.Web;
   4 ?7 n8 V* |+ b! ]/ w
9. using System.Web.WebSockets;
   
10. 
    
11. 
    
12. namespace WebApplicationWebsocketHandler
    ! L) H1 B6 ^. [+ X& D7 w
13. {
    
14.     /// <summary>
    
15.     /// 离线消息
    
16.     /// </summary>
    + h$ V8 D7 Q, z4 K( Y
17.     public class MessageInfo
    ) i# |  M* b6 V, ]. ?
18.     {
    
19.         public MessageInfo(DateTime _MsgTime, ArraySegment<byte> _MsgContent)
    ' R* ^) v$ {& Y- p5 z: _8 K, Z
20.         {
    
21.             MsgTime = _MsgTime;
    
22.             MsgContent = _MsgContent;
    
23.         }
    
24.         public DateTime MsgTime { get; set; }
    4 P6 a' Z/ N% v) j
25.         public ArraySegment<byte> MsgContent { get; set; }
    
26.     }
    
27. 
    
28. 
    $ \+ @" Q  D4 \+ y2 Z
29. 
    
30. 
    , ?' e1 E# E2 I; Y& s
31.     /// <summary>
    
32.     /// Handler1 的摘要说明
    
33.     /// </summary>
    " z! W0 ~8 y7 E% w  s7 z  N( X2 w
34.     public class Handler1 : IHttpHandler
    0 J2 a3 k5 b: E, j$ ]
35.     {
    
36.         private static Dictionary<string, WebSocket> CONNECT_POOL = new Dictionary<string, WebSocket>();//用户连接池
    ! d$ K, n& U" F9 l
37.         private static Dictionary<string, List<MessageInfo>> MESSAGE_POOL = new Dictionary<string, List<MessageInfo>>();//离线消息池
    # [4 A. m2 k  H: Z! |  i
38.         public void ProcessRequest(HttpContext context)
    
39.         {
    . r8 ?  t* h7 v9 d
40.             //context.Response.ContentType = "text/plain";
    
41.             //context.Response.Write("Hello World");
    
42.             if (context.IsWebSocketRequest)
    
43.             {
    4 s7 ]; [8 r9 d% N4 @
44.                 context.AcceptWebSocketRequest(ProcessChat);
    ! Y1 h# L+ [, n2 c  G. A$ E0 R
45.             }
    - T5 t5 O; @6 ^9 ~% Q+ l
46.         }
    0 ]' ]+ Y5 j# H0 g' f. t. @
47. 
    
48.         private async Task ProcessChat(AspNetWebSocketContext context)
    . ^1 b6 `* Z0 Q. m
49.         {
    
50.             WebSocket socket = context.WebSocket;
    5 `& N4 o  n. G4 W5 f
51.             string user = context.QueryString["user"].ToString();
    
52. 
    
53.             try
    
54.             {
    ( p. }/ z5 o7 X1 p" ~$ C. W
55.                 #region 用户添加连接池
    
56.                 //第一次open时，添加到连接池中
    
57.                 if (!CONNECT_POOL.ContainsKey(user))
    9 X" a' o" M8 F7 D& i. N4 ?8 L4 U5 t
58.                     CONNECT_POOL.Add(user, socket);//不存在，添加
    
59.                 else
    % Y0 f/ q# B' W+ S6 L+ q! ~
60.                     if (socket != CONNECT_POOL[user])//当前对象不一致，更新
    
61.                         CONNECT_POOL[user] = socket;
    
62.                 #endregion
    
63. 
    $ x* ?6 z# W# A: K3 u
64.                 #region 离线消息处理
    
65.                 if (MESSAGE_POOL.ContainsKey(user))
    ) W. c4 X8 A$ ~% \) R
66.                 {
    , S7 a. S' P5 V
67.                     List<MessageInfo> msgs = MESSAGE_POOL[user];
    
68.                     foreach (MessageInfo item in msgs)
    0 M* ~% _. @0 s+ g! r. j; ]
69.                     {
    
70.                         await socket.SendAsync(item.MsgContent, WebSocketMessageType.Text, true, CancellationToken.None);
    " u! e" J0 H( |. q% K: g; c
71.                     }
    
72.                     MESSAGE_POOL.Remove(user);//移除离线消息
    
73.                 }
    8 C5 G) w1 g& v
74.                 #endregion
    
75. 
    
76.                 string descUser = string.Empty;//目的用户
    / ^$ t: S4 f4 C
77.                 while (true)
    
78.                 {
    2 i6 @4 x: Y6 m$ [
79.                     if (socket.State == WebSocketState.Open)
    
80.                     {
    0 b0 I( [3 l' `3 Q- e$ D
81.                         ArraySegment<byte> buffer = new ArraySegment<byte>(new byte[2048]);
    5 h. C$ I% `* M2 L7 |, F0 g
82.                         WebSocketReceiveResult result = await socket.ReceiveAsync(buffer, CancellationToken.None);
    / L/ d$ c' r( [: R0 d$ }5 c
83. 
    & u& Y  n$ e2 j
84.                         #region 消息处理（字符截取、消息转发）
    + k% P" P/ g  d, w' L3 ^6 @% h
85.                         try
    # `- `' Q/ Q  e5 a, g# F/ W
86.                         {
    
87.                             #region 关闭Socket处理，删除连接池
    
88.                             if (socket.State != WebSocketState.Open)//连接关闭
    6 n1 h! p& ^$ \! n
89.                             {
      b5 ^5 t' V9 B1 m* _: k
90.                                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);//删除连接池
    $ F' C+ S- I9 t/ O9 u/ L8 z4 v
91.                                 break;
    ' o! r+ D5 H% h
92.                             }
    * k3 O3 C: i* n( @- `
93.                             #endregion
    ! G( G( ~: Q1 e( i. ~
94. 
    
95.                             string userMsg = Encoding.UTF8.GetString(buffer.Array, 0, result.Count);//发送过来的消息
    
96.                             string[] msgList = userMsg.Split('|');
    
97.                             if (msgList.Length == 2)
    
98.                             {
    
99.                                 if (msgList[0].Trim().Length > 0)
    # w8 [  I8 N9 z7 G7 h, B: E3 o
100.                                     descUser = msgList[0].Trim();//记录消息目的用户
     & U/ e% ?. m2 s5 n  @" z5 R
101.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(msgList[1]));
     : m& e& w! x1 z; _1 W" |
102.                             }
     
103.                             else
     
104.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(userMsg));
     ) W( _; s7 [. M" J( X" L* P4 o! h5 T3 ]
105. 
     4 C, x: C/ x8 M( D% n
106.                             if (CONNECT_POOL.ContainsKey(descUser))//判断客户端是否在线
     ' M( E- G& m9 Z3 F' Q
107.                             {
     " B4 _% F4 _0 q. t
108.                                 WebSocket destSocket = CONNECT_POOL[descUser];//目的客户端
     
109.                                 if (destSocket != null && destSocket.State == WebSocketState.Open)
     
110.                                     await destSocket.SendAsync(buffer, WebSocketMessageType.Text, true, CancellationToken.None);
     ) g! o. o7 Z  ]
111.                             }
     ) H  w% e+ C+ i, l
112.                             else
     
113.                             {
     
114.                                 Task.Run(() =>
     & K) x# j/ c8 C- q
115.                                 {
     $ ]0 d7 q! q' L) p% o4 \
116.                                     if (!MESSAGE_POOL.ContainsKey(descUser))//将用户添加至离线消息池中
     + P1 n+ [5 z, Q3 z- A5 C6 @
117.                                         MESSAGE_POOL.Add(descUser, new List<MessageInfo>());
     
118.                                     MESSAGE_POOL[descUser].Add(new MessageInfo(DateTime.Now, buffer));//添加离线消息
     ; |3 Q* F: p/ ]: a9 p3 X, ^& v
119.                                 });
     ' R8 F7 Z  p3 w" U
120.                             }
     
121.                         }
     # w) l/ B8 u$ L. z; U
122.                         catch (Exception exs)
     : U: |4 e; @' X: W/ Y  b
123.                         {
     + [" E& [+ h. H  J" J, i/ T$ ^
124.                             //消息转发异常处理，本次消息忽略 继续监听接下来的消息
     " \/ e4 Z& p5 K2 m8 N
125.                         }
     
126.                         #endregion
     
127.                     }
     " h& x5 j+ {1 W5 \
128.                     else
     : U- K; i! x, N- k' C: N
129.                     {
     9 `  c! M1 G4 k# \& F
130.                         break;
     3 E2 ^* r$ V- {# q7 q
131.                     }
     . y; m) z$ ^" {5 f7 v
132.                 }//while end
     + I, E& L( _5 K5 Q' G* F9 @5 z
133.             }
     
134.             catch (Exception ex)
     7 E/ a8 \3 c3 V8 {
135.             {
     
136.                 //整体异常处理
     
137.                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);
     
138.             }
     4 H4 g9 `2 S" c
139.         }
     1 M, ]; }; `; D  T% d3 O
140. 
     
141. 
     ! Y1 K- {# D; f" [' ~: @" z5 a& e
142.         public bool IsReusable
     
143.         {
     
144.             get
     
145.             {
     
146.                 return false;
     ' e; X) v' C. M' h/ @
147.             }
     
148.         }
     4 J2 e) \( F) K4 _0 t# ]
149.     }
     
150. }

复制代码

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

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

2 O% ]- X- I: F# _3 {