php实现websocket实时消息推送

admin

php实现websocket实时消息推送

6 L3 T" f5 o" ^/ b7 D
( c, ?( `2 J2 XSocketService.php

1. <?php
   
2. /**
   
3. * Created by xwx
   
4. * Date: 2017/10/18
   % a8 y  U* F3 p* G+ @( [4 g
5. * Time: 14:33
   . C5 S( u2 L0 J3 y5 N
6. */
   
7. 
   
8. class SocketService
   + M$ ]' d$ {$ H$ L
9. {
     \8 l4 V' }* L! D. ?  }0 r7 R& V
10.     private $address  = '0.0.0.0';
    
11.     private $port = 8083;
    
12.     private $_sockets;
    
13.     public function __construct($address = '', $port='')
    5 f. ~2 C9 @, I* e! p
14.     {
    1 m9 ^* L6 i9 o( w+ z8 I) ^
15.             if(!empty($address)){
    " D7 b" w- V7 ?8 [* B, g
16.                 $this->address = $address;
    
17.             }
    3 ~, @3 e+ K2 E# ^
18.             if(!empty($port)) {
    ) b( \& R9 t7 @1 a: g! A
19.                 $this->port = $port;
    
20.             }
    4 D5 e; h$ `! y* R! C- {
21.     }
    " U8 n5 a: b% L9 ^! S- m& ^
22. 
    
23.     public function service(){
    
24.         //获取tcp协议号码。
    
25.         $tcp = getprotobyname("tcp");
    
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    & m" W+ e6 S$ A1 `
28.         if($sock < 0)
    
29.         {
    
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    9 C4 _% Z* C& f1 R2 t9 D3 z: M
31.         }
    + p) N) J8 I. o" b5 W. [4 J
32.         socket_bind($sock, $this->address, $this->port);
    $ \- B8 z) ~5 g; z: X: f, `! i* e
33.         socket_listen($sock, $this->port);
    
34.         echo "listen on $this->address $this->port ... \n";
    , U  t2 \$ y. M- J/ u2 g; k3 T# m
35.         $this->_sockets = $sock;
    ! L3 i  v- C2 \* o
36.     }
    6 o7 W0 R$ ~2 @  W7 Z) V& c# U+ M
37. 
    + n9 L% G2 K6 e- J: ?
38.     public function run(){
    
39.         $this->service();
    
40.         $clients[] = $this->_sockets;
    8 c. T: d& r( p, y" a6 Z, c
41.         while (true){
    6 L. i* G; H# ?, G6 ^3 a6 H
42.             $changes = $clients;
    
43.             $write = NULL;
    
44.             $except = NULL;
    & W6 E$ g' A6 e
45.             socket_select($changes,  $write,  $except, NULL);
    ( m# _0 N# {4 i0 W) }
46.             foreach ($changes as $key => $_sock){
    
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    
48.                     if(($newClient = socket_accept($_sock))  === false){
    ! g1 `1 O; Z# S" x1 u
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    ' V, p; _! k4 A' Y+ }
50.                     }
    
51.                     $line = trim(socket_read($newClient, 1024));
    : w$ L% H% r. I* {" Z
52.                     $this->handshaking($newClient, $line);
    
53.                     //获取client ip
    
54.                     socket_getpeername ($newClient, $ip);
    # {8 A( h8 x" x6 t4 R; N% l: q6 C2 s
55.                     $clients[$ip] = $newClient;
    
56.                     echo  "Client ip:{$ip}   \n";
    
57.                     echo "Client msg:{$line} \n";
    / s8 {, |  Z9 z, q% Y2 j# F
58.                 } else {
    ' ~, z4 S9 S' W1 p' z& Q
59.                     socket_recv($_sock, $buffer,  2048, 0);
    4 D! O4 m- \3 u0 @  N
60.                     $msg = $this->message($buffer);
    
61.                     //在这里业务代码
    ( H1 O5 S# u$ E1 c7 j9 J
62.                     echo "{$key} clinet msg:",$msg,"\n";
    
63.                     fwrite(STDOUT, 'Please input a argument:');
    
64.                     $response = trim(fgets(STDIN));
    , S, _& a" y( j; J9 C
65.                     $this->send($_sock, $response);
    
66.                     echo "{$key} response to Client:".$response,"\n";
    * {1 E0 Q( {5 x, I
67.                 }
    * s4 E- r3 k3 x( }' P+ I
68.             }
    
69.         }
    2 A: g2 \1 c( ^- G; L7 K/ |+ G
70.     }
    * h6 q; }- E) e! P* ^, s
71. 
    
72.     /**
    
73.      * 握手处理
    
74.      * @param $newClient socket
    2 @  H/ M& M+ b# i7 o
75.      * @return int  接收到的信息
    
76.      */
    " S8 M, Q: q6 K% V5 v. b# _( H
77.     public function handshaking($newClient, $line){
    
78. 
    
79.         $headers = array();
    
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))
    - N. M. N. a' ^; G0 W
85.             {
    
86.                 $headers[$matches[1]] = $matches[2];
    ) E5 P6 a" Y0 p: q* x
87.             }
    
88.         }
    / P# r' X% Z/ X+ O0 E
89.         $secKey = $headers['Sec-WebSocket-Key'];
    
90.         $secAccept = base64_encode(pack('H*', sha1($secKey . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11')));
    + U# ]: z8 x  u5 T( h/ ^$ I" w- z2 n/ d
91.         $upgrade  = "HTTP/1.1 101 Web Socket Protocol Handshake\r\n" .
    5 {' ~4 W2 I* Q) O
92.             "Upgrade: websocket\r\n" .
    
93.             "Connection: Upgrade\r\n" .
    - C4 @5 M& U0 y9 G1 i/ x
94.             "WebSocket-Origin: $this->address\r\n" .
    
95.             "WebSocket-Location: ws://$this->address:$this->port/websocket/websocket\r\n".
    ( p* m1 ]  {- R( u& E6 ?: o9 g
96.             "Sec-WebSocket-Accept:$secAccept\r\n\r\n";
    
97.         return socket_write($newClient, $upgrade, strlen($upgrade));
    4 `, |/ h$ C. T% r1 w; e( `
98.     }
    ' T: ^" F; B; [: t+ K& V- Q" r5 J
99. 
    7 L) g. Q7 Y+ Y
100.     /**
     3 S. r/ m' z/ z) a# H, L8 A
101.      * 解析接收数据
     . `% ?; |; b6 z7 j1 T3 V- f
102.      * @param $buffer
     . I; e+ q2 t  B+ X
103.      * @return null|string
     " Z4 J' H: A# q( S
104.      */
     % F1 M7 j+ U/ X3 E
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);
     # X- s+ @# O1 [8 E
110.             $data = substr($buffer, 8);
     
111.         } else if ($len === 127)  {
     8 k$ A# H4 |: z* A" y' C* T
112.             $masks = substr($buffer, 10, 4);
     
113.             $data = substr($buffer, 14);
     
114.         } else  {
     
115.             $masks = substr($buffer, 2, 4);
     
116.             $data = substr($buffer, 6);
     
117.         }
     $ S0 G2 j: j5 p" A. a
118.         for ($index = 0; $index < strlen($data); $index++) {
     
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     - z% v# A4 ?) ?: `1 z. G
120.         }
     
121.         return $decoded;
     
122.     }
     
123. 
     
124.     /**
     ' A0 K+ [/ V5 C8 j2 @, H" k
125.      * 发送数据
     
126.      * @param $newClinet 新接入的socket
     ! b# B0 T# T# D4 y0 @1 M2 e5 c9 M
127.      * @param $msg   要发送的数据
     
128.      * @return int|string
     
129.      */
     2 n0 {* W4 S0 T* S
130.     public function send($newClinet, $msg){
     
131.         $msg = $this->frame($msg);
     8 V9 L. M% N' W4 ^3 {  [' Q1 c
132.         socket_write($newClinet, $msg, strlen($msg));
     
133.     }
     
134. 
     
135.     public function frame($s) {
     - Y: w! o/ c; a
136.         $a = str_split($s, 125);
     : m( A% c1 ^) t5 R8 U# p$ @4 d" p
137.         if (count($a) == 1) {
     
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     : J0 P6 b" d6 K( |
139.         }
     
140.         $ns = "";
     
141.         foreach ($a as $o) {
     . [' u- G0 S6 y2 G
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     
143.         }
     
144.         return $ns;
     
145.     }
     $ @/ K; `. b6 |/ l+ v3 H& @
146. 
     
147.     /**
     % r) g( d" c1 s5 ^5 f
148.      * 关闭socket
     ; k9 {6 x& A8 n1 u/ C7 r) U0 E( e$ j
149.      */
     7 W# B& g0 q! N/ t- D9 U7 L
150.     public function close(){
     
151.         return socket_close($this->_sockets);
     
152.     }
     4 m7 q/ G' x+ x) _- D5 k/ {; f# |
153. }
     
154. 
     # J! ^+ l/ O5 J- z
155. $sock = new SocketService();
     8 Q7 z* }5 I  m5 z- N& M
156. $sock->run();
     3 @- S4 w2 H2 [, z$ O. E5 ~
157. 
     

复制代码

web.html
& w9 r& v3 _! X$ O1 m1 ~" g

1. <!doctype html>
   
2. <html lang="en">
   4 Y. l( o8 ^0 ?/ Z! x
3. <head>
   
4.   <meta charset="UTF-8">
   
5.   <meta name="viewport" content="width=device-width,initial-scale=1, maximum-scale=1, user-scalable=no">
   8 V8 a" ?" v& p0 l7 X: u( N0 n
6.   <title>websocket</title>
   5 z) T& O4 V5 v0 f* g  y; ~
7. </head>
   . d! `+ F2 W9 L' M( R- j
8. <body>
   " ]$ C3 Y8 x7 B1 D) N
9. <input id="text" value="">
   ) @- N8 G! V; g. [: p
10. <input type="submit" value="send" onclick="start()">
    $ i% ^) j* O7 w. d1 V
11. <input type="submit" value="close" onclick="close()">
    
12. <div id="msg"></div>
    
13. <script>
    
14. /**
    
15. 0：未连接
    
16. 1：连接成功，可通讯
    ( M/ q/ Q6 b, _  H  F) M9 o
17. 2：正在关闭
    / u% w6 e* E4 a9 |
18. 3：连接已关闭或无法打开
    
19. */
    
20. 
    % s7 q8 u+ P- j8 g9 P
21.     //创建一个webSocket 实例
    4 N- @( z+ {% f" J( k/ n" V  r) ~; m8 ]
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    
23. 
    
24. 
    $ I# z( C" E2 K" |1 h
25.     webSocket.onerror = function (event){
    
26.         onError(event);
    & N, ?- ]6 d! x9 L) Q
27.     };
    
28. 
    
29.     // 打开websocket
    
30.     webSocket.onopen = function (event){
    3 T" f* b" ?1 V' Z3 w+ D
31.         onOpen(event);
    4 a% x" a+ a5 Z; Q$ T
32.     };
    * d0 x3 `; ?! A% j4 b
33. 
    
34.     //监听消息
    & @7 V/ c! z2 g* ^" B% v$ t
35.     webSocket.onmessage = function (event){
    & v5 n: \. V$ q9 @+ r: D7 F
36.         onMessage(event);
    
37.     };
    ) B9 K. Q9 _( w2 Z* `; C" ?+ M
38. 
    
39. 
    
40.     webSocket.onclose = function (event){
    
41.         onClose(event);
    
42.     }
    $ ]! ^7 M" t' o' g8 ^1 Z/ I
43. 
    
44.     //关闭监听websocket
      m3 s* e4 ?3 A  F
45.     function onError(event){
    * ]7 o  K5 e6 v0 w6 A6 s
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    : b2 @9 h5 e  l) C. N5 }0 o6 U7 f( z+ _
47.         console.log("error"+event.data);
    4 ^9 ]+ W) v$ c1 x0 _( J# [
48.     };
    
49. 
    ' o$ _; s6 }  e- B8 M& B% A) |" w$ Y
50.     function onOpen(event){
    
51.         console.log("open:"+sockState());
    ( |2 F) b. Q+ I3 _: C- z4 |7 Z4 `
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    $ O& f1 S2 N0 h
53.     };
    ' E! `1 r# z8 W( t0 J, A/ T
54.     function onMessage(event){
    , \* Z$ I3 p5 L, t
55.         console.log("onMessage");
    
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    # S6 H/ l) w7 V+ l% K
57.     };
    3 Z7 a8 S3 P8 H8 @2 ^) H' D% ^: }  S" k
58. 
    
59.     function onClose(event){
    - s3 k+ X" H# l& r
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
61.         console.log("close:"+sockState());
    
62.         webSocket.close();
    
63.     }
    
64. 
    ) p! x" _2 l: S$ {8 s
65.     function sockState(){
    2 \0 j) p) d! }3 T4 p: ]3 S/ b
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    
67.             return status[webSocket.readyState];
    $ }- }: k. i/ j+ T
68.     }
    
69. 
    ; A& x: c- R' ]: @( m0 r9 o: v
70. 
    
71. 
    
72. function start(event){
    4 r7 Y, B$ g9 o  a4 J
73.         console.log(webSocket);
    
74.         var msg = document.getElementById('text').value;
    
75.         document.getElementById('text').value = '';
    ; V' n/ s/ v5 w4 l; H3 P! Q* r
76.         console.log("send:"+sockState());
    ( T$ n* d; `' q+ }) i
77.         console.log("msg="+msg);
    
78.         webSocket.send("msg="+msg);
    8 l( F4 C$ @' a. N# {9 z# F, @6 R
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    
80.     };
    , j9 s/ [* ?) H0 W+ ]* E
81. 
    / A$ O# O4 @. }3 _" r/ P# G
82.     function close(event){
    8 H! x$ ~5 ~2 O6 G: f
83.         webSocket.close();
    9 p' H" u) _5 F8 A) U
84.     }
    1 D( D' T8 k! k8 ], v, q7 ~
85. </script>
    
86. </body>
    0 z* J/ i' g+ W8 x4 u9 X
87. </html>

复制代码

" T- a# X( V0 k( R