php实现websocket实时消息推送

admin

php实现websocket实时消息推送
% @9 Z9 ?$ }( e) Z+ b, c

SocketService.php
: i" j4 _  {  H2 h

1. <?php
   
2. /**
   
3. * Created by xwx
   
4. * Date: 2017/10/18
   
5. * Time: 14:33
   
6. */
   # \5 [, ^; U* n  Z9 f% F$ X
7. 
   ) ~" G- s4 U4 \0 p
8. class SocketService
   ! g7 y; j0 p& l1 J2 s
9. {
   ) A7 Q) Q3 c' E% P9 b& z
10.     private $address  = '0.0.0.0';
    
11.     private $port = 8083;
    
12.     private $_sockets;
    
13.     public function __construct($address = '', $port='')
    " C3 t1 Z" x/ z# x9 B# ]
14.     {
    3 G% Y* u7 v2 A
15.             if(!empty($address)){
    4 T7 B9 d# s/ F4 h) e+ m- t" ^
16.                 $this->address = $address;
    
17.             }
    $ p, \: `7 u, U* f6 t6 f8 k/ [# E) A
18.             if(!empty($port)) {
    1 u9 B4 g; H6 v$ n6 ?% c. v
19.                 $this->port = $port;
    0 s6 s, P3 z# F: @9 m7 V" a2 E
20.             }
    
21.     }
    # F/ ^# x) O8 H9 i7 G& K8 a) W
22. 
    
23.     public function service(){
    $ U: _4 }, I0 m( N0 F% v
24.         //获取tcp协议号码。
    
25.         $tcp = getprotobyname("tcp");
      C8 q7 i) P6 ?2 J* Z  ?6 B! h1 n7 m
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    " Q0 h5 @% ^6 Z1 T# V9 y& j  R
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    
28.         if($sock < 0)
    * ~3 [$ T6 ~8 A3 `0 |" V
29.         {
    
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    
31.         }
    
32.         socket_bind($sock, $this->address, $this->port);
    ' E* k7 a# C6 m. O' _# p
33.         socket_listen($sock, $this->port);
    . o* W5 Q3 ~- q$ E9 H
34.         echo "listen on $this->address $this->port ... \n";
    
35.         $this->_sockets = $sock;
    ! {- x- d9 R6 |
36.     }
    
37. 
    
38.     public function run(){
    % O- s3 M* H- R) N4 |( f7 N
39.         $this->service();
    
40.         $clients[] = $this->_sockets;
    $ d, \: z) K4 f3 S# }# U
41.         while (true){
    
42.             $changes = $clients;
    - S! K9 Z- R+ w
43.             $write = NULL;
    2 P+ f6 v2 W1 }" d: S! p4 y* K
44.             $except = NULL;
    
45.             socket_select($changes,  $write,  $except, NULL);
    
46.             foreach ($changes as $key => $_sock){
    . ]1 k1 N+ Z% {. e2 ~8 G% M3 G
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    
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));
    * \8 w- }9 H6 k1 }6 z; [& l
52.                     $this->handshaking($newClient, $line);
    & T) F' o" n0 @# E. H# m% a; N7 v
53.                     //获取client ip
    7 o% O# `. e! ~$ b+ H! O' p8 m
54.                     socket_getpeername ($newClient, $ip);
    9 v# }1 S; q7 }
55.                     $clients[$ip] = $newClient;
    
56.                     echo  "Client ip:{$ip}   \n";
    
57.                     echo "Client msg:{$line} \n";
    
58.                 } else {
    ) P. g& m3 P: I6 J1 [5 W" O
59.                     socket_recv($_sock, $buffer,  2048, 0);
    5 @5 J! {8 L4 Y8 J  T
60.                     $msg = $this->message($buffer);
    
61.                     //在这里业务代码
    2 _/ s& W# ]6 u: b7 s  w
62.                     echo "{$key} clinet msg:",$msg,"\n";
    1 Q. ^! H  e" p4 H. r& @
63.                     fwrite(STDOUT, 'Please input a argument:');
    
64.                     $response = trim(fgets(STDIN));
    / F- v0 S- h9 b( H5 K# }: L
65.                     $this->send($_sock, $response);
    ! h# @2 m% w2 N1 Y& V4 g
66.                     echo "{$key} response to Client:".$response,"\n";
    . a) ~5 l: W* ?3 t% B/ ^, Y
67.                 }
    
68.             }
    
69.         }
    2 d" I; A6 @: A0 N  Y8 d
70.     }
    $ R: d$ `- k5 q* }& Y& X4 c7 f
71. 
    ) F2 S2 J  @9 Z3 Z& h8 j; F' b
72.     /**
    
73.      * 握手处理
    
74.      * @param $newClient socket
    
75.      * @return int  接收到的信息
    0 x! w2 L/ o. H" F( A( M/ \- T- k4 I
76.      */
    - Z  _- z; V0 I* }
77.     public function handshaking($newClient, $line){
    2 g- a' C0 k: q
78. 
    & ?6 ~6 l0 r" m2 y( u2 D! @! D# _0 ]/ y
79.         $headers = array();
    
80.         $lines = preg_split("/\r\n/", $line);
    7 N) I. z- a# X
81.         foreach($lines as $line)
    % w1 ?  J; f% Z! l- I" @2 P
82.         {
    ( B0 F2 \2 ^" l3 E( D# E
83.             $line = chop($line);
    : X% h. Y# {, ~9 O% l8 s8 Z
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    ' k2 ^- a! Q/ a. F" V0 c9 X
85.             {
    * \/ T# B& \8 a6 p4 i% [0 j
86.                 $headers[$matches[1]] = $matches[2];
    
87.             }
    
88.         }
    
89.         $secKey = $headers['Sec-WebSocket-Key'];
    & e& x9 R5 _/ p. O
90.         $secAccept = base64_encode(pack('H*', sha1($secKey . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11')));
    0 u' I3 s) F7 s& P1 x0 D7 ]4 k
91.         $upgrade  = "HTTP/1.1 101 Web Socket Protocol Handshake\r\n" .
    $ _5 T+ Q  o# `; G8 t9 {# {
92.             "Upgrade: websocket\r\n" .
    ( ^) S4 K8 ~3 d6 }
93.             "Connection: Upgrade\r\n" .
    # K5 S9 S% i' S* @& E5 T5 J
94.             "WebSocket-Origin: $this->address\r\n" .
    - @; K. K/ w, z- k" ~; m7 n
95.             "WebSocket-Location: ws://$this->address:$this->port/websocket/websocket\r\n".
    
96.             "Sec-WebSocket-Accept:$secAccept\r\n\r\n";
    % \* V5 s' n9 [: L
97.         return socket_write($newClient, $upgrade, strlen($upgrade));
    
98.     }
    7 z" x; |0 ?1 O' l, R
99. 
    * W3 D" H6 f7 w$ o% E3 Y
100.     /**
     
101.      * 解析接收数据
     
102.      * @param $buffer
     & a- t- v+ l$ f/ Z$ ?4 l' E
103.      * @return null|string
     5 z8 }# A0 F) Q. G) Z$ G) w
104.      */
     0 M9 R2 u0 |2 y2 X
105.     public function message($buffer){
     $ _+ E2 J, e1 e. U0 K
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);
     + N( j8 C4 W7 K; a  c, [) U
111.         } else if ($len === 127)  {
     $ q' f: L' G- `. W, V6 _8 r
112.             $masks = substr($buffer, 10, 4);
     
113.             $data = substr($buffer, 14);
     9 l2 G0 }. }8 |5 j$ v3 F
114.         } else  {
     $ x) g* N. X6 i& e! V: k3 c
115.             $masks = substr($buffer, 2, 4);
     
116.             $data = substr($buffer, 6);
     
117.         }
     
118.         for ($index = 0; $index < strlen($data); $index++) {
     7 o3 S6 d: n4 n! D4 T1 J
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     7 M. j2 A: M+ P/ w( d4 A
120.         }
     ( Q% M+ |9 i- q+ O
121.         return $decoded;
     
122.     }
     
123. 
     ; G) M& B8 _/ |, E5 \% v
124.     /**
     8 J7 x- w& p$ J9 ^0 [
125.      * 发送数据
     , u3 N* W* n/ u- J9 s; x
126.      * @param $newClinet 新接入的socket
     ( k6 H6 F5 j/ h  z0 V7 W4 a5 [; \/ Y
127.      * @param $msg   要发送的数据
     
128.      * @return int|string
     3 b+ L! t  m2 _: i* z2 q2 C; o
129.      */
     " ?9 ]9 k( W( j( u* |6 G
130.     public function send($newClinet, $msg){
     ; w/ p" r- n! h
131.         $msg = $this->frame($msg);
     / q+ ^. L, }0 x4 j
132.         socket_write($newClinet, $msg, strlen($msg));
     , o) }) Z$ y- a3 x: g1 o
133.     }
     
134. 
     ' e2 @9 K" l- c; ]) ]6 q$ R
135.     public function frame($s) {
     ' a2 n, e. |0 W/ z
136.         $a = str_split($s, 125);
     
137.         if (count($a) == 1) {
     - N) B8 F) H3 z0 J
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     
139.         }
     
140.         $ns = "";
     ! m$ l% Q9 r: Y& G! }9 v$ Z
141.         foreach ($a as $o) {
     
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     # o" I2 J4 [: S( w
143.         }
     ! R% ^7 f+ J. L$ j0 {' G
144.         return $ns;
     
145.     }
     ) ~7 @; H$ C1 @  W! p6 a
146. 
     ' @1 G: G4 f, a% l4 H
147.     /**
     
148.      * 关闭socket
     * b! n, Y2 n. v3 \* b
149.      */
     ( m2 X! B  L' i# R. h5 T! G2 b1 f
150.     public function close(){
     
151.         return socket_close($this->_sockets);
     
152.     }
     
153. }
     8 W- g3 v- t. I" b' @5 i" ^% k
154. 
     
155. $sock = new SocketService();
     - X, ~  ?, m$ j! L# L/ `* j4 \
156. $sock->run();
     % B2 K8 _$ [" M* t) k- Y. h' A
157. 
     

复制代码

web.html

1. <!doctype html>
   " g  w. ?# Q2 A( v
2. <html lang="en">
   
3. <head>
   ) O% _  ^3 v; T
4.   <meta charset="UTF-8">
   
5.   <meta name="viewport" content="width=device-width,initial-scale=1, maximum-scale=1, user-scalable=no">
   0 g" j- A5 t+ l7 l
6.   <title>websocket</title>
   ; @2 o+ Y% U; u, Q  ^
7. </head>
   
8. <body>
   , @8 }! n! i9 [$ U
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>
    ( c3 v, [! [  x, i' ?, L+ _
13. <script>
    ! \# X) h# f! x8 x0 U  ^9 B
14. /**
    
15. 0：未连接
    ; J# c8 m& L7 Q/ x+ C* y
16. 1：连接成功，可通讯
    3 }& A9 b: J5 X
17. 2：正在关闭
    
18. 3：连接已关闭或无法打开
    
19. */
    3 J* W+ S# g/ n& ~# z2 l# |
20. 
    : Y& z3 W3 q; J/ e
21.     //创建一个webSocket 实例
    
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    : i$ ]# Z/ `' z# ]
23. 
    , K* R: u3 K9 }' R! U5 L  j$ |" W
24. 
    
25.     webSocket.onerror = function (event){
    ; ~1 C3 Y' G: D& q9 l1 V7 h6 T
26.         onError(event);
      ~# v5 ]; t- J+ W% R4 d  J
27.     };
    
28. 
    
29.     // 打开websocket
    
30.     webSocket.onopen = function (event){
    % |6 M8 c) M  B+ y. Z7 Q
31.         onOpen(event);
    7 K, z% ?! d! W: I! t7 i* k* D
32.     };
    ' n2 d" X% U4 g$ @! @' ]
33. 
    - r! I, g. y2 R8 e9 t
34.     //监听消息
    
35.     webSocket.onmessage = function (event){
    - Q+ q, F0 u  f# n+ I% Q; m) G7 `
36.         onMessage(event);
    
37.     };
    
38. 
    # s) G; N. }4 @& x3 ~
39. 
    # @5 Y1 O7 Q4 T( n7 q% U1 r' r
40.     webSocket.onclose = function (event){
    
41.         onClose(event);
    
42.     }
    
43. 
    % e: j' a5 l& \0 [7 k
44.     //关闭监听websocket
    
45.     function onError(event){
    % w4 n0 V  G  E3 `' N. b
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
47.         console.log("error"+event.data);
    " b5 f+ v% e7 N# a* ?! O0 X
48.     };
    
49. 
    9 s0 Y# k% f! O1 j0 a
50.     function onOpen(event){
    + l8 m; k' m  C/ R) p
51.         console.log("open:"+sockState());
    7 O1 E. e9 ^  |: _: M+ z7 d! R
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    * j$ L( _6 u( @9 i/ ]# G( X6 Z7 O
53.     };
    + v. P# H- W. n+ R# i/ ?" ]
54.     function onMessage(event){
    2 o3 M* D6 U6 H
55.         console.log("onMessage");
    
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
      S/ T7 P) j6 N# J% h* Z
57.     };
    
58. 
    ' o, x8 T8 j) ]' K# j) P- D
59.     function onClose(event){
    
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
61.         console.log("close:"+sockState());
    
62.         webSocket.close();
    
63.     }
    7 u- ?9 n: I  Z5 A1 v$ h# u
64. 
    
65.     function sockState(){
    
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    1 i6 t+ c3 |+ [7 L+ Q* A7 j
67.             return status[webSocket.readyState];
    
68.     }
    8 H5 m0 {" u# ?7 l6 p* @0 ~
69. 
    
70. 
    
71. 
    5 s7 b  C4 C) D2 p  ^% H" T
72. function start(event){
    
73.         console.log(webSocket);
    
74.         var msg = document.getElementById('text').value;
    
75.         document.getElementById('text').value = '';
    7 I6 P/ d/ \) {5 X
76.         console.log("send:"+sockState());
    
77.         console.log("msg="+msg);
    ! S% b, Z' p& H. E( [
78.         webSocket.send("msg="+msg);
    
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    
80.     };
      s5 V. W" C3 M! L' |; Q; O
81. 
    & w' @! a0 w+ y
82.     function close(event){
    7 }( g8 }; e' `
83.         webSocket.close();
    
84.     }
    
85. </script>
    & e$ e; K* X$ T* C0 X9 A; D1 M3 r6 j7 c
86. </body>
    ) M; t1 q) x6 W' N6 M
87. </html>

复制代码

0 m1 ]1 d* e4 u7 R