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

admin

服务器端代码编写

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

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

3.Handler1.ashx代码如下：

1. using System;
   ! e& ?0 Y. M8 e$ Z, l
2. using System.Collections.Generic;
   
3. using System.Linq;
   
4. using System.Net.WebSockets;
   
5. using System.Text;
   2 C3 l; Y& G9 A5 B0 x* v
6. using System.Threading;
   
7. using System.Threading.Tasks;
   
8. using System.Web;
   
9. using System.Web.WebSockets;
   1 g& f  s' ~* j4 ~# A9 x
10. 
    
11. 
    ! p; p" q+ n& p; Y, Q4 ~
12. namespace WebApplicationWebsocketHandler
    
13. {
    $ D, ~; J6 s6 x/ |- I* c) q4 N
14.     /// <summary>
    
15.     /// 离线消息
    
16.     /// </summary>
    4 v! i& H/ m' f/ T) U' p
17.     public class MessageInfo
    
18.     {
    
19.         public MessageInfo(DateTime _MsgTime, ArraySegment<byte> _MsgContent)
    
20.         {
    
21.             MsgTime = _MsgTime;
    ) Z. S( o' h8 \0 L" F. H6 u- R- e
22.             MsgContent = _MsgContent;
    2 V. o0 k* L, X( z5 g) d# Q  X
23.         }
    % g$ C+ \6 S+ `# N4 T
24.         public DateTime MsgTime { get; set; }
      B' |5 q$ H( Y$ M
25.         public ArraySegment<byte> MsgContent { get; set; }
    4 Y. y7 Y* }8 x4 ?3 M! Q
26.     }
    & I9 E1 H/ G9 ~8 r
27. 
    
28. 
    ) Z* e$ I3 E& u& B( [
29. 
    
30. 
    
31.     /// <summary>
    6 e: K2 h9 r6 ^
32.     /// Handler1 的摘要说明
      x3 N/ r( L7 T. ]
33.     /// </summary>
    * T+ y+ [" ?# L# Y" `
34.     public class Handler1 : IHttpHandler
    
35.     {
    
36.         private static Dictionary<string, WebSocket> CONNECT_POOL = new Dictionary<string, WebSocket>();//用户连接池
    
37.         private static Dictionary<string, List<MessageInfo>> MESSAGE_POOL = new Dictionary<string, List<MessageInfo>>();//离线消息池
    
38.         public void ProcessRequest(HttpContext context)
    9 Z2 [8 `0 \1 C! K: I
39.         {
    ; v" f& `( y* T. R% p$ T" D
40.             //context.Response.ContentType = "text/plain";
    
41.             //context.Response.Write("Hello World");
    : i; D$ f5 L) [2 s
42.             if (context.IsWebSocketRequest)
    
43.             {
    
44.                 context.AcceptWebSocketRequest(ProcessChat);
    
45.             }
    
46.         }
    
47. 
    
48.         private async Task ProcessChat(AspNetWebSocketContext context)
    ; Z7 Y  c9 \! c- B3 [& _8 T
49.         {
    / B5 I& Q* ~# l/ \, d# \
50.             WebSocket socket = context.WebSocket;
    0 {! p8 R6 [$ d: o; Y% O
51.             string user = context.QueryString["user"].ToString();
    
52. 
    / ~6 D& B) z- F. f; P
53.             try
    
54.             {
    & r9 D9 ^- n5 P+ U, t+ b7 L
55.                 #region 用户添加连接池
    8 N* G2 k2 n+ F0 e* |* Q
56.                 //第一次open时，添加到连接池中
    5 N2 ?- s& {8 x6 {- I
57.                 if (!CONNECT_POOL.ContainsKey(user))
    $ t0 a: Z) D" }) v* l. P: F
58.                     CONNECT_POOL.Add(user, socket);//不存在，添加
    3 x6 S" p  T, _1 Y! D
59.                 else
    : H- \% H' U/ z6 e& `3 Y& A
60.                     if (socket != CONNECT_POOL[user])//当前对象不一致，更新
    9 M" C& T+ C9 h/ @$ M. V
61.                         CONNECT_POOL[user] = socket;
    9 D! e. K+ B( ?/ o9 g4 M
62.                 #endregion
    4 ~% R& O7 K) ?
63. 
    
64.                 #region 离线消息处理
    ) l; n& F# N5 `: H
65.                 if (MESSAGE_POOL.ContainsKey(user))
    % g: u3 T9 c9 r& b' e  M8 L0 z. ~
66.                 {
    7 h8 ?2 \& k, `* K
67.                     List<MessageInfo> msgs = MESSAGE_POOL[user];
    
68.                     foreach (MessageInfo item in msgs)
    
69.                     {
    ) G2 l" O& R! Q( }+ ]5 q
70.                         await socket.SendAsync(item.MsgContent, WebSocketMessageType.Text, true, CancellationToken.None);
    
71.                     }
    # m4 i" y7 Y8 E+ {  B1 B- j) C
72.                     MESSAGE_POOL.Remove(user);//移除离线消息
    
73.                 }
    / Z6 Q6 ?. G9 O( p! e' B& Q. M
74.                 #endregion
    
75. 
    , H7 S- |2 r* U1 Q$ L
76.                 string descUser = string.Empty;//目的用户
    % g( U3 j7 E0 ]- o" L- T
77.                 while (true)
    7 s) ]+ z& @; u- U/ b
78.                 {
    
79.                     if (socket.State == WebSocketState.Open)
    2 b4 y! k7 D% J# P
80.                     {
    
81.                         ArraySegment<byte> buffer = new ArraySegment<byte>(new byte[2048]);
    
82.                         WebSocketReceiveResult result = await socket.ReceiveAsync(buffer, CancellationToken.None);
    
83. 
    
84.                         #region 消息处理（字符截取、消息转发）
    - f0 F! ^  S+ M
85.                         try
    2 ~; ]% Z0 `" }2 s
86.                         {
    
87.                             #region 关闭Socket处理，删除连接池
    - l+ x& c! K* h0 S) w! r- R
88.                             if (socket.State != WebSocketState.Open)//连接关闭
    
89.                             {
    
90.                                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);//删除连接池
    6 M4 m+ ?. m; F! K) T- ?
91.                                 break;
    5 Q) a% G4 V. b* d0 ]" j
92.                             }
    
93.                             #endregion
    
94. 
    + q! Z  W+ K/ V9 B8 M7 U, S8 x" d2 O
95.                             string userMsg = Encoding.UTF8.GetString(buffer.Array, 0, result.Count);//发送过来的消息
    * E( m' S# b! I1 K0 n
96.                             string[] msgList = userMsg.Split('|');
    
97.                             if (msgList.Length == 2)
    ; W: }% S) Q% E
98.                             {
    
99.                                 if (msgList[0].Trim().Length > 0)
    $ N: M/ n6 _! H8 B- T+ m7 \
100.                                     descUser = msgList[0].Trim();//记录消息目的用户
     
101.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(msgList[1]));
     + b2 R4 Z- P+ l( }& z4 W
102.                             }
     6 h. k; n6 w6 O$ y
103.                             else
     
104.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(userMsg));
     , H+ a; v+ U4 ^- [) A
105. 
     / n8 ?6 Y# C- b
106.                             if (CONNECT_POOL.ContainsKey(descUser))//判断客户端是否在线
     
107.                             {
     / C: z) P' M7 x% G' l: @- p/ Z
108.                                 WebSocket destSocket = CONNECT_POOL[descUser];//目的客户端
     , Q3 c) I6 S( c; I$ E, W8 K
109.                                 if (destSocket != null && destSocket.State == WebSocketState.Open)
     ; U) c) \9 p4 l' f/ ~& E
110.                                     await destSocket.SendAsync(buffer, WebSocketMessageType.Text, true, CancellationToken.None);
     
111.                             }
     
112.                             else
     
113.                             {
     
114.                                 Task.Run(() =>
     / v: M2 x5 M! ?
115.                                 {
     : J% i' C; H* Z2 }8 ~0 G
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.                                 });
     ; m% c' J$ o' Z) c
120.                             }
     
121.                         }
     
122.                         catch (Exception exs)
     $ q+ w' `* [- M- s
123.                         {
     
124.                             //消息转发异常处理，本次消息忽略 继续监听接下来的消息
     4 ^0 y; I8 V! y+ d( M3 y8 N
125.                         }
     # S2 E  p( L( |1 Q
126.                         #endregion
     " k7 G2 @( Z* b6 C7 H7 v# |
127.                     }
     
128.                     else
     0 W* V' I/ j4 D6 l' [
129.                     {
     
130.                         break;
     
131.                     }
     
132.                 }//while end
     
133.             }
     ! t8 f* w1 N4 N# f8 l$ B; M
134.             catch (Exception ex)
     
135.             {
     
136.                 //整体异常处理
     
137.                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);
     
138.             }
     ' T/ H' K  g3 W! V& ~+ O: y
139.         }
     
140. 
     
141. 
     
142.         public bool IsReusable
     
143.         {
     
144.             get
     
145.             {
     & v. ?/ B' z: n: L- t% _  O7 t
146.                 return false;
     7 u% F0 e* k3 g. n
147.             }
     
148.         }
     
149.     }
     / m# A, ~* G8 J+ j1 u
150. }

复制代码

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

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

: Y, W6 @) d+ L2 }- ?