php实现websocket实时消息推送

admin

php实现websocket实时消息推送
( I" [: p& C3 u! m. M* ^
6 W9 l+ W" Q7 [; ~
$ _9 P' B9 i6 D+ Q6 H/ O" u/ U, n8 FSocketService.php
% U8 L" T4 I' W: m2 ~

1. <?php
   
2. /**
   
3. * Created by xwx
   
4. * Date: 2017/10/18
   
5. * Time: 14:33
   
6. */
   8 i- K. F. {4 a- O
7. 
   
8. class SocketService
   
9. {
   
10.     private $address  = '0.0.0.0';
    
11.     private $port = 8083;
    
12.     private $_sockets;
    
13.     public function __construct($address = '', $port='')
    # ?; j5 q) m3 _& f3 d
14.     {
    
15.             if(!empty($address)){
    
16.                 $this->address = $address;
    
17.             }
    # Y0 k+ J6 a& A) E1 L; |
18.             if(!empty($port)) {
    1 h1 C  E1 F1 z2 A6 v) h
19.                 $this->port = $port;
    9 H( W, H5 F$ m# J
20.             }
    
21.     }
    , T2 k: F" ~( `6 x, A7 p
22. 
    0 a/ p' E) V3 s4 V. g1 R
23.     public function service(){
      @/ [2 M3 L$ H0 E
24.         //获取tcp协议号码。
    ; a& m! V4 H0 o3 _1 S, l7 @* W
25.         $tcp = getprotobyname("tcp");
    7 F' U& F  N3 W7 ]+ ]4 U
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    5 [4 Z8 {& m$ `: X
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    
28.         if($sock < 0)
    1 n, Q& w- ?4 T) [, d5 W' V& q1 l
29.         {
    ) b- v7 |+ ?7 |8 a& e( Z
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    2 r* N/ |# w+ r5 W9 q, m
31.         }
    
32.         socket_bind($sock, $this->address, $this->port);
      n4 F8 P; [, F# X7 K- I
33.         socket_listen($sock, $this->port);
    : j' R+ u5 J7 _$ ]1 {3 }; @
34.         echo "listen on $this->address $this->port ... \n";
    
35.         $this->_sockets = $sock;
    4 F0 |4 F# U" P
36.     }
    3 e* }1 |2 \# M# O9 s
37. 
    % `1 \  C! h1 l3 o
38.     public function run(){
    
39.         $this->service();
    4 L( Q# x1 Q5 D- K, ?
40.         $clients[] = $this->_sockets;
    3 j% r! l" b8 c2 F9 p0 d* _/ X3 R8 p
41.         while (true){
    ) c5 _0 b3 m, R9 ]
42.             $changes = $clients;
    ! w) K6 _; A5 ~! `
43.             $write = NULL;
    
44.             $except = NULL;
    
45.             socket_select($changes,  $write,  $except, NULL);
    
46.             foreach ($changes as $key => $_sock){
    * T. P8 g0 J) J
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    
48.                     if(($newClient = socket_accept($_sock))  === false){
    
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    5 _% N: v; F, E% E$ Q  I7 B( l
50.                     }
    , }# g+ I( G5 M7 b
51.                     $line = trim(socket_read($newClient, 1024));
    0 P/ t1 O/ Y3 m9 a9 q
52.                     $this->handshaking($newClient, $line);
    
53.                     //获取client ip
    
54.                     socket_getpeername ($newClient, $ip);
    
55.                     $clients[$ip] = $newClient;
    1 e/ b6 L8 \' s, M% T
56.                     echo  "Client ip:{$ip}   \n";
    8 \$ z+ R" h7 o! ?8 f( a2 D
57.                     echo "Client msg:{$line} \n";
    
58.                 } else {
    4 k. j4 f) V& {7 t
59.                     socket_recv($_sock, $buffer,  2048, 0);
    2 f! [& c2 H: r8 F" S2 T
60.                     $msg = $this->message($buffer);
    / K6 m" X* ~* D
61.                     //在这里业务代码
    
62.                     echo "{$key} clinet msg:",$msg,"\n";
    
63.                     fwrite(STDOUT, 'Please input a argument:');
    ( I3 v; [- f/ k: C4 k
64.                     $response = trim(fgets(STDIN));
    
65.                     $this->send($_sock, $response);
    $ a# W# n  F! D- I' [
66.                     echo "{$key} response to Client:".$response,"\n";
    
67.                 }
    5 v- w) I! L0 f0 o  f
68.             }
    
69.         }
    
70.     }
    ( c5 l# @( p3 x  e) N
71. 
    
72.     /**
    ; \4 j' S9 ?9 T* L9 U
73.      * 握手处理
    6 P1 q% i& K6 z$ u( r: i* I0 ?
74.      * @param $newClient socket
    
75.      * @return int  接收到的信息
    1 e$ M2 H7 t( @) @' ?  V3 ^4 l
76.      */
    
77.     public function handshaking($newClient, $line){
    
78. 
    ( e3 Y0 r# d+ f, i3 d
79.         $headers = array();
    
80.         $lines = preg_split("/\r\n/", $line);
    
81.         foreach($lines as $line)
    ; q0 i' S# p0 d" ^
82.         {
    
83.             $line = chop($line);
    * S. r2 V" t4 c' |8 l5 e( v
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    0 `2 i( d7 l+ h$ F; ~
85.             {
    
86.                 $headers[$matches[1]] = $matches[2];
    
87.             }
    
88.         }
    4 h3 i8 j' E# M3 ]
89.         $secKey = $headers['Sec-WebSocket-Key'];
    $ ^% u) o# L" m- ?) X
90.         $secAccept = base64_encode(pack('H*', sha1($secKey . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11')));
    
91.         $upgrade  = "HTTP/1.1 101 Web Socket Protocol Handshake\r\n" .
    
92.             "Upgrade: websocket\r\n" .
    * ^. k5 M, c  }, d# U
93.             "Connection: Upgrade\r\n" .
    
94.             "WebSocket-Origin: $this->address\r\n" .
    + U  ~8 z+ l& ~* K1 U8 j' X9 V
95.             "WebSocket-Location: ws://$this->address:$this->port/websocket/websocket\r\n".
    , K" G& w# v, C) T
96.             "Sec-WebSocket-Accept:$secAccept\r\n\r\n";
    
97.         return socket_write($newClient, $upgrade, strlen($upgrade));
    * J9 M0 r4 V5 k+ r
98.     }
    
99. 
    
100.     /**
     & [4 _5 n/ D9 ?! l# a
101.      * 解析接收数据
     
102.      * @param $buffer
     ( N4 ^* l. n9 r# M
103.      * @return null|string
     
104.      */
     
105.     public function message($buffer){
     
106.         $len = $masks = $data = $decoded = null;
     
107.         $len = ord($buffer[1]) & 127;
     
108.         if ($len === 126)  {
     
109.             $masks = substr($buffer, 4, 4);
     7 c6 z, c$ m9 L& G+ B, J! X6 t
110.             $data = substr($buffer, 8);
     : i( s' s) v8 ^- K; l
111.         } else if ($len === 127)  {
     # E0 B5 `2 k5 X1 ?6 n
112.             $masks = substr($buffer, 10, 4);
     
113.             $data = substr($buffer, 14);
     + s; K3 T4 j, h$ ~, A2 O* z& _: w9 V
114.         } else  {
     3 T- \/ L! K4 e( P* Z- @2 s
115.             $masks = substr($buffer, 2, 4);
     + t; j6 K) g4 ?% \% g
116.             $data = substr($buffer, 6);
     
117.         }
     7 G; }! Y" j& L% o. K' F
118.         for ($index = 0; $index < strlen($data); $index++) {
     
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     
120.         }
     0 @, H0 e* D; [1 j1 k
121.         return $decoded;
     ! o5 R+ n" r+ [3 M# M1 f! G2 Z
122.     }
     7 M  O# o( w0 l  w, Y
123. 
     
124.     /**
     * W9 O" ~/ G# v* a3 W
125.      * 发送数据
     : E* @8 j1 O6 ^+ Z. i7 A' h
126.      * @param $newClinet 新接入的socket
     4 X& Q% S, Q) M+ i. o
127.      * @param $msg   要发送的数据
     
128.      * @return int|string
     
129.      */
     
130.     public function send($newClinet, $msg){
     
131.         $msg = $this->frame($msg);
     
132.         socket_write($newClinet, $msg, strlen($msg));
     
133.     }
     
134. 
     0 K$ Q/ ?6 k2 V. e+ Y
135.     public function frame($s) {
     
136.         $a = str_split($s, 125);
     3 T6 G  ]/ d6 k! I9 [# {
137.         if (count($a) == 1) {
     - J4 }8 u& c7 z7 R( ]( w
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     $ [- ~9 z6 g- B( }; f
139.         }
     " e8 B5 d/ F  X& H
140.         $ns = "";
     ' P6 S7 x" K6 e  a& b7 f0 N
141.         foreach ($a as $o) {
     % l0 A1 A4 U% q4 k% i0 S
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     
143.         }
     
144.         return $ns;
     % q7 w1 [9 e6 R  b- Y. j
145.     }
     3 G" @( m. m& h3 N" ?1 d5 _
146. 
     , z& Z  p( I3 N  A+ w$ a+ g! H
147.     /**
     
148.      * 关闭socket
     
149.      */
     / P2 e# ?8 b* h- [
150.     public function close(){
     
151.         return socket_close($this->_sockets);
     
152.     }
     1 K- Z5 z* L, Z; I$ O
153. }
     
154. 
     / _5 n9 q  q( e* Y
155. $sock = new SocketService();
     8 p: E& n- x3 b$ [, J+ U0 r& v2 Y
156. $sock->run();
     3 h% k7 W. d: x1 O1 z
157. 
     

复制代码

web.html
. z2 T7 v5 E0 G+ F* O7 X

1. <!doctype html>
   5 ]. C3 R# k  M3 ?/ w# A( @/ f
2. <html lang="en">
   
3. <head>
   
4.   <meta charset="UTF-8">
   
5.   <meta name="viewport" content="width=device-width,initial-scale=1, maximum-scale=1, user-scalable=no">
   
6.   <title>websocket</title>
   
7. </head>
   
8. <body>
   ' m3 e0 _+ J3 O& ^
9. <input id="text" value="">
   0 w7 U# p; o5 o6 L
10. <input type="submit" value="send" onclick="start()">
    
11. <input type="submit" value="close" onclick="close()">
    8 G; [/ C4 J+ y  P! a2 I$ ^4 l1 H
12. <div id="msg"></div>
    
13. <script>
    
14. /**
    # Z  C! L- a5 l2 h- O
15. 0：未连接
    
16. 1：连接成功，可通讯
    
17. 2：正在关闭
    ! P; Q! X1 r# y2 l. T
18. 3：连接已关闭或无法打开
    
19. */
    
20. 
    $ L: O4 o6 G7 H# F/ i) b) a
21.     //创建一个webSocket 实例
    
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    + M( _# L# H% e) q% d' k+ N
23. 
    6 I: a, V1 m- u9 g- H* Q
24. 
    
25.     webSocket.onerror = function (event){
    
26.         onError(event);
    
27.     };
    
28. 
    
29.     // 打开websocket
    3 l! A# ]6 u/ R: `7 P0 c
30.     webSocket.onopen = function (event){
    : p! A, q' l! z* j
31.         onOpen(event);
    
32.     };
    $ p0 Q% v9 p- s$ O$ f% k
33. 
    # H9 e! y8 I! a3 o
34.     //监听消息
    
35.     webSocket.onmessage = function (event){
    
36.         onMessage(event);
    - D3 T- ~& d7 G; r. g( j" V
37.     };
    
38. 
    6 u; g6 G/ P3 I5 R( B3 l! d3 R
39. 
    
40.     webSocket.onclose = function (event){
    : N+ ~$ j  v0 [) {: c3 J
41.         onClose(event);
    5 z8 _; y( O* n7 f
42.     }
    / Y0 Q1 c) q8 r9 |" G
43. 
    
44.     //关闭监听websocket
    
45.     function onError(event){
    
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    ' C1 y8 R% E: t: w4 g- ]  r
47.         console.log("error"+event.data);
    
48.     };
    6 `; y+ B& X& o: A0 S
49. 
    7 I" b+ r3 S( J/ H; L/ @$ ~
50.     function onOpen(event){
    ! N5 k" q* C0 G. ~+ }
51.         console.log("open:"+sockState());
    
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    
53.     };
    + y# g0 {7 d, e# S9 u
54.     function onMessage(event){
    
55.         console.log("onMessage");
    ! M: T8 S% q/ x$ M& t) M
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    
57.     };
    
58. 
    4 m3 _, t; r4 C6 `" ]
59.     function onClose(event){
    ; Z# e/ P' V1 ?, ?9 ]( }
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    7 i2 X8 L5 _+ K+ }
61.         console.log("close:"+sockState());
    0 ~  o) g9 j2 C5 p
62.         webSocket.close();
    
63.     }
    
64. 
    ' l* |- V- U. ]% l* ~" w( G
65.     function sockState(){
    * R3 w4 _  O) j7 g6 t  L
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    / `" T, u' o4 `$ T
67.             return status[webSocket.readyState];
    * L& ]# N* I4 ?# p7 {% g( b
68.     }
    
69. 
    
70. 
    
71. 
    
72. function start(event){
    
73.         console.log(webSocket);
    ; B( V5 V: C/ O+ g/ s
74.         var msg = document.getElementById('text').value;
    
75.         document.getElementById('text').value = '';
    
76.         console.log("send:"+sockState());
    $ P2 W* A4 Q- p. B: ?. p
77.         console.log("msg="+msg);
    ) [8 |  F5 c5 g' j) Q
78.         webSocket.send("msg="+msg);
    / {8 q, ?5 j* ~5 N) m3 e) q) i
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    # d2 p) b' y* Z0 ]2 c
80.     };
    2 |; x& j& j$ [* [. j4 W
81. 
    ) F- m  P* |# V
82.     function close(event){
    3 R* c) I' l- q/ e% c( c
83.         webSocket.close();
    " ~& J6 n5 @  t! Y
84.     }
    3 {  q0 ?$ {5 |4 w+ o
85. </script>
    
86. </body>
    8 H: O6 W; y& x# K( |
87. </html>

复制代码

, t: w1 d* Q+ S& {