php实现websocket实时消息推送

admin

php实现websocket实时消息推送


* U8 o: I  ]( D/ ~" @SocketService.php
) I# S% i$ }' w4 t

1. <?php
   8 T: y; F8 p. d) `; y/ N
2. /**
   
3. * Created by xwx
   
4. * Date: 2017/10/18
   
5. * Time: 14:33
   5 H7 W% [" O- `8 f' P
6. */
   
7. 
     s6 m: u) B) q1 X
8. class SocketService
   8 s# x: T5 c. O3 Z7 P
9. {
   7 o, }2 M% J) n& Q: l. p& u( ?' l4 c
10.     private $address  = '0.0.0.0';
    # q) X0 U! G4 }6 h/ e8 q' @, W
11.     private $port = 8083;
    ( g% A* Z& y  ]& p( m
12.     private $_sockets;
    
13.     public function __construct($address = '', $port='')
    
14.     {
    6 B$ f' N$ m5 P; l0 V( S
15.             if(!empty($address)){
    # [) n, c# [( W: _: |# {3 Z" B
16.                 $this->address = $address;
    
17.             }
    
18.             if(!empty($port)) {
    
19.                 $this->port = $port;
    $ S( c; f( A7 e7 J0 j2 I
20.             }
    
21.     }
    ; B0 e0 X3 s, g, f; I
22. 
    
23.     public function service(){
    
24.         //获取tcp协议号码。
    6 I% c/ f6 l2 Y1 w% ?7 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)
    
29.         {
    # f6 J) G4 R3 E; o# `
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    
31.         }
    , }4 l5 N& W2 S% X6 X6 W  E
32.         socket_bind($sock, $this->address, $this->port);
    
33.         socket_listen($sock, $this->port);
    
34.         echo "listen on $this->address $this->port ... \n";
    4 ~  D: I5 j# i  d
35.         $this->_sockets = $sock;
    
36.     }
    
37. 
    % R, A2 E) L; ?4 q" F3 }" D" }
38.     public function run(){
    & v. b) \# b2 }0 B
39.         $this->service();
    5 j( \* X' H6 B9 d0 i
40.         $clients[] = $this->_sockets;
    
41.         while (true){
    
42.             $changes = $clients;
    
43.             $write = NULL;
    ) i& \+ _. v0 V
44.             $except = NULL;
    ) I! f- h4 x( t: y$ K+ U0 [/ G
45.             socket_select($changes,  $write,  $except, NULL);
    
46.             foreach ($changes as $key => $_sock){
    ) d9 M3 ]# ^9 `7 g+ s" j
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    $ ^) r. d2 X% X' a  J0 v; H3 k
48.                     if(($newClient = socket_accept($_sock))  === false){
    
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
      ?  t6 m$ O. V* [* f
50.                     }
    
51.                     $line = trim(socket_read($newClient, 1024));
    ( z9 v6 c; ^; R" ]" _6 s
52.                     $this->handshaking($newClient, $line);
    
53.                     //获取client ip
    $ C% m$ k1 K7 k2 t2 w" H
54.                     socket_getpeername ($newClient, $ip);
    ! ~# V  d8 x3 _
55.                     $clients[$ip] = $newClient;
    0 ~& X) R& v. n9 L( d
56.                     echo  "Client ip:{$ip}   \n";
    2 m: g: w( H5 W' Y7 Y& L* I& ^! j! h
57.                     echo "Client msg:{$line} \n";
    
58.                 } else {
    % b# F, {0 r: S; H) \4 D
59.                     socket_recv($_sock, $buffer,  2048, 0);
    & s$ R# [" i) `
60.                     $msg = $this->message($buffer);
    
61.                     //在这里业务代码
    ) P/ N' q+ I; ^$ B0 `
62.                     echo "{$key} clinet msg:",$msg,"\n";
    
63.                     fwrite(STDOUT, 'Please input a argument:');
    + u# m7 w- E! \+ {2 b) Z# J
64.                     $response = trim(fgets(STDIN));
    ; F% W0 v+ Q, V% \+ n
65.                     $this->send($_sock, $response);
    ! ^9 D- J' G$ V0 M0 K0 R" v) X" U
66.                     echo "{$key} response to Client:".$response,"\n";
    
67.                 }
    2 a) _1 j4 e; ~2 i
68.             }
    0 X# u! |  z$ l
69.         }
    9 ^- j/ k5 _! f5 N6 L
70.     }
    
71. 
    
72.     /**
    ( \* ?* ]8 C' A/ A6 c% L% f5 f
73.      * 握手处理
    - F' E2 I$ u$ Q8 ]* _
74.      * @param $newClient socket
    : E! R5 U, U& i' _/ G! q- |1 w4 r
75.      * @return int  接收到的信息
    6 m2 {. K3 D1 r( K
76.      */
    . P9 e) v. e0 M+ z" F; b
77.     public function handshaking($newClient, $line){
    ; q" k' W. _2 ^/ h' E
78. 
    8 D' x7 \2 q% \5 r: S' m! }
79.         $headers = array();
    / t/ a1 q; ^4 v
80.         $lines = preg_split("/\r\n/", $line);
    
81.         foreach($lines as $line)
    ( }. Y  r0 c+ b. N) K
82.         {
    
83.             $line = chop($line);
    7 J' v" O1 Z+ y( {' ]2 w9 N
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    
85.             {
    + m' X3 V( D7 b
86.                 $headers[$matches[1]] = $matches[2];
    
87.             }
    
88.         }
    
89.         $secKey = $headers['Sec-WebSocket-Key'];
    
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" .
    / X0 Z) u; U9 n* y6 X
92.             "Upgrade: websocket\r\n" .
    
93.             "Connection: Upgrade\r\n" .
    
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.     }
    2 t* ~$ }7 ~. M" F" F
99. 
    8 }0 }' Q" E9 d  }' X$ Z
100.     /**
     
101.      * 解析接收数据
     
102.      * @param $buffer
     
103.      * @return null|string
     " j# s$ [/ i; E2 u/ ^& j3 X. U- Y
104.      */
     ) B: w; [/ u( x0 D% ~) 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);
     + e$ O% o8 F" A% l
110.             $data = substr($buffer, 8);
     
111.         } else if ($len === 127)  {
     8 s' D' \/ j- Z
112.             $masks = substr($buffer, 10, 4);
     5 H5 ?" o' C# W5 ]- H; T# O. e$ R
113.             $data = substr($buffer, 14);
     2 m0 }4 m. i( |& s1 l
114.         } else  {
     % P8 I, Q9 @  V: S/ S* w3 w3 l0 k
115.             $masks = substr($buffer, 2, 4);
     
116.             $data = substr($buffer, 6);
     9 N1 b" G" w" w9 F/ B
117.         }
     # `( e4 `( o  q
118.         for ($index = 0; $index < strlen($data); $index++) {
     
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     
120.         }
     
121.         return $decoded;
     : Y1 h( E$ ~/ z
122.     }
     & ?" n) d8 E/ S$ O/ n/ I* C
123. 
     3 }5 S/ P0 _5 p. |1 i8 S% x6 Y% Z$ x4 a
124.     /**
     9 Q) x5 q* ?/ `0 U5 a/ d" W
125.      * 发送数据
     0 y/ M1 f, }+ t/ |
126.      * @param $newClinet 新接入的socket
     " J' R/ j7 z0 V% @+ m
127.      * @param $msg   要发送的数据
     
128.      * @return int|string
     
129.      */
     
130.     public function send($newClinet, $msg){
     , @! m2 ^2 ]5 [8 L! x5 B# j
131.         $msg = $this->frame($msg);
     
132.         socket_write($newClinet, $msg, strlen($msg));
     ) R0 H6 W" T3 ?: }0 b8 b8 _
133.     }
     4 ^7 V, L: \# G  M
134. 
     
135.     public function frame($s) {
     % H/ G$ I) O8 j& x. e1 A
136.         $a = str_split($s, 125);
     ) G; w  p- L- @+ O; `; R
137.         if (count($a) == 1) {
     0 D+ ^; N3 w# |  C) P. q; y. W
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     * }. z0 p, V$ I6 I- x+ i
139.         }
     
140.         $ns = "";
     ! Y$ E1 H$ t. _2 T# g
141.         foreach ($a as $o) {
     
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     0 \. k2 |4 X0 T5 s5 S* Z. a
143.         }
     / X) L: E" k# M8 Y9 w
144.         return $ns;
     
145.     }
     1 d5 h0 V2 v3 y2 X& A2 S6 H- I: T
146. 
     % ]+ G' e* p+ F" m. N, B
147.     /**
     
148.      * 关闭socket
     
149.      */
     3 d/ r$ O" z& O8 M5 J& C
150.     public function close(){
     
151.         return socket_close($this->_sockets);
     4 r7 q3 B% h. t4 P
152.     }
     # w1 B: k4 Q, P/ ]: ~0 k- S
153. }
     
154. 
     
155. $sock = new SocketService();
     $ J, F8 J! n. ^" R2 b* H6 i0 d
156. $sock->run();
     , |" w( V: n) Y7 t8 h) u
157. 
     

复制代码

web.html

1. <!doctype html>
   9 k+ v( l# G. d* T, g
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>
   3 Y2 Y* P* E4 b) n
7. </head>
   
8. <body>
   % F. u: K2 w$ v4 f" R# u$ K
9. <input id="text" value="">
     p7 T+ O6 D$ }, x5 ?8 o
10. <input type="submit" value="send" onclick="start()">
    
11. <input type="submit" value="close" onclick="close()">
    
12. <div id="msg"></div>
    9 L/ {9 x. m1 c8 ?2 _
13. <script>
    5 L. m% r# ]' t1 M* n
14. /**
    . L! e4 L( C* U) m
15. 0：未连接
    # C4 e- t5 U6 y2 C, o4 [" E) K
16. 1：连接成功，可通讯
    
17. 2：正在关闭
    2 o" Y6 |1 f) i2 G) Y
18. 3：连接已关闭或无法打开
    % s6 M) P6 l' K: x: u8 B6 l* f
19. */
    
20. 
    
21.     //创建一个webSocket 实例
    # g5 j* u! f( M, j/ i; r+ Q
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    
23. 
    2 x" K. I4 m7 Z" O7 p7 n
24. 
    & Y7 ]; _! i5 y$ b" n
25.     webSocket.onerror = function (event){
    
26.         onError(event);
    2 u) \8 P+ R, a# z' v! q
27.     };
    " y: a( U4 K$ t8 r3 R. W4 U
28. 
    3 q- I. w. U$ Y, Z! Q$ p& w
29.     // 打开websocket
      s4 K! t2 Q( U
30.     webSocket.onopen = function (event){
    
31.         onOpen(event);
    
32.     };
    
33. 
      }3 Y9 B0 _$ r
34.     //监听消息
    ( M1 a$ A3 ?4 l2 T
35.     webSocket.onmessage = function (event){
    
36.         onMessage(event);
    
37.     };
    & G& B7 p4 u0 h+ P8 u* D
38. 
    - k# o- Z& J# C6 e) R
39. 
    
40.     webSocket.onclose = function (event){
    
41.         onClose(event);
    
42.     }
    
43. 
    % n. P3 e. m; c$ R
44.     //关闭监听websocket
    
45.     function onError(event){
    
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    ; y2 Z* f! p# {) u& [4 i
47.         console.log("error"+event.data);
    
48.     };
    / p( {9 b3 {9 Y5 \& T, ~; P3 p
49. 
    0 W0 t- Y1 A: p
50.     function onOpen(event){
      _1 _" c) f2 F" C1 H) H) M. o
51.         console.log("open:"+sockState());
    
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    
53.     };
    . w3 {3 a0 m" V/ I. ?
54.     function onMessage(event){
    " ?6 R- ~) q/ e- O8 p# L
55.         console.log("onMessage");
    ( j- Y7 ^: P9 I7 ^/ a+ z
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    
57.     };
    * b+ t& @1 O" }9 D$ T+ L! F
58. 
    " s1 T6 N3 W' b2 j
59.     function onClose(event){
    
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
61.         console.log("close:"+sockState());
    
62.         webSocket.close();
    / q( H3 P! A2 f- k6 P
63.     }
    
64. 
    
65.     function sockState(){
    
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    
67.             return status[webSocket.readyState];
    ) ]6 I, R' U" U5 c' g$ @* V7 E
68.     }
    ; r" [( T; T! ~% X$ S" t7 V5 O
69. 
    
70. 
    
71. 
    ( U  k4 _' ?; H9 x- A. U% ]
72. function start(event){
    & @$ |4 T- g- {* f- [+ O
73.         console.log(webSocket);
    : d8 z, G' s9 X2 ^0 P
74.         var msg = document.getElementById('text').value;
    
75.         document.getElementById('text').value = '';
    
76.         console.log("send:"+sockState());
    # E! b/ B- H; S: ^- @& S
77.         console.log("msg="+msg);
    
78.         webSocket.send("msg="+msg);
    
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    
80.     };
    5 K2 V! o, s: E; O
81. 
    + c6 O! f) z3 B' H7 {
82.     function close(event){
    
83.         webSocket.close();
    
84.     }
    
85. </script>
    
86. </body>
    
87. </html>

复制代码

+ J) r/ J  A; `