php实现websocket实时消息推送

admin

php实现websocket实时消息推送
4 v5 j# R# `' |9 z$ X8 V
7 g0 l  o2 I' m$ W+ ^' u
SocketService.php
9 ?8 A; J5 H4 R, y9 b

1. <?php
   
2. /**
   ; e8 w% C: p7 N$ p
3. * Created by xwx
   6 ?6 ?" B7 S7 |* Q' N$ n
4. * Date: 2017/10/18
   
5. * Time: 14:33
   
6. */
   
7. 
   
8. class SocketService
   * }' j4 p% p0 q9 a1 ?7 s
9. {
   * G. N& x' J: d
10.     private $address  = '0.0.0.0';
    
11.     private $port = 8083;
    2 j$ U& _1 |$ A
12.     private $_sockets;
    ! A2 f4 o6 @. X% u' ]/ d6 Z
13.     public function __construct($address = '', $port='')
    
14.     {
    & N4 a$ a/ y9 w0 X/ S
15.             if(!empty($address)){
    
16.                 $this->address = $address;
    
17.             }
    
18.             if(!empty($port)) {
    6 k* h1 Y. Q  S. P; m
19.                 $this->port = $port;
    
20.             }
    $ Q# k$ @: F; Z$ t6 C
21.     }
    
22. 
    
23.     public function service(){
    9 Q3 d( D- u# s( ^3 m4 z7 ?' i
24.         //获取tcp协议号码。
    5 r( V: t. B7 z/ j1 R$ e; V6 P3 u
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)
    : g  M; T* [" ]+ c% ?) `( Q
29.         {
    9 l% B5 s; j6 g2 s3 f3 L
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    ; i. O. \% r  _0 S- l1 ]9 u/ _
31.         }
    5 {+ O7 A1 }* T1 I7 b
32.         socket_bind($sock, $this->address, $this->port);
    
33.         socket_listen($sock, $this->port);
    2 q9 K  R" _; q; Y0 r4 w7 r, f
34.         echo "listen on $this->address $this->port ... \n";
    
35.         $this->_sockets = $sock;
    
36.     }
    ' U, a$ Z4 T0 C9 I# U
37. 
    " i6 ^* Y' W. v4 H* p$ ?
38.     public function run(){
    
39.         $this->service();
    * x( v$ \/ Q1 C6 M5 s
40.         $clients[] = $this->_sockets;
    % n$ W1 q1 y% a; _
41.         while (true){
    
42.             $changes = $clients;
    ' p' c" V# H2 A9 H7 z& [
43.             $write = NULL;
    5 r0 A! E/ F) T9 h! W
44.             $except = NULL;
    
45.             socket_select($changes,  $write,  $except, NULL);
    9 K; P  v; Z) q# F) a& Z
46.             foreach ($changes as $key => $_sock){
    ' F( T$ o4 h9 A. J0 Y' x
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    9 A+ c; o2 ^! r1 A) [) T" M
48.                     if(($newClient = socket_accept($_sock))  === false){
    
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    
50.                     }
    
51.                     $line = trim(socket_read($newClient, 1024));
    ( Y( `5 ~% v& T9 S/ j, m
52.                     $this->handshaking($newClient, $line);
    % h/ A3 }- x% X" q# ~$ v+ x- w
53.                     //获取client ip
    - Y- t4 v+ H8 Q+ d+ m$ w5 K" V# A
54.                     socket_getpeername ($newClient, $ip);
    # W' G7 L! [9 ]: x
55.                     $clients[$ip] = $newClient;
    ; W* G0 ~% V6 h- W
56.                     echo  "Client ip:{$ip}   \n";
    
57.                     echo "Client msg:{$line} \n";
    - f" }9 Z# d, Z
58.                 } else {
    
59.                     socket_recv($_sock, $buffer,  2048, 0);
    
60.                     $msg = $this->message($buffer);
    
61.                     //在这里业务代码
    9 f2 F6 O7 Q2 c' `' g$ P' X  R
62.                     echo "{$key} clinet msg:",$msg,"\n";
    
63.                     fwrite(STDOUT, 'Please input a argument:');
    8 N) C6 f& U  a1 z" }, X2 p$ m
64.                     $response = trim(fgets(STDIN));
    7 C5 V7 s2 G, Q% N& M; ?! I0 k: s
65.                     $this->send($_sock, $response);
    2 E& E: T' @! M4 _* J$ I+ A
66.                     echo "{$key} response to Client:".$response,"\n";
    ; k$ y8 \" a1 Z) }4 Q+ f* H# l7 S
67.                 }
    
68.             }
    & L0 v: z8 D; j! }" Q: g
69.         }
    & {$ M2 T4 O. L: E& q5 k' `7 T
70.     }
    . ?* m' X5 ]6 N0 Y' _& m+ [+ W
71. 
    2 u  }8 p- D3 E+ Y
72.     /**
    $ J! ~5 ?: g  c7 w) q( W5 p
73.      * 握手处理
    4 K4 s* i, k* R) P. E! q6 X
74.      * @param $newClient socket
    
75.      * @return int  接收到的信息
    
76.      */
    
77.     public function handshaking($newClient, $line){
    
78. 
    
79.         $headers = array();
    
80.         $lines = preg_split("/\r\n/", $line);
    ; i1 U: m: [: H4 d' k0 G( H& H: F! W
81.         foreach($lines as $line)
    8 ]9 a. Z+ z* i; Q
82.         {
    $ K% g: m/ r% ]. D. o+ h  A$ R& m
83.             $line = chop($line);
    1 i& A8 a' B+ c4 w1 J
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    
85.             {
    
86.                 $headers[$matches[1]] = $matches[2];
    
87.             }
    
88.         }
    - T' A. z8 O; w0 ?! q/ i# e: P/ c! B
89.         $secKey = $headers['Sec-WebSocket-Key'];
    ) g1 ~1 t  s# i
90.         $secAccept = base64_encode(pack('H*', sha1($secKey . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11')));
    $ x! |' F, g: M
91.         $upgrade  = "HTTP/1.1 101 Web Socket Protocol Handshake\r\n" .
    0 w$ j3 H& F3 m: X5 A. d
92.             "Upgrade: websocket\r\n" .
    
93.             "Connection: Upgrade\r\n" .
    % W3 z1 p, k, K. |: ~, c* `
94.             "WebSocket-Origin: $this->address\r\n" .
    
95.             "WebSocket-Location: ws://$this->address:$this->port/websocket/websocket\r\n".
    ; ]2 e, W% Q! |5 ]
96.             "Sec-WebSocket-Accept:$secAccept\r\n\r\n";
    
97.         return socket_write($newClient, $upgrade, strlen($upgrade));
    ; U3 a. F& Y  O- b' t6 P/ s8 B
98.     }
    
99. 
    
100.     /**
     5 ~* H7 Q  J3 X6 g: P
101.      * 解析接收数据
     & Q( r( K- P) Z2 d
102.      * @param $buffer
     & x1 p* ?5 s7 e$ F# v3 }5 U
103.      * @return null|string
     
104.      */
     # L: o0 T/ b' B" Z) j* m
105.     public function message($buffer){
     
106.         $len = $masks = $data = $decoded = null;
     
107.         $len = ord($buffer[1]) & 127;
     
108.         if ($len === 126)  {
     6 s( _6 Z3 ~: m1 L
109.             $masks = substr($buffer, 4, 4);
     % q6 N6 e( C* Z" i
110.             $data = substr($buffer, 8);
     4 I' n' D- m& |- b3 w
111.         } else if ($len === 127)  {
     
112.             $masks = substr($buffer, 10, 4);
     
113.             $data = substr($buffer, 14);
     
114.         } else  {
     2 K8 T) `* a  y
115.             $masks = substr($buffer, 2, 4);
     ; Z. l* Z/ r2 [$ d# K, M* Q
116.             $data = substr($buffer, 6);
     / J3 A1 ]: ^- w" l0 ^7 A% s
117.         }
     
118.         for ($index = 0; $index < strlen($data); $index++) {
     2 G% y( a& O6 {  Z
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     4 R3 g$ p- N- s6 H7 g3 C
120.         }
     / J0 u5 y# t7 S/ c7 F  N
121.         return $decoded;
     
122.     }
     
123. 
     5 q. y1 k6 U7 R. H+ b1 {
124.     /**
     
125.      * 发送数据
     
126.      * @param $newClinet 新接入的socket
     
127.      * @param $msg   要发送的数据
     
128.      * @return int|string
     - o9 G2 ?3 K+ n8 L9 W1 S7 B7 Y
129.      */
     5 r3 I& @0 O3 d) N3 y: V
130.     public function send($newClinet, $msg){
     & A5 M0 C/ y' t$ h
131.         $msg = $this->frame($msg);
     ' R) u! v8 j  w# _
132.         socket_write($newClinet, $msg, strlen($msg));
     
133.     }
     
134. 
     ' I( d4 z+ z# D- z, m) y# @5 C3 O: r
135.     public function frame($s) {
     
136.         $a = str_split($s, 125);
     4 b  Z6 M( ?  X! u) t
137.         if (count($a) == 1) {
     
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     4 x0 d: s3 d* L% o2 j
139.         }
     ; ~9 [% Q3 M- R) W- e$ u; u
140.         $ns = "";
     
141.         foreach ($a as $o) {
     
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     % B" d9 S7 p9 w$ N! Z  a
143.         }
     
144.         return $ns;
     8 b& L1 a: D  `  u8 A/ j
145.     }
     
146. 
     
147.     /**
       [# Z! S3 H8 f* t
148.      * 关闭socket
     # Q0 p2 w. r2 F$ f* }- m1 y
149.      */
     " @8 [  Z: t) M' ~: U0 H
150.     public function close(){
     
151.         return socket_close($this->_sockets);
     $ n( d4 I9 t& Q% x. i1 H
152.     }
     
153. }
     . @- c6 }$ j% W; Z' M' y
154. 
     
155. $sock = new SocketService();
     
156. $sock->run();
     
157. 
     

复制代码

web.html
& `, Y. [7 z5 {0 w9 @. I

1. <!doctype html>
   
2. <html lang="en">
   " M* j( w  X- j; C* a
3. <head>
   1 w1 f+ B2 W* m7 n
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>
   3 w$ Z  D' H! F. L. m4 O7 T
8. <body>
   
9. <input id="text" value="">
   ( v6 _9 f3 B* j
10. <input type="submit" value="send" onclick="start()">
    & o2 c2 J4 n# L
11. <input type="submit" value="close" onclick="close()">
    
12. <div id="msg"></div>
    + T7 |' R# i# g$ W6 q+ B0 C: {
13. <script>
    . c( i: N& I+ |+ R. S
14. /**
      Y. q8 Y" N! |$ y- b) G$ F/ _5 ^. X
15. 0：未连接
    # H* b1 J2 f- f- O3 ?9 {
16. 1：连接成功，可通讯
    
17. 2：正在关闭
    
18. 3：连接已关闭或无法打开
    ) A, I1 }) V' n+ }
19. */
    # ~7 s' ?( A; T3 c
20. 
    * ?2 l& ?3 w- G( W/ U3 X
21.     //创建一个webSocket 实例
    
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    % o+ z% X; [3 T# Z" \- i" P' w, i
23. 
    
24. 
    
25.     webSocket.onerror = function (event){
    
26.         onError(event);
    ; O. s2 H9 J" t5 G5 o4 ^9 x/ M2 P$ E
27.     };
    
28. 
    
29.     // 打开websocket
    
30.     webSocket.onopen = function (event){
    
31.         onOpen(event);
    - c( p; c/ e& t4 L- x
32.     };
    % y/ D# U8 i0 e- {, p. `+ v3 R3 S
33. 
    : I% X5 i2 a* C: w3 h& m
34.     //监听消息
    % v- f8 t; ^( T& w- K3 O
35.     webSocket.onmessage = function (event){
    # E" D( |  }* d; {
36.         onMessage(event);
    + r  f4 q) }" f. t1 R/ S
37.     };
    2 l& r  s* f$ G
38. 
    + Q  k" ~* f+ j+ E" B& ~- }: C, J
39. 
    # `" \; J7 y* q' Y% [0 I
40.     webSocket.onclose = function (event){
    
41.         onClose(event);
    ! P3 ^- t3 v/ v: ?+ M0 g! O+ G
42.     }
    5 s) i. {8 Y/ q) J. A( ^1 z: T8 Q
43. 
    0 m  H  b5 }7 e" S& V* Z2 \& n
44.     //关闭监听websocket
    
45.     function onError(event){
    5 K0 \+ l3 y" I' `
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    7 c3 i6 U4 `  I# _' H
47.         console.log("error"+event.data);
    
48.     };
    5 a; w  ?( S$ i7 Z* V8 l! Y: |
49. 
    
50.     function onOpen(event){
    5 t( f( f6 d2 N. L
51.         console.log("open:"+sockState());
    1 T/ s7 k2 @/ `# [) Q3 j$ U
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    $ X- ^3 T# Q3 A' ~& d( o7 h& O: t
53.     };
    
54.     function onMessage(event){
    ! V+ C/ p7 W/ s% w7 L. \9 m8 c0 D
55.         console.log("onMessage");
    6 K% r$ s2 i- y  @! B, @% o
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    , y' \9 r( r, t* G
57.     };
    
58. 
    
59.     function onClose(event){
    
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    + Q+ I1 i9 H( ^* B
61.         console.log("close:"+sockState());
    
62.         webSocket.close();
    1 b, a" g  {6 O& e
63.     }
    + s- l1 w7 G8 I' z5 h' x
64. 
    
65.     function sockState(){
    
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    
67.             return status[webSocket.readyState];
    % l8 P+ g) u7 |% K- ]
68.     }
    
69. 
    # m! I1 S* e+ T) a0 h% J
70. 
    
71. 
    # C9 K+ r% ?2 ~/ I  r) h  P
72. function start(event){
    
73.         console.log(webSocket);
    
74.         var msg = document.getElementById('text').value;
    ' g; n1 e0 J& Y; f, V. x/ t
75.         document.getElementById('text').value = '';
    8 x$ P4 h/ K7 }  a. d+ g" j
76.         console.log("send:"+sockState());
    ; C- U0 A  ]2 X1 p
77.         console.log("msg="+msg);
    
78.         webSocket.send("msg="+msg);
    
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    
80.     };
    : J, V6 e8 f1 ]4 {: a' Y8 J
81. 
    
82.     function close(event){
    
83.         webSocket.close();
    $ W0 I! N4 P& J  o( D
84.     }
    
85. </script>
    , h% u2 o& p, p; c1 F
86. </body>
    
87. </html>

复制代码

& |9 a. k" s, s. d1 R0 g) I1 L
$ n5 F+ V: G7 v2 t/ q