php实现websocket实时消息推送

admin

php实现websocket实时消息推送
3 G: ^9 \! K) [0 I4 s/ e

% f" Z. v% A4 s9 NSocketService.php

1. <?php
   
2. /**
   ! X; ?+ p1 v1 ]3 b% z" g
3. * Created by xwx
   $ p3 h2 p2 `4 ~) ?
4. * Date: 2017/10/18
   2 s) i, a- Z2 F4 |! H$ I+ }
5. * Time: 14:33
   5 n5 U; T% W2 ?7 M. g, `
6. */
   . M/ X( I& |! ^7 S  T4 y
7. 
   ! R# t2 `& _4 H* r
8. class SocketService
   
9. {
   & E( P/ n* l- B  C0 w
10.     private $address  = '0.0.0.0';
    
11.     private $port = 8083;
    
12.     private $_sockets;
    ( r' X1 n* W. n4 w2 k
13.     public function __construct($address = '', $port='')
    , _9 x* o9 j4 G
14.     {
    
15.             if(!empty($address)){
    
16.                 $this->address = $address;
    
17.             }
    
18.             if(!empty($port)) {
    ' J. f+ o0 r  }6 K
19.                 $this->port = $port;
    
20.             }
      d; L* l) v/ ^4 I  W! h0 l% ]4 \/ k; J
21.     }
    
22. 
    ( d2 R& W4 u8 T6 r
23.     public function service(){
      @9 h+ T9 `& h4 g. o7 }" v; r$ U
24.         //获取tcp协议号码。
    8 g, \" z0 e& j# e; }
25.         $tcp = getprotobyname("tcp");
    . X  O( s, e( b( F* v* v
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    
28.         if($sock < 0)
    4 s" Y) s' O3 W3 n
29.         {
    $ V% Y: A' {- d' ]5 m, L
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    
31.         }
    : C! Z! {2 `! ]% ^) D9 ^
32.         socket_bind($sock, $this->address, $this->port);
    
33.         socket_listen($sock, $this->port);
    * O5 }0 T5 }3 c' f$ R  y: w
34.         echo "listen on $this->address $this->port ... \n";
    
35.         $this->_sockets = $sock;
    . R1 \$ D5 s4 n8 u" r
36.     }
    
37. 
    4 a+ f) [. l: y* U- N0 C
38.     public function run(){
    6 r- c; z' L; A( l6 d$ s5 i0 L
39.         $this->service();
    
40.         $clients[] = $this->_sockets;
    ; t1 w' [' Q% K- ^# }" ?
41.         while (true){
    ; j# D5 J' ]$ e$ L+ ]$ t# V  U
42.             $changes = $clients;
    , e( W/ y, n* l# ?4 ]
43.             $write = NULL;
    
44.             $except = NULL;
    
45.             socket_select($changes,  $write,  $except, NULL);
    
46.             foreach ($changes as $key => $_sock){
    ' f  B' L7 R2 z1 o6 q
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    * z! c% U, b. R0 A' y/ g- K  _
48.                     if(($newClient = socket_accept($_sock))  === false){
    2 D& d" y; u) Z9 _
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    . o1 M1 F0 U5 L5 ~( Z: z4 q
50.                     }
    
51.                     $line = trim(socket_read($newClient, 1024));
    
52.                     $this->handshaking($newClient, $line);
    
53.                     //获取client ip
    + P1 |6 i' J* H" _# ~0 @
54.                     socket_getpeername ($newClient, $ip);
    
55.                     $clients[$ip] = $newClient;
    
56.                     echo  "Client ip:{$ip}   \n";
    
57.                     echo "Client msg:{$line} \n";
    
58.                 } else {
    
59.                     socket_recv($_sock, $buffer,  2048, 0);
    . x2 E3 o5 P$ X2 _) x1 b# l) t! b$ D
60.                     $msg = $this->message($buffer);
    . k  \$ X9 d" Q" {7 M; v* c, r
61.                     //在这里业务代码
    " g+ X1 h0 r' _9 X
62.                     echo "{$key} clinet msg:",$msg,"\n";
    
63.                     fwrite(STDOUT, 'Please input a argument:');
    
64.                     $response = trim(fgets(STDIN));
    
65.                     $this->send($_sock, $response);
    
66.                     echo "{$key} response to Client:".$response,"\n";
    
67.                 }
    
68.             }
    
69.         }
    9 K6 s1 T6 E# }4 \. a
70.     }
    
71. 
    # r/ H8 r1 T' b' t$ Y# I
72.     /**
    
73.      * 握手处理
    & p9 ~& D8 b1 [6 D1 _0 m4 b8 |; C
74.      * @param $newClient socket
    $ d$ E4 }5 V# T
75.      * @return int  接收到的信息
    
76.      */
    & L$ u8 T; K8 I( I
77.     public function handshaking($newClient, $line){
    
78. 
    
79.         $headers = array();
    
80.         $lines = preg_split("/\r\n/", $line);
    6 x2 J' F$ y9 M
81.         foreach($lines as $line)
    
82.         {
    ' f& h* [4 H" h$ M0 g9 L
83.             $line = chop($line);
    ) s+ B# Q; b6 V- X4 `5 s
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    ( s( ], s/ o% g! S/ I  K9 J
85.             {
    + r: D( l6 {2 t) U9 i/ w
86.                 $headers[$matches[1]] = $matches[2];
    
87.             }
    
88.         }
    
89.         $secKey = $headers['Sec-WebSocket-Key'];
    " ?5 K* l" a9 a
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" .
    
92.             "Upgrade: websocket\r\n" .
    3 r9 d8 b% h& M
93.             "Connection: Upgrade\r\n" .
    
94.             "WebSocket-Origin: $this->address\r\n" .
    " B4 K' a2 a% h% S- A
95.             "WebSocket-Location: ws://$this->address:$this->port/websocket/websocket\r\n".
    
96.             "Sec-WebSocket-Accept:$secAccept\r\n\r\n";
    3 P0 ]/ V7 m- T9 ^4 h& j7 T! G
97.         return socket_write($newClient, $upgrade, strlen($upgrade));
    & X3 P! \3 o" Q1 T$ U+ I
98.     }
    1 S5 j+ x+ Q4 K- L# J
99. 
    
100.     /**
     ; K, U7 @0 [9 h! E4 j6 |
101.      * 解析接收数据
     8 G" E& |8 E8 v. b2 p0 K0 d
102.      * @param $buffer
     
103.      * @return null|string
     2 H) g# l: u2 U) X- x
104.      */
     6 w$ ^" x6 u# r0 I' X" p. {
105.     public function message($buffer){
     4 |/ ], G$ o$ K& \3 W
106.         $len = $masks = $data = $decoded = null;
     
107.         $len = ord($buffer[1]) & 127;
     
108.         if ($len === 126)  {
     
109.             $masks = substr($buffer, 4, 4);
     
110.             $data = substr($buffer, 8);
     3 _# m% W( q" d2 y/ ~/ T
111.         } else if ($len === 127)  {
     
112.             $masks = substr($buffer, 10, 4);
     0 F9 K2 s% ~; G3 Q4 a
113.             $data = substr($buffer, 14);
     
114.         } else  {
     $ ^; G0 o8 s3 m! m( X/ A. c
115.             $masks = substr($buffer, 2, 4);
     # m* O; S* x- X9 h
116.             $data = substr($buffer, 6);
     
117.         }
     
118.         for ($index = 0; $index < strlen($data); $index++) {
     
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     ! [6 U, x1 K) s
120.         }
     4 a0 J- t' y) p3 t
121.         return $decoded;
     ( W7 I* X) R) W3 h! I1 p
122.     }
     / S; ?# Q0 j+ Q+ ]- I5 v+ b7 ?
123. 
     
124.     /**
     # e1 x  Y/ p& W: ]1 G3 A7 A8 O- d
125.      * 发送数据
     8 X. `2 N) Y$ b* o
126.      * @param $newClinet 新接入的socket
     
127.      * @param $msg   要发送的数据
     5 h" i5 r- k- D# q
128.      * @return int|string
     
129.      */
     
130.     public function send($newClinet, $msg){
     + y7 [( s' J$ X( a1 G
131.         $msg = $this->frame($msg);
       w$ B+ f9 v$ _* A- L
132.         socket_write($newClinet, $msg, strlen($msg));
     
133.     }
     & m% b3 \$ {* M+ j( R( `3 H' j: m- }
134. 
     
135.     public function frame($s) {
     - ^$ v" n" _1 E3 v/ i
136.         $a = str_split($s, 125);
     0 U  F+ S) N3 t! L: m1 T; K, B' p
137.         if (count($a) == 1) {
     ( C5 ]. ~0 i* i4 n" w. P$ Z
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     2 f2 P& O3 B, d/ h0 g
139.         }
     
140.         $ns = "";
     6 \9 q/ }8 H5 L6 a' i* Z3 J2 w
141.         foreach ($a as $o) {
     / {: i0 S5 h2 z8 n' {- v: @
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     
143.         }
     ' u; R, k5 L' t; `, ~$ d
144.         return $ns;
     
145.     }
     0 E( q4 R, Y) V' ~' p
146. 
     ( E: ]7 `& O$ d
147.     /**
     9 A* `$ q8 z% s. Y# O8 H) X' B
148.      * 关闭socket
     
149.      */
     
150.     public function close(){
     : ?; S3 |: j; a" N( D
151.         return socket_close($this->_sockets);
     
152.     }
     6 ~9 }" u5 [7 y9 a! [
153. }
     7 V2 ]7 O  z% l$ e; `
154. 
     
155. $sock = new SocketService();
     : M7 R8 w$ `+ s, r1 x9 w4 v0 o
156. $sock->run();
     2 E; h8 ~2 e: k9 S
157. 
     , `$ w4 k0 m+ l# z+ _

复制代码

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">
   8 ]# q- v9 `3 N7 x0 [, e# `! n
6.   <title>websocket</title>
   - p; ~0 e& y1 s4 Y3 g: F! ~5 S
7. </head>
   ; h  a& C, Q3 l8 C  j; s# ^
8. <body>
   
9. <input id="text" value="">
   7 E& z+ d! F: ?) o) @" @! Q- f
10. <input type="submit" value="send" onclick="start()">
    5 O6 h# r1 V" P1 X% a; u
11. <input type="submit" value="close" onclick="close()">
    ; O. h/ ^6 p, H
12. <div id="msg"></div>
    # G# z6 `  c/ |" f1 M
13. <script>
    5 Z1 ]  l# V0 I4 s+ q% e' n* W/ p
14. /**
    
15. 0：未连接
    
16. 1：连接成功，可通讯
    ' U: v: t. N% E! T. f6 O5 L7 |7 k
17. 2：正在关闭
    
18. 3：连接已关闭或无法打开
    : X! V& Z, |0 f( R& O
19. */
    ' M% K  D: b; `! ]6 d
20. 
    
21.     //创建一个webSocket 实例
    
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    
23. 
    : u: e+ P0 \8 [  G; s
24. 
    
25.     webSocket.onerror = function (event){
    1 `) K4 D3 W6 c
26.         onError(event);
    
27.     };
    
28. 
    
29.     // 打开websocket
    4 t. p# q3 h  A6 G0 J4 C# k/ w; Q
30.     webSocket.onopen = function (event){
    
31.         onOpen(event);
    5 B4 p# c, @3 K# |2 B, t8 n' c
32.     };
    & Z$ d6 j8 C0 ]3 d8 G3 `/ Y0 X" R
33. 
    
34.     //监听消息
    
35.     webSocket.onmessage = function (event){
    - ^6 M: c8 K4 k. M: B3 Z0 m$ Q
36.         onMessage(event);
    
37.     };
    2 \+ O& n: {) D. x' o& r! k
38. 
    
39. 
    
40.     webSocket.onclose = function (event){
    2 r* v  u$ ^! [
41.         onClose(event);
    
42.     }
    : }% p. ^' \4 e
43. 
    , ?6 s) e/ r. f# F
44.     //关闭监听websocket
    ( w- l! U3 n. q% B& V3 y
45.     function onError(event){
    : [6 D/ O. J/ `% |: @7 v7 O0 {6 h0 n
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    8 a# y$ e8 c9 x. \
47.         console.log("error"+event.data);
    8 W/ ]2 D; A9 ~0 S- q
48.     };
    
49. 
    + I) K3 B& B' ^+ _. S
50.     function onOpen(event){
    
51.         console.log("open:"+sockState());
    
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    * {! _4 y% r' R
53.     };
    ( _% K4 v" K% `: S/ z- }+ e
54.     function onMessage(event){
    
55.         console.log("onMessage");
    & G# C6 }) S; N2 x) @7 g) \3 o3 m, N
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    
57.     };
    
58. 
    
59.     function onClose(event){
    , M! _  d" S& J7 ^4 q* Y
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    2 w" d2 m6 d) N# R' u" H
61.         console.log("close:"+sockState());
    
62.         webSocket.close();
    5 s& m7 m$ x. `+ n* L. M7 N1 O
63.     }
    6 m) c. T; P5 W, C' e
64. 
    
65.     function sockState(){
    $ f1 [1 Q, m! P$ S$ i+ }8 p
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    . t" d1 f: p+ p' T! N5 o4 B8 U
67.             return status[webSocket.readyState];
    
68.     }
    - }9 e# t2 Y( T8 }+ b
69. 
    ! E* {, ~/ D( d, q* |& r6 _) D  v2 \
70. 
    
71. 
    8 d0 y6 d# Y6 R3 d* _( e1 V
72. function start(event){
    6 K3 [2 S! q# A9 H7 }" f
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);
    
78.         webSocket.send("msg="+msg);
    
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    9 a* X4 n! {- u; V1 u0 u- s
80.     };
    7 U6 d3 H9 z- ~6 R& R3 P
81. 
    
82.     function close(event){
    ( t2 a" \# A' q
83.         webSocket.close();
    : L0 E0 J6 A0 L& F2 o
84.     }
    / _0 ?8 l; v% U) Y( g, W
85. </script>
    
86. </body>
    $ c2 l1 M1 q  g( l$ i3 K% Q% x
87. </html>

复制代码

' M; ]% x; Q& o6 ^