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

admin

服务器端代码编写

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

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

3.Handler1.ashx代码如下：

1. using System;
   , O. n* A3 k! D3 C6 n' Y) p1 y
2. using System.Collections.Generic;
   , K% H/ t$ v8 I3 V4 z' S/ a) \1 b+ {
3. using System.Linq;
   
4. using System.Net.WebSockets;
   & Z, x4 _3 v& d: o3 L# O) b
5. using System.Text;
   
6. using System.Threading;
   
7. using System.Threading.Tasks;
   
8. using System.Web;
   6 v8 U: T' K) E/ @! b, |/ m
9. using System.Web.WebSockets;
   
10. 
    
11. 
    : [/ q- c* D( q! D% v  z- A
12. namespace WebApplicationWebsocketHandler
    / Y4 A( y* H3 D5 M1 B
13. {
    
14.     /// <summary>
    2 H; v# }2 S. f9 F; z- a
15.     /// 离线消息
    
16.     /// </summary>
    2 h+ p* e3 P/ J! M: Y
17.     public class MessageInfo
    
18.     {
    
19.         public MessageInfo(DateTime _MsgTime, ArraySegment<byte> _MsgContent)
    
20.         {
    
21.             MsgTime = _MsgTime;
    
22.             MsgContent = _MsgContent;
    , q9 X1 L) h1 q
23.         }
    8 l! _" @- t/ r  Q/ m, J
24.         public DateTime MsgTime { get; set; }
    # A1 l- E) A7 c' j! _
25.         public ArraySegment<byte> MsgContent { get; set; }
    
26.     }
    
27. 
    
28. 
    % d# Z- q$ e* O; K
29. 
    3 w2 o5 I6 ?! W
30. 
    
31.     /// <summary>
    % N# b' r! z- L2 d
32.     /// Handler1 的摘要说明
    
33.     /// </summary>
    0 M/ L8 S( _, Q  I
34.     public class Handler1 : IHttpHandler
    - i# @3 O8 S0 N" d0 a0 Q1 ^. A
35.     {
    
36.         private static Dictionary<string, WebSocket> CONNECT_POOL = new Dictionary<string, WebSocket>();//用户连接池
    ( Y1 a3 R% O/ o- v: i
37.         private static Dictionary<string, List<MessageInfo>> MESSAGE_POOL = new Dictionary<string, List<MessageInfo>>();//离线消息池
    , f8 j5 y: e# _! w) C4 x; p
38.         public void ProcessRequest(HttpContext context)
    
39.         {
    
40.             //context.Response.ContentType = "text/plain";
    * ~/ a; r/ X- I( d9 P. e. e
41.             //context.Response.Write("Hello World");
    
42.             if (context.IsWebSocketRequest)
      T1 ?- X9 u3 }+ Q
43.             {
    . J/ \# {1 e4 v$ m$ m# F
44.                 context.AcceptWebSocketRequest(ProcessChat);
    ! g9 G" }( n* {% t$ C
45.             }
    , I. P$ l  G+ b
46.         }
    5 `, x( A6 m! ?- ?4 p0 r2 {
47. 
    + u! F, b2 S/ K( k* A1 `& a& J
48.         private async Task ProcessChat(AspNetWebSocketContext context)
    9 f7 c0 `; X, r9 v( @
49.         {
    * u0 K8 R) e# l0 B
50.             WebSocket socket = context.WebSocket;
    
51.             string user = context.QueryString["user"].ToString();
    : S1 l; t5 s3 F8 a" F
52. 
    # r, {- F& O+ f0 W
53.             try
    * y9 I6 ?3 R9 v+ K2 N) G& C5 I+ _' d
54.             {
    
55.                 #region 用户添加连接池
    
56.                 //第一次open时，添加到连接池中
    
57.                 if (!CONNECT_POOL.ContainsKey(user))
    " v* L- g4 j' w7 Q+ w9 g: C+ X  I
58.                     CONNECT_POOL.Add(user, socket);//不存在，添加
    
59.                 else
    
60.                     if (socket != CONNECT_POOL[user])//当前对象不一致，更新
    ; Q! t; r  N# e' Y
61.                         CONNECT_POOL[user] = socket;
    3 |8 j/ H" W. o3 S) Q8 N
62.                 #endregion
    
63. 
    ' H4 c1 M, T/ x( k: G' [- ^
64.                 #region 离线消息处理
    
65.                 if (MESSAGE_POOL.ContainsKey(user))
    
66.                 {
    ! O0 X1 e6 g; z0 U( D- P/ p
67.                     List<MessageInfo> msgs = MESSAGE_POOL[user];
    9 [  M' X3 u/ h' G
68.                     foreach (MessageInfo item in msgs)
    
69.                     {
    5 D0 }. d4 w! z; x9 S) h/ e; ?, S
70.                         await socket.SendAsync(item.MsgContent, WebSocketMessageType.Text, true, CancellationToken.None);
    , [$ X9 C' Y4 F/ G% |
71.                     }
    
72.                     MESSAGE_POOL.Remove(user);//移除离线消息
    # U$ k  O' P0 B# J. I
73.                 }
    
74.                 #endregion
    
75. 
    7 z* [/ _8 @# g1 V. A; p% d
76.                 string descUser = string.Empty;//目的用户
    
77.                 while (true)
    
78.                 {
    4 {. `$ Z+ q. f, c  b) E; i4 h) |
79.                     if (socket.State == WebSocketState.Open)
    
80.                     {
    ( J0 C. k( K( g- ]; F+ Y# e
81.                         ArraySegment<byte> buffer = new ArraySegment<byte>(new byte[2048]);
    
82.                         WebSocketReceiveResult result = await socket.ReceiveAsync(buffer, CancellationToken.None);
    
83. 
    
84.                         #region 消息处理（字符截取、消息转发）
    
85.                         try
    ' Y8 M3 h; O# B- g6 k
86.                         {
    & L& u8 y$ Z: x7 ~8 Y* S
87.                             #region 关闭Socket处理，删除连接池
    
88.                             if (socket.State != WebSocketState.Open)//连接关闭
    2 E8 h$ \5 p& U5 U7 v: Z% R/ g. o
89.                             {
    
90.                                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);//删除连接池
    + W% P/ j* @0 N, Q
91.                                 break;
    5 b! O% z- ~$ Q% e% m% z
92.                             }
    
93.                             #endregion
    
94. 
    3 i( z. t* s/ D# D4 m" d+ L! e
95.                             string userMsg = Encoding.UTF8.GetString(buffer.Array, 0, result.Count);//发送过来的消息
    
96.                             string[] msgList = userMsg.Split('|');
    
97.                             if (msgList.Length == 2)
    ' j( P5 g2 k  [  q9 t& I9 W
98.                             {
    4 m) N5 l9 S& t4 i  P
99.                                 if (msgList[0].Trim().Length > 0)
    
100.                                     descUser = msgList[0].Trim();//记录消息目的用户
     
101.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(msgList[1]));
     ; g9 k/ A' v9 i2 ]
102.                             }
     ( x/ o6 S" B" b7 f
103.                             else
     
104.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(userMsg));
     
105. 
     
106.                             if (CONNECT_POOL.ContainsKey(descUser))//判断客户端是否在线
     - P+ H* o; ]: [
107.                             {
     
108.                                 WebSocket destSocket = CONNECT_POOL[descUser];//目的客户端
     
109.                                 if (destSocket != null && destSocket.State == WebSocketState.Open)
     ! n# l+ T5 E1 Z2 x8 i$ S
110.                                     await destSocket.SendAsync(buffer, WebSocketMessageType.Text, true, CancellationToken.None);
     0 Z. y* q& s" T1 Y( D
111.                             }
     
112.                             else
     # a1 w# r) w& \+ j) ?
113.                             {
     
114.                                 Task.Run(() =>
     
115.                                 {
     
116.                                     if (!MESSAGE_POOL.ContainsKey(descUser))//将用户添加至离线消息池中
     4 l% g/ G  ~! J3 o! }
117.                                         MESSAGE_POOL.Add(descUser, new List<MessageInfo>());
     7 {! x( t6 v/ C. N& n
118.                                     MESSAGE_POOL[descUser].Add(new MessageInfo(DateTime.Now, buffer));//添加离线消息
     
119.                                 });
     $ \' E+ O  ~4 r9 ]
120.                             }
     
121.                         }
     
122.                         catch (Exception exs)
     
123.                         {
     $ d4 n: ]/ G, [; @% S0 E& c8 w
124.                             //消息转发异常处理，本次消息忽略 继续监听接下来的消息
     $ [7 h: @+ D6 j7 F
125.                         }
     
126.                         #endregion
     
127.                     }
     
128.                     else
     ! f, H# @. W* M8 v% V+ g# Q
129.                     {
     
130.                         break;
     7 o9 ~9 o" w3 n$ y9 o
131.                     }
     
132.                 }//while end
     
133.             }
     
134.             catch (Exception ex)
     
135.             {
     1 f1 s' b6 c# b% a, ~9 Y
136.                 //整体异常处理
     , s* j. Y% y9 v( }# p4 i$ y/ ?, G
137.                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);
     
138.             }
     : j. ]  N5 J0 \, f
139.         }
     : g6 d$ N# V; O- Y0 ?0 S. U
140. 
     2 U; {3 U( P9 ?2 O5 d
141. 
     
142.         public bool IsReusable
     
143.         {
     
144.             get
     
145.             {
     " c1 O) A& I% q. P8 G9 D
146.                 return false;
     / f  i! ^6 n8 A& e% G) G) d
147.             }
     
148.         }
     # P! O" {' w3 q8 L; n& `
149.     }
     
150. }

复制代码

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

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

  r4 g! [* E) c4 E