php实现websocket实时消息推送

admin

php实现websocket实时消息推送


SocketService.php
( l0 f% ?" j6 t

1. <?php
   
2. /**
   
3. * Created by xwx
   : Z; N" j. T6 \  o* w  Z/ w3 a
4. * Date: 2017/10/18
   
5. * Time: 14:33
   
6. */
   6 [6 N1 i& [8 y" n0 o. m
7. 
   
8. class SocketService
   
9. {
   ) k1 H3 K2 C: U: [
10.     private $address  = '0.0.0.0';
    ! b8 ~+ @. d) ]9 |3 Q( Z  W
11.     private $port = 8083;
    
12.     private $_sockets;
    
13.     public function __construct($address = '', $port='')
    
14.     {
    ( f  D; N* `& g# C% _
15.             if(!empty($address)){
    
16.                 $this->address = $address;
    8 p& K: n4 ]* @
17.             }
    
18.             if(!empty($port)) {
    7 W, R, g$ i1 X/ t" V: m
19.                 $this->port = $port;
    ; O$ H; a3 r* d1 G+ {
20.             }
    ' v3 S0 n, C* u! `
21.     }
    
22. 
    ! B$ w" E# a' B1 ]$ d7 d
23.     public function service(){
    
24.         //获取tcp协议号码。
    8 x2 U. f# ]: v1 O# F, O( R. x
25.         $tcp = getprotobyname("tcp");
    ! p0 i) I1 f# U! z4 h9 i3 o& }
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    ! o8 s7 k: {8 O( s& U  k$ Q2 O
28.         if($sock < 0)
    
29.         {
    7 u" ~$ l7 X+ f" m3 N
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    
31.         }
    . Y# |! d- l6 T
32.         socket_bind($sock, $this->address, $this->port);
    
33.         socket_listen($sock, $this->port);
    1 W4 y& |% D0 ~, l
34.         echo "listen on $this->address $this->port ... \n";
    
35.         $this->_sockets = $sock;
    
36.     }
    
37. 
    
38.     public function run(){
    ) K4 M: f/ |5 _/ d9 n
39.         $this->service();
    
40.         $clients[] = $this->_sockets;
    . Z. Z- L2 F! Y1 G: X. |
41.         while (true){
    8 j, h$ W5 R/ M
42.             $changes = $clients;
    
43.             $write = NULL;
    0 V2 m) ]4 _; u0 `* _
44.             $except = NULL;
    
45.             socket_select($changes,  $write,  $except, NULL);
    3 `/ v, h% C: j% [
46.             foreach ($changes as $key => $_sock){
    ' P7 K2 u% W. W3 |7 }
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    5 q4 b: U- U1 f% Q
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));
    3 f9 c# z" c4 S3 B4 R$ K
52.                     $this->handshaking($newClient, $line);
    
53.                     //获取client ip
    " O. s) Y  D2 D! `$ O0 B
54.                     socket_getpeername ($newClient, $ip);
    6 L7 P# O# S6 N
55.                     $clients[$ip] = $newClient;
    
56.                     echo  "Client ip:{$ip}   \n";
    
57.                     echo "Client msg:{$line} \n";
    # _* t, m0 M8 {0 `' ?2 v' t
58.                 } else {
    
59.                     socket_recv($_sock, $buffer,  2048, 0);
    
60.                     $msg = $this->message($buffer);
    
61.                     //在这里业务代码
    
62.                     echo "{$key} clinet msg:",$msg,"\n";
    + c! M8 O5 T, D5 W( F9 C
63.                     fwrite(STDOUT, 'Please input a argument:');
    
64.                     $response = trim(fgets(STDIN));
    . e" @, N$ S4 h
65.                     $this->send($_sock, $response);
    
66.                     echo "{$key} response to Client:".$response,"\n";
    
67.                 }
      j+ ~. E. A' w; C9 O7 t
68.             }
    / W3 {; v# C* r" o, J( A
69.         }
    5 F* i, ~1 D- X$ S& J; k
70.     }
    
71. 
    
72.     /**
    5 `9 j- A- i; e1 t# b
73.      * 握手处理
    ! K2 _, O- X# C( Y
74.      * @param $newClient socket
    
75.      * @return int  接收到的信息
    
76.      */
    
77.     public function handshaking($newClient, $line){
    
78. 
    
79.         $headers = array();
    
80.         $lines = preg_split("/\r\n/", $line);
    
81.         foreach($lines as $line)
    
82.         {
    1 o* t; M/ W+ W
83.             $line = chop($line);
    
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    
85.             {
    
86.                 $headers[$matches[1]] = $matches[2];
    + Q* E. X$ G2 |* U5 y. [
87.             }
    
88.         }
    6 A4 C! q4 u1 t0 z1 f/ e3 C
89.         $secKey = $headers['Sec-WebSocket-Key'];
    * ~, ?6 j( h; R# g6 J& c( E
90.         $secAccept = base64_encode(pack('H*', sha1($secKey . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11')));
    ( n/ v( ]( ~8 G4 r
91.         $upgrade  = "HTTP/1.1 101 Web Socket Protocol Handshake\r\n" .
    
92.             "Upgrade: websocket\r\n" .
    . o; N2 G0 Q6 o( s9 p' s& ?6 V( E) 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";
      \% z/ r5 P; D+ y5 {6 S' Z' t& j
97.         return socket_write($newClient, $upgrade, strlen($upgrade));
    # q+ b. S. b$ n* R: x8 g: @
98.     }
    8 L7 X# @9 i9 h- E% f3 g
99. 
    
100.     /**
     
101.      * 解析接收数据
     ! M1 E, Q5 ?/ `4 N. H  Q* l0 R3 V' R
102.      * @param $buffer
     
103.      * @return null|string
     
104.      */
     ; [4 w( r  ^) a' f
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);
     
110.             $data = substr($buffer, 8);
     0 r2 \, J% {1 L. R/ q) ~: z
111.         } else if ($len === 127)  {
     # }  M: a5 A/ Q* {
112.             $masks = substr($buffer, 10, 4);
     
113.             $data = substr($buffer, 14);
     
114.         } else  {
     4 Z& Q# o, E  X+ J( L
115.             $masks = substr($buffer, 2, 4);
     & a4 ^- }7 M/ w. n4 s+ |3 {
116.             $data = substr($buffer, 6);
     ! b! Q0 @* C$ a+ k, R
117.         }
     % r! p9 V  z; W, ^4 J2 P8 d5 p  T
118.         for ($index = 0; $index < strlen($data); $index++) {
     
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     
120.         }
     
121.         return $decoded;
     
122.     }
     
123. 
     
124.     /**
     
125.      * 发送数据
     
126.      * @param $newClinet 新接入的socket
     
127.      * @param $msg   要发送的数据
     
128.      * @return int|string
     % B8 E- w# H# }; z7 x( N7 P4 z5 h
129.      */
     $ G( J# S+ y. V0 F! b* p, @. \! O
130.     public function send($newClinet, $msg){
     5 i% e: x0 Q, b8 m
131.         $msg = $this->frame($msg);
     
132.         socket_write($newClinet, $msg, strlen($msg));
     - o8 D  e4 E! v! h) g4 k/ ], M- W% f
133.     }
     0 a# I. V9 n0 {0 {
134. 
     0 \4 `1 J8 r/ b: c1 M8 E& C+ d
135.     public function frame($s) {
     , t  a2 o! R: _* U
136.         $a = str_split($s, 125);
     ( Q2 R6 E; G: r+ T& T# h+ H0 F0 {
137.         if (count($a) == 1) {
     
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     
139.         }
     9 [8 |: R4 q7 r5 k
140.         $ns = "";
     
141.         foreach ($a as $o) {
     5 w1 m  n0 m' I- Q1 j( l9 |
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     0 ~  s2 y, W% A
143.         }
     " j0 f' ]+ S1 R0 Y
144.         return $ns;
     8 ~: x8 }& w) l. L
145.     }
     
146. 
     
147.     /**
     6 m" k$ @' Z6 d: `& P7 C4 Q" K
148.      * 关闭socket
     ' B5 T6 v, X$ y" G7 ]5 f
149.      */
     * I( c# i* ~! R8 {
150.     public function close(){
     " T0 b# X1 N) \, r
151.         return socket_close($this->_sockets);
     
152.     }
     8 ]& S# K4 h$ i7 O  s
153. }
     8 T2 X: p4 L  r
154. 
     
155. $sock = new SocketService();
     
156. $sock->run();
     , @: @, O* t: R8 a0 q5 O# e
157. 
     6 n. p0 r/ O4 K' S  l6 t0 N

复制代码

web.html

1. <!doctype html>
   6 e% ^. j4 v$ S, T
2. <html lang="en">
   7 g$ q5 i, h. z& x  B! g) {9 D
3. <head>
   8 ]- p" x! O) i9 {2 E
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>
   6 }+ m, d& H8 ]: X# v, d& ?
7. </head>
   % i, w, u9 x- `+ n' j/ n, B
8. <body>
   . C/ [" ~: E, @$ k# O8 f- E4 p
9. <input id="text" value="">
   
10. <input type="submit" value="send" onclick="start()">
    ! [- X  }& V4 W3 Z' W
11. <input type="submit" value="close" onclick="close()">
    
12. <div id="msg"></div>
    ( n# p2 y0 |' b+ ?1 f% T
13. <script>
    
14. /**
    ( h8 ?- K2 N" m1 i  n1 l
15. 0：未连接
    ! E( ]) o7 d. R- b; o: z' [1 v8 K
16. 1：连接成功，可通讯
    / A9 F# u4 x$ v
17. 2：正在关闭
    
18. 3：连接已关闭或无法打开
    , `! i% g, l7 u' R2 g% l
19. */
      w2 M' Y& l( f# E, T8 {
20. 
    
21.     //创建一个webSocket 实例
    " d: V: J% U0 U4 o3 S+ g
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    
23. 
    0 q; B# _8 v" I6 u9 i6 \; }
24. 
    
25.     webSocket.onerror = function (event){
    
26.         onError(event);
    ! U3 @+ m0 F- m+ @$ N1 b
27.     };
    
28. 
    
29.     // 打开websocket
    
30.     webSocket.onopen = function (event){
    
31.         onOpen(event);
    
32.     };
    2 o' q& @( V* X$ s2 Y7 `) |% B
33. 
    
34.     //监听消息
    
35.     webSocket.onmessage = function (event){
    
36.         onMessage(event);
    7 @' r! x+ T" ?% E" B
37.     };
    8 e& R0 m. Y- _0 E& s
38. 
      Q+ w4 }7 O0 ^1 b+ R
39. 
    + n& [1 p3 l# E' P7 q$ W
40.     webSocket.onclose = function (event){
    0 [4 L9 n6 D4 F6 m1 o
41.         onClose(event);
    
42.     }
    $ y3 a  ?; ^% e9 k/ f' ^! H
43. 
    . B& _6 p6 ?- E/ {8 f( w7 i
44.     //关闭监听websocket
    
45.     function onError(event){
    
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
47.         console.log("error"+event.data);
    : K& W6 X$ R. i, F6 u( W, R
48.     };
    
49. 
    ( M: }& s; p# a8 x
50.     function onOpen(event){
    4 F0 P) x7 G& F* A. }" X3 G, e
51.         console.log("open:"+sockState());
    % {3 u! w' D" N
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    
53.     };
    1 a7 Z2 J* a/ v; G8 a0 i$ n* Q1 B/ e
54.     function onMessage(event){
    : Z/ b/ t- C/ |/ X  p- H& b
55.         console.log("onMessage");
    
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    
57.     };
    : N$ a8 c! s/ l
58. 
    # ], _) m& E* v6 ?
59.     function onClose(event){
    ! Q! }8 ?4 f( `! b
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
61.         console.log("close:"+sockState());
    2 C6 C5 T$ G$ p
62.         webSocket.close();
    
63.     }
    
64. 
    
65.     function sockState(){
    
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    
67.             return status[webSocket.readyState];
    ! h* `% d  X% {3 A5 l- C
68.     }
    
69. 
    + ^& P1 W  f; Y. D; M# c
70. 
    : c4 N0 z3 ]% P4 b
71. 
    ' T0 H3 T" ?9 S4 F
72. function start(event){
    
73.         console.log(webSocket);
    % a, C' X' E, u1 @9 T$ t  X* y
74.         var msg = document.getElementById('text').value;
    
75.         document.getElementById('text').value = '';
    
76.         console.log("send:"+sockState());
    
77.         console.log("msg="+msg);
    6 S! C1 y0 ]2 V4 Q0 p
78.         webSocket.send("msg="+msg);
    & ~" X, W' f" j$ V! j
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    $ h. f$ _/ L' L* Y' H( s, a
80.     };
    ; M5 S2 ~* a. G
81. 
    
82.     function close(event){
    ' ^# V0 P3 {6 s- ]9 c
83.         webSocket.close();
    
84.     }
    + q& B( K' J5 @% V
85. </script>
    , O& l1 n9 H4 D6 o1 ^: Z
86. </body>
    $ k2 o- t# \& w! g+ Z' c
87. </html>

复制代码

4 @' L4 V: @. a7 X8 t' L! C, N