php实现websocket实时消息推送

admin

php实现websocket实时消息推送


/ w; y) _2 B7 z+ G9 OSocketService.php

1. <?php
   + k+ U( o0 `* B
2. /**
   6 y. b$ b, g7 G) h4 t. K) P* U
3. * Created by xwx
   8 R/ N5 J* a- d: r
4. * Date: 2017/10/18
   $ Y: F, Q, T* v) g& |+ k
5. * Time: 14:33
   
6. */
   
7. 
   ) U# P& X4 V! I* R2 L
8. class SocketService
   9 b3 f* F; ]' f! o
9. {
   - l2 J: O/ v8 W7 _
10.     private $address  = '0.0.0.0';
    
11.     private $port = 8083;
      ?# b4 Y! S+ J, P
12.     private $_sockets;
    5 j  h' s  Y- F
13.     public function __construct($address = '', $port='')
    0 ~* i& ?8 n$ A
14.     {
    
15.             if(!empty($address)){
    - O# ]  ~( I3 o1 Y' U
16.                 $this->address = $address;
    ( o( n0 Q. k: p% @
17.             }
    
18.             if(!empty($port)) {
    
19.                 $this->port = $port;
    
20.             }
    
21.     }
    
22. 
    3 U- K0 d/ D0 z% |! f
23.     public function service(){
    + S3 J- i( q, ^: J- l$ t4 x/ `/ c: F
24.         //获取tcp协议号码。
    ) j! U. h6 v, W  {0 L( U/ }
25.         $tcp = getprotobyname("tcp");
    ) G/ _0 @" s; [" _* b# U5 [
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    8 O# L+ f# M0 q6 L. D
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    ( i. O  C* f1 A# T+ q# {, u7 v
28.         if($sock < 0)
      I# d1 c! J6 r( C& j7 V
29.         {
    * m5 w# K0 j0 K5 W
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    
31.         }
    
32.         socket_bind($sock, $this->address, $this->port);
    7 g7 A3 L( n5 }1 S" B
33.         socket_listen($sock, $this->port);
    ! [6 W1 e8 K; u3 f$ Z
34.         echo "listen on $this->address $this->port ... \n";
    2 ?1 m# P" ?' E: h$ G
35.         $this->_sockets = $sock;
    ' M9 m& o0 S& @4 I+ `
36.     }
    % V% t- [) j/ M3 y0 i; p+ F4 K1 w
37. 
    
38.     public function run(){
    7 e- Y5 q% A2 R/ o( i; _* g8 @3 L
39.         $this->service();
    
40.         $clients[] = $this->_sockets;
    
41.         while (true){
    + n" ~+ y8 K) ?# @
42.             $changes = $clients;
    
43.             $write = NULL;
    
44.             $except = NULL;
    " Z  V. i! ~6 K# H/ D$ T% |/ n
45.             socket_select($changes,  $write,  $except, NULL);
    
46.             foreach ($changes as $key => $_sock){
    0 e0 ?) c8 f+ `- Y
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
      E; ~7 w) ~3 z8 b" O! G
48.                     if(($newClient = socket_accept($_sock))  === false){
    . e' T% T1 s+ |. u9 N7 s
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    
50.                     }
    3 [, @1 ~2 k; O$ X7 R
51.                     $line = trim(socket_read($newClient, 1024));
    
52.                     $this->handshaking($newClient, $line);
    
53.                     //获取client ip
    
54.                     socket_getpeername ($newClient, $ip);
    
55.                     $clients[$ip] = $newClient;
    5 v, s. F! ], n3 v
56.                     echo  "Client ip:{$ip}   \n";
    0 o/ A- P+ w2 h4 a3 x
57.                     echo "Client msg:{$line} \n";
    
58.                 } else {
    # ^4 ?0 y2 ?( y- ?) g
59.                     socket_recv($_sock, $buffer,  2048, 0);
    
60.                     $msg = $this->message($buffer);
    
61.                     //在这里业务代码
    + U) _$ I! g4 r8 S1 B& l2 A
62.                     echo "{$key} clinet msg:",$msg,"\n";
    ; W0 P' Y) |4 T4 _7 W
63.                     fwrite(STDOUT, 'Please input a argument:');
    
64.                     $response = trim(fgets(STDIN));
    
65.                     $this->send($_sock, $response);
    ! U) p4 X' \! c) \6 R& @
66.                     echo "{$key} response to Client:".$response,"\n";
    % W. w" H5 ~4 W$ ?  \' }; e, n
67.                 }
    
68.             }
    
69.         }
    : F& _9 T! m- M4 a  Q* c0 r
70.     }
    
71. 
    * j% \5 Q" i+ ?# I* e2 Z5 ?1 H2 {
72.     /**
    ( y: ]9 B2 e9 k9 d' J( p; d
73.      * 握手处理
    
74.      * @param $newClient socket
    # \* E) R+ f/ f  G* w
75.      * @return int  接收到的信息
    
76.      */
    
77.     public function handshaking($newClient, $line){
    
78. 
    & U2 W; Z* y& M& U6 c
79.         $headers = array();
    
80.         $lines = preg_split("/\r\n/", $line);
    1 X, N; Y# ?% ~1 Y# ^/ f
81.         foreach($lines as $line)
    
82.         {
    3 J7 Q9 G8 d4 ?
83.             $line = chop($line);
    
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    - O: ~" O5 H0 ~1 j4 Y- N: H- F
85.             {
    " B  q: p1 W: u8 x: V3 _; t
86.                 $headers[$matches[1]] = $matches[2];
    
87.             }
    
88.         }
    
89.         $secKey = $headers['Sec-WebSocket-Key'];
    $ p  R& W( L2 p# o. S( |4 y+ X
90.         $secAccept = base64_encode(pack('H*', sha1($secKey . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11')));
    ; E; M7 ^: S6 ?% ~) ^& f" Q1 ~$ K
91.         $upgrade  = "HTTP/1.1 101 Web Socket Protocol Handshake\r\n" .
    
92.             "Upgrade: websocket\r\n" .
    , K- t: s# z2 D4 ~5 C1 S% x9 {
93.             "Connection: Upgrade\r\n" .
    
94.             "WebSocket-Origin: $this->address\r\n" .
    * ?) ]+ R. m" q+ ?
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));
    & N: I# N# q' A( k
98.     }
    8 q& Y" j2 r" h5 G5 ^0 W# @
99. 
    
100.     /**
     
101.      * 解析接收数据
     
102.      * @param $buffer
     
103.      * @return null|string
     
104.      */
     
105.     public function message($buffer){
     
106.         $len = $masks = $data = $decoded = null;
     ' U+ N" y7 \  t7 M4 k( O/ m) S* H# `
107.         $len = ord($buffer[1]) & 127;
     
108.         if ($len === 126)  {
     
109.             $masks = substr($buffer, 4, 4);
     # t) i7 l3 |6 G* Y
110.             $data = substr($buffer, 8);
     
111.         } else if ($len === 127)  {
     6 x0 t( f# |5 ~+ h0 ~6 X( G: }
112.             $masks = substr($buffer, 10, 4);
     / ?  f3 u' W; E" `
113.             $data = substr($buffer, 14);
     
114.         } else  {
     : C; R7 v4 C6 Y( u; v
115.             $masks = substr($buffer, 2, 4);
     / H( T0 m2 G' V6 O
116.             $data = substr($buffer, 6);
     
117.         }
     9 b5 [/ }1 D+ N3 D" l. V, W
118.         for ($index = 0; $index < strlen($data); $index++) {
     
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     ' M' r/ x" k( q
120.         }
     " F+ v2 E# w4 G  N% O: r" F4 |
121.         return $decoded;
     * h6 Y0 E( k2 v/ F+ i* {* z; C
122.     }
     , f: o6 q6 p  H  L  f
123. 
     , [' b4 ]: u: z+ ^
124.     /**
     
125.      * 发送数据
     8 e# `6 c! b  B# L
126.      * @param $newClinet 新接入的socket
     
127.      * @param $msg   要发送的数据
       M& |5 `& U' K$ I/ Q
128.      * @return int|string
     
129.      */
     ! u0 g- ^% z$ j3 i* W9 Z" l2 B
130.     public function send($newClinet, $msg){
     4 }3 m- P* b* B6 ], P! m. b
131.         $msg = $this->frame($msg);
     ) v& _9 F1 ?$ |! L5 y. r' v
132.         socket_write($newClinet, $msg, strlen($msg));
     8 }" B! d: C: i" c3 i7 U0 A8 j
133.     }
     0 |% ]" L7 T# w2 |! M1 J% L
134. 
     
135.     public function frame($s) {
     
136.         $a = str_split($s, 125);
     
137.         if (count($a) == 1) {
     ' R. A/ R# Q) m9 n
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     
139.         }
     6 U! d$ s- W4 V6 N( C
140.         $ns = "";
     
141.         foreach ($a as $o) {
     
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     
143.         }
     
144.         return $ns;
     
145.     }
     
146. 
     6 E5 h, k. E  k: Y, o7 R
147.     /**
     
148.      * 关闭socket
     
149.      */
     0 G& h& |4 E& V! p
150.     public function close(){
     / ?- Q3 g3 m  i5 g; W2 q& V5 m
151.         return socket_close($this->_sockets);
     * w1 H- m- }% Q7 X* E2 z
152.     }
     
153. }
     
154. 
     
155. $sock = new SocketService();
     + f6 @% p4 j7 c' W7 x5 M
156. $sock->run();
     ; n6 }, i8 n- q# W
157. 
     

复制代码

web.html

1. <!doctype html>
   2 T. r1 B9 P# G# h% M- L
2. <html lang="en">
   
3. <head>
   
4.   <meta charset="UTF-8">
   : Z4 N2 G( Y' N: l- E3 n
5.   <meta name="viewport" content="width=device-width,initial-scale=1, maximum-scale=1, user-scalable=no">
   + _) C, t$ ~) {2 D1 u" l# h* L5 J" w8 N
6.   <title>websocket</title>
   
7. </head>
   
8. <body>
   . j1 n% V4 g7 b1 N" m: c& k
9. <input id="text" value="">
   
10. <input type="submit" value="send" onclick="start()">
    ! w, ?, x) J, f( }4 \: o
11. <input type="submit" value="close" onclick="close()">
    
12. <div id="msg"></div>
    
13. <script>
    . Q5 e. v! f* r  O% s. g
14. /**
    4 q: V: a4 x' Q3 ~0 `9 u% O
15. 0：未连接
    5 S0 V+ T+ l+ D* J
16. 1：连接成功，可通讯
    
17. 2：正在关闭
    
18. 3：连接已关闭或无法打开
    6 v4 i. t1 u3 N. C3 K( L0 W
19. */
    
20. 
    0 w. Z3 Y* O7 j  ^
21.     //创建一个webSocket 实例
    
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    - ?$ o1 b* T" T  W- J6 Z
23. 
    
24. 
    
25.     webSocket.onerror = function (event){
    
26.         onError(event);
      ^0 K- |6 W# G2 X# y# z
27.     };
    
28. 
    7 ~: o: X. j" p. E
29.     // 打开websocket
    9 f' O2 @9 r# w% E& s
30.     webSocket.onopen = function (event){
    " w% a; e' n+ H! W* I2 l; r
31.         onOpen(event);
    ( [9 T$ f& M" d& r
32.     };
    
33. 
    
34.     //监听消息
    : X& N; C: d1 X, x4 [3 m2 R
35.     webSocket.onmessage = function (event){
    
36.         onMessage(event);
    
37.     };
    
38. 
    
39. 
    
40.     webSocket.onclose = function (event){
    
41.         onClose(event);
    0 A/ |/ i1 R5 F; P4 s7 Q
42.     }
    ' N) _4 M8 x' w! V
43. 
    & @9 s( M$ ]8 o! a
44.     //关闭监听websocket
    7 Q! |9 ~6 u( ?1 y% a
45.     function onError(event){
    
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
47.         console.log("error"+event.data);
    
48.     };
    
49. 
    
50.     function onOpen(event){
    4 w  C7 d! T, m1 p+ O
51.         console.log("open:"+sockState());
    
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    & G: o6 M/ _6 f
53.     };
    - U  O8 A7 }/ U5 s9 Y: @) j; ^
54.     function onMessage(event){
    * \# f: S# X' D1 n% p
55.         console.log("onMessage");
    $ f: \6 O  F9 K% Q7 u( W* q
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
      {& A0 a2 Y  R3 u$ T1 [
57.     };
    
58. 
    
59.     function onClose(event){
    2 _0 ~( Z' N) L5 i
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    / G  `) Y6 f5 l1 X* j( K; t
61.         console.log("close:"+sockState());
    3 l) t4 n5 x' d' A+ |$ Z/ f+ p% e  X
62.         webSocket.close();
    
63.     }
    3 W2 L( {! f9 H6 j
64. 
    2 S6 l6 q: g7 o5 H: u& |4 K
65.     function sockState(){
    
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    7 ~3 v: U9 s' ?+ I1 |# V
67.             return status[webSocket.readyState];
    
68.     }
    
69. 
    
70. 
    $ o1 p8 R" p2 g4 a
71. 
    
72. function start(event){
    
73.         console.log(webSocket);
    # _. B( G; w3 y4 F8 ?  ^
74.         var msg = document.getElementById('text').value;
    % s' G$ N2 `6 [* v/ ^
75.         document.getElementById('text').value = '';
    
76.         console.log("send:"+sockState());
    7 s0 y7 P# U' L/ Y9 B
77.         console.log("msg="+msg);
    
78.         webSocket.send("msg="+msg);
    # d' B) o: {$ T- K
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    ! F% Q' \8 s5 P
80.     };
    
81. 
    
82.     function close(event){
    
83.         webSocket.close();
    
84.     }
    
85. </script>
    . L  H" \* D7 d8 J
86. </body>
    
87. </html>

复制代码