php实现websocket实时消息推送

admin

php实现websocket实时消息推送
" Z9 k/ }0 ?* F

SocketService.php
9 ^7 a* h5 ^, r6 d( N

1. <?php
   / D4 R7 W! W$ }& @  m
2. /**
   
3. * Created by xwx
   1 f3 i: D4 f4 I+ V4 x& h2 K
4. * Date: 2017/10/18
   
5. * Time: 14:33
   + L' p- ]2 N* n
6. */
     ]/ d& X9 ^, ]! n
7. 
   
8. class SocketService
   : y- X& D3 T0 p
9. {
   ' \' Y( z# B1 i  K9 ?- j  ]6 M" y0 N
10.     private $address  = '0.0.0.0';
    4 |( Y8 Y# Z6 C+ D/ u0 r1 X' K
11.     private $port = 8083;
    8 C% X, v5 h1 V; j
12.     private $_sockets;
    
13.     public function __construct($address = '', $port='')
    
14.     {
    ! h$ ]5 U: P  L
15.             if(!empty($address)){
    
16.                 $this->address = $address;
    
17.             }
    
18.             if(!empty($port)) {
    
19.                 $this->port = $port;
    * ^# K/ ]" `" F7 I, s# O
20.             }
    . e% W1 U. }$ U3 H
21.     }
    
22. 
    
23.     public function service(){
    
24.         //获取tcp协议号码。
    
25.         $tcp = getprotobyname("tcp");
    9 d7 |' q$ l( B
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    
28.         if($sock < 0)
    
29.         {
    
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    
31.         }
    % z% m6 l7 ]2 [  T- [" F) ]
32.         socket_bind($sock, $this->address, $this->port);
    $ w! [# B, c6 t/ c* z6 s
33.         socket_listen($sock, $this->port);
    
34.         echo "listen on $this->address $this->port ... \n";
    . _8 |0 z) l7 w
35.         $this->_sockets = $sock;
    ) T, ]8 Y  t4 \$ Y& r7 N% R  D
36.     }
    
37. 
    
38.     public function run(){
    
39.         $this->service();
    ) y4 ~( L! y2 l9 ^# `
40.         $clients[] = $this->_sockets;
    # N* C/ S! i* W, \) o$ U
41.         while (true){
    $ C% m: o+ i' i2 M2 {$ ]
42.             $changes = $clients;
    " K* p; L) k7 Y
43.             $write = NULL;
    
44.             $except = NULL;
    
45.             socket_select($changes,  $write,  $except, NULL);
    
46.             foreach ($changes as $key => $_sock){
      i$ l9 b7 J- ^  R
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    ' k8 q6 f+ E. n; Y
48.                     if(($newClient = socket_accept($_sock))  === false){
      F$ B! L, i  k
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    
50.                     }
    ! ^" D- u2 ^2 z+ ?' Q9 W0 M7 w
51.                     $line = trim(socket_read($newClient, 1024));
    % \& r) Y" _) c( M. H
52.                     $this->handshaking($newClient, $line);
    
53.                     //获取client ip
    + m! q! T3 K# S" p/ W( O
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);
    3 Z1 d# a: T3 ?
60.                     $msg = $this->message($buffer);
    * H" T8 ?' v. [7 R! Q  t
61.                     //在这里业务代码
    
62.                     echo "{$key} clinet msg:",$msg,"\n";
    
63.                     fwrite(STDOUT, 'Please input a argument:');
    8 Y$ R6 ^( E3 f9 r
64.                     $response = trim(fgets(STDIN));
    4 K% G9 m$ h7 f
65.                     $this->send($_sock, $response);
    
66.                     echo "{$key} response to Client:".$response,"\n";
    5 d, D: @9 H( w& P+ ~. b" q( ?
67.                 }
    3 K$ g7 g* z- b; f- p6 X  y
68.             }
    - _8 S, m% a0 R% s
69.         }
    
70.     }
    
71. 
    % S: A! z; N  {7 t( B$ e
72.     /**
    
73.      * 握手处理
    
74.      * @param $newClient socket
    . f) g( g5 d* ^
75.      * @return int  接收到的信息
    - ~. i6 t. Q; f2 F/ @
76.      */
    
77.     public function handshaking($newClient, $line){
    
78. 
    3 D! [; B0 d( {% l0 g# w) c" q5 g
79.         $headers = array();
    9 M( l# I" t# ~6 s, x% C
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))
    2 _! p% V: n% i. q0 o
85.             {
    
86.                 $headers[$matches[1]] = $matches[2];
    7 B/ e) [3 i+ {* d" E$ y: }5 t
87.             }
    7 [, r3 C# ], ^4 D, L5 [8 v6 d4 X
88.         }
    
89.         $secKey = $headers['Sec-WebSocket-Key'];
    
90.         $secAccept = base64_encode(pack('H*', sha1($secKey . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11')));
    ; X9 \0 U% I9 ?. Z  `2 x0 \
91.         $upgrade  = "HTTP/1.1 101 Web Socket Protocol Handshake\r\n" .
    1 R# r1 E2 B1 q7 Q& r- ^
92.             "Upgrade: websocket\r\n" .
    8 ]6 s+ B. J7 i! h* D: |' h
93.             "Connection: Upgrade\r\n" .
    ; V( K* ?) _* O$ R/ ^
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";
    
97.         return socket_write($newClient, $upgrade, strlen($upgrade));
    
98.     }
    " I) H" Q  G; {4 k, R7 q+ `* v  j6 N( ]
99. 
    " \% N4 D8 [% @2 v9 H+ l0 A! y0 c
100.     /**
     0 `, t( j& \- B  i% N( |0 j* L5 ?
101.      * 解析接收数据
     , P2 g) l- }+ h( o! b
102.      * @param $buffer
     
103.      * @return null|string
     
104.      */
     7 @5 l+ |/ t1 _8 C! P
105.     public function message($buffer){
     8 a/ p& G! h* K7 `0 U
106.         $len = $masks = $data = $decoded = null;
     
107.         $len = ord($buffer[1]) & 127;
     
108.         if ($len === 126)  {
     
109.             $masks = substr($buffer, 4, 4);
     . B+ J+ A- b, t0 L+ M  `
110.             $data = substr($buffer, 8);
     
111.         } else if ($len === 127)  {
     + U9 [$ R% `5 q3 A* x, T1 t" U
112.             $masks = substr($buffer, 10, 4);
     
113.             $data = substr($buffer, 14);
     
114.         } else  {
     9 p! F9 E% u' p& p
115.             $masks = substr($buffer, 2, 4);
     9 z# c* s1 N7 D4 ~( O1 }2 N; p
116.             $data = substr($buffer, 6);
     
117.         }
     
118.         for ($index = 0; $index < strlen($data); $index++) {
     8 M& g% d# \" Q' g
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     
120.         }
     
121.         return $decoded;
     
122.     }
     
123. 
     # T1 u/ M! {$ s. F' U- O- ^
124.     /**
     & {* ]( e  K3 v% {/ j7 T: T( w( @
125.      * 发送数据
     
126.      * @param $newClinet 新接入的socket
     
127.      * @param $msg   要发送的数据
     
128.      * @return int|string
       D% _$ V) m4 M3 {
129.      */
     
130.     public function send($newClinet, $msg){
     ; M' W- C! X0 x5 S; c
131.         $msg = $this->frame($msg);
     
132.         socket_write($newClinet, $msg, strlen($msg));
     
133.     }
     
134. 
       q3 S. T6 H' \2 E% {% n; T
135.     public function frame($s) {
     ( K$ U9 d! m4 `  c( h. b2 L5 G7 P/ K( z- d
136.         $a = str_split($s, 125);
     
137.         if (count($a) == 1) {
     
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     
139.         }
     
140.         $ns = "";
     0 P$ t  [7 G6 P, V3 K
141.         foreach ($a as $o) {
     3 O6 [3 D7 `% B. q
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     " z9 }* q( j. W+ f' u, ]" V
143.         }
     
144.         return $ns;
     1 I. Y( B; w4 K
145.     }
     
146. 
     % G, |; Y1 _. k% O3 {3 a1 g" h2 {2 F
147.     /**
     5 \6 G/ {1 D: q
148.      * 关闭socket
     
149.      */
     $ _6 w2 S& c# e; V, D# d4 S
150.     public function close(){
     
151.         return socket_close($this->_sockets);
     
152.     }
     
153. }
     
154. 
     7 A- v$ Q) X8 K8 x# W+ _
155. $sock = new SocketService();
     
156. $sock->run();
     
157. 
     

复制代码

web.html

1. <!doctype html>
   
2. <html lang="en">
   0 K$ W" i, }/ T- k$ k& C$ t
3. <head>
   * U: V! M3 j+ Y
4.   <meta charset="UTF-8">
   ; k2 I/ L8 H1 M4 b/ Q+ }3 p3 j
5.   <meta name="viewport" content="width=device-width,initial-scale=1, maximum-scale=1, user-scalable=no">
   ! {6 X# g* B! o* Z# |) ~9 c
6.   <title>websocket</title>
   0 x& \, G0 V5 y  g# X3 L( W3 i, L
7. </head>
   9 i6 I: @* H. }* ], i
8. <body>
   
9. <input id="text" value="">
   
10. <input type="submit" value="send" onclick="start()">
    ) [2 K6 b+ U8 v% \5 y1 h- C
11. <input type="submit" value="close" onclick="close()">
    
12. <div id="msg"></div>
    ' g& o& c# u% y0 n$ p
13. <script>
    
14. /**
    
15. 0：未连接
    
16. 1：连接成功，可通讯
    
17. 2：正在关闭
    3 U" r! h7 r# h
18. 3：连接已关闭或无法打开
    
19. */
    
20. 
    # {' \$ g. h& ]+ n# u
21.     //创建一个webSocket 实例
    
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    
23. 
    4 d6 C' j2 t/ @) l5 y5 z2 ^
24. 
    1 z# w* C! F  e& u/ h: P# f- A4 a
25.     webSocket.onerror = function (event){
    
26.         onError(event);
    / v. W. d2 X' O2 ~, B7 B
27.     };
    7 A" t# l  J6 r1 }3 n  B7 O/ r
28. 
    6 x) K: J4 l) y
29.     // 打开websocket
    
30.     webSocket.onopen = function (event){
    
31.         onOpen(event);
    3 Z4 C8 X% g+ ]6 t2 s; [* I
32.     };
    8 t. y2 \, k" f4 D6 ]. N4 G: t% Z
33. 
    
34.     //监听消息
    8 t, r" f9 E( M* M- o9 H1 N
35.     webSocket.onmessage = function (event){
    
36.         onMessage(event);
    ! C( ]6 v/ ]$ {: L' J0 s( X  ^
37.     };
    5 N# b- e1 D# y
38. 
    $ c- i9 v( W/ r. |% b
39. 
    
40.     webSocket.onclose = function (event){
    3 ]+ f8 O( k2 T) a2 }% F" P  J
41.         onClose(event);
    
42.     }
    + ^" x) k' N4 N3 j+ M! {7 w
43. 
    ' s9 u6 }# M' ~
44.     //关闭监听websocket
    * T! _1 ]! C" ^# U
45.     function onError(event){
    ( T2 R" X% o- K# C: q; z1 x
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
47.         console.log("error"+event.data);
    
48.     };
    1 E  G* _1 Z7 C! ^1 S$ L
49. 
    7 C" h2 w! o3 M& N$ V" ?
50.     function onOpen(event){
    - Y; a# T; X& U/ w
51.         console.log("open:"+sockState());
    4 a7 s0 U$ p0 S% c" i; A7 Q0 i
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    5 f, n6 }. N- I! V2 f$ \- C7 I1 A0 a% e
53.     };
    1 f3 r* b* N. c! g7 A+ n
54.     function onMessage(event){
    
55.         console.log("onMessage");
    
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    % p9 l" F  Y: o; n& Q
57.     };
    & F1 J( ?( \, q6 g3 C1 s' F
58. 
    
59.     function onClose(event){
    & @' m2 D# c. ^  w- T' Y
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
61.         console.log("close:"+sockState());
    
62.         webSocket.close();
    + p4 ]/ [% K* c/ G% U1 h1 d5 E
63.     }
    
64. 
    4 w8 T9 C/ E1 T' t4 n" [
65.     function sockState(){
    + |8 P/ O' c3 n
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    + P5 x: l; z9 @. @* d$ S
67.             return status[webSocket.readyState];
    
68.     }
    
69. 
    0 w; v& J' ]0 L7 ]7 t9 c0 Y6 B
70. 
    
71. 
    
72. function start(event){
    
73.         console.log(webSocket);
    ; j9 Z6 x" g3 \  S$ B4 \1 T) V8 `3 ]
74.         var msg = document.getElementById('text').value;
    9 ^& P) b4 ^, H& o
75.         document.getElementById('text').value = '';
    $ D$ I4 b  d( M
76.         console.log("send:"+sockState());
    
77.         console.log("msg="+msg);
    
78.         webSocket.send("msg="+msg);
    / H, m# a' |9 a, o
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    
80.     };
      E; p2 \8 M3 _8 z) M6 g1 ~
81. 
    
82.     function close(event){
    7 d, @: [% b  e: @4 R( ^
83.         webSocket.close();
    
84.     }
    8 b* m2 f/ A8 {) I2 Y0 h
85. </script>
    
86. </body>
    9 ~6 j  D" c- [1 ^9 p
87. </html>

复制代码