php实现websocket实时消息推送

admin

php实现websocket实时消息推送
1 U' y* {$ p9 U- a& g
3 h$ W! I8 _- Z: E1 I) c" y
SocketService.php

1. <?php
   - |' t0 s5 V2 Z% r
2. /**
   
3. * Created by xwx
   
4. * Date: 2017/10/18
   
5. * Time: 14:33
   
6. */
   
7. 
   
8. class SocketService
   1 J) {0 n- S" W8 d1 V
9. {
   2 C; O% \6 H: y% D& p
10.     private $address  = '0.0.0.0';
    
11.     private $port = 8083;
    
12.     private $_sockets;
    6 M9 @; m, ~" P5 P3 Z
13.     public function __construct($address = '', $port='')
    , l4 B" u. q$ t( p
14.     {
    ( Z: V0 C- U, m, s
15.             if(!empty($address)){
    % _7 U" a1 |! d
16.                 $this->address = $address;
    
17.             }
    ) Q" ~+ t" U5 V) f% n& \, q
18.             if(!empty($port)) {
    ( L8 C. I/ P$ F- Z8 K% a
19.                 $this->port = $port;
    
20.             }
    
21.     }
    
22. 
    3 I# E5 I2 X7 P
23.     public function service(){
    . K  `  f' G9 y( M" r  ?* D
24.         //获取tcp协议号码。
    ; Z) n7 b+ D0 `  G
25.         $tcp = getprotobyname("tcp");
    
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    
28.         if($sock < 0)
    ! Z/ W, ~4 ]# K9 v! f7 \! H
29.         {
    
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    
31.         }
    ; p" ^) `+ U7 L( |1 ^4 c% M
32.         socket_bind($sock, $this->address, $this->port);
    . j) i6 t1 v: B- a  R4 ~$ W
33.         socket_listen($sock, $this->port);
    
34.         echo "listen on $this->address $this->port ... \n";
    
35.         $this->_sockets = $sock;
    
36.     }
    
37. 
    
38.     public function run(){
    3 C% Y4 N* d1 x( m
39.         $this->service();
    
40.         $clients[] = $this->_sockets;
    
41.         while (true){
    / W4 ^2 \9 ^, W6 W( e' I
42.             $changes = $clients;
    % Y; n" H1 F9 h( d
43.             $write = NULL;
    1 [5 k9 d; j3 C' |  }) r3 s
44.             $except = NULL;
    9 z6 r+ L+ e9 l" u8 E
45.             socket_select($changes,  $write,  $except, NULL);
      G1 h* n  z5 A9 w
46.             foreach ($changes as $key => $_sock){
    0 ~0 U% n' v& o4 ^; V; G
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    
48.                     if(($newClient = socket_accept($_sock))  === false){
      `& Y; b  Y) D8 o+ ]( ~) `
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    ; L3 i" y5 i% g* K# u
50.                     }
    
51.                     $line = trim(socket_read($newClient, 1024));
    
52.                     $this->handshaking($newClient, $line);
    
53.                     //获取client ip
    0 `6 r4 P# Y' B+ t2 V# Y0 [
54.                     socket_getpeername ($newClient, $ip);
    
55.                     $clients[$ip] = $newClient;
    ( ^1 m8 M" @6 o
56.                     echo  "Client ip:{$ip}   \n";
    
57.                     echo "Client msg:{$line} \n";
    1 b% d" {% C& G
58.                 } else {
    ( ]0 y1 ^: S9 L/ n2 r) _' K; ~
59.                     socket_recv($_sock, $buffer,  2048, 0);
    
60.                     $msg = $this->message($buffer);
    
61.                     //在这里业务代码
    
62.                     echo "{$key} clinet msg:",$msg,"\n";
    3 S& y* X9 @1 U
63.                     fwrite(STDOUT, 'Please input a argument:');
    ' Q2 H4 O+ [6 s4 m
64.                     $response = trim(fgets(STDIN));
    5 |6 j' H, ~- B
65.                     $this->send($_sock, $response);
    
66.                     echo "{$key} response to Client:".$response,"\n";
    ' v" ~4 Z( v4 n% r6 y
67.                 }
    
68.             }
    % U/ z! m  ?" ^* \
69.         }
    $ C( A% ?9 u6 o5 K( U
70.     }
    + m# v  \* W' a8 r: ?" S! U
71. 
    
72.     /**
    * k$ s. e8 k, r0 c3 u- U2 u
73.      * 握手处理
    
74.      * @param $newClient socket
    
75.      * @return int  接收到的信息
    
76.      */
    
77.     public function handshaking($newClient, $line){
    ' D6 L7 @- j$ ?. ~8 ?
78. 
    
79.         $headers = array();
    : [8 I, j2 }/ B! W4 h# k( ^0 @9 E
80.         $lines = preg_split("/\r\n/", $line);
    
81.         foreach($lines as $line)
    
82.         {
    
83.             $line = chop($line);
    
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    3 o. X' N* U/ N* H" k  h  k$ g
85.             {
    
86.                 $headers[$matches[1]] = $matches[2];
    
87.             }
    
88.         }
    4 E2 {  C/ z  e4 f
89.         $secKey = $headers['Sec-WebSocket-Key'];
    5 n% Q4 m- F' ^8 \) }
90.         $secAccept = base64_encode(pack('H*', sha1($secKey . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11')));
    : v: `; \+ T9 R. a& _/ O! g
91.         $upgrade  = "HTTP/1.1 101 Web Socket Protocol Handshake\r\n" .
    
92.             "Upgrade: websocket\r\n" .
    . @- |8 L) d) J# n6 Y
93.             "Connection: Upgrade\r\n" .
    
94.             "WebSocket-Origin: $this->address\r\n" .
    6 \! D5 V5 s0 ~1 g9 Z( N0 k# F
95.             "WebSocket-Location: ws://$this->address:$this->port/websocket/websocket\r\n".
    
96.             "Sec-WebSocket-Accept:$secAccept\r\n\r\n";
    
97.         return socket_write($newClient, $upgrade, strlen($upgrade));
    
98.     }
    
99. 
    
100.     /**
     % H: W% G- V7 H. v( D% Z
101.      * 解析接收数据
     
102.      * @param $buffer
     
103.      * @return null|string
     
104.      */
     
105.     public function message($buffer){
     
106.         $len = $masks = $data = $decoded = null;
     + ?4 O1 C8 C; r5 A( c* D
107.         $len = ord($buffer[1]) & 127;
     
108.         if ($len === 126)  {
     - P, G. b; j( I( \
109.             $masks = substr($buffer, 4, 4);
     
110.             $data = substr($buffer, 8);
     ' [& a( I' u- k6 V
111.         } else if ($len === 127)  {
     
112.             $masks = substr($buffer, 10, 4);
     ( P, @+ Q8 H+ P. B, y+ B8 @
113.             $data = substr($buffer, 14);
     / q5 `( z! o& E* ^% t; \, }
114.         } else  {
     
115.             $masks = substr($buffer, 2, 4);
     
116.             $data = substr($buffer, 6);
     
117.         }
     
118.         for ($index = 0; $index < strlen($data); $index++) {
     4 y3 F: \9 s9 U& `: A
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     
120.         }
     ( E! w. z' N) T! y4 {" r. b
121.         return $decoded;
     
122.     }
     
123. 
     
124.     /**
     5 u4 M' P! n8 N/ c% D$ t0 M
125.      * 发送数据
     8 d) ?/ g1 t) a  Y
126.      * @param $newClinet 新接入的socket
     ; Z+ b% _: o5 v8 C% Z- \* w
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.     }
     9 {6 V" L) O2 h$ ?1 M+ Q
134. 
     ! }, L2 c; d  V9 M* r/ z
135.     public function frame($s) {
     6 j+ V- P% Q( l- S$ F6 U# ?
136.         $a = str_split($s, 125);
     : M$ v& p6 \- v- Z  B& N6 }
137.         if (count($a) == 1) {
     ; v# e( `3 |' u
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     
139.         }
     & F# J2 |5 z- T, `4 ~6 n' X
140.         $ns = "";
     7 V. ^2 k$ @3 _- V2 `, _' N
141.         foreach ($a as $o) {
     0 v: \8 I2 F2 \3 t" _% F! x
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     % g8 ~( \8 W$ V; p
143.         }
     
144.         return $ns;
     
145.     }
     / V& K' H% r" M- i- A; s
146. 
     $ b, c9 E8 B! u% G
147.     /**
     + C) D+ {" P4 A4 O. L) z
148.      * 关闭socket
     * L" M# d  t& S+ t5 H! Q
149.      */
     
150.     public function close(){
     * ?; ~* _1 j" D5 O
151.         return socket_close($this->_sockets);
     # Y' i( i0 `4 J2 k0 x. g) p
152.     }
     
153. }
     * W! h- e- ?4 m# A
154. 
     4 F- I; o  L" y+ [
155. $sock = new SocketService();
     9 H: S) w" W- C$ v8 J0 x; s, Y
156. $sock->run();
     
157. 
     

复制代码

web.html

1. <!doctype html>
   
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">
   $ q" Y3 h, Z, W1 x  f
6.   <title>websocket</title>
   
7. </head>
   
8. <body>
   
9. <input id="text" value="">
   
10. <input type="submit" value="send" onclick="start()">
    
11. <input type="submit" value="close" onclick="close()">
    7 P- D& g& r' f8 K1 B* U9 q& I+ @
12. <div id="msg"></div>
    + f- I$ X0 `8 p* _
13. <script>
    
14. /**
    ; {; `4 n7 _: I. z) Z$ c5 p: J
15. 0：未连接
    : l4 f+ U5 O6 ]1 Z) ~* ^0 T
16. 1：连接成功，可通讯
    
17. 2：正在关闭
    + f: P1 k7 G$ H$ M4 Q3 n
18. 3：连接已关闭或无法打开
    3 e1 w* D; ?' V5 K" n
19. */
    
20. 
    : D; x6 x/ C/ E6 Z- R8 j0 n2 Q( u; ]
21.     //创建一个webSocket 实例
    # J- V' E2 q6 k' N) U! P" d  ?
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    
23. 
    
24. 
    
25.     webSocket.onerror = function (event){
    
26.         onError(event);
    
27.     };
    
28. 
    $ N0 M" q: Z% K+ x' u2 v% {
29.     // 打开websocket
    4 h. _( c! y1 z" M5 r
30.     webSocket.onopen = function (event){
    
31.         onOpen(event);
    ' A' Q+ s4 Y, y2 A# ~
32.     };
    
33. 
    ; W) k( V5 H3 ^. p2 f
34.     //监听消息
      G( K/ A& X. d
35.     webSocket.onmessage = function (event){
    " t$ C# P5 S) T) Z+ U" U: w* r
36.         onMessage(event);
    . @0 j( {% b8 f& C" [
37.     };
    
38. 
    
39. 
    ! f& R! k# b# Z+ X8 G) h4 G
40.     webSocket.onclose = function (event){
    
41.         onClose(event);
    ! `2 }4 Y) j3 F1 P5 K# e" t: f9 @
42.     }
    
43. 
    / z" K7 i! ?3 D3 \, l
44.     //关闭监听websocket
    
45.     function onError(event){
    ( c2 g/ E7 }. G9 w
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
47.         console.log("error"+event.data);
    
48.     };
    
49. 
    
50.     function onOpen(event){
      q1 _$ }/ y3 Y# b7 W
51.         console.log("open:"+sockState());
    , d. e& ^4 J( J6 X; H9 M( {$ t7 p
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    8 K5 i! o+ X4 E3 I/ k" v' W0 v
53.     };
    ( L$ o# ^% D7 |6 l# \% @1 @
54.     function onMessage(event){
    ( B" Z9 D  T5 C( p& }( F
55.         console.log("onMessage");
    
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    & ^: i4 Q, |0 N7 ]0 V) D" f; |
57.     };
    
58. 
    
59.     function onClose(event){
    
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
61.         console.log("close:"+sockState());
    ) u& c( S$ M1 f0 _* l8 p2 L
62.         webSocket.close();
    
63.     }
    . G2 c$ g: _* M* \& |. ?" ]; j$ ]
64. 
    0 ]; H/ g* H& g2 H2 c
65.     function sockState(){
    
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    
67.             return status[webSocket.readyState];
    . }, u) L( p2 z
68.     }
    
69. 
    
70. 
    , i' v: C: Z8 T+ O, O
71. 
    / ?2 K  W& K" v& l8 a* H2 j8 l
72. function start(event){
    
73.         console.log(webSocket);
    3 q/ V* e/ c9 S, ~2 y- g3 L
74.         var msg = document.getElementById('text').value;
    & a, F1 k2 {: L9 ?$ `
75.         document.getElementById('text').value = '';
    
76.         console.log("send:"+sockState());
    ; _! n  I) _& h5 @! A; N
77.         console.log("msg="+msg);
    
78.         webSocket.send("msg="+msg);
    $ P# a* Z9 E; r
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    * U) [: W% P. l0 t" m
80.     };
    
81. 
    
82.     function close(event){
    7 D) L+ |4 E5 p6 V2 C
83.         webSocket.close();
    4 n9 w, k4 i+ t5 p: H3 S, m
84.     }
    
85. </script>
    
86. </body>
    # a1 S( m+ `: E, |
87. </html>

复制代码

9 d3 X( C1 _% R# B( A
9 q' v8 }, x: H