php实现websocket实时消息推送

admin

php实现websocket实时消息推送
9 v0 E& ?. ?1 H- q) }& n4 N

, a5 p( q4 C& h" g6 o+ p  |  L2 \SocketService.php
3 l" ?  r* b6 {! z9 u/ ~* N+ n9 j

1. <?php
   % n4 B8 M: ?. W/ C4 G2 G
2. /**
   & g: f# k4 p! }5 s8 ~9 b
3. * Created by xwx
   6 u# ~; U+ j8 B+ X+ `
4. * Date: 2017/10/18
   
5. * Time: 14:33
   
6. */
   $ t0 a' q( l  e9 B. V3 P0 ~
7. 
   
8. class SocketService
   : t5 S. X7 K% t+ W& G! R
9. {
   
10.     private $address  = '0.0.0.0';
    
11.     private $port = 8083;
    
12.     private $_sockets;
    
13.     public function __construct($address = '', $port='')
    
14.     {
    7 X3 l: S8 N( h
15.             if(!empty($address)){
    
16.                 $this->address = $address;
    " X- t) |+ z* ?9 p' K' d9 g
17.             }
    : G# B" `. P# x6 b% m! X
18.             if(!empty($port)) {
    
19.                 $this->port = $port;
    
20.             }
      X& T$ D3 _4 I  j& B
21.     }
    5 x3 v, k% f: ]( Y
22. 
    ' V: X' o9 Q' s* P& ?5 t
23.     public function service(){
    
24.         //获取tcp协议号码。
    
25.         $tcp = getprotobyname("tcp");
    ; p! W6 _& N3 H& c
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    
28.         if($sock < 0)
    
29.         {
    4 \6 y/ _$ [( A
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    
31.         }
    . F" v) r0 L& Q& r+ i5 ?! M* Z
32.         socket_bind($sock, $this->address, $this->port);
    ! H3 @. K9 Q0 b" J# Y0 N% y
33.         socket_listen($sock, $this->port);
    # s, ^9 [' }7 b. W8 c
34.         echo "listen on $this->address $this->port ... \n";
    . h9 w- e* Q& o! n: n& C
35.         $this->_sockets = $sock;
    
36.     }
    
37. 
    1 n; }) o: d) p: s8 A$ g
38.     public function run(){
    3 A. s) \1 K/ y2 \, g8 G
39.         $this->service();
    
40.         $clients[] = $this->_sockets;
    
41.         while (true){
    
42.             $changes = $clients;
    & u2 E6 m$ ?# Y7 `1 |
43.             $write = NULL;
    
44.             $except = NULL;
    
45.             socket_select($changes,  $write,  $except, NULL);
    6 }1 H: A! h3 @0 b( o* C
46.             foreach ($changes as $key => $_sock){
    8 n% d/ m$ `- ^
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    6 j/ B4 B, C* Q' c% ?9 L4 E
48.                     if(($newClient = socket_accept($_sock))  === false){
    & R6 T$ D0 Y: Q
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    9 H* @# W9 R5 \0 k- L
50.                     }
    
51.                     $line = trim(socket_read($newClient, 1024));
    4 ?- k  S4 l% [4 N) ~3 z
52.                     $this->handshaking($newClient, $line);
    ; O6 n  R0 {( y% q3 {* q
53.                     //获取client ip
    4 d/ i9 c  F" h. N; T  m& e( ~4 i
54.                     socket_getpeername ($newClient, $ip);
    
55.                     $clients[$ip] = $newClient;
    3 j4 a* T2 F' [! D* g# T- L
56.                     echo  "Client ip:{$ip}   \n";
    ! {4 o$ g* t6 a$ \
57.                     echo "Client msg:{$line} \n";
    
58.                 } else {
    
59.                     socket_recv($_sock, $buffer,  2048, 0);
    2 t6 X0 R- m5 W
60.                     $msg = $this->message($buffer);
    $ ~, ?% c6 a6 E: f( S0 O% d3 H3 B
61.                     //在这里业务代码
    ' `, H; I4 P0 Y. D% g( y, ]3 [
62.                     echo "{$key} clinet msg:",$msg,"\n";
    
63.                     fwrite(STDOUT, 'Please input a argument:');
    . C+ g& a- s) U8 W5 v* l+ Z
64.                     $response = trim(fgets(STDIN));
    
65.                     $this->send($_sock, $response);
    
66.                     echo "{$key} response to Client:".$response,"\n";
    
67.                 }
    
68.             }
    
69.         }
    & }6 A8 z1 l9 x. |6 B$ o
70.     }
    9 m# X# n! e7 D2 W! x1 x7 o
71. 
    ! R( L; t; L+ G# s* q3 m
72.     /**
    ' S! Y/ M( l8 ~9 p3 n
73.      * 握手处理
    ( P! p$ J6 j( X$ A2 L- I
74.      * @param $newClient socket
    
75.      * @return int  接收到的信息
    
76.      */
    
77.     public function handshaking($newClient, $line){
    ! W- \! F; Y+ f, G$ ^. b# Z
78. 
    4 K- W+ M& L' X8 b6 k8 q. L
79.         $headers = array();
    
80.         $lines = preg_split("/\r\n/", $line);
    
81.         foreach($lines as $line)
      P7 c- L1 u6 G1 p& \) _6 q" ^
82.         {
    
83.             $line = chop($line);
    5 {4 p; E9 c; O! U! o; X8 S+ m
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    0 c8 }2 D- G' g7 B" \4 Q- K
85.             {
    
86.                 $headers[$matches[1]] = $matches[2];
    
87.             }
    
88.         }
    3 d& S' O3 J$ |( j8 a
89.         $secKey = $headers['Sec-WebSocket-Key'];
    
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" .
    ! m/ {' D3 V, ], c7 Y, G, K
92.             "Upgrade: websocket\r\n" .
    
93.             "Connection: Upgrade\r\n" .
      D4 ]  n0 T8 Z8 y
94.             "WebSocket-Origin: $this->address\r\n" .
    
95.             "WebSocket-Location: ws://$this->address:$this->port/websocket/websocket\r\n".
    
96.             "Sec-WebSocket-Accept:$secAccept\r\n\r\n";
    " a) d. \' D: L
97.         return socket_write($newClient, $upgrade, strlen($upgrade));
    
98.     }
    
99. 
    
100.     /**
     
101.      * 解析接收数据
     9 d6 s3 Q  u- a( i
102.      * @param $buffer
     & @% ]3 w# O2 q/ F) L
103.      * @return null|string
     
104.      */
     & F9 A0 F! ?6 ?% e
105.     public function message($buffer){
     
106.         $len = $masks = $data = $decoded = null;
     
107.         $len = ord($buffer[1]) & 127;
     ' h, y9 K# R) p7 p
108.         if ($len === 126)  {
     4 m* }  y7 V: @5 @
109.             $masks = substr($buffer, 4, 4);
     % n# l# T: i% g- w7 L# E4 q
110.             $data = substr($buffer, 8);
     
111.         } else if ($len === 127)  {
     
112.             $masks = substr($buffer, 10, 4);
     
113.             $data = substr($buffer, 14);
     ) A1 u" E) m5 b* u" ^
114.         } else  {
     ) B- S3 T: L' ~+ Y; }& C
115.             $masks = substr($buffer, 2, 4);
     6 X2 {0 o: ?8 V2 g9 c# Q2 g
116.             $data = substr($buffer, 6);
     
117.         }
     
118.         for ($index = 0; $index < strlen($data); $index++) {
     
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     
120.         }
     
121.         return $decoded;
     
122.     }
     
123. 
     4 w+ D! J1 F& ?2 ^$ n: b
124.     /**
     9 N. F; ]3 S/ a
125.      * 发送数据
     
126.      * @param $newClinet 新接入的socket
     3 B+ |7 s8 M1 }! a) Q
127.      * @param $msg   要发送的数据
     
128.      * @return int|string
     8 H( l' I3 X# i8 V3 g& v2 D: ?
129.      */
     / [2 g% K; I0 {. B* U
130.     public function send($newClinet, $msg){
     5 P+ {" x  x5 u3 z1 t0 |: Z
131.         $msg = $this->frame($msg);
     1 `* _3 I& b4 F# i4 l
132.         socket_write($newClinet, $msg, strlen($msg));
     
133.     }
     
134. 
     : N. l8 k5 K' r. E; P
135.     public function frame($s) {
     
136.         $a = str_split($s, 125);
     
137.         if (count($a) == 1) {
     . u! v7 {# e: u* C1 l& k0 N
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     3 }; W& Q+ Z/ t) l
139.         }
     
140.         $ns = "";
     
141.         foreach ($a as $o) {
     
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     
143.         }
     
144.         return $ns;
     , b9 R- F9 B$ B) @2 J$ Y0 j6 u8 D
145.     }
     
146. 
     
147.     /**
     0 f! _8 |2 V+ M: A8 Q
148.      * 关闭socket
     * c" ]/ P; d1 n' z5 e% j
149.      */
     ; r1 m( c( {; l& d. ~) @1 S
150.     public function close(){
     
151.         return socket_close($this->_sockets);
     
152.     }
     : d( m" D! A6 N; f" M# t4 U
153. }
     2 |9 ~% ]2 C3 Z: e0 `1 r
154. 
     
155. $sock = new SocketService();
     
156. $sock->run();
     
157. 
     

复制代码

web.html
# C0 x# h# I, |0 ^; `$ \, G

1. <!doctype html>
     N; e' x3 C4 x0 m& w
2. <html lang="en">
   
3. <head>
   0 A! N: C2 v' s# e* ?
4.   <meta charset="UTF-8">
   0 U6 B, D; N& P3 {' n. h
5.   <meta name="viewport" content="width=device-width,initial-scale=1, maximum-scale=1, user-scalable=no">
   
6.   <title>websocket</title>
   2 X* L8 h5 W& t3 u
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()">
    
12. <div id="msg"></div>
    
13. <script>
    9 E; G! H! p. `- q$ T
14. /**
    
15. 0：未连接
    
16. 1：连接成功，可通讯
    
17. 2：正在关闭
    
18. 3：连接已关闭或无法打开
    $ u' l0 p4 W) v7 e5 s
19. */
    
20. 
    
21.     //创建一个webSocket 实例
    7 l6 `! ~  c: x. P. O) S
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    . l$ h7 t$ y/ Y9 e+ {
23. 
    0 g- U! ]9 U7 Q9 o
24. 
    
25.     webSocket.onerror = function (event){
    / Z  s- [, n! h! Z
26.         onError(event);
    3 I3 a0 q% z+ C2 A# E7 m
27.     };
    5 r* S) U( j0 j7 y6 ]
28. 
    ) ]( m) d, w% i
29.     // 打开websocket
    ) k7 B# v* g8 q0 D
30.     webSocket.onopen = function (event){
    : d2 X+ Q8 T! N5 D
31.         onOpen(event);
    ; _6 z" G) ]( K8 L5 V$ m
32.     };
    
33. 
    ! \, T8 S! f- @2 r
34.     //监听消息
      O6 h: V" l. e. E+ |! I
35.     webSocket.onmessage = function (event){
    , R# y) [& f) _3 g) R
36.         onMessage(event);
    / D& ]( v7 v8 w0 J* r
37.     };
    # _6 S6 X- }& u, f/ S+ m4 P2 q7 `
38. 
    4 u1 D* a/ p8 @# g; a
39. 
    6 p7 \) t  g3 S8 n( G
40.     webSocket.onclose = function (event){
    
41.         onClose(event);
    ) ^1 E4 o0 _* h' u) z& `
42.     }
    
43. 
    . ?6 o: M# O1 W( q  v2 w# h
44.     //关闭监听websocket
    
45.     function onError(event){
      ^; G! t. `( R5 a9 n; v
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    0 H! }0 }$ w8 W/ t1 `+ _
47.         console.log("error"+event.data);
    3 n4 z4 R& G" r8 Y% e1 t7 S: m6 c
48.     };
    , q' I7 ^- B! Y9 F# k
49. 
    
50.     function onOpen(event){
    ; {9 V6 J: e# a% V/ P
51.         console.log("open:"+sockState());
    ; y; T' |1 s& `# J, [  J
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    
53.     };
    7 ~  s5 b% Y4 Y9 S/ k
54.     function onMessage(event){
    4 {' y* C7 M+ u+ a) ?
55.         console.log("onMessage");
    
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    
57.     };
    
58. 
    # |, |( H' M1 }' S' ]
59.     function onClose(event){
    
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
61.         console.log("close:"+sockState());
    * f6 e6 E- Q8 z; ^- ?
62.         webSocket.close();
    2 x# {% ^2 x" i! q& r& I- o" Z
63.     }
    
64. 
    1 F: Z9 _4 _) h9 m+ z
65.     function sockState(){
    ) _, C) f! E$ T) r
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    
67.             return status[webSocket.readyState];
    
68.     }
    
69. 
    & K6 C2 w( G% G# D8 C/ {: P8 k3 C
70. 
    
71. 
    1 _0 U+ g" c. V3 b( w
72. function start(event){
    " i" ~, j8 Y3 V) ~! ]# Q! g
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);
    " a. {3 l8 n9 i- E* M
78.         webSocket.send("msg="+msg);
    2 S! w% E' w4 H* G0 d- z3 z" Q
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    & Z8 ^5 k& I  }6 E
80.     };
    ' n, s! V3 `& t3 B
81. 
    ( F; c) i: v/ F$ |
82.     function close(event){
    
83.         webSocket.close();
    9 v3 P1 t; P+ c
84.     }
    6 u& V/ z$ H- s8 W$ \- ?& [) W
85. </script>
    ) D7 J5 ]0 Y+ L# O/ i1 z3 h
86. </body>
    
87. </html>

复制代码

9 g6 _% r3 _$ e% c7 B! T: g
8 L' j6 T" g% e0 J& ]+ P. Z; o" v5 T
  `+ L) S# b! c$ Q1 i& p