php实现websocket实时消息推送

admin

php实现websocket实时消息推送
9 `3 b/ p+ F* ~0 L& V7 G, x; |

SocketService.php

1. <?php
   ; N3 O9 h" c$ T1 o% M% d9 y
2. /**
   
3. * Created by xwx
   
4. * Date: 2017/10/18
   
5. * Time: 14:33
   4 G6 i: ]% X: {0 P' L# s% n
6. */
   ! U% r) O3 a5 x
7. 
   8 ?+ y% b- O% }
8. class SocketService
   
9. {
   ' _& J3 g# t1 v9 M) v( D: H+ Y' I6 g  k
10.     private $address  = '0.0.0.0';
    - B1 F! D9 z. t% K8 k
11.     private $port = 8083;
    / ]+ T8 l; H' O4 c9 a$ W, T
12.     private $_sockets;
    
13.     public function __construct($address = '', $port='')
    # K' _$ ^+ [3 t; E' [1 T, L/ v* M
14.     {
    4 j; s+ E, H& Z* q3 J
15.             if(!empty($address)){
    3 [+ M2 Y: B# `& J0 x8 P
16.                 $this->address = $address;
    
17.             }
    : d* ?5 _7 l4 B6 D+ `
18.             if(!empty($port)) {
    
19.                 $this->port = $port;
    
20.             }
    - O9 j2 K. O% h, D  P
21.     }
    
22. 
    / f# e+ }$ F; e3 x3 o
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);
    
28.         if($sock < 0)
    . @6 U) o0 p) j3 i" m1 K
29.         {
    
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    5 E9 f- B- F- T2 q; P) z1 l/ M
31.         }
    
32.         socket_bind($sock, $this->address, $this->port);
    ' C! S4 i; B) f$ O% Z  W
33.         socket_listen($sock, $this->port);
    5 d, \- U. @+ I# f& @7 R" {; i
34.         echo "listen on $this->address $this->port ... \n";
    
35.         $this->_sockets = $sock;
    
36.     }
    - i% S4 A( ?/ K& K1 q1 r5 l
37. 
    + m7 R7 M) {" t3 D9 Q3 D, J
38.     public function run(){
    
39.         $this->service();
    , l& _6 S: g& h& R
40.         $clients[] = $this->_sockets;
    
41.         while (true){
    
42.             $changes = $clients;
    4 H$ G' V$ H8 U; W* {! R
43.             $write = NULL;
    ' C, O) d; w% n6 e( T2 b
44.             $except = NULL;
    
45.             socket_select($changes,  $write,  $except, NULL);
    
46.             foreach ($changes as $key => $_sock){
    # z5 O5 _+ u% t, F  X
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    
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));
    
52.                     $this->handshaking($newClient, $line);
    
53.                     //获取client ip
    
54.                     socket_getpeername ($newClient, $ip);
    ! E  A5 _7 o- d: v! F
55.                     $clients[$ip] = $newClient;
    # \3 p$ x* `9 }  g4 r: _5 U9 X
56.                     echo  "Client ip:{$ip}   \n";
    ( S+ u2 o" v% I6 y9 }, b6 D
57.                     echo "Client msg:{$line} \n";
    
58.                 } else {
    6 l2 N4 K! i5 r7 n: S  A1 p5 ]
59.                     socket_recv($_sock, $buffer,  2048, 0);
    
60.                     $msg = $this->message($buffer);
    ; S- [! X! X! v/ u2 b$ L
61.                     //在这里业务代码
    
62.                     echo "{$key} clinet msg:",$msg,"\n";
    5 r. b# Z% r2 W6 k
63.                     fwrite(STDOUT, 'Please input a argument:');
    . Y5 p5 ~( a% e4 l8 R7 `7 N
64.                     $response = trim(fgets(STDIN));
    ; E) j) P; C6 ^: n& X  k, ^
65.                     $this->send($_sock, $response);
    
66.                     echo "{$key} response to Client:".$response,"\n";
    
67.                 }
    ' y( V" o" ]+ G* [
68.             }
    5 o+ t6 w* X9 {; V2 j3 f+ X& C
69.         }
    
70.     }
    
71. 
    
72.     /**
    
73.      * 握手处理
    
74.      * @param $newClient socket
    
75.      * @return int  接收到的信息
    
76.      */
    
77.     public function handshaking($newClient, $line){
    
78. 
    
79.         $headers = array();
    9 ]& u0 v2 P, @; l- |
80.         $lines = preg_split("/\r\n/", $line);
    1 p7 w5 A* h( M9 i* l8 _$ a( A
81.         foreach($lines as $line)
    
82.         {
    2 U2 q, \: A9 G! t6 o
83.             $line = chop($line);
    
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    * V& T; E6 d/ ?2 ?
85.             {
    5 h) r  |, @: U4 g3 s
86.                 $headers[$matches[1]] = $matches[2];
    # d( H% s( x6 U8 k: x
87.             }
    
88.         }
    
89.         $secKey = $headers['Sec-WebSocket-Key'];
    - |6 s1 {" k+ v: D
90.         $secAccept = base64_encode(pack('H*', sha1($secKey . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11')));
      s9 ]5 W* O. {: N/ ^5 v/ F
91.         $upgrade  = "HTTP/1.1 101 Web Socket Protocol Handshake\r\n" .
    
92.             "Upgrade: websocket\r\n" .
    ' d1 X5 b; R6 [0 ^. P
93.             "Connection: Upgrade\r\n" .
    
94.             "WebSocket-Origin: $this->address\r\n" .
    
95.             "WebSocket-Location: ws://$this->address:$this->port/websocket/websocket\r\n".
    3 ^/ f5 u: i1 A; r9 L+ G+ h% R
96.             "Sec-WebSocket-Accept:$secAccept\r\n\r\n";
    
97.         return socket_write($newClient, $upgrade, strlen($upgrade));
    * N9 K1 @3 K+ x' y
98.     }
    
99. 
    
100.     /**
     
101.      * 解析接收数据
     
102.      * @param $buffer
     7 U7 K9 E) q9 R: a* o) \4 z
103.      * @return null|string
     
104.      */
     " U7 o9 }6 u' ?4 `7 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);
     % G/ _7 f# Q0 b* q9 v
110.             $data = substr($buffer, 8);
     / ^3 G  G$ D" C. Y+ J- k! j
111.         } else if ($len === 127)  {
     
112.             $masks = substr($buffer, 10, 4);
     9 c4 Y4 `! }- B! E
113.             $data = substr($buffer, 14);
     
114.         } else  {
     + T# i1 w/ Z; W5 I, e
115.             $masks = substr($buffer, 2, 4);
     
116.             $data = substr($buffer, 6);
     
117.         }
     / k& y2 X( q: {; ?
118.         for ($index = 0; $index < strlen($data); $index++) {
     
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     3 r) U2 I3 A% y. O. q& U
120.         }
     
121.         return $decoded;
     2 t5 j) ?# G! z# L% w9 g
122.     }
     8 N  |* d" u/ e" ?5 q
123. 
     5 k: o3 U' g" Z( `5 K
124.     /**
     # l- T7 g( i0 N* f$ |2 D0 U
125.      * 发送数据
     
126.      * @param $newClinet 新接入的socket
     
127.      * @param $msg   要发送的数据
     " l& m: B, h: U  @2 Y* m
128.      * @return int|string
     6 t  O' o# u7 b- D0 ?* o& {
129.      */
     
130.     public function send($newClinet, $msg){
     3 c7 d) p4 q2 K  A5 h9 D/ W
131.         $msg = $this->frame($msg);
     0 \' b( T6 b8 h$ e! C  {+ R( _
132.         socket_write($newClinet, $msg, strlen($msg));
     
133.     }
     + v, Z+ ~0 s5 S5 K4 n: ~+ ^
134. 
     : m$ i5 _: I6 W; |8 _3 B8 l
135.     public function frame($s) {
     ; @" W/ \/ z3 _4 ^
136.         $a = str_split($s, 125);
     
137.         if (count($a) == 1) {
     7 z3 A  l/ i, a4 T1 B
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     
139.         }
     
140.         $ns = "";
     
141.         foreach ($a as $o) {
     9 \  c9 d% Y$ t' }
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     
143.         }
     & j6 B$ t, l8 k- B" b" d
144.         return $ns;
     # s, p+ s5 E) c
145.     }
     
146. 
     
147.     /**
     
148.      * 关闭socket
     
149.      */
     
150.     public function close(){
     ; M: x: ^8 g0 ?- y" x; {7 t: s( u" r' ]
151.         return socket_close($this->_sockets);
     2 H; A5 k: S( G$ w8 I; y
152.     }
     
153. }
     
154. 
     
155. $sock = new SocketService();
     
156. $sock->run();
     
157. 
     7 e1 ]8 q% d% h

复制代码

web.html
$ f: t$ Z/ y, }) E' T$ _

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">
   
6.   <title>websocket</title>
   
7. </head>
   
8. <body>
   ! ^( n, j3 y  I5 J/ D$ v
9. <input id="text" value="">
   ( }7 \. X' j. |5 T% l' M$ x& @
10. <input type="submit" value="send" onclick="start()">
    , Q; M% O* l2 x. y3 X0 u5 V
11. <input type="submit" value="close" onclick="close()">
    
12. <div id="msg"></div>
    
13. <script>
    . r% a9 P# k3 E. F1 t1 K5 G
14. /**
      F( n* t0 V+ G* ^+ W7 F
15. 0：未连接
    1 y1 `7 Q# q. ^! h
16. 1：连接成功，可通讯
    
17. 2：正在关闭
    3 T" d" ~+ G* P3 g$ ]8 z
18. 3：连接已关闭或无法打开
    
19. */
    0 s9 x/ Z2 q( w2 f8 F8 C; l
20. 
    1 ^" t4 i) B1 O% A4 u
21.     //创建一个webSocket 实例
    
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    ) v& c- |  S  X' o# W. S6 V  @& T
23. 
    
24. 
    
25.     webSocket.onerror = function (event){
    1 `8 r* B0 p3 M, m9 C: o+ H( ^
26.         onError(event);
    " |. r; Z+ R( ~4 \6 p+ F( x& e7 G
27.     };
    
28. 
    
29.     // 打开websocket
    6 r* {( N0 u7 ?& y8 {- e
30.     webSocket.onopen = function (event){
    : Z5 v( ~8 h0 p
31.         onOpen(event);
    : D" f7 U& g: J: L3 t  K4 b: }7 w6 S
32.     };
    ) u: s+ D1 e, ~5 _1 }% O/ A6 ]. T
33. 
    5 w, Z- D  q' {' W, t
34.     //监听消息
    
35.     webSocket.onmessage = function (event){
    0 f3 T2 n  F8 H1 w! r9 X. a2 E
36.         onMessage(event);
    1 l: b5 C) i3 _0 I! D% E
37.     };
    
38. 
    # S* B: L, r9 m7 f8 _
39. 
    
40.     webSocket.onclose = function (event){
    
41.         onClose(event);
    
42.     }
    
43. 
    & ~8 R! H3 Y7 e) a+ I
44.     //关闭监听websocket
    
45.     function onError(event){
    ' [* d1 ~* x0 A$ \  Q
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    2 C$ O& W3 t& ?) ?" T0 A( b( i& [0 K
47.         console.log("error"+event.data);
    ( r) l' v7 x* ~; u1 D: |( |4 z
48.     };
    
49. 
    
50.     function onOpen(event){
    
51.         console.log("open:"+sockState());
      \" K4 W3 ~9 U  x2 c2 V6 c+ y0 k
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    
53.     };
    1 r9 x5 V% H0 y( j5 C7 u
54.     function onMessage(event){
    
55.         console.log("onMessage");
    2 T9 Z1 u, A% ?7 C
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    
57.     };
    
58. 
    
59.     function onClose(event){
    + k$ x7 C  s3 i0 s  r3 z
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    3 y7 J& a% Z9 B1 ]
61.         console.log("close:"+sockState());
    $ v" q: ?# b. h& X$ Y, ~  D
62.         webSocket.close();
    
63.     }
    # b( F# j  S1 D4 [# ]6 t
64. 
    
65.     function sockState(){
    
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    
67.             return status[webSocket.readyState];
    
68.     }
    
69. 
    
70. 
    % j4 i, _# Q5 n1 F/ f6 \
71. 
    ; {' x  Y: M) D4 C
72. function start(event){
    
73.         console.log(webSocket);
    * s! v/ j0 {7 u
74.         var msg = document.getElementById('text').value;
    
75.         document.getElementById('text').value = '';
    
76.         console.log("send:"+sockState());
    
77.         console.log("msg="+msg);
    4 e  k( j* \- f  d2 Z& i" c  |
78.         webSocket.send("msg="+msg);
    / I- K" A: |' ^; j
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    
80.     };
    
81. 
    
82.     function close(event){
    3 D8 x8 q* d* ^3 E' _: ]& _
83.         webSocket.close();
    # D& N  h5 J& ^, i  @! D( {
84.     }
    9 t# w  Z* ]6 g* u2 m
85. </script>
    
86. </body>
    ' `. \! K- u1 l2 e  U. v
87. </html>

复制代码

  B( c) @' N, \  q# l0 }( y* s& B
7 z2 T6 T( o5 f7 u# ]+ U, d5 c