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

admin

服务器端代码编写

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

  D- c, {2 \- M, {/ U$ T% y

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

3.Handler1.ashx代码如下：

1. using System;
   
2. using System.Collections.Generic;
   
3. using System.Linq;
   ! e+ b  H+ h5 j; f  e; G; W, q
4. using System.Net.WebSockets;
   
5. using System.Text;
   ! q* C3 ?& t: @2 ?! F/ T
6. using System.Threading;
   
7. using System.Threading.Tasks;
   8 j& L4 V& U) T4 h6 Y
8. using System.Web;
   . H5 n4 L4 M: _/ P: B
9. using System.Web.WebSockets;
   8 B8 J8 K( D2 o
10. 
    . Y. c0 t; l" C1 Q, v
11. 
    # I  O- {  h+ E, s" }4 O4 e
12. namespace WebApplicationWebsocketHandler
    % u; `: N/ t, O! R% F1 V$ r% ?
13. {
    * P7 \7 M4 K4 n5 I
14.     /// <summary>
    8 N, _% R% ?$ R- f' a* D, ?/ J
15.     /// 离线消息
    6 T, P/ O$ `. G* a' Y9 J
16.     /// </summary>
    
17.     public class MessageInfo
    6 V2 d: F# |; B9 C
18.     {
    
19.         public MessageInfo(DateTime _MsgTime, ArraySegment<byte> _MsgContent)
    
20.         {
    / q! k/ j7 @& }" Y7 I
21.             MsgTime = _MsgTime;
    $ L- }7 d- \+ ^/ B
22.             MsgContent = _MsgContent;
    
23.         }
    
24.         public DateTime MsgTime { get; set; }
    
25.         public ArraySegment<byte> MsgContent { get; set; }
    
26.     }
    1 H- t, X1 G5 Q4 G* S
27. 
    
28. 
    0 S% ^3 N: ]9 b6 W/ T4 \: G; ?
29. 
    
30. 
    . ^: ~4 E6 c+ Y( u6 ?' F; O. {
31.     /// <summary>
    
32.     /// Handler1 的摘要说明
    6 T; c8 r' k) I9 {4 m
33.     /// </summary>
    ! n" T( c+ w- B, F( Q$ s
34.     public class Handler1 : IHttpHandler
      `# K" i2 o2 C( f( k
35.     {
    & _9 L: R0 `( R0 b' V( D! s# f
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)
    
39.         {
    
40.             //context.Response.ContentType = "text/plain";
    4 N( J. W9 C7 X# v3 x0 I
41.             //context.Response.Write("Hello World");
    
42.             if (context.IsWebSocketRequest)
    $ b. z  ?- V" Q
43.             {
    " m- ]1 U9 b7 Q' O$ U5 W4 T
44.                 context.AcceptWebSocketRequest(ProcessChat);
    6 G9 |& Q4 n  y5 I. }+ A- }
45.             }
    ( K9 o6 Q% {' J: ~' W/ N
46.         }
    , Y# r3 H3 p; q  R) `
47. 
    6 e" ~9 S+ c8 _4 s! X( s% O
48.         private async Task ProcessChat(AspNetWebSocketContext context)
    . k( a+ z+ J* H9 \# R* h) {) z9 b
49.         {
    
50.             WebSocket socket = context.WebSocket;
    , B1 |3 N9 E* C  _1 g3 M
51.             string user = context.QueryString["user"].ToString();
    
52. 
    
53.             try
    ! W& c! @  @# Y
54.             {
    ! n$ |1 x& Y# s
55.                 #region 用户添加连接池
    ! L3 J" o6 H9 }* T4 Q. u
56.                 //第一次open时，添加到连接池中
    , x2 k* N/ L9 H3 X
57.                 if (!CONNECT_POOL.ContainsKey(user))
    4 S( @; ^, V/ \. S
58.                     CONNECT_POOL.Add(user, socket);//不存在，添加
    
59.                 else
    
60.                     if (socket != CONNECT_POOL[user])//当前对象不一致，更新
    + y, X' H8 Z6 W% \$ }3 ]" E
61.                         CONNECT_POOL[user] = socket;
    
62.                 #endregion
    
63. 
    
64.                 #region 离线消息处理
    , I/ w. h" n* z3 z
65.                 if (MESSAGE_POOL.ContainsKey(user))
    
66.                 {
    " y- V7 f+ L; C0 f/ _0 P& o6 @
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);
    5 b6 o/ h! i& l1 l( ^1 c: j5 \! L& m
71.                     }
    / X2 A8 `0 b( }$ s- Q) ^
72.                     MESSAGE_POOL.Remove(user);//移除离线消息
    " E+ |; K5 r1 O! O0 c
73.                 }
    3 D; b! @- ^% U. F8 ]8 [
74.                 #endregion
    
75. 
    
76.                 string descUser = string.Empty;//目的用户
    6 |/ K6 R% i  o: Q; x
77.                 while (true)
    , H; q& m/ |- V# M3 `
78.                 {
    + |% ^, U; s* y, O0 _" f  U8 o4 B
79.                     if (socket.State == WebSocketState.Open)
    
80.                     {
    0 ^  o: O' a" D' Z1 ^* Q& T
81.                         ArraySegment<byte> buffer = new ArraySegment<byte>(new byte[2048]);
    , X& _2 d/ j2 I1 ^9 K
82.                         WebSocketReceiveResult result = await socket.ReceiveAsync(buffer, CancellationToken.None);
    $ d' u; G4 D# H% K4 x
83. 
    9 e- R; s* l# i; M, T
84.                         #region 消息处理（字符截取、消息转发）
    
85.                         try
    ; U# O# D, h  e5 [3 B1 q& C3 g
86.                         {
    , c2 f! l+ \; ~
87.                             #region 关闭Socket处理，删除连接池
    
88.                             if (socket.State != WebSocketState.Open)//连接关闭
    - ^$ r# ]; j2 i  \/ h
89.                             {
    8 X4 y3 i; z$ A/ a; u
90.                                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);//删除连接池
    , A& c. C2 R! d  o0 @
91.                                 break;
    7 W: e9 X# l: m6 u0 r5 {* o& x
92.                             }
    7 ^6 f: h7 A  y) l! M
93.                             #endregion
    - D2 X) ^% E+ Q2 I( d- N* y0 W
94. 
    
95.                             string userMsg = Encoding.UTF8.GetString(buffer.Array, 0, result.Count);//发送过来的消息
    
96.                             string[] msgList = userMsg.Split('|');
    
97.                             if (msgList.Length == 2)
    
98.                             {
    7 I% H( O' D$ {# o' F$ \
99.                                 if (msgList[0].Trim().Length > 0)
    
100.                                     descUser = msgList[0].Trim();//记录消息目的用户
     4 r. S- z# S. |, u+ P# G4 H
101.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(msgList[1]));
     
102.                             }
     4 n* S9 h+ Z- V+ I$ a+ ~- c( x
103.                             else
     
104.                                 buffer = new ArraySegment<byte>(Encoding.UTF8.GetBytes(userMsg));
     
105. 
     ! O" U# F3 W* }6 q- h* `% t
106.                             if (CONNECT_POOL.ContainsKey(descUser))//判断客户端是否在线
     
107.                             {
     
108.                                 WebSocket destSocket = CONNECT_POOL[descUser];//目的客户端
     / F3 z) w3 x4 e! ]" L; [8 h* @
109.                                 if (destSocket != null && destSocket.State == WebSocketState.Open)
     , `( D  K) g. \+ |
110.                                     await destSocket.SendAsync(buffer, WebSocketMessageType.Text, true, CancellationToken.None);
     + a% G! D/ v3 ^/ t
111.                             }
     
112.                             else
     
113.                             {
     
114.                                 Task.Run(() =>
     
115.                                 {
     : A+ f( j/ {* \! g% D" ]4 v
116.                                     if (!MESSAGE_POOL.ContainsKey(descUser))//将用户添加至离线消息池中
     ) H: }7 j# N4 P2 O- h1 n" Q
117.                                         MESSAGE_POOL.Add(descUser, new List<MessageInfo>());
     
118.                                     MESSAGE_POOL[descUser].Add(new MessageInfo(DateTime.Now, buffer));//添加离线消息
     
119.                                 });
     ) \1 X: C7 c2 v( `
120.                             }
     
121.                         }
     
122.                         catch (Exception exs)
     
123.                         {
     
124.                             //消息转发异常处理，本次消息忽略 继续监听接下来的消息
     . ^2 }2 o, T6 d  L) Y
125.                         }
     
126.                         #endregion
     ( }8 H& d$ ^/ c
127.                     }
     
128.                     else
     
129.                     {
     
130.                         break;
     
131.                     }
     # G: q& |8 L$ ^6 ?
132.                 }//while end
     / o' G) S! [8 h4 k; \- b9 }8 z/ e
133.             }
     7 B  Y& J; c3 r  E+ W
134.             catch (Exception ex)
     . W, c" @* _# t- A  \
135.             {
     
136.                 //整体异常处理
     ( a3 v* {. M2 n, j0 q5 S6 u
137.                 if (CONNECT_POOL.ContainsKey(user)) CONNECT_POOL.Remove(user);
       N5 v) w0 t0 t- j1 f4 s
138.             }
     
139.         }
     
140. 
     ; y/ i1 ]! L1 x* w1 N- T
141. 
     
142.         public bool IsReusable
     8 f& ]. Y1 \$ y; Q) D6 f0 X# z
143.         {
     
144.             get
     3 ]. W$ Y, A2 ~% {3 J+ A
145.             {
     8 C& B( o, P' [4 s. G1 o
146.                 return false;
     9 M$ G' W. \" P( u, j+ G; l
147.             }
     
148.         }
     ( J& Y' z, C: `8 i; q
149.     }
     
150. }

复制代码

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

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

6 c1 _! i' e0 W6 x5 Z* }