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

admin

服务器端代码编写

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

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

3.Handler1.ashx代码如下：

1. using System;
   : g2 D" n$ ^& J- m6 k
2. using System.Collections.Generic;
   
3. using System.Linq;
   4 D; r( |' B4 E- D
4. using System.Net.WebSockets;
   7 A; e1 K! g+ @% `: k
5. using System.Text;
   , l, J& Y: y: c
6. using System.Threading;
   / j/ z, S& G# @* ]( {: ]: X- k
7. using System.Threading.Tasks;
   
8. using System.Web;
   
9. using System.Web.WebSockets;
   0 f$ I7 B% u: x$ o, {% K& h; h
10. 
    6 c4 |4 G7 R; f
11. 
    
12. namespace WebApplicationWebsocketHandler
    ( b* h, L) d, R
13. {
    
14.     /// <summary>
    # G/ {7 h9 J& B) a1 T( p
15.     /// 离线消息
    6 ^  b% I" o9 E* r6 T: [
16.     /// </summary>
    + w# |. o8 p3 T1 |
17.     public class MessageInfo
    0 J' r- L% ~8 _: o3 G0 z5 z; G
18.     {
    ; c/ b4 n9 E# ]3 J
19.         public MessageInfo(DateTime _MsgTime, ArraySegment<byte> _MsgContent)
    
20.         {
    
21.             MsgTime = _MsgTime;
    
22.             MsgContent = _MsgContent;
    
23.         }
    3 v' _& H# u  j
24.         public DateTime MsgTime { get; set; }
    7 B. P1 f2 k6 p- C9 Y2 ]. H- K
25.         public ArraySegment<byte> MsgContent { get; set; }
    
26.     }
    
27. 
    
28. 
    1 [1 `/ B- j; ]
29. 
    
30. 
    
31.     /// <summary>
    
32.     /// Handler1 的摘要说明
    
33.     /// </summary>
    
34.     public class Handler1 : IHttpHandler
    
35.     {
    ) _2 e* T4 P4 v5 Z: ^
36.         private static Dictionary<string, WebSocket> CONNECT_POOL = new Dictionary<string, WebSocket>();//用户连接池
    8 }( w0 Z3 E3 z  s
37.         private static Dictionary<string, List<MessageInfo>> MESSAGE_POOL = new Dictionary<string, List<MessageInfo>>();//离线消息池
    
38.         public void ProcessRequest(HttpContext context)
    ! h$ X4 L- O6 Q; p
39.         {
    
40.             //context.Response.ContentType = "text/plain";
    
41.             //context.Response.Write("Hello World");
    8 W6 k: s1 Y5 k8 R
42.             if (context.IsWebSocketRequest)
    
43.             {
    
44.                 context.AcceptWebSocketRequest(ProcessChat);
    
45.             }
    
46.         }
    ; d  J5 a. }: K
47. 
    
48.         private async Task ProcessChat(AspNetWebSocketContext context)
    ; p: [& W+ H- N  T
49.         {
    ; D/ l  R* Q% T
50.             WebSocket socket = context.WebSocket;
    & l6 b" j0 m  x! D$ w0 U0 A9 O
51.             string user = context.QueryString["user"].ToString();
    
52. 
    
53.             try
    
54.             {
    8 C+ ]+ d. H4 l% [# _) P) k0 D
55.                 #region 用户添加连接池
    5 b! i5 q& W  @7 r& _
56.                 //第一次open时，添加到连接池中
    
57.                 if (!CONNECT_POOL.ContainsKey(user))
    
58.                     CONNECT_POOL.Add(user, socket);//不存在，添加
    7 V5 D) S3 P9 |8 b- l6 u
59.                 else
    ( k% ?+ m/ e2 Y- R0 L7 B$ p" r# M; V
60.                     if (socket != CONNECT_POOL[user])//当前对象不一致，更新
    
61.                         CONNECT_POOL[user] = socket;
    
62.                 #endregion
    - S) N% H. w5 Z, Y6 p6 g/ j
63. 
    / ~% H- N% z8 j$ n
64.                 #region 离线消息处理
    : `0 V- i% q# k% G9 G
65.                 if (MESSAGE_POOL.ContainsKey(user))
    
66.                 {
    2 ~. ^4 f# Y1 Q
67.                     List<MessageInfo> msgs = MESSAGE_POOL[user];
    8 [- T# Y( N$ h4 u7 h
68.                     foreach (MessageInfo item in msgs)
    
69.                     {
    
70.                         await socket.SendAsync(item.MsgContent, WebSocketMessageType.Text, true, CancellationToken.None);
    , _' q" R1 s# X6 }- y1 A! O! L
71.                     }
    
72.                     MESSAGE_POOL.Remove(user);//移除离线消息
    + r3 M0 k; C: j- C9 d1 v2 C* q1 E
73.                 }
    ( Q( |/ B: l; y: C, h1 N
74.                 #endregion
    
75. 
    
76.                 string descUser = string.Empty;//目的用户
    
77.                 while (true)
    
78.                 {
    
79.                     if (socket.State == WebSocketState.Open)
    , {2 E6 o% q- k( U' M
80.                     {
    8 G: M3 L; ]; Z- h! s5 @! J
81.                         ArraySegment<byte> buffer = new ArraySegment<byte>(new byte[2048]);
    % t1 D' y* u9 C0 C$ M
82.                         WebSocketReceiveResult result = await socket.ReceiveAsync(buffer, CancellationToken.None);
    6 Q8 ?. @8 ~  g) a
83. 
    ; L! l, d& d  ^* L6 ]' }
84.                         #region 消息处理（字符截取、消息转发）
    
85.                         try
    
86.                         {
    
87.                             #region 关闭Socket处理，删除连接池
    
88.                             if (socket.State != WebSocketState.Open)//连接关闭
    " N4 s; d/ v" \
89.                             {
    6 O5 r! K/ T: R4 R
90.                                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);//删除连接池
    + Q# a$ S* L0 w: H) a4 U& x
91.                                 break;
    
92.                             }
    " F" c" o3 N, O3 p# ~8 P# M- J
93.                             #endregion
    & C9 S' j, l) W- [1 s
94. 
    # C) j* q3 D' i
95.                             string userMsg = Encoding.UTF8.GetString(buffer.Array, 0, result.Count);//发送过来的消息
    
96.                             string[] msgList = userMsg.Split('|');
    
97.                             if (msgList.Length == 2)
    * d. t) G3 j3 ^1 _& I: _/ d
98.                             {
    
99.                                 if (msgList[0].Trim().Length > 0)
    
100.                                     descUser = msgList[0].Trim();//记录消息目的用户
     
101.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(msgList[1]));
     2 m- i' X: v; d* _
102.                             }
       @$ _. ]6 v. }- s) I4 J4 e
103.                             else
     & S7 D' h" m, T' m; c5 J+ x
104.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(userMsg));
     # Q7 L# ]7 Q0 n# I
105. 
     
106.                             if (CONNECT_POOL.ContainsKey(descUser))//判断客户端是否在线
     0 n4 r$ h. |: t4 x5 O
107.                             {
     
108.                                 WebSocket destSocket = CONNECT_POOL[descUser];//目的客户端
     6 U+ U4 B% g' R" k8 h8 ~
109.                                 if (destSocket != null && destSocket.State == WebSocketState.Open)
     
110.                                     await destSocket.SendAsync(buffer, WebSocketMessageType.Text, true, CancellationToken.None);
     
111.                             }
       u! k( C3 I2 g; B: W
112.                             else
     , j( @; C$ I9 O, b' ^! l; B; O
113.                             {
     
114.                                 Task.Run(() =>
     9 Z1 d; {2 w7 v$ ^+ W
115.                                 {
     
116.                                     if (!MESSAGE_POOL.ContainsKey(descUser))//将用户添加至离线消息池中
     ) i) \  s# H0 t  B  x) }* X4 D
117.                                         MESSAGE_POOL.Add(descUser, new List<MessageInfo>());
     
118.                                     MESSAGE_POOL[descUser].Add(new MessageInfo(DateTime.Now, buffer));//添加离线消息
     5 O6 }6 x+ ^0 L# V5 I# {! G$ `
119.                                 });
     
120.                             }
     
121.                         }
     
122.                         catch (Exception exs)
     5 m4 w8 U0 c- Q9 s4 A. t" {: w
123.                         {
     / h0 S4 `7 o& ]! b
124.                             //消息转发异常处理，本次消息忽略 继续监听接下来的消息
     0 T" O# k' o7 |) z: W* u
125.                         }
     : `! k9 B# P0 V7 M
126.                         #endregion
     
127.                     }
     2 }" J2 n6 S, \( b3 B' W) g
128.                     else
     ) g" |4 N4 o: Q& R( ~# f
129.                     {
     
130.                         break;
     
131.                     }
     / l8 a+ \4 r9 @3 Q& W
132.                 }//while end
     3 j5 V3 u# W5 _: J5 D+ {& W
133.             }
     
134.             catch (Exception ex)
     
135.             {
     
136.                 //整体异常处理
     , C5 i# U/ x7 H$ Q4 z
137.                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);
     7 N$ z. a3 X5 _6 u: \+ x+ s: ~
138.             }
     
139.         }
     4 c$ `: ]) P' @/ h8 x" F
140. 
     9 R2 z' U9 G9 E+ z$ }+ ^  i: e
141. 
     
142.         public bool IsReusable
     
143.         {
     # O) x. Y2 X  _0 ?" |
144.             get
     
145.             {
     
146.                 return false;
     & n. ~/ w- ~  ~  k. ^4 ~
147.             }
     
148.         }
     
149.     }
     
150. }

复制代码

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

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

# Y9 `0 X1 c6 n% Q- H