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

admin

服务器端代码编写

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

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

3.Handler1.ashx代码如下：

1. using System;
   
2. using System.Collections.Generic;
   & D  t8 K; }9 s1 S$ m
3. using System.Linq;
   $ I# b, B/ j. X, ~
4. using System.Net.WebSockets;
   
5. using System.Text;
   & U, z. M& b& i5 x
6. using System.Threading;
   * u9 J% a% b  ~; i0 v9 i' P* \
7. using System.Threading.Tasks;
   " z. E3 [  L( b9 f  @5 X
8. using System.Web;
   
9. using System.Web.WebSockets;
   ( q  F  @% g! [3 I" s
10. 
    
11. 
    $ k# N" f6 L9 X' \
12. namespace WebApplicationWebsocketHandler
    4 d0 D# _* g$ Y& s9 i
13. {
    
14.     /// <summary>
      ~% v/ M* }( b/ b5 }8 M6 O: l$ o
15.     /// 离线消息
    
16.     /// </summary>
    2 b* R, m3 Z# N) l; @' E! |$ D
17.     public class MessageInfo
    
18.     {
    
19.         public MessageInfo(DateTime _MsgTime, ArraySegment<byte> _MsgContent)
    
20.         {
    
21.             MsgTime = _MsgTime;
    
22.             MsgContent = _MsgContent;
    
23.         }
    0 u+ A- R" c* J0 v8 z5 _5 Q2 v
24.         public DateTime MsgTime { get; set; }
    ) W* Z) H/ g/ @6 k+ F! _0 [
25.         public ArraySegment<byte> MsgContent { get; set; }
    
26.     }
    + b* A! m# v" {# o8 G. R
27. 
    
28. 
    5 _! a9 w, B8 U6 G* N: V$ g
29. 
    ; ^" C: ^0 i, Z8 e
30. 
    
31.     /// <summary>
    
32.     /// Handler1 的摘要说明
    
33.     /// </summary>
    
34.     public class Handler1 : IHttpHandler
    8 Y# R* F8 I- |9 [
35.     {
      [! I% K: `7 h: `0 @1 I/ q3 h9 ~3 A
36.         private static Dictionary<string, WebSocket> CONNECT_POOL = new Dictionary<string, WebSocket>();//用户连接池
    0 q* O1 Q* i4 E5 |2 a9 p" \/ y8 `
37.         private static Dictionary<string, List<MessageInfo>> MESSAGE_POOL = new Dictionary<string, List<MessageInfo>>();//离线消息池
    
38.         public void ProcessRequest(HttpContext context)
    
39.         {
    : v, N1 A; D. ?( S
40.             //context.Response.ContentType = "text/plain";
    
41.             //context.Response.Write("Hello World");
    
42.             if (context.IsWebSocketRequest)
    
43.             {
    
44.                 context.AcceptWebSocketRequest(ProcessChat);
    . _. g  H; j6 r9 o6 A0 A
45.             }
    
46.         }
    ( d: i; N" M0 j/ U
47. 
    
48.         private async Task ProcessChat(AspNetWebSocketContext context)
    : j6 k5 x6 Z. z# V8 C
49.         {
    
50.             WebSocket socket = context.WebSocket;
    : q1 P/ O" L6 z, q. ]# H% |. p+ q
51.             string user = context.QueryString["user"].ToString();
    
52. 
    % k" a; i* l7 z2 ^* A6 T9 a
53.             try
    # P5 ~% s: ]! U; ]0 _) G
54.             {
    
55.                 #region 用户添加连接池
    
56.                 //第一次open时，添加到连接池中
    4 @( j/ O, C$ \
57.                 if (!CONNECT_POOL.ContainsKey(user))
    ( P( e& C4 l6 c7 t4 o: T
58.                     CONNECT_POOL.Add(user, socket);//不存在，添加
    
59.                 else
    
60.                     if (socket != CONNECT_POOL[user])//当前对象不一致，更新
    
61.                         CONNECT_POOL[user] = socket;
    1 y( j: A  e2 n
62.                 #endregion
    " m2 \1 x) u( ]% a% z0 N7 h1 @: V
63. 
    
64.                 #region 离线消息处理
    2 @+ ^+ Q  p7 k' ?
65.                 if (MESSAGE_POOL.ContainsKey(user))
    
66.                 {
    
67.                     List<MessageInfo> msgs = MESSAGE_POOL[user];
    
68.                     foreach (MessageInfo item in msgs)
    
69.                     {
    
70.                         await socket.SendAsync(item.MsgContent, WebSocketMessageType.Text, true, CancellationToken.None);
    
71.                     }
    ) ?' q7 u# i4 k. U/ E
72.                     MESSAGE_POOL.Remove(user);//移除离线消息
    
73.                 }
    
74.                 #endregion
    - v+ C2 {- P( Z
75. 
    
76.                 string descUser = string.Empty;//目的用户
    
77.                 while (true)
    ; R: P) S! o& x6 G8 d
78.                 {
    
79.                     if (socket.State == WebSocketState.Open)
    * \) x' R* I. y
80.                     {
    ! ?0 O2 ^' V5 B- q; f, w
81.                         ArraySegment<byte> buffer = new ArraySegment<byte>(new byte[2048]);
    : x9 ~* h# c4 A
82.                         WebSocketReceiveResult result = await socket.ReceiveAsync(buffer, CancellationToken.None);
    * o9 b) s- d( Q1 x* ?
83. 
    ' \/ [% |: B4 ?  I3 [- r
84.                         #region 消息处理（字符截取、消息转发）
    
85.                         try
    
86.                         {
    " L$ `% l0 d  Y/ Q5 [% K
87.                             #region 关闭Socket处理，删除连接池
    
88.                             if (socket.State != WebSocketState.Open)//连接关闭
    
89.                             {
    
90.                                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);//删除连接池
    * |! B2 G6 B( y  F/ v' p
91.                                 break;
    
92.                             }
    
93.                             #endregion
    
94. 
    8 w7 q/ y& U8 @
95.                             string userMsg = Encoding.UTF8.GetString(buffer.Array, 0, result.Count);//发送过来的消息
    
96.                             string[] msgList = userMsg.Split('|');
    % d$ c# t( @2 i( N" ~+ L! s  f
97.                             if (msgList.Length == 2)
    . i! `( t1 {* d/ ^# b3 ~
98.                             {
    * V9 S! {! n: e) Q1 J
99.                                 if (msgList[0].Trim().Length > 0)
    
100.                                     descUser = msgList[0].Trim();//记录消息目的用户
     3 k; h, B( x, s! \2 T  ^; I7 c
101.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(msgList[1]));
     
102.                             }
     
103.                             else
     , H) e# m; F0 t1 l( E
104.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(userMsg));
     * V6 E5 x! u5 n4 P  }1 i) U
105. 
     2 _0 K4 T5 l; C+ r' C
106.                             if (CONNECT_POOL.ContainsKey(descUser))//判断客户端是否在线
     
107.                             {
     
108.                                 WebSocket destSocket = CONNECT_POOL[descUser];//目的客户端
     
109.                                 if (destSocket != null && destSocket.State == WebSocketState.Open)
     1 [: _4 z7 w+ F4 F' I
110.                                     await destSocket.SendAsync(buffer, WebSocketMessageType.Text, true, CancellationToken.None);
     
111.                             }
     
112.                             else
     
113.                             {
     
114.                                 Task.Run(() =>
     
115.                                 {
     
116.                                     if (!MESSAGE_POOL.ContainsKey(descUser))//将用户添加至离线消息池中
     9 c+ K4 d$ K; n1 O
117.                                         MESSAGE_POOL.Add(descUser, new List<MessageInfo>());
     
118.                                     MESSAGE_POOL[descUser].Add(new MessageInfo(DateTime.Now, buffer));//添加离线消息
     4 Q1 W/ K6 O9 q% Y2 O  k
119.                                 });
     
120.                             }
     5 E6 m/ T* u& ^! S. k
121.                         }
     
122.                         catch (Exception exs)
     : E, H) X9 Y' {: x
123.                         {
     7 f% \% x7 [4 I7 H$ L
124.                             //消息转发异常处理，本次消息忽略 继续监听接下来的消息
     
125.                         }
     7 f! R+ ^* F, v( X: m' M0 o" ^1 f
126.                         #endregion
       G& A' o. S$ V! f$ ~
127.                     }
     
128.                     else
     5 t0 K* f% @6 T$ k
129.                     {
     
130.                         break;
     1 W& H3 ?2 @; `: w/ N9 \; x
131.                     }
     
132.                 }//while end
     
133.             }
     # H1 _& q; d; b, D* v8 w# ~
134.             catch (Exception ex)
     . ?, B; k+ V$ Q7 k' ~
135.             {
     
136.                 //整体异常处理
     ( F" D/ i# q$ U
137.                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);
     * O! h6 Z; T9 Z; t
138.             }
     
139.         }
     
140. 
     
141. 
     
142.         public bool IsReusable
     5 s# ~0 x' A# m  T% ~% |
143.         {
     
144.             get
     
145.             {
     
146.                 return false;
     6 w6 N: t5 K9 f3 H
147.             }
     ) n# P" V: c  t8 y& N; C7 x
148.         }
     # X, a" w' b, m, q& f  U6 G
149.     }
     ) r. n7 i% B, @* Y
150. }

复制代码

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

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