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

admin

服务器端代码编写

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

/ [8 O# L* B- V( z0 z" i

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

3.Handler1.ashx代码如下：

1. using System;
   
2. using System.Collections.Generic;
   0 ~! U6 e: x/ t3 \7 h% R9 P
3. using System.Linq;
   - X& v' K' E( ~5 q- z3 S
4. using System.Net.WebSockets;
   
5. using System.Text;
   8 a) K+ B% t: ~! p9 {3 O: z& B2 {
6. using System.Threading;
   9 r2 W4 K7 h  g! a: ^" v$ M
7. using System.Threading.Tasks;
   
8. using System.Web;
   0 Y+ V  F- ^, F
9. using System.Web.WebSockets;
   
10. 
    ; n# L) ^, ]# h7 P- F5 V8 |
11. 
    : H! P/ n7 Y9 [$ z
12. namespace WebApplicationWebsocketHandler
    
13. {
    / Y: B% `1 X% X2 Q1 A: H1 d/ ~
14.     /// <summary>
    
15.     /// 离线消息
      c& w  b6 ^" ]0 T
16.     /// </summary>
    : }6 l4 ?- q; o: w' O1 `% R
17.     public class MessageInfo
    
18.     {
    
19.         public MessageInfo(DateTime _MsgTime, ArraySegment<byte> _MsgContent)
    
20.         {
    * Y* ?8 c9 C( @9 d' ^
21.             MsgTime = _MsgTime;
    ! W. m. X; @) O+ s
22.             MsgContent = _MsgContent;
    9 @( F) j7 D; @; v& E: Y
23.         }
    3 `) x/ d4 I  n6 R' z
24.         public DateTime MsgTime { get; set; }
    
25.         public ArraySegment<byte> MsgContent { get; set; }
    
26.     }
    & k0 [* J1 i8 N' c8 ?2 p
27. 
    
28. 
    : z( ^3 o2 c7 X+ P$ y' f  t9 {
29. 
    ) I5 M! B. d0 h: ^( E
30. 
      G2 r, k# J2 i/ K% q; T
31.     /// <summary>
    , f1 ]9 d. k/ I$ g! m- ~& f
32.     /// Handler1 的摘要说明
    
33.     /// </summary>
    
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)
    - z0 M2 B# n3 A; T: J- t  Y/ A* a2 l
39.         {
    5 f/ H, _  Z& }. [
40.             //context.Response.ContentType = "text/plain";
      r4 ~. o& ~* t5 E- V0 e
41.             //context.Response.Write("Hello World");
    
42.             if (context.IsWebSocketRequest)
    
43.             {
    
44.                 context.AcceptWebSocketRequest(ProcessChat);
    ) K: w1 ^, {; s, j) j+ p8 W: i
45.             }
    
46.         }
    5 c- U0 a$ v- t  h5 c7 Q! Y- W8 ?
47. 
    
48.         private async Task ProcessChat(AspNetWebSocketContext context)
    
49.         {
    - x: e0 o7 J% ]; j+ j2 m) ]/ J
50.             WebSocket socket = context.WebSocket;
    
51.             string user = context.QueryString["user"].ToString();
    : L5 Y- e  A* [0 Y
52. 
    . i/ t! b( Y  l7 z
53.             try
    ' k& U( ]% R8 e1 u% `  [: _+ l4 l2 M
54.             {
    
55.                 #region 用户添加连接池
    
56.                 //第一次open时，添加到连接池中
    
57.                 if (!CONNECT_POOL.ContainsKey(user))
    * r6 r$ L, j, ~. Z- z- n
58.                     CONNECT_POOL.Add(user, socket);//不存在，添加
    + A1 |6 g- u3 p4 f
59.                 else
    ) k3 Z) V+ n; t
60.                     if (socket != CONNECT_POOL[user])//当前对象不一致，更新
    ( r3 U9 x6 \, f1 b3 n6 H5 ~5 M
61.                         CONNECT_POOL[user] = socket;
    8 A; }1 z! D& S4 l* T8 I* e! L
62.                 #endregion
    6 S, l: L" R) t& Q- h
63. 
    
64.                 #region 离线消息处理
    $ `) [- p' _4 Z6 C3 W
65.                 if (MESSAGE_POOL.ContainsKey(user))
    
66.                 {
    5 s  K8 j9 w' W2 F  S" l
67.                     List<MessageInfo> msgs = MESSAGE_POOL[user];
    0 U0 o+ f8 W, s2 R
68.                     foreach (MessageInfo item in msgs)
    / H7 x8 O3 K  u, z
69.                     {
    
70.                         await socket.SendAsync(item.MsgContent, WebSocketMessageType.Text, true, CancellationToken.None);
    
71.                     }
    
72.                     MESSAGE_POOL.Remove(user);//移除离线消息
    
73.                 }
    " ^+ O( S) R7 u
74.                 #endregion
    
75. 
    - u, r4 S3 D. e6 X, @( K! x* t9 A
76.                 string descUser = string.Empty;//目的用户
    ( M, F+ R7 w) E
77.                 while (true)
    ! `0 E: Q9 k9 J$ Y! a+ d0 t( C, I  S
78.                 {
    # Y7 _( ^7 g1 c/ a3 O' X6 ?
79.                     if (socket.State == WebSocketState.Open)
    & `$ e; s5 Y0 K1 Q4 |
80.                     {
    4 b; ]% P" [$ s4 K& x# B- k+ q# P
81.                         ArraySegment<byte> buffer = new ArraySegment<byte>(new byte[2048]);
    
82.                         WebSocketReceiveResult result = await socket.ReceiveAsync(buffer, CancellationToken.None);
    
83. 
    
84.                         #region 消息处理（字符截取、消息转发）
    
85.                         try
      b# [* l, n0 E9 o3 E
86.                         {
    
87.                             #region 关闭Socket处理，删除连接池
    6 G# B& Y2 p+ |3 N" h  D3 s
88.                             if (socket.State != WebSocketState.Open)//连接关闭
    + U4 T+ c9 R2 O% S1 E
89.                             {
    
90.                                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);//删除连接池
      o5 W/ z: |) U' L9 J, I5 G7 ^" h
91.                                 break;
    
92.                             }
    6 g2 Z& M$ M9 C/ a7 n- o, }. B
93.                             #endregion
    " x1 t" \% L& |6 L1 q: f1 m
94. 
    
95.                             string userMsg = Encoding.UTF8.GetString(buffer.Array, 0, result.Count);//发送过来的消息
    
96.                             string[] msgList = userMsg.Split('|');
    4 I' v1 w% ?: N/ Y+ L
97.                             if (msgList.Length == 2)
    
98.                             {
    % b1 k7 V0 M, }4 F9 i+ p
99.                                 if (msgList[0].Trim().Length > 0)
    * p7 V- R8 x$ A# u
100.                                     descUser = msgList[0].Trim();//记录消息目的用户
     
101.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(msgList[1]));
     6 H9 X3 x- s. c4 k
102.                             }
     " o3 L  \  h0 U. u% @! J
103.                             else
     
104.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(userMsg));
     
105. 
       V9 b3 P; R' C% j! A; r6 s; q* N
106.                             if (CONNECT_POOL.ContainsKey(descUser))//判断客户端是否在线
     
107.                             {
     6 I7 e" c( Y7 _$ e, f
108.                                 WebSocket destSocket = CONNECT_POOL[descUser];//目的客户端
     4 s' L) c' ~5 E' U% {/ }% @* p  h
109.                                 if (destSocket != null && destSocket.State == WebSocketState.Open)
     
110.                                     await destSocket.SendAsync(buffer, WebSocketMessageType.Text, true, CancellationToken.None);
     
111.                             }
     
112.                             else
     , @3 R1 G- z2 c& }' v; j, j4 Y
113.                             {
     5 V: |& [8 f/ ~: [+ V
114.                                 Task.Run(() =>
     1 _$ u# G! N, l& P6 D1 L
115.                                 {
     
116.                                     if (!MESSAGE_POOL.ContainsKey(descUser))//将用户添加至离线消息池中
     
117.                                         MESSAGE_POOL.Add(descUser, new List<MessageInfo>());
     3 Z' ]. B+ C2 w- V+ B
118.                                     MESSAGE_POOL[descUser].Add(new MessageInfo(DateTime.Now, buffer));//添加离线消息
     
119.                                 });
     0 D$ X2 G( t! ?0 V1 k( k& d% }
120.                             }
     0 Z, ~& H6 v2 g3 `" `- n+ d# m9 l
121.                         }
     
122.                         catch (Exception exs)
     ' r. n1 O9 m3 d. }6 }
123.                         {
     $ M, |" \- ~1 ?9 N' w% R
124.                             //消息转发异常处理，本次消息忽略 继续监听接下来的消息
     : V8 r& f2 d* K7 G  R1 b
125.                         }
     - ?/ F! t0 T# {
126.                         #endregion
     
127.                     }
     + o: P4 C) Q! n$ _* R3 t- c( t
128.                     else
     ! D3 P) N- j( n
129.                     {
     ! u% e" M6 {( {4 c) b
130.                         break;
     
131.                     }
     1 G! L- j: u. Z/ Z9 @
132.                 }//while end
     $ c: i7 ~+ w" e% D1 H) A
133.             }
     
134.             catch (Exception ex)
     7 @$ F! X0 d- [
135.             {
     . E# n1 B: h$ [: i2 X
136.                 //整体异常处理
     
137.                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);
     
138.             }
     : d' i6 ~& O& n4 d5 Y
139.         }
     8 s# s. @' \* X. T+ ^- E
140. 
     
141. 
     3 t4 c1 |, `* h% g
142.         public bool IsReusable
     4 `* G8 L2 p) _1 R2 ]
143.         {
     9 Y/ }- Q- W8 S% C  N1 s5 }
144.             get
     - K+ q% D1 j: l; M
145.             {
     
146.                 return false;
     
147.             }
     ; Z/ O* v5 `% f- C' O
148.         }
     " ]0 J' w" e5 d
149.     }
     ! N9 [- ^: I# ^% q; x, {' \' G; }7 c
150. }

复制代码

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

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