php实现websocket实时消息推送

admin

php实现websocket实时消息推送
& L; Z2 [+ }( X" E9 f2 z0 m
. z# F) P- |5 n  ^+ L6 U
+ o! y5 f, ^: E7 J: `5 U% A( U5 D1 BSocketService.php
" I+ x, t1 S- J$ f5 P5 w

1. <?php
   
2. /**
   % G1 a* I7 F3 @1 Y
3. * Created by xwx
   
4. * Date: 2017/10/18
   - V& ]8 g+ `3 {, [- h) J
5. * Time: 14:33
   
6. */
   " o, n, P; {- {; z7 `1 L8 D
7. 
   
8. class SocketService
   
9. {
   
10.     private $address  = '0.0.0.0';
    8 ]* Z3 _. D& y  i5 ~
11.     private $port = 8083;
    + g, w" N; T! L8 _6 _! Y
12.     private $_sockets;
    
13.     public function __construct($address = '', $port='')
    
14.     {
    . s  P( v: ]4 F3 a& k& m9 F
15.             if(!empty($address)){
    2 C: ^1 h5 o1 v: V: ?. e
16.                 $this->address = $address;
    
17.             }
    
18.             if(!empty($port)) {
    
19.                 $this->port = $port;
    
20.             }
    
21.     }
    % H0 e  B* E; g' }  \& l' N8 x
22. 
    1 S0 R% }2 i' ]; {
23.     public function service(){
    % W, ?0 N2 x: ^/ ]; {3 ]# Y$ W
24.         //获取tcp协议号码。
    6 T1 A$ r& S2 F) d; v
25.         $tcp = getprotobyname("tcp");
    $ c1 `3 n) s0 [
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    ( |& w# d4 A- W( v
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    : i8 }. S% [+ m# I
28.         if($sock < 0)
    ; z9 f* ]3 h4 d' ^# f, E! Q4 w: V
29.         {
    8 E/ U% g4 N) ^; M( i3 l4 `. |
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    
31.         }
    
32.         socket_bind($sock, $this->address, $this->port);
    
33.         socket_listen($sock, $this->port);
    
34.         echo "listen on $this->address $this->port ... \n";
      U- s9 E* V3 y3 Y1 |
35.         $this->_sockets = $sock;
    4 V. x- d. a/ G6 ]1 b1 f5 v
36.     }
    
37. 
    
38.     public function run(){
    
39.         $this->service();
    $ p3 m2 a# p$ F$ l" p1 g0 ^
40.         $clients[] = $this->_sockets;
    
41.         while (true){
    ) J. O5 r/ U! D) @; Z
42.             $changes = $clients;
    0 E% M& p  ]7 z; d2 [
43.             $write = NULL;
    
44.             $except = NULL;
    3 n8 h7 y8 C8 _$ ~% m% M& P! P5 G
45.             socket_select($changes,  $write,  $except, NULL);
    ! \. L- o& m# a8 k2 K/ c
46.             foreach ($changes as $key => $_sock){
    ! f& s% B: P7 z/ Q6 c
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    $ X. d- w: F& g' F- B7 n
48.                     if(($newClient = socket_accept($_sock))  === false){
    
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    ; w4 r7 T5 ], a9 J2 F4 ~
50.                     }
    
51.                     $line = trim(socket_read($newClient, 1024));
    
52.                     $this->handshaking($newClient, $line);
    
53.                     //获取client ip
    
54.                     socket_getpeername ($newClient, $ip);
    8 t0 s5 O5 k+ b( P4 L7 t+ I2 g% O+ {
55.                     $clients[$ip] = $newClient;
    ' q" D7 u7 C, Q- G7 [/ H  V: ^
56.                     echo  "Client ip:{$ip}   \n";
    ! X2 v+ `/ w. G, \5 j. C$ p+ ?
57.                     echo "Client msg:{$line} \n";
    9 S/ n. S5 s/ Z9 _' x( C
58.                 } else {
    
59.                     socket_recv($_sock, $buffer,  2048, 0);
    ' U% A- E3 x. m' w) a2 ]* d
60.                     $msg = $this->message($buffer);
    
61.                     //在这里业务代码
    
62.                     echo "{$key} clinet msg:",$msg,"\n";
      o, c) f9 `( O' h# E: w, ^$ h
63.                     fwrite(STDOUT, 'Please input a argument:');
    
64.                     $response = trim(fgets(STDIN));
    
65.                     $this->send($_sock, $response);
    
66.                     echo "{$key} response to Client:".$response,"\n";
    
67.                 }
    & |7 L$ ^+ L1 l. s
68.             }
    
69.         }
    
70.     }
    
71. 
    
72.     /**
    
73.      * 握手处理
    / V4 }$ T/ d( L6 }& ^# b' Y0 A
74.      * @param $newClient socket
    
75.      * @return int  接收到的信息
    
76.      */
    , ^/ f. {: _* D  N9 i1 k$ w3 o
77.     public function handshaking($newClient, $line){
    & f) ^- v5 \( I6 K
78. 
    . N2 w# t; T# R( |
79.         $headers = array();
    
80.         $lines = preg_split("/\r\n/", $line);
    5 J; l4 P0 O' m" U! g5 g
81.         foreach($lines as $line)
    
82.         {
    
83.             $line = chop($line);
    ( M) ~; _/ V1 b: D
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    
85.             {
    
86.                 $headers[$matches[1]] = $matches[2];
    
87.             }
    6 u  e# F) n: h0 z" o9 U% N
88.         }
    " O" R% W$ e: S$ ?
89.         $secKey = $headers['Sec-WebSocket-Key'];
    
90.         $secAccept = base64_encode(pack('H*', sha1($secKey . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11')));
    5 A+ Y) N& [5 n# b; ^! a+ r
91.         $upgrade  = "HTTP/1.1 101 Web Socket Protocol Handshake\r\n" .
    0 b, ?$ [2 Z! u( X8 C: V
92.             "Upgrade: websocket\r\n" .
    / W* n4 R) r$ Z; D6 k& n8 |
93.             "Connection: Upgrade\r\n" .
    
94.             "WebSocket-Origin: $this->address\r\n" .
    6 s2 K8 q  n- k! N8 G. X* K
95.             "WebSocket-Location: ws://$this->address:$this->port/websocket/websocket\r\n".
    ( o6 \$ t9 ~* \% I0 v5 E; Q
96.             "Sec-WebSocket-Accept:$secAccept\r\n\r\n";
    ' I( x: }. T% C# W, z; @8 e
97.         return socket_write($newClient, $upgrade, strlen($upgrade));
    
98.     }
    
99. 
      X6 U5 A2 k3 P
100.     /**
     : y1 W. g/ L+ Z$ M. h8 V  V
101.      * 解析接收数据
     9 m- }! a  i: G& s
102.      * @param $buffer
     ; j$ U/ o, X0 w* G8 C6 ^
103.      * @return null|string
     - C0 A- ], e3 q" }
104.      */
     
105.     public function message($buffer){
     - Y6 E! }3 u! _0 Q# X7 ^
106.         $len = $masks = $data = $decoded = null;
     
107.         $len = ord($buffer[1]) & 127;
     6 |3 M) i' C9 R! _0 Q% F7 L
108.         if ($len === 126)  {
     0 ]" }+ Y) f+ y; B: W- y; D
109.             $masks = substr($buffer, 4, 4);
     
110.             $data = substr($buffer, 8);
     
111.         } else if ($len === 127)  {
     
112.             $masks = substr($buffer, 10, 4);
       C- u- ?& G% h  F; o
113.             $data = substr($buffer, 14);
     + g+ V1 h/ x6 K$ x9 y
114.         } else  {
     3 Z# U0 a* l, ~$ |/ V
115.             $masks = substr($buffer, 2, 4);
     
116.             $data = substr($buffer, 6);
     
117.         }
     
118.         for ($index = 0; $index < strlen($data); $index++) {
     & ~: G1 Z$ {& R3 I
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     
120.         }
     4 j6 W! }  }; h. K8 |& g2 e5 `
121.         return $decoded;
     
122.     }
     
123. 
     " N( S( b; P; t2 \
124.     /**
     
125.      * 发送数据
     ' r6 Z# F2 x$ \, ]- E2 |
126.      * @param $newClinet 新接入的socket
     # |  T* K# |3 ~" Q4 C1 h2 z
127.      * @param $msg   要发送的数据
     
128.      * @return int|string
     
129.      */
     
130.     public function send($newClinet, $msg){
     7 |2 C9 w2 k3 i( u( t
131.         $msg = $this->frame($msg);
     / K1 y2 `( U! k" Q0 H
132.         socket_write($newClinet, $msg, strlen($msg));
     4 ?! B5 X/ E, y+ X. u' W- i, ~
133.     }
     
134. 
     : c9 f' G$ o% i
135.     public function frame($s) {
     4 {) Y- Q+ l% J, v$ D2 p; g, C
136.         $a = str_split($s, 125);
     
137.         if (count($a) == 1) {
     7 \9 y; p8 o2 Y4 R
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     * R* E  }; f1 I. ?4 Z
139.         }
     ) E$ u5 M0 w5 L% N% o$ j, W1 Y0 U
140.         $ns = "";
     
141.         foreach ($a as $o) {
     : q5 q5 f- S& Y* r4 M
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     
143.         }
     ( R0 |7 Z0 L3 h
144.         return $ns;
     
145.     }
     
146. 
     ! p! f3 Z! C, T7 e  T0 g4 Y: V2 @0 V
147.     /**
     
148.      * 关闭socket
     
149.      */
     + Z+ K/ B. l2 K2 k1 s1 {% ~
150.     public function close(){
     & o4 G! r7 l$ ]9 b- t
151.         return socket_close($this->_sockets);
     
152.     }
     & c! w' R+ N8 j; g5 p6 Y5 ~
153. }
     
154. 
     
155. $sock = new SocketService();
     
156. $sock->run();
     
157. 
     

复制代码

web.html

1. <!doctype html>
   
2. <html lang="en">
   * Y' _% ]& m, K1 S* E' L0 H
3. <head>
   % Z0 Y* V4 {; C: Y) @
4.   <meta charset="UTF-8">
   ! }  t6 w& n5 a1 V( o8 a
5.   <meta name="viewport" content="width=device-width,initial-scale=1, maximum-scale=1, user-scalable=no">
   
6.   <title>websocket</title>
   
7. </head>
   " w! z+ h, Y/ e6 s" T! T6 b
8. <body>
   
9. <input id="text" value="">
   
10. <input type="submit" value="send" onclick="start()">
    
11. <input type="submit" value="close" onclick="close()">
    $ L  _8 _3 t" _! K- I, C( z6 F
12. <div id="msg"></div>
    : T( G9 m$ O7 }1 y. O
13. <script>
    
14. /**
    + g9 o* M/ H9 S% c! O! t) r
15. 0：未连接
    
16. 1：连接成功，可通讯
    $ L0 Z. R) N0 P: n7 ^
17. 2：正在关闭
    3 Y6 V6 g% j/ A
18. 3：连接已关闭或无法打开
    
19. */
    $ R7 k" c* b# @8 l5 o' N6 [
20. 
    
21.     //创建一个webSocket 实例
    " ?+ {4 F. X- m. p& p
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    
23. 
    
24. 
    , w$ \4 J; D% D' B0 V
25.     webSocket.onerror = function (event){
    " O! f4 ?- i+ O; I/ D) ]3 N6 u
26.         onError(event);
    ) W7 g# y6 a$ f$ O' {0 b8 c
27.     };
    
28. 
    
29.     // 打开websocket
    8 r6 \4 _$ Y: C5 p% O+ W% r) [
30.     webSocket.onopen = function (event){
    7 v/ i. A$ z' o( \: f9 \" A# }
31.         onOpen(event);
    
32.     };
      r2 u( t7 ]. m. ~$ W; x8 k
33. 
    
34.     //监听消息
    8 Y: t; @2 d. G/ ]
35.     webSocket.onmessage = function (event){
    : @+ y* |+ p: V0 j1 M! @2 ]
36.         onMessage(event);
    ) C/ J6 X$ }8 }
37.     };
    
38. 
      }( h, c7 [9 o- v. X8 C
39. 
    ; _9 X1 a- X+ }  C* u  }
40.     webSocket.onclose = function (event){
    ) p6 r8 q: S& b# C9 h3 b3 \
41.         onClose(event);
    
42.     }
    + R, Z0 T2 `3 g% [1 I& Q
43. 
    
44.     //关闭监听websocket
    
45.     function onError(event){
    9 E! o$ U" O% P6 S( z& O6 u: B
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
47.         console.log("error"+event.data);
    . S' S3 [/ L  ]5 n( r# _; P6 U  D
48.     };
    
49. 
    
50.     function onOpen(event){
    + L6 a0 g4 c! w/ \6 }3 o! ]1 s
51.         console.log("open:"+sockState());
    
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    
53.     };
    * I; q$ P: H! `+ M* Q
54.     function onMessage(event){
    
55.         console.log("onMessage");
    
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    
57.     };
    
58. 
    ! d9 Z% K4 M* t  u/ K
59.     function onClose(event){
    
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    0 C( P( B, K6 |2 {! B
61.         console.log("close:"+sockState());
    7 T7 F. ^- W5 ]! w8 l
62.         webSocket.close();
    " j+ y' S* S3 z3 ^
63.     }
    
64. 
    
65.     function sockState(){
    . \" C: t! ^+ g5 _
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    3 f; b6 b/ ]' Q% `- g
67.             return status[webSocket.readyState];
    
68.     }
    
69. 
    2 K* K: a9 X! r$ z
70. 
    $ q$ a9 f( W. v! `# O# \; D  B$ t
71. 
    
72. function start(event){
    
73.         console.log(webSocket);
    
74.         var msg = document.getElementById('text').value;
    
75.         document.getElementById('text').value = '';
    , ]9 r( N/ t' r+ j1 D
76.         console.log("send:"+sockState());
    ( S4 [) }8 [) X. P9 D1 Q7 P
77.         console.log("msg="+msg);
    . {4 m6 N* L# p4 Z( j# Q  z' ^; j/ Q
78.         webSocket.send("msg="+msg);
    & y" s$ U0 M7 {( [! r! B
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
      {; n3 S" \+ N0 g+ g3 Q
80.     };
    
81. 
    
82.     function close(event){
    1 a8 G  i) `9 q  `% P2 Q! {
83.         webSocket.close();
    
84.     }
    & H8 ?: Z. m' y1 e) d% W
85. </script>
      N2 B  v% n$ h8 M  A& Y. W  ~* h
86. </body>
    
87. </html>

复制代码

) ^/ K4 Q8 Z3 B$ g2 E& a. j