php实现websocket实时消息推送

admin

php实现websocket实时消息推送
; ?2 }$ f4 M. l$ P% v

SocketService.php

1. <?php
   # p' w: G# I7 {$ }2 ?
2. /**
   2 }$ @5 F: `1 Q3 g5 E- m6 g
3. * Created by xwx
   
4. * Date: 2017/10/18
   . a+ Y( H7 a7 l7 w- a
5. * Time: 14:33
   * X/ \$ w8 j0 i+ B  R1 D+ {
6. */
   1 H* `4 n; |9 `, k' @7 [+ i: l
7. 
   2 ^3 D. n; h" ]: c* c2 q& D8 A
8. class SocketService
   8 Z2 Z+ X% `: T! ?
9. {
   
10.     private $address  = '0.0.0.0';
    
11.     private $port = 8083;
    $ ~# y. q4 l9 M  I' Q9 F
12.     private $_sockets;
    
13.     public function __construct($address = '', $port='')
    
14.     {
    / @2 Z* L# m- f5 s3 b# {. p6 H, Y
15.             if(!empty($address)){
    9 Y( s8 f6 E1 ^# }8 Z, f9 R
16.                 $this->address = $address;
    
17.             }
    6 D: \4 F7 ~+ Y: v+ e0 @& b+ u
18.             if(!empty($port)) {
    ! X$ a6 Q7 t0 q5 O1 X( _
19.                 $this->port = $port;
    
20.             }
    4 F0 T3 X9 z( ~- W* k3 C. W
21.     }
    
22. 
    4 f+ k+ L# G9 f0 ^' i
23.     public function service(){
    2 h5 K) q- ]. k$ O' I/ L* @
24.         //获取tcp协议号码。
    
25.         $tcp = getprotobyname("tcp");
    * a3 m! Y. Z  P6 \. `- j7 a1 R
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");
    / f4 k# e, H0 v4 _- k; m' D+ k
31.         }
    
32.         socket_bind($sock, $this->address, $this->port);
    + _8 e  u2 u3 l% v
33.         socket_listen($sock, $this->port);
    
34.         echo "listen on $this->address $this->port ... \n";
    $ i; ?, I  K' x' h5 l8 y9 P
35.         $this->_sockets = $sock;
    4 o! q0 I) }5 p
36.     }
    
37. 
    
38.     public function run(){
    , A8 }! B. b% N6 s+ ]
39.         $this->service();
    
40.         $clients[] = $this->_sockets;
    , s7 }  a8 L, o2 ~0 G% {# y
41.         while (true){
    
42.             $changes = $clients;
    ) w9 _  I6 f" C8 C3 F( B
43.             $write = NULL;
    7 }& _2 z8 l  D! \; R5 b
44.             $except = NULL;
    
45.             socket_select($changes,  $write,  $except, NULL);
    
46.             foreach ($changes as $key => $_sock){
    
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    ( B5 \& }% e/ ~3 D( {/ U4 ~) W
48.                     if(($newClient = socket_accept($_sock))  === false){
    ' ?0 N  m5 _+ v6 Q
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    ; c  X  H1 t* t$ o
50.                     }
    6 N" i, B. w) O8 b3 D# J. w
51.                     $line = trim(socket_read($newClient, 1024));
    
52.                     $this->handshaking($newClient, $line);
    ! |9 q; D& B6 j% t+ Q' S
53.                     //获取client ip
    
54.                     socket_getpeername ($newClient, $ip);
    # ^1 {& [+ C" s5 i7 y& X' `
55.                     $clients[$ip] = $newClient;
    9 s7 q2 ^/ m' V% N( o
56.                     echo  "Client ip:{$ip}   \n";
    
57.                     echo "Client msg:{$line} \n";
    
58.                 } else {
    4 P7 A# K( i# }) U$ k: K: B
59.                     socket_recv($_sock, $buffer,  2048, 0);
      l- e- c. ]; F- H- \: X/ }1 V
60.                     $msg = $this->message($buffer);
    
61.                     //在这里业务代码
    
62.                     echo "{$key} clinet msg:",$msg,"\n";
    8 O9 C& k0 L. }' g# b$ N
63.                     fwrite(STDOUT, 'Please input a argument:');
    , M/ z, Q3 e6 P1 T. n4 a
64.                     $response = trim(fgets(STDIN));
    ! f- M% X8 A- l7 h0 ^4 ]! x8 \
65.                     $this->send($_sock, $response);
    
66.                     echo "{$key} response to Client:".$response,"\n";
    
67.                 }
    
68.             }
    
69.         }
    0 x( m  k) \. x* S( q* K1 h6 t
70.     }
    
71. 
    
72.     /**
    : @. F" d9 I5 @; V
73.      * 握手处理
    
74.      * @param $newClient socket
    
75.      * @return int  接收到的信息
    / R. a9 h2 [& ~! V6 |
76.      */
    
77.     public function handshaking($newClient, $line){
    
78. 
    / S0 `: X0 z$ v" v& Z
79.         $headers = array();
    ! ~2 i8 O8 B, M. Q
80.         $lines = preg_split("/\r\n/", $line);
    
81.         foreach($lines as $line)
    5 x& c$ I4 S8 m4 Q; `1 u
82.         {
    
83.             $line = chop($line);
    
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    
85.             {
    9 |; D* k* e9 {( z
86.                 $headers[$matches[1]] = $matches[2];
    + }1 l; j! E% U" ~; E6 ^& _
87.             }
    # |6 \. ^) t* A: ?
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" .
    2 E& N( t0 u+ m% P5 k7 j
92.             "Upgrade: websocket\r\n" .
    
93.             "Connection: Upgrade\r\n" .
    6 G" ^& m& ?# j
94.             "WebSocket-Origin: $this->address\r\n" .
    " ]6 P. R# b+ D3 F
95.             "WebSocket-Location: ws://$this->address:$this->port/websocket/websocket\r\n".
    * i) }4 q0 e6 d: l
96.             "Sec-WebSocket-Accept:$secAccept\r\n\r\n";
    & I& m# Q0 E0 E2 b
97.         return socket_write($newClient, $upgrade, strlen($upgrade));
    0 T; O% N# p; V$ ]
98.     }
    8 g  K+ P; n1 c1 a9 P
99. 
    
100.     /**
     0 U$ G1 [6 f& V9 K
101.      * 解析接收数据
     
102.      * @param $buffer
     . j( k# y2 Z3 |2 n1 G, ?( x+ x
103.      * @return null|string
     
104.      */
     
105.     public function message($buffer){
     
106.         $len = $masks = $data = $decoded = null;
     8 v) t  f5 t: U7 L4 g# ]
107.         $len = ord($buffer[1]) & 127;
     6 x& G( U9 ~3 y' t: y4 Z. m
108.         if ($len === 126)  {
     
109.             $masks = substr($buffer, 4, 4);
     $ X7 M+ J; w* b0 O: H7 j
110.             $data = substr($buffer, 8);
     
111.         } else if ($len === 127)  {
     
112.             $masks = substr($buffer, 10, 4);
     
113.             $data = substr($buffer, 14);
       S4 U3 ~$ e0 g% N: G+ i
114.         } else  {
     3 i: r* P6 J4 a
115.             $masks = substr($buffer, 2, 4);
     
116.             $data = substr($buffer, 6);
     1 _  y0 b( Z" F' g
117.         }
     + W9 s/ }. x- \. a
118.         for ($index = 0; $index < strlen($data); $index++) {
     1 b' B0 ?0 u. _) X
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     $ A+ d" o, h7 N2 S7 D
120.         }
     / Z4 X- Q) Y3 ?! d2 a
121.         return $decoded;
     
122.     }
     9 F! t: X9 Z& y- x0 p- X% [
123. 
     
124.     /**
     ! S) I* f7 O- _4 {
125.      * 发送数据
     0 L7 T  T+ s8 Y! `7 C7 J
126.      * @param $newClinet 新接入的socket
     
127.      * @param $msg   要发送的数据
     $ u* E! G* h  p, W: R& U# J) Z
128.      * @return int|string
     1 l: S4 c9 `8 O+ |) ?
129.      */
     4 ]4 a8 A3 q8 W' P  m# f* G
130.     public function send($newClinet, $msg){
     
131.         $msg = $this->frame($msg);
     
132.         socket_write($newClinet, $msg, strlen($msg));
     
133.     }
     ( I, k5 _$ V; L1 F6 o
134. 
     
135.     public function frame($s) {
     
136.         $a = str_split($s, 125);
     2 W6 }* U+ r0 o, I; Y. A% H
137.         if (count($a) == 1) {
     
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     
139.         }
     
140.         $ns = "";
     ; R" n5 e+ J0 v) C
141.         foreach ($a as $o) {
     ' w" Y, h0 a% m9 D
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     
143.         }
     8 X- _$ ~6 i- |" D5 q0 e. L: T4 Z
144.         return $ns;
     
145.     }
     
146. 
     % {% B! z  R$ l4 H- f4 q& X
147.     /**
     
148.      * 关闭socket
     3 [" K% T  v; o: w
149.      */
     # d* r3 i" t! f9 k
150.     public function close(){
     
151.         return socket_close($this->_sockets);
     
152.     }
     
153. }
     ; ~6 n6 z& n% t& X  M$ T* l, n5 e
154. 
     
155. $sock = new SocketService();
     
156. $sock->run();
     
157. 
     

复制代码

web.html
8 u5 c% h2 t4 i

1. <!doctype html>
   
2. <html lang="en">
   6 K6 C" a5 z, j2 h
3. <head>
   3 Q5 E; S6 b) O9 F0 n
4.   <meta charset="UTF-8">
   
5.   <meta name="viewport" content="width=device-width,initial-scale=1, maximum-scale=1, user-scalable=no">
   + V" R0 k' N2 Z7 P
6.   <title>websocket</title>
   
7. </head>
   ; v# s6 ~! {/ _7 [  M" M4 `
8. <body>
   
9. <input id="text" value="">
   
10. <input type="submit" value="send" onclick="start()">
    
11. <input type="submit" value="close" onclick="close()">
    - U$ N7 S' G$ H4 A3 l8 c
12. <div id="msg"></div>
    
13. <script>
    
14. /**
    
15. 0：未连接
    
16. 1：连接成功，可通讯
    
17. 2：正在关闭
    5 U# m3 G. Z/ Z. a
18. 3：连接已关闭或无法打开
    
19. */
    
20. 
    
21.     //创建一个webSocket 实例
    
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    1 E+ q4 e  g$ a$ S
23. 
    
24. 
    
25.     webSocket.onerror = function (event){
    
26.         onError(event);
    ( a) V2 o; M& v: c2 ?2 `$ f# X- h
27.     };
    $ ^% z0 b, O$ [' T0 o, ]5 A/ J) K3 z
28. 
    . a: B4 M8 @2 I: c  [& E
29.     // 打开websocket
    
30.     webSocket.onopen = function (event){
    
31.         onOpen(event);
    3 X: B% l/ u8 X( `2 _0 U$ d( e6 H
32.     };
    
33. 
    $ ]' h& P5 W0 o( a/ ]
34.     //监听消息
    
35.     webSocket.onmessage = function (event){
    ( u0 W6 x0 |" Q2 h& R
36.         onMessage(event);
    
37.     };
    
38. 
    
39. 
    + ?/ R0 b0 `% Y  l% G3 _6 Z1 k- I* @- X
40.     webSocket.onclose = function (event){
    : ]! s4 B. b0 r0 }2 E2 {
41.         onClose(event);
    
42.     }
    * V7 d; m) N6 t' Y/ j% N
43. 
    
44.     //关闭监听websocket
    % f9 s! Z! Z  }  u
45.     function onError(event){
    ( A9 c  R5 y! M+ M! w9 p# O9 }
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
47.         console.log("error"+event.data);
    
48.     };
    
49. 
    ) h2 x) `0 k& B5 P* H
50.     function onOpen(event){
    
51.         console.log("open:"+sockState());
    
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    ) j% o0 a4 ]3 M# Q' l3 [
53.     };
    3 X# V% Y$ G7 c% N+ C
54.     function onMessage(event){
    
55.         console.log("onMessage");
    / e, A! H/ c5 E. U: x
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    1 _" r- r2 c8 x, @/ \2 D; l, O
57.     };
    
58. 
    
59.     function onClose(event){
    ; \% b3 o- P, O& R2 f) f
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
61.         console.log("close:"+sockState());
    , S' c( w9 f" i% l; r
62.         webSocket.close();
    
63.     }
    
64. 
    
65.     function sockState(){
    0 f/ i: T! u* o. S- E
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    8 l7 I1 I* U) g/ G: ~' y' t9 X/ X) g
67.             return status[webSocket.readyState];
    & I# K6 l) w5 n5 y3 F
68.     }
    ( g/ v4 `6 J5 s  M
69. 
    & G: `0 ~0 m2 Q+ j, Q
70. 
    # P; B' K; t; q/ {, S4 A4 F
71. 
    7 Z* T+ \5 U/ P( _  ?9 F4 z
72. function start(event){
    8 p& Q8 a# P$ u7 Y8 O; r2 F! f
73.         console.log(webSocket);
    # j% Q/ |9 R$ h; x6 ^7 a* C3 r; A  o
74.         var msg = document.getElementById('text').value;
    
75.         document.getElementById('text').value = '';
    . G# u8 N) t4 a3 B
76.         console.log("send:"+sockState());
    / t9 }& v* ^4 t: V
77.         console.log("msg="+msg);
    " w  r8 A& `7 B+ g
78.         webSocket.send("msg="+msg);
    : u3 o! X( P; l2 c2 @3 \
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    9 O$ i. ~' M4 G' D
80.     };
    
81. 
    * p$ t; X! V3 u/ \2 \1 _; Y( D
82.     function close(event){
    
83.         webSocket.close();
    * ~8 @2 @" O. E1 F# Z( x$ @
84.     }
    * D4 I6 v2 I8 |
85. </script>
    
86. </body>
    2 ]' W( Y, \5 [6 w5 E* j1 M1 l
87. </html>

复制代码

5 Y# T6 A) a; l; B