php实现websocket实时消息推送

admin

php实现websocket实时消息推送
. {+ K# a2 a& O- @6 v" J

SocketService.php
$ L& B1 b0 E$ p1 K4 t$ h

1. <?php
   
2. /**
   
3. * Created by xwx
   
4. * Date: 2017/10/18
   
5. * Time: 14:33
   ) Z( i) [' Y! o/ e) e; h- U
6. */
   
7. 
   # [; h: X1 e' |- J# w
8. class SocketService
   $ @) |; r  H' B2 Z+ \4 V0 s
9. {
   
10.     private $address  = '0.0.0.0';
    ; u0 e1 L# Y6 a/ a2 U
11.     private $port = 8083;
    
12.     private $_sockets;
    
13.     public function __construct($address = '', $port='')
    
14.     {
    ' B2 _7 Q2 [  n- J
15.             if(!empty($address)){
    8 L* T3 w; e$ T' c* B9 d9 ~, X1 c
16.                 $this->address = $address;
    
17.             }
    
18.             if(!empty($port)) {
    
19.                 $this->port = $port;
    
20.             }
    5 t# `/ e8 J  W6 Z! h% C
21.     }
    / X: @8 l0 R5 m+ B& H
22. 
    
23.     public function service(){
    
24.         //获取tcp协议号码。
    
25.         $tcp = getprotobyname("tcp");
    & C2 Z; C, |% ?( x# a4 W4 ]- ~+ ]
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    , H/ V( S6 A  P* E# a6 ]
28.         if($sock < 0)
    % W: r9 M9 k( B& w0 c2 X
29.         {
    
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    3 Q' p* V: ?! E% a$ V
31.         }
    
32.         socket_bind($sock, $this->address, $this->port);
    : n& A" P. c0 l- G. C. E
33.         socket_listen($sock, $this->port);
    
34.         echo "listen on $this->address $this->port ... \n";
    1 Q9 P& l; o4 ]
35.         $this->_sockets = $sock;
    
36.     }
      T7 u" @9 T0 J/ d, k& Z$ U
37. 
    , y2 _- C' N5 N6 A
38.     public function run(){
    
39.         $this->service();
    # B5 q, S( G$ _$ |+ h/ Z8 Q: E. i
40.         $clients[] = $this->_sockets;
    
41.         while (true){
    : a! q( t! W% D" k1 f* ]
42.             $changes = $clients;
    
43.             $write = NULL;
    # U$ p" a" e) T! ?
44.             $except = NULL;
    
45.             socket_select($changes,  $write,  $except, NULL);
    
46.             foreach ($changes as $key => $_sock){
    % N* N4 s$ R. E3 Q& p
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    
48.                     if(($newClient = socket_accept($_sock))  === false){
    
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    2 N( ^# G( [4 c, j* l
50.                     }
    
51.                     $line = trim(socket_read($newClient, 1024));
    # ~' N$ M$ Z* T. b) T1 h8 C! L
52.                     $this->handshaking($newClient, $line);
    
53.                     //获取client ip
    . r# L( O: w! \3 ^# c0 t
54.                     socket_getpeername ($newClient, $ip);
    
55.                     $clients[$ip] = $newClient;
    0 j7 |5 n7 h5 q) f' C
56.                     echo  "Client ip:{$ip}   \n";
    ' _% }# o# j7 s: N' z9 k8 Q/ V
57.                     echo "Client msg:{$line} \n";
    % U+ G/ \/ g7 b, l: }
58.                 } else {
      d3 A3 I1 O0 x8 i3 |
59.                     socket_recv($_sock, $buffer,  2048, 0);
    
60.                     $msg = $this->message($buffer);
    / u. c( r, P, S0 S0 M
61.                     //在这里业务代码
    " j+ l7 u- e$ c6 d: L& e
62.                     echo "{$key} clinet msg:",$msg,"\n";
    
63.                     fwrite(STDOUT, 'Please input a argument:');
    
64.                     $response = trim(fgets(STDIN));
    
65.                     $this->send($_sock, $response);
      h& Z$ Q4 f, N9 @8 r# W  i
66.                     echo "{$key} response to Client:".$response,"\n";
    . o4 U* Y% d0 H: P
67.                 }
    
68.             }
    
69.         }
    $ Y% W+ C; v( t1 Y5 r# t
70.     }
    
71. 
    ) H3 e. j2 A- e
72.     /**
    % B0 a2 h6 z- v8 W6 [2 [, r: G
73.      * 握手处理
    
74.      * @param $newClient socket
    
75.      * @return int  接收到的信息
    ! x' H. P# Y% b; B. c9 Z
76.      */
    
77.     public function handshaking($newClient, $line){
    # w* k* P- [* }, s6 Z# v
78. 
    " ~3 D0 J( {) j5 n2 U' N
79.         $headers = array();
    & f6 C+ B$ O  q- I& n6 [2 U
80.         $lines = preg_split("/\r\n/", $line);
    6 Q* x  G) U! [: G- B: k- [
81.         foreach($lines as $line)
    + z3 R1 A1 n. H" @! O
82.         {
    
83.             $line = chop($line);
    ; C. o8 O. f% `9 X4 t
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    
85.             {
    
86.                 $headers[$matches[1]] = $matches[2];
    : C/ I# ?+ P3 D  i
87.             }
    4 T8 c  d2 l. E. P! k
88.         }
    / D. ]! T1 u# ~: l6 Z1 l
89.         $secKey = $headers['Sec-WebSocket-Key'];
    ! B& x% u/ y* L$ o6 e" l
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" .
    
92.             "Upgrade: websocket\r\n" .
    
93.             "Connection: Upgrade\r\n" .
    
94.             "WebSocket-Origin: $this->address\r\n" .
    % U- C) B6 a/ D. a; N# b
95.             "WebSocket-Location: ws://$this->address:$this->port/websocket/websocket\r\n".
    * j$ {) W, e5 F
96.             "Sec-WebSocket-Accept:$secAccept\r\n\r\n";
    
97.         return socket_write($newClient, $upgrade, strlen($upgrade));
    
98.     }
    
99. 
    
100.     /**
     3 b( k% K: v4 d3 j
101.      * 解析接收数据
     
102.      * @param $buffer
     % u6 w; M- b7 f) B0 l4 S# @
103.      * @return null|string
     
104.      */
     
105.     public function message($buffer){
     0 f7 v& Q- Z& @% J7 D2 y
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);
     ! a$ {7 ]# C+ ]4 a& }
111.         } else if ($len === 127)  {
       Z2 L' i  n9 o' U: i3 W( ^1 Q
112.             $masks = substr($buffer, 10, 4);
       Z# q, Z7 _9 B4 ?, D6 ]
113.             $data = substr($buffer, 14);
     
114.         } else  {
     9 q# ?% ^% C5 m) U) z, w% F) @! a
115.             $masks = substr($buffer, 2, 4);
     6 U3 e4 A2 f' S# ~' K
116.             $data = substr($buffer, 6);
       Z$ B7 `; X! q) A
117.         }
     
118.         for ($index = 0; $index < strlen($data); $index++) {
     ( D/ A( I* ?2 o5 q9 d, J) y
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     
120.         }
     
121.         return $decoded;
     
122.     }
     
123. 
     , W* S( R& q) N% u# K
124.     /**
     
125.      * 发送数据
     
126.      * @param $newClinet 新接入的socket
     
127.      * @param $msg   要发送的数据
     
128.      * @return int|string
     
129.      */
     * i4 F9 Y" t7 a" b
130.     public function send($newClinet, $msg){
     
131.         $msg = $this->frame($msg);
     
132.         socket_write($newClinet, $msg, strlen($msg));
     
133.     }
     2 u' [* y4 q( x
134. 
     
135.     public function frame($s) {
     ; d. I* c4 L6 O( C
136.         $a = str_split($s, 125);
     
137.         if (count($a) == 1) {
     
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     * I7 z2 O8 _, h6 Q. V0 l- B4 K2 ?
139.         }
       O# j4 Y9 g2 P- l0 v
140.         $ns = "";
     
141.         foreach ($a as $o) {
     
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     " t( f6 K# |+ V: V
143.         }
     , G& ^7 l% z% p# Q
144.         return $ns;
     
145.     }
     
146. 
     ( p# s& p: i5 k# S  E9 d
147.     /**
     
148.      * 关闭socket
     
149.      */
     ) T( \* a& L* p' M% m! h
150.     public function close(){
     $ H( G7 X% S; v0 O, C
151.         return socket_close($this->_sockets);
     5 j- X" |! ^& l/ H0 K$ A* ~
152.     }
     , D5 j! }, @$ p: [) I, ]0 [# z
153. }
     
154. 
     
155. $sock = new SocketService();
     ; \$ E: n& d) D3 i
156. $sock->run();
     
157. 
     6 x) g! b( P- ?

复制代码

web.html

1. <!doctype html>
   8 _' l/ x7 O$ |* |
2. <html lang="en">
   4 n4 M: ^% n+ L, ]
3. <head>
   4 ^- g+ d2 v$ z
4.   <meta charset="UTF-8">
   
5.   <meta name="viewport" content="width=device-width,initial-scale=1, maximum-scale=1, user-scalable=no">
   . {6 B+ V5 x  A; q
6.   <title>websocket</title>
   
7. </head>
   ; c/ B4 {/ p2 @( [1 K% _3 B
8. <body>
   7 n9 Y5 V9 W$ i
9. <input id="text" value="">
   9 Y. _' B, t$ b3 {1 `
10. <input type="submit" value="send" onclick="start()">
    ' d: L0 A3 m0 u
11. <input type="submit" value="close" onclick="close()">
    
12. <div id="msg"></div>
    
13. <script>
    
14. /**
    
15. 0：未连接
    
16. 1：连接成功，可通讯
    # W& E# g  Z1 A9 s/ x* Q
17. 2：正在关闭
    - v+ H1 p" E8 X" T! |& T( K
18. 3：连接已关闭或无法打开
    
19. */
    1 A5 T! H; V( `7 L4 ~
20. 
    
21.     //创建一个webSocket 实例
    
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    ( o$ h9 f1 x; {; {  T) q
23. 
    2 c4 N) i4 M' b8 D2 s
24. 
    * u, ^$ u& q% c3 t; L
25.     webSocket.onerror = function (event){
    : I4 N0 }7 f7 y6 g$ r3 o6 i9 K7 \
26.         onError(event);
    
27.     };
    
28. 
    1 A5 h# J+ \% V5 {
29.     // 打开websocket
    
30.     webSocket.onopen = function (event){
    . N% ?  [, P* K
31.         onOpen(event);
    
32.     };
    8 |( ]6 g) k9 }/ `1 \! @$ h
33. 
    
34.     //监听消息
    
35.     webSocket.onmessage = function (event){
      q: }) B/ [# o# s
36.         onMessage(event);
      Z8 l6 d. w9 R/ N# t7 k
37.     };
    
38. 
    # u2 R0 C) n7 F7 b& j/ H
39. 
    ; a$ l* u9 k3 s
40.     webSocket.onclose = function (event){
    7 o2 r% x: ?$ e0 h. I5 C
41.         onClose(event);
    
42.     }
    ' n! c0 R. y. g; H" G; e
43. 
    
44.     //关闭监听websocket
    
45.     function onError(event){
    ( s! B" y7 V' }* [& _5 R
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
47.         console.log("error"+event.data);
    
48.     };
    . j/ i& b$ f; G* w
49. 
    : }3 f' c# g) k% @1 d2 \
50.     function onOpen(event){
    " i6 y) l" e2 @1 B2 f6 ~4 h, [% ^
51.         console.log("open:"+sockState());
    . E" [2 R! d) K  S/ U9 j
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    
53.     };
    ) X; }; T8 E3 ]' a) A
54.     function onMessage(event){
    
55.         console.log("onMessage");
    - G5 g4 S8 y5 w! q5 ~3 V
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    
57.     };
    ) g2 {% q. N% _% J# C) G6 k: s
58. 
    0 W- e" @8 z& S0 ~5 D2 k
59.     function onClose(event){
    7 k. i; U  G! o* p# ]
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    8 Z, E3 I, W( @( B% P, P# j: X
61.         console.log("close:"+sockState());
    
62.         webSocket.close();
    - v! c% E- b( z: x" r0 K0 l
63.     }
    / a! l6 {% Y, h- j. C8 B- A
64. 
    * z4 q! ^  z# ~+ D, I
65.     function sockState(){
    4 M9 n2 w* Y0 ]2 G5 m( h- |  [$ }
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    
67.             return status[webSocket.readyState];
    & k' z/ B* A* c
68.     }
    
69. 
    5 T2 Y" f+ N5 W( R" n# k4 ]
70. 
    2 |9 \3 {9 i% ~: B3 `
71. 
    , l- r! ]! _: t- Z& G# s
72. function start(event){
    & \. x1 K* y4 E; I" N
73.         console.log(webSocket);
    : V& P5 W1 `5 r  A5 A! v
74.         var msg = document.getElementById('text').value;
    
75.         document.getElementById('text').value = '';
    
76.         console.log("send:"+sockState());
    
77.         console.log("msg="+msg);
    
78.         webSocket.send("msg="+msg);
      K! j. ?/ Q/ b6 d, l  Y5 D9 i
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    
80.     };
    ' [; Y6 |, ]* d/ e8 ^
81. 
    6 V7 V8 ~  G3 i3 U" Q5 I
82.     function close(event){
    ; j: z- Z! K) t" P2 @! T# D( P
83.         webSocket.close();
    ! G1 M# F3 o6 ]; h5 R
84.     }
    ; j8 c6 Y' G5 `: V9 R3 B4 {7 N
85. </script>
    
86. </body>
    9 U# ?: f9 w% b8 A! l9 R
87. </html>

复制代码

3 l8 Z+ Y, ^. x4 H' Y- G& Q/ H

8 p" P; }  G8 n% {8 c0 h6 ~