php实现websocket实时消息推送

admin

php实现websocket实时消息推送
' L3 B1 D5 N4 `! u; e" l( ?

SocketService.php
* f& N$ U  K& A+ L" j

1. <?php
   
2. /**
   5 q4 K9 ]* `# N9 Z+ C; D
3. * Created by xwx
   , v* c* R  |9 `0 X( ]' {
4. * Date: 2017/10/18
   
5. * Time: 14:33
   + q9 K  p0 C" r: L
6. */
   
7. 
   6 B0 I6 m* c2 y1 ^4 m
8. class SocketService
   6 N3 ?8 L" ]  n/ u
9. {
   
10.     private $address  = '0.0.0.0';
    
11.     private $port = 8083;
    
12.     private $_sockets;
    
13.     public function __construct($address = '', $port='')
    
14.     {
    , z- k9 p" a# a, d% S
15.             if(!empty($address)){
    3 }. {! }: b' f8 n$ O
16.                 $this->address = $address;
    
17.             }
    ( l! j, E" q- u
18.             if(!empty($port)) {
    - P# l6 \# \2 D' A2 Y& `" L3 C9 F
19.                 $this->port = $port;
    0 u, X# y4 b; s( }6 f
20.             }
    
21.     }
    
22. 
    " M, q* H' {7 [
23.     public function service(){
    
24.         //获取tcp协议号码。
    3 x. I7 L% j: D$ N7 @
25.         $tcp = getprotobyname("tcp");
      T+ k3 r5 O( W3 ?
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    
28.         if($sock < 0)
    6 p7 y7 e2 ~$ `! T' X  p4 L. q
29.         {
    
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    
31.         }
    
32.         socket_bind($sock, $this->address, $this->port);
    8 l" x; f, R1 I* D2 h  g9 F
33.         socket_listen($sock, $this->port);
    
34.         echo "listen on $this->address $this->port ... \n";
    6 f, A# Y8 c: {: @3 A$ z' [
35.         $this->_sockets = $sock;
    
36.     }
    
37. 
    " g# f% K3 b7 a' R6 I! g( `. l3 r
38.     public function run(){
    
39.         $this->service();
    
40.         $clients[] = $this->_sockets;
    # h1 H& ^( q" i5 Y& }* G
41.         while (true){
    - I& g( J! }( z0 A4 n9 |* Z
42.             $changes = $clients;
      i0 v( Y# }6 v
43.             $write = NULL;
    
44.             $except = NULL;
    
45.             socket_select($changes,  $write,  $except, NULL);
    
46.             foreach ($changes as $key => $_sock){
    
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    9 V0 |- n$ [5 l1 v  D3 Y$ t
48.                     if(($newClient = socket_accept($_sock))  === false){
    % J# O% C1 ~( T# W: U
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    
50.                     }
    . s2 t9 {) J5 x) \. `, o! d
51.                     $line = trim(socket_read($newClient, 1024));
    
52.                     $this->handshaking($newClient, $line);
    
53.                     //获取client ip
    4 y4 r- S5 E: O" \
54.                     socket_getpeername ($newClient, $ip);
    
55.                     $clients[$ip] = $newClient;
    
56.                     echo  "Client ip:{$ip}   \n";
    
57.                     echo "Client msg:{$line} \n";
    ( q* Z8 b6 C9 a' f( x
58.                 } else {
    
59.                     socket_recv($_sock, $buffer,  2048, 0);
    9 P. f% h$ }9 h3 j+ ^
60.                     $msg = $this->message($buffer);
    , a' q" x: i2 ~9 K5 n* E
61.                     //在这里业务代码
    1 N6 k0 V1 \/ @' v; M
62.                     echo "{$key} clinet msg:",$msg,"\n";
    
63.                     fwrite(STDOUT, 'Please input a argument:');
    7 }! a5 y2 E: L* v" }* Q
64.                     $response = trim(fgets(STDIN));
    
65.                     $this->send($_sock, $response);
    
66.                     echo "{$key} response to Client:".$response,"\n";
    
67.                 }
    
68.             }
    0 D1 W- q; }0 [0 [
69.         }
    $ O3 n% b% h5 c; H; _& H: X! O
70.     }
    
71. 
    
72.     /**
    ( j0 H6 e' G! j7 @: `4 s5 K
73.      * 握手处理
    
74.      * @param $newClient socket
    " u/ L1 Z; Z. u+ b8 @+ j; d( y
75.      * @return int  接收到的信息
    
76.      */
    " Z0 J2 u( c. o, W& F5 R1 R
77.     public function handshaking($newClient, $line){
    
78. 
    
79.         $headers = array();
    
80.         $lines = preg_split("/\r\n/", $line);
    ! r6 Q/ n! c# M' ?. f# h( J
81.         foreach($lines as $line)
    4 U' ]$ S  y& K6 h% ~+ T2 _
82.         {
    + ~7 H/ M# M7 G1 h" ?) M* b" k
83.             $line = chop($line);
    
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    1 ?# A6 e7 O0 }7 U1 R
85.             {
    
86.                 $headers[$matches[1]] = $matches[2];
    . k( b. F1 d# X# ?
87.             }
    * A& v: U. l" R, y' y, h
88.         }
    
89.         $secKey = $headers['Sec-WebSocket-Key'];
    - a- c1 a, f6 h9 X6 K
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" .
    1 B( _! r, x, i1 D
93.             "Connection: Upgrade\r\n" .
    5 s: B; I" d2 s& _
94.             "WebSocket-Origin: $this->address\r\n" .
    
95.             "WebSocket-Location: ws://$this->address:$this->port/websocket/websocket\r\n".
    5 k' n3 |8 x5 S; h4 |5 t' v
96.             "Sec-WebSocket-Accept:$secAccept\r\n\r\n";
    9 [5 V  C4 }$ x& F& W6 U
97.         return socket_write($newClient, $upgrade, strlen($upgrade));
    
98.     }
    / C& ~' l6 \1 R1 y) X* G  }
99. 
    
100.     /**
     
101.      * 解析接收数据
     
102.      * @param $buffer
     
103.      * @return null|string
     
104.      */
     8 o! k2 K8 m- o% Q5 {7 t0 e
105.     public function message($buffer){
     
106.         $len = $masks = $data = $decoded = null;
     
107.         $len = ord($buffer[1]) & 127;
     * R0 N9 E$ L& a% ?" t
108.         if ($len === 126)  {
     
109.             $masks = substr($buffer, 4, 4);
     6 f, e( {3 y" {( x; P1 C, e' c; a/ B
110.             $data = substr($buffer, 8);
     
111.         } else if ($len === 127)  {
     0 W- K% ~3 l  j, [9 n
112.             $masks = substr($buffer, 10, 4);
     
113.             $data = substr($buffer, 14);
     ; A$ B& j! s' ?0 C# ~) `& N
114.         } else  {
     + o: j; l2 H( x: F6 L& `% O( W- Z
115.             $masks = substr($buffer, 2, 4);
     3 k$ G: I" ]+ i! Q
116.             $data = substr($buffer, 6);
     
117.         }
     ; W1 U6 g' k0 a2 y; K
118.         for ($index = 0; $index < strlen($data); $index++) {
     
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     
120.         }
     5 `( z, e; e; a  M& O# ~5 a
121.         return $decoded;
     4 l0 U" ?+ q* m+ d$ A% l+ Z
122.     }
     4 Q4 }4 Y+ v. g" v" p  j( G6 s3 r
123. 
     & g  {% X4 L  n) H6 ^3 l+ v
124.     /**
     1 a/ O* K* t1 N1 s2 Z- g
125.      * 发送数据
     
126.      * @param $newClinet 新接入的socket
     * o$ E; @* F7 \8 t( @* x5 V0 M
127.      * @param $msg   要发送的数据
     4 `+ P7 k3 p' w6 l; i8 O
128.      * @return int|string
     
129.      */
     / K4 }4 O+ d* a* Z8 K. d
130.     public function send($newClinet, $msg){
     
131.         $msg = $this->frame($msg);
     
132.         socket_write($newClinet, $msg, strlen($msg));
     " w5 R) n' A. j) y$ h  C) _- @) o
133.     }
     
134. 
     
135.     public function frame($s) {
     
136.         $a = str_split($s, 125);
     . {& x/ r5 A/ }3 F) p* G
137.         if (count($a) == 1) {
     
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     9 L  C! A$ M; i1 A" A
139.         }
     ( e4 y* q1 m$ j9 R1 b# v& V  V
140.         $ns = "";
     ! l$ @* H1 g- K% V( ]& @% |
141.         foreach ($a as $o) {
     5 c, W5 g0 b9 b6 t: b
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     ; P, h, j& s1 b: H6 J5 ]1 x9 G! a
143.         }
     ! i5 F( `3 w1 \. ~: }) J/ p
144.         return $ns;
     
145.     }
     0 C- ?. s& n* B% G0 ]( L
146. 
     " G: d, L& q% z" R; o+ b7 _2 A
147.     /**
     
148.      * 关闭socket
       J$ ~) n* k- C% S  V9 s+ Z& Y
149.      */
     
150.     public function close(){
     7 d  o& l( A3 n5 t, b; C
151.         return socket_close($this->_sockets);
     
152.     }
     
153. }
     
154. 
     
155. $sock = new SocketService();
     ) }: r' _& d. S9 L  l* }
156. $sock->run();
     % D- }8 C: _7 y9 l
157. 
     

复制代码

web.html
2 F# c1 a  A; @2 Q' e8 N

1. <!doctype html>
   
2. <html lang="en">
   1 \7 m  ?9 I7 r7 X" |8 D6 x" }
3. <head>
   * H/ h* m4 G1 [8 c# b8 x; b
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>
   / h+ q  w' o0 m+ I6 I
8. <body>
   
9. <input id="text" value="">
   9 F; c( _+ i- x" D& |$ K* @2 V& J1 B
10. <input type="submit" value="send" onclick="start()">
    4 @& |4 G6 `9 ]7 b' [' q
11. <input type="submit" value="close" onclick="close()">
    7 E' |6 |  z& b. b! }/ }
12. <div id="msg"></div>
    
13. <script>
    7 `* S( }$ E$ n
14. /**
    
15. 0：未连接
    
16. 1：连接成功，可通讯
    
17. 2：正在关闭
    
18. 3：连接已关闭或无法打开
    
19. */
    , p+ r/ C3 z' Q; k" P
20. 
    5 t( w1 q, g$ t" M% Y+ V
21.     //创建一个webSocket 实例
    $ [) t2 G; J3 l- `  y4 g' G' Z
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    
23. 
    
24. 
    
25.     webSocket.onerror = function (event){
    
26.         onError(event);
    * A& _- Q  @+ u" O. P" K2 S' x
27.     };
    : |% _6 z- Z) f8 e
28. 
    
29.     // 打开websocket
    
30.     webSocket.onopen = function (event){
    4 v/ \0 c* K: Y6 _% k0 I
31.         onOpen(event);
    
32.     };
    
33. 
    
34.     //监听消息
    
35.     webSocket.onmessage = function (event){
    & S2 O5 q. O$ l; H8 C: \$ s" e, L
36.         onMessage(event);
    
37.     };
    1 b' r+ T7 C' M8 ^5 v6 v
38. 
    " C5 M/ E: w- p5 }" r3 b* h' a- o
39. 
    
40.     webSocket.onclose = function (event){
    0 o* O8 D& [# `; J, P3 s$ B/ T' Y
41.         onClose(event);
    
42.     }
    
43. 
    
44.     //关闭监听websocket
    
45.     function onError(event){
    
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    4 k. b4 W& r. B9 }, p% v
47.         console.log("error"+event.data);
    ) y/ h( Y5 @7 g+ G* {& D
48.     };
    
49. 
    
50.     function onOpen(event){
    + d6 ^0 Q- U' ?
51.         console.log("open:"+sockState());
    ' d8 W) _* W8 T" d% [
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    ' q' F$ C6 s" W0 z$ k+ O. `+ T5 V, h) O
53.     };
    
54.     function onMessage(event){
    ; e8 ^5 Q. X  y! X
55.         console.log("onMessage");
      o8 T% l: d, K- O3 W6 I
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    
57.     };
    
58. 
    1 ?# d3 k, q3 B4 N( X/ a$ S. h" a
59.     function onClose(event){
    8 M% ~) u% M1 H" a# l+ r( E  @6 e
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    7 j7 r% t% c) q# B
61.         console.log("close:"+sockState());
      H% c/ t7 e6 r8 C  N- |" V
62.         webSocket.close();
    ' m. t7 T! I" H3 W/ u$ \0 \
63.     }
    
64. 
    
65.     function sockState(){
    9 o% `( ?1 i5 B, b; ^8 q' r
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    7 w0 a4 F( F1 C9 [/ R* Z. s$ ]
67.             return status[webSocket.readyState];
    
68.     }
    * h: P9 m) a; J  C& {* |: I: a9 k7 D$ @
69. 
    - Y+ K% |3 E- `1 q9 w/ ?+ w' g
70. 
    3 w% z. q0 Q1 w3 [6 E) V! E
71. 
    
72. function start(event){
    
73.         console.log(webSocket);
    
74.         var msg = document.getElementById('text').value;
    . c, t- o1 O- H$ a6 x" \
75.         document.getElementById('text').value = '';
    
76.         console.log("send:"+sockState());
    
77.         console.log("msg="+msg);
    , ~  U- E- _: i$ s
78.         webSocket.send("msg="+msg);
    
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    & ]) d4 R2 y6 v. X
80.     };
    6 `( U9 h1 E1 {* v  E5 ?* b
81. 
    
82.     function close(event){
    & I1 E5 e) q, T/ n, a) L$ J
83.         webSocket.close();
    
84.     }
    ) l! [6 E$ s, ~( [& |. O: {# m, [; `
85. </script>
    7 I- H3 k  w) K. i- d& q3 m
86. </body>
    + e* s' U% A1 f& C
87. </html>

复制代码

1 V7 |' c' _6 S
, Y7 d$ g/ q% [/ r
' `6 n4 a1 R: R% X- V" `& {, Q  b