php实现websocket实时消息推送

admin

php实现websocket实时消息推送
' }* f% D- p, k

& B  u* M; \6 ~2 z% U7 BSocketService.php

1. <?php
   
2. /**
   6 G1 I" B* [8 ]
3. * Created by xwx
   
4. * Date: 2017/10/18
   ) l$ Y# @' N0 v  ]# S9 I) }% ~
5. * Time: 14:33
   # ~7 U' E, ~5 f, d/ D9 Y
6. */
   
7. 
   & r5 Q3 L* s! D
8. class SocketService
   
9. {
   $ f; ~3 h, ^2 x/ p
10.     private $address  = '0.0.0.0';
    " u: s( v3 H+ `8 M
11.     private $port = 8083;
    
12.     private $_sockets;
    
13.     public function __construct($address = '', $port='')
    + X+ Q4 |8 G' p  \% R
14.     {
    
15.             if(!empty($address)){
      i1 t" o. J- j1 |. Q, O2 l
16.                 $this->address = $address;
    0 G( Y. o1 [: E
17.             }
    
18.             if(!empty($port)) {
    
19.                 $this->port = $port;
    
20.             }
    ( S; }5 Y6 y$ o) v' l
21.     }
    5 x& y: K( v1 k& [( W2 v2 q/ ?
22. 
    
23.     public function service(){
    . z: ]- }6 a- |: l& R8 P5 M
24.         //获取tcp协议号码。
    5 H/ E: H) L1 Y- M5 k3 b  S
25.         $tcp = getprotobyname("tcp");
    / c) K8 z, |" G# D2 w9 [2 D
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");
    8 i+ |, ?& `  N+ I( F
31.         }
    
32.         socket_bind($sock, $this->address, $this->port);
    ' a9 s! S4 |2 Q1 b. O0 h
33.         socket_listen($sock, $this->port);
    
34.         echo "listen on $this->address $this->port ... \n";
    
35.         $this->_sockets = $sock;
    
36.     }
    9 c5 F6 H* X7 M! Q* ?
37. 
      c* j  ^) }: ~( _) y) i
38.     public function run(){
    + T3 v! Q/ n; J* w7 C
39.         $this->service();
    # y1 k1 e( z* h% c( Z: b7 }' r* i
40.         $clients[] = $this->_sockets;
    ) ~5 O" k) ?/ Q0 |- }
41.         while (true){
    
42.             $changes = $clients;
    # z. J8 X$ V% R# j7 Q/ h
43.             $write = NULL;
    & C* h) `* B5 b' O# Q8 X1 ^+ z
44.             $except = NULL;
    
45.             socket_select($changes,  $write,  $except, NULL);
    + g5 o2 h& f- t% A
46.             foreach ($changes as $key => $_sock){
    
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    
48.                     if(($newClient = socket_accept($_sock))  === false){
    
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    
50.                     }
    " s4 X; Y  R+ ], h
51.                     $line = trim(socket_read($newClient, 1024));
    9 r* v: X. U- G. @  v3 i! M
52.                     $this->handshaking($newClient, $line);
    : p( `  r: @4 k/ G# B
53.                     //获取client ip
    
54.                     socket_getpeername ($newClient, $ip);
    
55.                     $clients[$ip] = $newClient;
    5 U. a' ^% v7 G
56.                     echo  "Client ip:{$ip}   \n";
    ! i! S5 g5 z0 u) u. }6 P/ U( f  ]
57.                     echo "Client msg:{$line} \n";
    
58.                 } else {
    
59.                     socket_recv($_sock, $buffer,  2048, 0);
    4 \' F- Q, E# v6 w+ d
60.                     $msg = $this->message($buffer);
    - j5 p+ x7 \" v2 [$ J. h1 ^8 A' R3 C
61.                     //在这里业务代码
    
62.                     echo "{$key} clinet msg:",$msg,"\n";
    ! a, t; ?0 M& M9 d9 d0 n  L. m5 b
63.                     fwrite(STDOUT, 'Please input a argument:');
    6 }! h3 ]+ Q# w/ h- h) s" |
64.                     $response = trim(fgets(STDIN));
    5 A+ q; _$ u: w. O! t
65.                     $this->send($_sock, $response);
    
66.                     echo "{$key} response to Client:".$response,"\n";
    
67.                 }
    + m2 B* j+ ]+ t6 A' Y* [
68.             }
    
69.         }
    
70.     }
    
71. 
    
72.     /**
    5 P( |+ M' ~7 l4 k  n) e/ A
73.      * 握手处理
    , s+ c* K( i- k; x0 d4 y6 l
74.      * @param $newClient socket
    
75.      * @return int  接收到的信息
    
76.      */
    ) a2 o% t: Q9 i; @: v/ ^% z; C
77.     public function handshaking($newClient, $line){
    + b5 B1 I; H2 Z0 b8 U
78. 
    % ?2 y3 v$ y/ f3 J0 O' j) ?* [
79.         $headers = array();
    $ U, J- }) v! O8 @  M; G- ]
80.         $lines = preg_split("/\r\n/", $line);
    + B4 J! i  r- T* F
81.         foreach($lines as $line)
    
82.         {
    
83.             $line = chop($line);
    / R# l& {5 r2 |
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    & z% k3 ~3 Q+ a3 c+ h. D
85.             {
    
86.                 $headers[$matches[1]] = $matches[2];
    0 x" w4 @4 o9 U
87.             }
    
88.         }
    0 V- d4 `7 D. \5 Y3 g: p% x" v
89.         $secKey = $headers['Sec-WebSocket-Key'];
    
90.         $secAccept = base64_encode(pack('H*', sha1($secKey . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11')));
    , g1 b2 o' J7 b0 m, d6 m* g8 p3 [: ?
91.         $upgrade  = "HTTP/1.1 101 Web Socket Protocol Handshake\r\n" .
    4 E5 S* t! v) ]5 h6 _9 O
92.             "Upgrade: websocket\r\n" .
    ' B2 C1 C; Q! ~( X7 H$ R
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.     }
    ! H  t- I$ o. m3 ~
99. 
    
100.     /**
     ( Q* r# ]9 B  [: O" y
101.      * 解析接收数据
     
102.      * @param $buffer
     6 i( Y$ i6 A4 v, c. |! {* x3 x1 @* k1 F
103.      * @return null|string
     - P5 y" I6 X- X# P7 o
104.      */
     1 }* }& h- A) c+ a: K+ b5 Y7 S
105.     public function message($buffer){
     
106.         $len = $masks = $data = $decoded = null;
     : J  \. k& s' q/ \% x- J" D5 W
107.         $len = ord($buffer[1]) & 127;
     
108.         if ($len === 126)  {
     * Z0 D" ^# L6 b% Z3 N
109.             $masks = substr($buffer, 4, 4);
     ! A+ p  G+ X( c+ y' Y0 d
110.             $data = substr($buffer, 8);
     
111.         } else if ($len === 127)  {
     
112.             $masks = substr($buffer, 10, 4);
     . V. {* c6 r$ H8 N  J( ?2 W5 e7 I
113.             $data = substr($buffer, 14);
     5 B+ |/ A! x' @: L$ {
114.         } else  {
     . o6 ]+ Y9 c. S: B+ E) d0 y. G
115.             $masks = substr($buffer, 2, 4);
     
116.             $data = substr($buffer, 6);
     
117.         }
     5 f* p9 d4 M6 B1 y
118.         for ($index = 0; $index < strlen($data); $index++) {
     
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     7 |4 z; Q- r; V4 a3 y$ h6 w  E6 L9 _
120.         }
     , n5 p" k6 l9 W. h- e2 L% r
121.         return $decoded;
     
122.     }
     
123. 
     
124.     /**
     
125.      * 发送数据
     : i0 R  o4 K) ?9 S3 _; m" _2 }" m. n! }
126.      * @param $newClinet 新接入的socket
     + N1 j6 ^+ W6 d" c4 Q' ?/ w
127.      * @param $msg   要发送的数据
     : [" g# v! @5 p5 v' B. [8 l
128.      * @return int|string
     ( {1 c3 g  W6 z( V- [  ]  E7 V9 T
129.      */
     
130.     public function send($newClinet, $msg){
     9 t; z7 X! r6 i6 |9 q  x7 P+ U
131.         $msg = $this->frame($msg);
     ( l5 a8 K' B% k/ E/ s5 P
132.         socket_write($newClinet, $msg, strlen($msg));
     
133.     }
     & e$ L1 m$ Y0 ?, G9 P2 W  W
134. 
     
135.     public function frame($s) {
     # `( Y5 N! N/ \1 t8 H7 o
136.         $a = str_split($s, 125);
     
137.         if (count($a) == 1) {
     / A6 n/ e* F: Q" L' ^
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     
139.         }
     
140.         $ns = "";
     
141.         foreach ($a as $o) {
     
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     3 ~8 F* p+ N$ U$ {# f+ U/ `
143.         }
     
144.         return $ns;
     * H& a- c1 X* N  K- Q% W8 o3 ^
145.     }
     / r( A" y: E0 m* X3 z
146. 
     
147.     /**
     
148.      * 关闭socket
     
149.      */
     
150.     public function close(){
     
151.         return socket_close($this->_sockets);
     
152.     }
     5 H) S! M( a1 c2 I$ u  `1 H
153. }
     9 J$ O% b6 \; Y! w+ W& ~% L8 l
154. 
     . J9 G4 N, T+ G6 g5 ^" y
155. $sock = new SocketService();
     * a5 X8 r/ p- Y5 q4 s7 s
156. $sock->run();
     ; E3 Y* q6 T4 M# Q; r9 [
157. 
     / M  g3 p! E& T# b* L" f/ C" K

复制代码

web.html
" V' T4 K2 |! K7 Z

1. <!doctype html>
   
2. <html lang="en">
   
3. <head>
   
4.   <meta charset="UTF-8">
   0 \5 D0 A7 b$ [  L' o4 m# g( G
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>
   
9. <input id="text" value="">
   . A5 o$ ^& T; |  r4 J
10. <input type="submit" value="send" onclick="start()">
    1 U) J/ L+ }- U+ G0 ^
11. <input type="submit" value="close" onclick="close()">
    
12. <div id="msg"></div>
    
13. <script>
    ; C+ ?6 n* n" G
14. /**
    . Z) x! @1 k) ?" B2 {& _( _
15. 0：未连接
    
16. 1：连接成功，可通讯
    
17. 2：正在关闭
    " T( N3 g) t( N% v) C
18. 3：连接已关闭或无法打开
    0 o0 e- |) T4 Z5 P& {
19. */
    5 M( A  W! L, E
20. 
    
21.     //创建一个webSocket 实例
    ; A7 H* }+ e( v- W  f( s5 i6 K3 U* Z
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    # o, k* w3 t. A% M$ D, T- U( b3 o* o
23. 
    5 z$ e9 T# B/ d3 `3 k7 b1 O& v; J
24. 
    
25.     webSocket.onerror = function (event){
    & G0 G+ p7 W/ A
26.         onError(event);
    
27.     };
    
28. 
    , Y" X$ C) V* N4 g7 P9 C' Z
29.     // 打开websocket
    
30.     webSocket.onopen = function (event){
    
31.         onOpen(event);
    5 h8 |0 m9 U/ ~5 B1 c4 @) R. G/ a
32.     };
    
33. 
    
34.     //监听消息
    6 B$ }; m- ~4 Y' M
35.     webSocket.onmessage = function (event){
    
36.         onMessage(event);
    ( T+ k+ X! l5 N. x: z
37.     };
    
38. 
    * r( k  p" Y' L& s/ L- {( M
39. 
    
40.     webSocket.onclose = function (event){
    
41.         onClose(event);
    2 M. G& s+ L% a" ~. t: j2 a0 x* `
42.     }
    
43. 
    
44.     //关闭监听websocket
    
45.     function onError(event){
    
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    $ O8 q% ]( ]+ }0 O" q
47.         console.log("error"+event.data);
      u' H4 @0 `! `  |$ V
48.     };
    
49. 
    * O' Y/ h9 K3 S
50.     function onOpen(event){
      H& _/ N  I$ T: ^, H, i9 X
51.         console.log("open:"+sockState());
    
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    ( {0 a3 C1 c! P0 y1 G
53.     };
    7 i/ H0 v, ^7 L' `5 c8 e' x7 c3 S0 M, a
54.     function onMessage(event){
    
55.         console.log("onMessage");
    
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    
57.     };
    
58. 
    6 A, M% y% t! L+ f! M/ o% m7 [2 d7 k
59.     function onClose(event){
    " Z7 S8 x  W) W' n$ ~, ?' Z* w
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    3 `7 v+ o) C1 ]
61.         console.log("close:"+sockState());
    
62.         webSocket.close();
    
63.     }
    2 k3 ~0 ]1 c+ ^) N, u. N( @
64. 
    / S6 [0 W8 T! P) q9 E
65.     function sockState(){
    
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    
67.             return status[webSocket.readyState];
    
68.     }
    , m# M4 i; C' w7 E1 F
69. 
    , q+ l( v) l$ P; w
70. 
    ( n6 L1 l% E7 s4 J# I7 F" \
71. 
    1 j# Y. @# V* d* w0 m7 I
72. function start(event){
    ' x# X$ h8 s1 K. w
73.         console.log(webSocket);
    4 `/ X* L5 N  h; S. I" W
74.         var msg = document.getElementById('text').value;
    " r# O+ I$ b5 M6 V
75.         document.getElementById('text').value = '';
    3 b; M" @) ]/ d  t
76.         console.log("send:"+sockState());
    
77.         console.log("msg="+msg);
    . i' x1 X# H6 |0 _3 l
78.         webSocket.send("msg="+msg);
    
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    
80.     };
    
81. 
    8 _- V/ G3 E- s6 q! l6 q
82.     function close(event){
    
83.         webSocket.close();
    0 Z* S( v3 H. }8 V, W- f
84.     }
    
85. </script>
    ) `0 G( B% H$ z( y1 y. }! v
86. </body>
    3 J2 ^5 t! A1 h0 j4 b; M( J2 n
87. </html>

复制代码

; l, ?# W: T2 z