php实现websocket实时消息推送

admin

php实现websocket实时消息推送
% Z; _9 J7 ?; [  R% d
! C) a. G0 @8 k* ?
SocketService.php

1. <?php
   5 P1 \6 o) r2 z4 y
2. /**
   & ~/ Y2 o& B( ]) \$ b
3. * Created by xwx
   
4. * Date: 2017/10/18
   
5. * Time: 14:33
   - Q" P) I+ x" B. D8 `
6. */
   
7. 
   4 {: Q  j  o. Y" O
8. class SocketService
   
9. {
   
10.     private $address  = '0.0.0.0';
    * c* X* r" O9 Z
11.     private $port = 8083;
    
12.     private $_sockets;
    
13.     public function __construct($address = '', $port='')
    7 A8 O& i, f( }$ E( {& i+ f1 Z( o
14.     {
    ! h7 Y' j8 \0 {, c
15.             if(!empty($address)){
    
16.                 $this->address = $address;
    $ t/ E  s6 {! I0 v* _* s! u
17.             }
    0 ?5 x* k1 T/ ^
18.             if(!empty($port)) {
    
19.                 $this->port = $port;
    
20.             }
    
21.     }
    2 ?  T) g* c: C: ?
22. 
    
23.     public function service(){
    
24.         //获取tcp协议号码。
    9 K5 C; }1 o+ W/ V( g; X4 J7 p
25.         $tcp = getprotobyname("tcp");
    
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    % [3 ?* e8 t8 _$ y( |. s8 Y
28.         if($sock < 0)
    ' _4 E$ I7 {( F3 j
29.         {
    
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    
31.         }
    8 O8 o& S; O5 a
32.         socket_bind($sock, $this->address, $this->port);
    
33.         socket_listen($sock, $this->port);
    
34.         echo "listen on $this->address $this->port ... \n";
    
35.         $this->_sockets = $sock;
    - ]! w9 |' b% P1 H5 G+ v0 A1 q# \
36.     }
    , A: u8 O. M+ L- D0 D. K9 E, e
37. 
    % i& I$ i/ w- F# y( G6 e- b+ I, @
38.     public function run(){
    ! f5 ~( i+ u- E+ A& w+ g
39.         $this->service();
    8 T5 F4 n% ]! v) T# l
40.         $clients[] = $this->_sockets;
    
41.         while (true){
    
42.             $changes = $clients;
    0 @5 [: ~3 R( k3 B- z
43.             $write = NULL;
    6 j3 Y) M) B* X8 g3 Z
44.             $except = NULL;
    
45.             socket_select($changes,  $write,  $except, NULL);
    
46.             foreach ($changes as $key => $_sock){
    
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    5 X3 I1 C% i" u; R
48.                     if(($newClient = socket_accept($_sock))  === false){
    
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    $ J5 @2 ~$ ^- k- B* u
50.                     }
    # w9 N; T6 M" L" T
51.                     $line = trim(socket_read($newClient, 1024));
    + P! ]8 J9 a& T' s- I
52.                     $this->handshaking($newClient, $line);
    
53.                     //获取client ip
    . B# Z. W& l% S1 E; v
54.                     socket_getpeername ($newClient, $ip);
    
55.                     $clients[$ip] = $newClient;
    
56.                     echo  "Client ip:{$ip}   \n";
    & K1 ~% a3 h# L% s, f: ]
57.                     echo "Client msg:{$line} \n";
    
58.                 } else {
    6 G. t0 V7 t- M5 \. f( H, I
59.                     socket_recv($_sock, $buffer,  2048, 0);
    8 q/ y3 f2 h- R, f
60.                     $msg = $this->message($buffer);
    ; X( u5 X% E; S$ ^) Y
61.                     //在这里业务代码
    % T; n+ u! S3 j# W4 }7 \# X
62.                     echo "{$key} clinet msg:",$msg,"\n";
    * a# Z6 z5 ]2 m! w& |" P
63.                     fwrite(STDOUT, 'Please input a argument:');
    ' y: `! o# K3 E0 i5 W) n8 S( {8 D
64.                     $response = trim(fgets(STDIN));
    6 P4 o) d2 d8 w. x
65.                     $this->send($_sock, $response);
    
66.                     echo "{$key} response to Client:".$response,"\n";
    5 w9 R. Y) t; r  |$ t; P
67.                 }
    
68.             }
    
69.         }
    5 L5 a; c0 W0 m. {. e9 K, e
70.     }
    
71. 
    & x3 r2 q2 u; Q& S* ^
72.     /**
    
73.      * 握手处理
    
74.      * @param $newClient socket
    
75.      * @return int  接收到的信息
    
76.      */
    ! E  R7 a+ S; j/ `
77.     public function handshaking($newClient, $line){
    
78. 
    0 N2 d. g) K& _$ d" l1 X. ~( P
79.         $headers = array();
    
80.         $lines = preg_split("/\r\n/", $line);
    
81.         foreach($lines as $line)
    
82.         {
    6 h; d8 w! w  V5 I+ ]4 d& W
83.             $line = chop($line);
    ! ^( T5 K: C* k( S/ J. V6 |
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    9 H  h3 o6 u- G$ P; ^( P9 ~
85.             {
    
86.                 $headers[$matches[1]] = $matches[2];
    , l' B# h$ T- Y* S; _# g) `
87.             }
    
88.         }
    
89.         $secKey = $headers['Sec-WebSocket-Key'];
    . z" f* H. C: [$ y- q. |4 T
90.         $secAccept = base64_encode(pack('H*', sha1($secKey . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11')));
    . P8 |5 K3 h' `' C. G' R
91.         $upgrade  = "HTTP/1.1 101 Web Socket Protocol Handshake\r\n" .
    
92.             "Upgrade: websocket\r\n" .
    
93.             "Connection: Upgrade\r\n" .
    * v" c7 @% o" S+ g$ M* J1 z
94.             "WebSocket-Origin: $this->address\r\n" .
    9 A6 r- h& a) q) H
95.             "WebSocket-Location: ws://$this->address:$this->port/websocket/websocket\r\n".
    2 B" U; Y! [* U% `) d% Y
96.             "Sec-WebSocket-Accept:$secAccept\r\n\r\n";
    
97.         return socket_write($newClient, $upgrade, strlen($upgrade));
    
98.     }
    ( z0 i9 z8 Z4 B! ], _8 d" _3 G
99. 
    . \0 Z% r$ R; P0 ~; M' F; w2 q
100.     /**
     ' i# C  X! r; w# [! ^3 y
101.      * 解析接收数据
     1 Y$ r3 i0 B- L) J$ N; |% g' |# q
102.      * @param $buffer
     
103.      * @return null|string
     
104.      */
     7 F+ x9 s* T; a" o4 _, V" \
105.     public function message($buffer){
     0 {! N/ S! a/ _' S6 ]  y
106.         $len = $masks = $data = $decoded = null;
     $ t: |. _' s9 Z) @! J; {/ _! B, i
107.         $len = ord($buffer[1]) & 127;
     - n+ i% t. A. |( p8 U% S6 H9 I
108.         if ($len === 126)  {
     
109.             $masks = substr($buffer, 4, 4);
     
110.             $data = substr($buffer, 8);
     . n, ^+ I5 d3 |2 `0 \  Z
111.         } else if ($len === 127)  {
     4 \1 ^9 V+ D& ]; O# _, B
112.             $masks = substr($buffer, 10, 4);
     % m. v4 f+ V3 \. x0 ]9 n: y
113.             $data = substr($buffer, 14);
     
114.         } else  {
     : k+ A0 B  T  K% m
115.             $masks = substr($buffer, 2, 4);
     ' g, N9 d% P1 t# T1 x$ f
116.             $data = substr($buffer, 6);
     
117.         }
     $ K4 ?8 g- [7 M# h
118.         for ($index = 0; $index < strlen($data); $index++) {
     - a3 S2 S4 ]: w) H: F$ Y- [
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     ; @# A# l# s; F4 Y
120.         }
     
121.         return $decoded;
     - V' G- m- {$ b5 e9 Q! V% Z
122.     }
     
123. 
     ) @. b# d0 S0 R1 {( E
124.     /**
     
125.      * 发送数据
     * |3 S6 R7 _: R: x
126.      * @param $newClinet 新接入的socket
     . _% m* J* N9 B6 ~2 [$ K+ S+ U
127.      * @param $msg   要发送的数据
     
128.      * @return int|string
     9 W2 _& _0 E* ^! G
129.      */
     
130.     public function send($newClinet, $msg){
     
131.         $msg = $this->frame($msg);
     / a2 U; s$ k( Y: n6 ]! A8 [
132.         socket_write($newClinet, $msg, strlen($msg));
     " [8 S; j4 c( w1 L
133.     }
     # _; {! X: ]; g. w6 m7 i2 X8 a
134. 
     
135.     public function frame($s) {
     * k+ u; N5 j' @  O4 Y2 S
136.         $a = str_split($s, 125);
     , |) k5 C7 k: i  F
137.         if (count($a) == 1) {
     
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     
139.         }
     
140.         $ns = "";
     5 a0 Y, [, K* v' U+ g
141.         foreach ($a as $o) {
     
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     : s% Z( ^1 d8 \/ g
143.         }
     4 _; }2 d' q; J! i  n
144.         return $ns;
     
145.     }
     # q6 L# P2 C" S
146. 
     
147.     /**
     
148.      * 关闭socket
     * x6 f7 n- R7 Q: c5 G  Y
149.      */
     
150.     public function close(){
     
151.         return socket_close($this->_sockets);
     
152.     }
     % Q2 ?) n" g% }. g, C2 Q
153. }
     7 i# e, u7 J' C; K1 F
154. 
     
155. $sock = new SocketService();
     . W/ y! {6 Y! O' x) G
156. $sock->run();
     ) Y2 Y# s1 `6 P0 O' V
157. 
     

复制代码

web.html
' s4 ~: z- T# e; q+ H3 t. [+ F! J

1. <!doctype html>
   
2. <html lang="en">
   
3. <head>
   + }5 U+ K  s' _( D( a
4.   <meta charset="UTF-8">
   6 u9 C  T% [# p  v8 b; m/ z+ V
5.   <meta name="viewport" content="width=device-width,initial-scale=1, maximum-scale=1, user-scalable=no">
   
6.   <title>websocket</title>
   : ?# O5 e6 o* ~# L& P1 D3 r
7. </head>
   
8. <body>
   1 N# X2 @. j4 E+ U% m7 I2 S
9. <input id="text" value="">
   
10. <input type="submit" value="send" onclick="start()">
    
11. <input type="submit" value="close" onclick="close()">
    1 {8 s7 N- u; }- N& [9 |
12. <div id="msg"></div>
      z: ~0 l' o3 P' J: J
13. <script>
    $ x9 w: o( u6 m1 |; G
14. /**
    
15. 0：未连接
    
16. 1：连接成功，可通讯
    
17. 2：正在关闭
    
18. 3：连接已关闭或无法打开
    
19. */
    
20. 
    # C% \: C: C' r# F) ~% k, v
21.     //创建一个webSocket 实例
    
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    0 |2 Y3 E0 d4 |" x8 ~4 h
23. 
    , U2 E! m) @/ Z1 ?% \4 }) G
24. 
    , v. W! u- ]0 j  l9 A/ A
25.     webSocket.onerror = function (event){
    
26.         onError(event);
    * J0 Y$ V/ `  _9 U8 O, E8 q
27.     };
    4 A' R* v6 R' V+ R5 V$ T4 a1 y* K
28. 
    
29.     // 打开websocket
    - @: Z+ _& U. H" n+ P
30.     webSocket.onopen = function (event){
    
31.         onOpen(event);
    
32.     };
    
33. 
    , B$ }& H- x2 s7 m/ H" E
34.     //监听消息
    8 V. V$ B8 L8 E8 i; W4 O
35.     webSocket.onmessage = function (event){
    6 D, z6 S% y" y- `
36.         onMessage(event);
    
37.     };
    : B* {) @: y# ?& e* Q' ~6 p! \
38. 
    
39. 
    
40.     webSocket.onclose = function (event){
    1 f( h* i5 a0 d/ A4 N
41.         onClose(event);
    
42.     }
    
43. 
    , b; c/ ]0 B5 x' A8 O
44.     //关闭监听websocket
    9 P# _1 `  E! e* r; n: a$ O
45.     function onError(event){
    
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    7 x0 r# g. O  a1 B7 v
47.         console.log("error"+event.data);
    " s% e9 N# _0 B$ n8 n
48.     };
    - X6 X6 u4 e' P$ a; m, S
49. 
    
50.     function onOpen(event){
      N) ~% l, Z) K- a" V+ ~
51.         console.log("open:"+sockState());
    
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    
53.     };
    " l. u& }$ U6 V- L0 q% N0 w. x
54.     function onMessage(event){
    
55.         console.log("onMessage");
    * N3 z7 W2 B: E) c& E$ s
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    
57.     };
    $ H# q! C# e/ w% \, E1 O2 u$ V4 g% d
58. 
    
59.     function onClose(event){
    ! e- u; j3 \7 m  B' X7 |
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
61.         console.log("close:"+sockState());
    8 F. W/ v+ J, b, w" b! z
62.         webSocket.close();
    " L' ~( N8 o" j  h
63.     }
    
64. 
    
65.     function sockState(){
    
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
      r6 `% l0 `: n8 z1 ]! q1 S6 L$ F" v
67.             return status[webSocket.readyState];
    
68.     }
    
69. 
    : A* ~6 R, ~8 k( J1 o/ a1 S
70. 
    + z. t& m) H  a: }1 G, C1 f
71. 
    & i" B$ w, Q( V- `7 [  V7 f
72. function start(event){
    : Z8 B/ J) x" I9 ^& }, P8 A1 T: \% L0 h
73.         console.log(webSocket);
    
74.         var msg = document.getElementById('text').value;
    
75.         document.getElementById('text').value = '';
    
76.         console.log("send:"+sockState());
    
77.         console.log("msg="+msg);
    2 K+ A! f5 H# F1 i1 {& C
78.         webSocket.send("msg="+msg);
    / O8 R+ q, Q. }) u$ i) I6 u5 s
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    
80.     };
    ( y: v1 |0 a3 l! F0 u1 R
81. 
    
82.     function close(event){
    # B2 A" m; G! n& I( x5 W4 t: {- n
83.         webSocket.close();
    9 H! w3 d) A3 `/ ?; u: w
84.     }
    ; }4 D& ]' {- i
85. </script>
    
86. </body>
    3 g5 s6 W( g$ e8 m3 O, j, `
87. </html>

复制代码

) X# N% Y. X& p0 p* [0 a