php实现websocket实时消息推送

admin

php实现websocket实时消息推送
* I, T: T- s! V, S* r6 {
- |# Z0 q' g0 B% q" @
SocketService.php

1. <?php
   9 |% m$ [" [1 ?* N& [1 V
2. /**
   ! o6 U. q) a% c( y+ s0 V3 S
3. * Created by xwx
   # Y& ?# Y& [8 x6 E# Q
4. * Date: 2017/10/18
   0 i  v9 r: g% o5 b
5. * Time: 14:33
   . M) c- I" i+ z9 v1 i
6. */
   . S( ?: L0 _0 I, O4 q
7. 
   
8. class SocketService
   
9. {
   
10.     private $address  = '0.0.0.0';
    
11.     private $port = 8083;
    : q$ r3 e8 G' N3 f# s  @" u  v* B$ e! F
12.     private $_sockets;
    " e# ~1 h- V2 l1 n6 h/ V: H
13.     public function __construct($address = '', $port='')
    
14.     {
    8 [0 E$ _# F  Q  R+ ~% `
15.             if(!empty($address)){
    % z& s! r' P! S" [$ K- S. H
16.                 $this->address = $address;
    & o  I3 m, Y( `8 t* D" A
17.             }
    
18.             if(!empty($port)) {
    
19.                 $this->port = $port;
    
20.             }
    
21.     }
    % W# C. f  o6 h0 G7 H
22. 
    
23.     public function service(){
      V* \4 p: \$ S3 r2 N5 d" a& z
24.         //获取tcp协议号码。
    & m. \1 J1 M. k$ W9 @
25.         $tcp = getprotobyname("tcp");
    
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    7 r; ^8 O. A  |
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    
28.         if($sock < 0)
    
29.         {
    7 w# P/ z$ o2 F$ Z+ N5 @( l( |
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    
31.         }
    6 B; I1 P, Y( x3 h8 r
32.         socket_bind($sock, $this->address, $this->port);
    9 h( m% l- i, d( u9 g# I! H
33.         socket_listen($sock, $this->port);
    7 x; A4 J( I4 ?' @3 L
34.         echo "listen on $this->address $this->port ... \n";
    
35.         $this->_sockets = $sock;
    
36.     }
    6 c$ z$ a4 i2 O( N! C) F4 R7 K+ W
37. 
    / E$ K% @# q1 Z0 F- |
38.     public function run(){
      ~, ]0 V% \) p: [9 t  C$ b
39.         $this->service();
    
40.         $clients[] = $this->_sockets;
    , n  n( @. E8 H) s
41.         while (true){
    
42.             $changes = $clients;
    
43.             $write = NULL;
    ; i) Z/ n7 D0 T- S' F2 t8 f3 e
44.             $except = NULL;
    
45.             socket_select($changes,  $write,  $except, NULL);
    5 S2 M" \3 F9 j0 \7 A7 l1 ^7 p% h
46.             foreach ($changes as $key => $_sock){
    3 S% o8 s* c8 F" D& P
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    
48.                     if(($newClient = socket_accept($_sock))  === false){
    
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    7 f/ d0 x) \/ u7 v
50.                     }
    5 ~: ]. A# B, n1 e* d! P
51.                     $line = trim(socket_read($newClient, 1024));
    7 k7 I1 A( Q. U! g/ ~
52.                     $this->handshaking($newClient, $line);
    
53.                     //获取client ip
    ! d, J( ^: Z; |' A
54.                     socket_getpeername ($newClient, $ip);
    
55.                     $clients[$ip] = $newClient;
    
56.                     echo  "Client ip:{$ip}   \n";
    5 ~. B7 t) {, q3 ^" A
57.                     echo "Client msg:{$line} \n";
    
58.                 } else {
    
59.                     socket_recv($_sock, $buffer,  2048, 0);
    
60.                     $msg = $this->message($buffer);
    0 B+ Y$ w5 m' o' v  T% T/ X/ |
61.                     //在这里业务代码
    
62.                     echo "{$key} clinet msg:",$msg,"\n";
    
63.                     fwrite(STDOUT, 'Please input a argument:');
    $ T7 b7 f* J+ {+ @# K& z
64.                     $response = trim(fgets(STDIN));
    4 I: k$ W' l! ~' S; ^
65.                     $this->send($_sock, $response);
    
66.                     echo "{$key} response to Client:".$response,"\n";
    " X; a4 x. ?& ~4 i
67.                 }
    
68.             }
    
69.         }
    
70.     }
    
71. 
    
72.     /**
    
73.      * 握手处理
    
74.      * @param $newClient socket
    & h' D0 T$ H; W' W
75.      * @return int  接收到的信息
    
76.      */
    
77.     public function handshaking($newClient, $line){
    ( Q3 j5 z+ |' o! ~& h: ~: X, _, N/ Z
78. 
    
79.         $headers = array();
    
80.         $lines = preg_split("/\r\n/", $line);
    
81.         foreach($lines as $line)
    $ z: x- x2 I9 M# W
82.         {
    
83.             $line = chop($line);
    
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    
85.             {
    
86.                 $headers[$matches[1]] = $matches[2];
      W8 I$ ^3 Z7 B# Z+ k! [
87.             }
    ; P0 R! T$ P0 s% w
88.         }
    " R6 ?4 n; m1 |4 t* n- Q
89.         $secKey = $headers['Sec-WebSocket-Key'];
    : }  o7 ~+ X6 Y6 x
90.         $secAccept = base64_encode(pack('H*', sha1($secKey . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11')));
      z0 {( l+ {' W, K
91.         $upgrade  = "HTTP/1.1 101 Web Socket Protocol Handshake\r\n" .
    2 V4 @' h1 U5 R7 M
92.             "Upgrade: websocket\r\n" .
    ! c+ B6 ~7 L" {7 t9 J- Y
93.             "Connection: Upgrade\r\n" .
    3 M* p, H0 W2 K1 q8 a
94.             "WebSocket-Origin: $this->address\r\n" .
    + F* @8 ^7 N1 x) O; I2 X
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.     }
    
99. 
    
100.     /**
     
101.      * 解析接收数据
     
102.      * @param $buffer
     * H. |8 J# G8 e/ v1 ~
103.      * @return null|string
     ' H9 `6 ~) L4 V1 J& i1 b
104.      */
     
105.     public function message($buffer){
     
106.         $len = $masks = $data = $decoded = null;
     
107.         $len = ord($buffer[1]) & 127;
     5 c% p4 ~# q3 \
108.         if ($len === 126)  {
     $ ^% P! j. |* v: \( k, K
109.             $masks = substr($buffer, 4, 4);
     2 b- {6 i4 Y, H1 o! s5 w
110.             $data = substr($buffer, 8);
     # i* L" D8 P/ `, C
111.         } else if ($len === 127)  {
     
112.             $masks = substr($buffer, 10, 4);
     
113.             $data = substr($buffer, 14);
     # n# @( Y  h' H, s" u1 g
114.         } else  {
     
115.             $masks = substr($buffer, 2, 4);
     
116.             $data = substr($buffer, 6);
     
117.         }
     * m# K. {' z; l  M" C% n3 M* @
118.         for ($index = 0; $index < strlen($data); $index++) {
     2 m3 }9 n+ J# B( z; A8 ^9 _
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     . }0 f! v+ Q/ S& b: @
120.         }
     
121.         return $decoded;
     
122.     }
     
123. 
     # G" N, s/ n6 p  z6 ^+ \
124.     /**
     5 c8 o6 }& e% g0 \% p
125.      * 发送数据
     
126.      * @param $newClinet 新接入的socket
     
127.      * @param $msg   要发送的数据
     . T. s) y* p7 r2 |: f% J( B$ v
128.      * @return int|string
     
129.      */
     - q7 V* P. q3 T9 Z1 x: b
130.     public function send($newClinet, $msg){
     
131.         $msg = $this->frame($msg);
     
132.         socket_write($newClinet, $msg, strlen($msg));
     
133.     }
     
134. 
     
135.     public function frame($s) {
       n/ ~) u9 F" O! i6 ]+ M
136.         $a = str_split($s, 125);
     , \( O) R' [' }& @& |
137.         if (count($a) == 1) {
     5 {- M& H$ i, [- \4 Z' ]
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     
139.         }
     1 K7 ~1 T3 D, K
140.         $ns = "";
     1 u6 q. k" M! E
141.         foreach ($a as $o) {
     
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     
143.         }
     
144.         return $ns;
     
145.     }
     8 z. d  \9 Y* W' a2 u
146. 
     % u! v6 |$ r4 J- |
147.     /**
     
148.      * 关闭socket
     
149.      */
     
150.     public function close(){
     / M( k- U/ N3 L
151.         return socket_close($this->_sockets);
     
152.     }
     
153. }
     ; I/ I! Y. w8 y0 D* j' H
154. 
     , {9 F  L" x& o( ~
155. $sock = new SocketService();
     ' `7 N# A7 A1 u. ^* u4 n6 }  H
156. $sock->run();
     
157. 
     

复制代码

web.html

1. <!doctype html>
     f) e/ `* N) R! E, k. `
2. <html lang="en">
   
3. <head>
   ) |" \8 C) u& Y2 t/ q+ ]4 q: a+ o) l. [1 V
4.   <meta charset="UTF-8">
   + I" E+ V& {! C8 V1 O
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>
   . Q" M0 t) Z; l6 f6 s0 `
9. <input id="text" value="">
   
10. <input type="submit" value="send" onclick="start()">
    
11. <input type="submit" value="close" onclick="close()">
    
12. <div id="msg"></div>
    
13. <script>
    9 x( v% m. I9 Y6 B: l% y% H: Q
14. /**
    . @- J1 ^6 b+ O" x  E, {6 j
15. 0：未连接
    
16. 1：连接成功，可通讯
    
17. 2：正在关闭
    + k0 u/ Q8 Y  E- [  J# e) G
18. 3：连接已关闭或无法打开
    
19. */
    
20. 
    
21.     //创建一个webSocket 实例
    
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    
23. 
    
24. 
    
25.     webSocket.onerror = function (event){
    
26.         onError(event);
    
27.     };
    : k) V/ P( z+ D* F; c9 t7 N6 w
28. 
    
29.     // 打开websocket
    ! k3 C5 V$ ~9 ]& v8 ^5 f5 x1 h) K: c' P: I
30.     webSocket.onopen = function (event){
    
31.         onOpen(event);
    " [* S4 F5 e  Z" V0 i
32.     };
    
33. 
    6 }# {/ `+ \+ a, Y$ T' B' C! ~
34.     //监听消息
    
35.     webSocket.onmessage = function (event){
    
36.         onMessage(event);
      F4 b9 L' `% b4 z
37.     };
    * @7 o' ?0 f; k6 ~. z5 I: t" U* r' g
38. 
    , ^( L: A& V6 O. q+ V; g8 S7 r
39. 
    6 ]5 a7 m* h( t# W
40.     webSocket.onclose = function (event){
    
41.         onClose(event);
    
42.     }
    
43. 
    / c  ]& B7 Z+ W6 w. p
44.     //关闭监听websocket
    # ?- O+ J8 w2 v# f7 @/ \
45.     function onError(event){
    ! f4 o6 c% ^) R2 K% A3 \
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    " q+ }, R& \: W$ h  Z$ {+ ?7 T
47.         console.log("error"+event.data);
    & Q' m) f/ z6 F4 d3 c  @
48.     };
    
49. 
    * e8 T! x" \4 ^. L6 x, C  Q. w
50.     function onOpen(event){
    
51.         console.log("open:"+sockState());
    : j: r7 q7 V( g/ b
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    1 Q% L5 }0 |3 e" Z) I) }4 |
53.     };
    
54.     function onMessage(event){
    2 U0 [% M: |5 e8 x' ^
55.         console.log("onMessage");
    * [& _, e0 b) Y: p2 D9 e, B
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    
57.     };
    
58. 
    
59.     function onClose(event){
    
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
61.         console.log("close:"+sockState());
    8 O: }; |8 p) \8 s9 e# p( u
62.         webSocket.close();
    $ g# p3 `5 [# j) u5 }; P
63.     }
    ! ~: x2 y& v! R& t
64. 
    ) N2 N( S$ R' [& Q- K
65.     function sockState(){
    
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    
67.             return status[webSocket.readyState];
    
68.     }
    
69. 
    : n" n/ T* Y: ?6 f
70. 
    9 m* S* C6 I* T0 j( b8 I$ f
71. 
    - G9 y$ W$ O. d! h
72. function start(event){
    ! L+ Z% N& P7 j+ t5 _3 ?
73.         console.log(webSocket);
    
74.         var msg = document.getElementById('text').value;
    * D+ A5 w9 R7 W7 B3 r
75.         document.getElementById('text').value = '';
    
76.         console.log("send:"+sockState());
    
77.         console.log("msg="+msg);
    / i5 }, f$ A9 K: d) @
78.         webSocket.send("msg="+msg);
    
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    
80.     };
    9 ^% O7 v( U4 J3 y
81. 
    9 v4 G0 ^1 a- a* _
82.     function close(event){
    5 Y6 f3 F4 b8 L$ x5 P; x8 Y
83.         webSocket.close();
    , m" x/ X7 m2 L6 `% r
84.     }
    - i* Y+ @7 Y1 L9 }+ j0 W7 T6 x  u7 W1 }
85. </script>
    
86. </body>
    
87. </html>

复制代码

- C5 `( K, y0 x% C+ A  H' V& I( I
4 M) g0 \4 L, d! V; H, ^