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

admin

服务器端代码编写

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

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

3.Handler1.ashx代码如下：

1. using System;
   3 c& S3 c" K$ i. T) f
2. using System.Collections.Generic;
   ; Q# F- s$ I) V6 W' Y
3. using System.Linq;
   
4. using System.Net.WebSockets;
   % M, H" G( J0 h5 ?
5. using System.Text;
   
6. using System.Threading;
   
7. using System.Threading.Tasks;
   # b  l5 c$ r) ~6 `& k
8. using System.Web;
   + l' n: f" Z0 i& v+ M6 A
9. using System.Web.WebSockets;
   0 b8 S4 M$ C( \% P3 m
10. 
    
11. 
    # y. C& i' {4 z
12. namespace WebApplicationWebsocketHandler
    
13. {
    # h% O1 b4 j0 j
14.     /// <summary>
    
15.     /// 离线消息
    ' R; [2 K( d! r
16.     /// </summary>
    8 B1 Y" j5 b, B2 u- S
17.     public class MessageInfo
    + \7 w. u7 L( c5 c$ D5 X
18.     {
    
19.         public MessageInfo(DateTime _MsgTime, ArraySegment<byte> _MsgContent)
    4 J; C: C  N' \. t4 B# ?
20.         {
    % {8 L6 t: S% @: {5 n# _
21.             MsgTime = _MsgTime;
      V, x: h- U  b; X" K: L
22.             MsgContent = _MsgContent;
    , p4 ~, r2 T; t! ~
23.         }
    ) F! S* r7 V0 `* c5 N2 }0 C
24.         public DateTime MsgTime { get; set; }
    
25.         public ArraySegment<byte> MsgContent { get; set; }
    
26.     }
    0 A( B5 x/ L# z& G' h
27. 
    ) U* V$ [/ H# \: w/ \/ V
28. 
    
29. 
    
30. 
    ' G1 I  I+ q" W7 }# `# d# @$ a' w
31.     /// <summary>
    
32.     /// Handler1 的摘要说明
    . C3 H0 I7 Z7 Z' v. g4 `
33.     /// </summary>
    
34.     public class Handler1 : IHttpHandler
    
35.     {
    
36.         private static Dictionary<string, WebSocket> CONNECT_POOL = new Dictionary<string, WebSocket>();//用户连接池
    ! o3 c% t+ e' a  U
37.         private static Dictionary<string, List<MessageInfo>> MESSAGE_POOL = new Dictionary<string, List<MessageInfo>>();//离线消息池
    ( W5 N; a6 {7 e# s" E5 ?2 \
38.         public void ProcessRequest(HttpContext context)
    / v$ e! j- D. ~
39.         {
    : I7 F; x# E2 e4 k2 X9 t  @3 ]
40.             //context.Response.ContentType = "text/plain";
    
41.             //context.Response.Write("Hello World");
    
42.             if (context.IsWebSocketRequest)
    
43.             {
    
44.                 context.AcceptWebSocketRequest(ProcessChat);
    : r, v7 W  b3 U3 D3 w% O8 S, }
45.             }
    2 n) P% c6 ~: e* q; y! d+ Z' k
46.         }
    ' n! R  `4 D' g5 {# F: u
47. 
    " V# p+ w- O  {
48.         private async Task ProcessChat(AspNetWebSocketContext context)
    / u/ D0 I+ z$ i/ f$ }7 s1 P  ~
49.         {
    5 g. I. I* P" r7 _) U5 e. t2 [0 @
50.             WebSocket socket = context.WebSocket;
    9 e7 @# b* P; D) G0 b  V
51.             string user = context.QueryString["user"].ToString();
    
52. 
    
53.             try
    & v9 b/ q: {1 `# Y/ f& L3 z
54.             {
    . \' w8 ?! w% R6 r' c
55.                 #region 用户添加连接池
    6 i# q+ _! b. R' R% f
56.                 //第一次open时，添加到连接池中
    
57.                 if (!CONNECT_POOL.ContainsKey(user))
    2 y8 Z: s; `! J, g
58.                     CONNECT_POOL.Add(user, socket);//不存在，添加
    
59.                 else
    
60.                     if (socket != CONNECT_POOL[user])//当前对象不一致，更新
    / [# a. l! X( ~: p
61.                         CONNECT_POOL[user] = socket;
    ' a0 D- c7 }; t4 }6 h
62.                 #endregion
      t7 ~# X- r8 ?7 k+ I9 V+ n. _
63. 
    
64.                 #region 离线消息处理
    3 q$ X5 h( P5 N% w
65.                 if (MESSAGE_POOL.ContainsKey(user))
    3 ?( Q9 w& s5 f" K1 f
66.                 {
    / q9 m7 P9 y2 P) s" w5 p. @( \
67.                     List<MessageInfo> msgs = MESSAGE_POOL[user];
    . M7 Y, ~! D% o+ v1 U" @, c
68.                     foreach (MessageInfo item in msgs)
    1 d) T. P3 L7 ?1 n/ k' F7 ?- b' r" S
69.                     {
    
70.                         await socket.SendAsync(item.MsgContent, WebSocketMessageType.Text, true, CancellationToken.None);
    
71.                     }
    4 o: P* C( T" Z
72.                     MESSAGE_POOL.Remove(user);//移除离线消息
    
73.                 }
    2 A! o. }* f5 J2 q0 M
74.                 #endregion
    
75. 
    8 f4 b/ Z' p3 \% ]  i& ~; s" i5 ^4 ?2 [
76.                 string descUser = string.Empty;//目的用户
    ( d( Q7 @$ ~& s: V
77.                 while (true)
    
78.                 {
    
79.                     if (socket.State == WebSocketState.Open)
    1 S) R6 o5 s' P) I  J8 @
80.                     {
    ; o2 x9 P- h( i% n
81.                         ArraySegment<byte> buffer = new ArraySegment<byte>(new byte[2048]);
    
82.                         WebSocketReceiveResult result = await socket.ReceiveAsync(buffer, CancellationToken.None);
    
83. 
    
84.                         #region 消息处理（字符截取、消息转发）
    
85.                         try
    
86.                         {
    
87.                             #region 关闭Socket处理，删除连接池
    
88.                             if (socket.State != WebSocketState.Open)//连接关闭
    
89.                             {
    2 Y- D6 q: X$ G9 p0 q3 `7 {2 u3 G
90.                                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);//删除连接池
    
91.                                 break;
    5 c/ u" |$ i- B. r2 q: g/ f- z
92.                             }
    
93.                             #endregion
    
94. 
    % c4 L5 h8 Q1 F; X" q
95.                             string userMsg = Encoding.UTF8.GetString(buffer.Array, 0, result.Count);//发送过来的消息
    
96.                             string[] msgList = userMsg.Split('|');
    ; e' y3 V( V8 t8 k
97.                             if (msgList.Length == 2)
    8 [& u0 n* l( }; G, j) {
98.                             {
    
99.                                 if (msgList[0].Trim().Length > 0)
    # X; O/ k/ R7 R9 S) e4 i
100.                                     descUser = msgList[0].Trim();//记录消息目的用户
     
101.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(msgList[1]));
     - ~+ j. M; f. g. ?; l
102.                             }
     
103.                             else
     $ I9 _# q* M$ u7 a
104.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(userMsg));
     ! e: \& x1 {' P, l7 d+ R# c$ I9 v
105. 
     1 q7 I& S+ d) \# S& Q
106.                             if (CONNECT_POOL.ContainsKey(descUser))//判断客户端是否在线
     8 m1 W' |& _' z: u/ e! t
107.                             {
     1 I! R3 K1 Z% B1 r% K! T* W
108.                                 WebSocket destSocket = CONNECT_POOL[descUser];//目的客户端
     
109.                                 if (destSocket != null && destSocket.State == WebSocketState.Open)
     9 P% s# D# N+ l6 `0 Y
110.                                     await destSocket.SendAsync(buffer, WebSocketMessageType.Text, true, CancellationToken.None);
     
111.                             }
     
112.                             else
     / P* e' T+ @0 g3 n
113.                             {
     5 z" g, v8 ]7 p: c* ], z
114.                                 Task.Run(() =>
     
115.                                 {
     7 y. M4 q! k9 L) ~) w/ q9 k
116.                                     if (!MESSAGE_POOL.ContainsKey(descUser))//将用户添加至离线消息池中
     
117.                                         MESSAGE_POOL.Add(descUser, new List<MessageInfo>());
     
118.                                     MESSAGE_POOL[descUser].Add(new MessageInfo(DateTime.Now, buffer));//添加离线消息
     
119.                                 });
     3 @6 v. ^9 U: J
120.                             }
     2 c/ X/ {2 m, ?' G1 c
121.                         }
     
122.                         catch (Exception exs)
     
123.                         {
     0 \, a# H9 w4 s( x1 M
124.                             //消息转发异常处理，本次消息忽略 继续监听接下来的消息
     4 L1 v& D% J) d1 Z& D( P# Q
125.                         }
     8 @& S) W5 X6 o% M8 j$ d5 m
126.                         #endregion
     
127.                     }
       k, v$ P) t0 t+ o/ T: Y
128.                     else
     
129.                     {
     
130.                         break;
     $ e+ G% U3 T+ z2 s
131.                     }
     " h. V7 h5 @0 {1 B8 ]/ R
132.                 }//while end
     # E+ H/ p" U6 e" U- g3 _
133.             }
     " f+ c& K( w& \. k& d8 \2 ~
134.             catch (Exception ex)
     1 y$ H  ~2 h" z3 f0 ^) P
135.             {
     , p4 Q' t6 ?* _. n% W
136.                 //整体异常处理
     
137.                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);
     
138.             }
     
139.         }
     " ~5 B5 W# @  l2 j; p+ }
140. 
     ! ?5 c- W7 X' D
141. 
     , w0 f: |- m  J+ s6 Y, B0 M$ v
142.         public bool IsReusable
     2 A, z: U+ i/ Y1 A9 U9 q( b
143.         {
     
144.             get
     + o! l1 A4 V/ ^4 ~4 [( m; a! k
145.             {
     : O# y6 {" o' A* w; B8 H) |
146.                 return false;
     
147.             }
     & f$ O& q* E+ k% t% y; ~
148.         }
     
149.     }
     
150. }

复制代码

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

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

) N) k3 ]% u% M/ |5 e