php实现websocket实时消息推送

admin

php实现websocket实时消息推送
; M' O( g) j  p5 i9 Y3 `6 H

SocketService.php
1 A! C7 P4 }0 x1 P0 O

1. <?php
   : M" d$ Y3 q6 X
2. /**
   % e, w: \, P) z; v' {# ]# O
3. * Created by xwx
   
4. * Date: 2017/10/18
   
5. * Time: 14:33
   - [1 b4 {/ B: R1 M0 \0 m
6. */
   # W- X- C' B! M9 y6 {
7. 
   - j$ a3 r8 s  E$ H4 h8 ]/ B
8. class SocketService
   
9. {
   # j+ B8 R9 m( R& B3 X
10.     private $address  = '0.0.0.0';
    
11.     private $port = 8083;
    
12.     private $_sockets;
    ! B) k7 [$ Z/ b/ f3 w* X- i
13.     public function __construct($address = '', $port='')
    
14.     {
    2 F6 g# q5 n, \! F3 i* M" W8 B
15.             if(!empty($address)){
    0 ~4 G" `' L' G$ `) Y0 a* r! z7 @, v- O
16.                 $this->address = $address;
    0 I$ [) X8 y! I  T, O: G9 S- C( a
17.             }
    - H/ }  c4 [5 U% N  r
18.             if(!empty($port)) {
    * \2 i. s) F3 C$ X
19.                 $this->port = $port;
    
20.             }
    
21.     }
    3 ?  z6 p) \8 C9 ]
22. 
    - W! _7 d. \% D' R  a. _7 k: F
23.     public function service(){
    % z3 D, Q6 A- x
24.         //获取tcp协议号码。
    ! ]' F  \$ P2 D  X3 q" w+ r
25.         $tcp = getprotobyname("tcp");
    
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    / I0 Q, H6 e4 x5 p) H
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    5 L: G7 z- K, t) B8 z/ ]
28.         if($sock < 0)
    
29.         {
    
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    9 `6 T+ N% Q' O
31.         }
    % [$ F* w$ c6 B: @
32.         socket_bind($sock, $this->address, $this->port);
    
33.         socket_listen($sock, $this->port);
    
34.         echo "listen on $this->address $this->port ... \n";
    1 X9 V, e  Q3 ?& H# g4 Z8 N
35.         $this->_sockets = $sock;
    7 v3 p, o9 k. h1 w: n
36.     }
    
37. 
    
38.     public function run(){
    
39.         $this->service();
    0 H7 J( W8 H, m! k8 y
40.         $clients[] = $this->_sockets;
    
41.         while (true){
    
42.             $changes = $clients;
    ( `" V( T# U0 C
43.             $write = NULL;
    ' w' q# Y6 J8 M7 Z; R) Q
44.             $except = NULL;
    ' \% t" x, m, g" Z$ ]% l1 a
45.             socket_select($changes,  $write,  $except, NULL);
    
46.             foreach ($changes as $key => $_sock){
    
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    
48.                     if(($newClient = socket_accept($_sock))  === false){
    0 i# @/ t. k, G, G- S3 r8 b! S* w
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    
50.                     }
    
51.                     $line = trim(socket_read($newClient, 1024));
    0 w! Q$ u6 _( H
52.                     $this->handshaking($newClient, $line);
    $ j1 ~8 ]" a* K( y8 d5 S2 a
53.                     //获取client ip
    , I. \& ?9 u/ G. ~
54.                     socket_getpeername ($newClient, $ip);
    
55.                     $clients[$ip] = $newClient;
    
56.                     echo  "Client ip:{$ip}   \n";
    
57.                     echo "Client msg:{$line} \n";
    
58.                 } else {
    ' [' H8 L8 w$ ^6 j- F- m
59.                     socket_recv($_sock, $buffer,  2048, 0);
    ) V. M" k7 S2 E/ x/ {
60.                     $msg = $this->message($buffer);
    
61.                     //在这里业务代码
    
62.                     echo "{$key} clinet msg:",$msg,"\n";
    
63.                     fwrite(STDOUT, 'Please input a argument:');
    8 z% r" Q! T3 u8 w
64.                     $response = trim(fgets(STDIN));
    
65.                     $this->send($_sock, $response);
    ' [# f4 n8 c0 b
66.                     echo "{$key} response to Client:".$response,"\n";
    7 Y0 Z! Z2 Z& s% T, F
67.                 }
    
68.             }
    
69.         }
    
70.     }
    
71. 
    
72.     /**
    
73.      * 握手处理
    
74.      * @param $newClient socket
    
75.      * @return int  接收到的信息
    
76.      */
    
77.     public function handshaking($newClient, $line){
    
78. 
    
79.         $headers = array();
    
80.         $lines = preg_split("/\r\n/", $line);
      o! s( _7 I  Q! `& b
81.         foreach($lines as $line)
    % c" }: g, ]: |0 s  K# A
82.         {
    
83.             $line = chop($line);
    
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    
85.             {
    2 |9 s- O% P, L8 x1 L
86.                 $headers[$matches[1]] = $matches[2];
    * i2 [) e# c: s; @# t* O
87.             }
    
88.         }
    
89.         $secKey = $headers['Sec-WebSocket-Key'];
    0 Z! F+ ?$ K- k# u
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" .
    , O) N3 j+ m2 g5 x" W9 y, b
92.             "Upgrade: websocket\r\n" .
    $ p# U. b4 h4 f2 Q2 _4 q  A2 h
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";
    7 y8 o$ m" s" f* d) L
97.         return socket_write($newClient, $upgrade, strlen($upgrade));
    , k8 r  ^3 l* `! @0 j
98.     }
    
99. 
    
100.     /**
     
101.      * 解析接收数据
     7 R* D; q7 u# u8 @6 t" I0 Y) I
102.      * @param $buffer
     
103.      * @return null|string
     
104.      */
     
105.     public function message($buffer){
     
106.         $len = $masks = $data = $decoded = null;
     
107.         $len = ord($buffer[1]) & 127;
     
108.         if ($len === 126)  {
     
109.             $masks = substr($buffer, 4, 4);
     0 y* k* T0 |% k( N2 h
110.             $data = substr($buffer, 8);
     8 U! ?/ G( H" W
111.         } else if ($len === 127)  {
     ( F0 Q8 D) y) ?- W
112.             $masks = substr($buffer, 10, 4);
     
113.             $data = substr($buffer, 14);
     & B% t' r8 `# S& d, {4 n
114.         } else  {
     $ H/ H) u& D. S- R
115.             $masks = substr($buffer, 2, 4);
     6 s1 {9 i$ i; V" _' h4 v- l" f7 @
116.             $data = substr($buffer, 6);
     
117.         }
     
118.         for ($index = 0; $index < strlen($data); $index++) {
     
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     
120.         }
     ' c, G- K3 i# u8 Y5 }
121.         return $decoded;
     + A) [! G# m5 m( H2 I) T2 ^! q1 v3 ^
122.     }
     
123. 
     % R6 \% t9 I7 F  X  t5 p
124.     /**
     : P9 T) B, K% q+ z- u. K/ ^1 M$ ]
125.      * 发送数据
     
126.      * @param $newClinet 新接入的socket
     
127.      * @param $msg   要发送的数据
     
128.      * @return int|string
     
129.      */
     
130.     public function send($newClinet, $msg){
     1 n. @6 y) ^- x
131.         $msg = $this->frame($msg);
     
132.         socket_write($newClinet, $msg, strlen($msg));
     0 e6 t7 k" _9 u# h; V3 Y5 ?% _
133.     }
     & j: l. p0 {7 H: l3 u
134. 
     
135.     public function frame($s) {
       T# P8 |$ P9 w$ B3 b  b' K5 P
136.         $a = str_split($s, 125);
     
137.         if (count($a) == 1) {
     
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     
139.         }
     
140.         $ns = "";
     : n6 K4 I. W% W
141.         foreach ($a as $o) {
     
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     
143.         }
     0 ]* D: d# D$ Q0 Z3 K3 K
144.         return $ns;
     7 f" b# h& I* X1 o4 F9 r
145.     }
     
146. 
     
147.     /**
     ) C. G  Z# ?# l9 L( B+ l
148.      * 关闭socket
     
149.      */
     : a+ y* d. e0 N
150.     public function close(){
     / G/ V1 B  r) `) S7 |
151.         return socket_close($this->_sockets);
     
152.     }
     6 n  x7 I  q" Q" m$ f! L: a
153. }
     3 L9 r* V9 L1 h1 W, y
154. 
     0 m) J1 A" }4 i* t4 b8 N( d
155. $sock = new SocketService();
     
156. $sock->run();
     " l5 W) ^& x1 f7 _: o; [4 l
157. 
     

复制代码

web.html

1. <!doctype html>
   
2. <html lang="en">
   
3. <head>
   6 t0 X, X  @2 F$ n" D$ S
4.   <meta charset="UTF-8">
   ' G! \6 D& \, Q* c# n% H
5.   <meta name="viewport" content="width=device-width,initial-scale=1, maximum-scale=1, user-scalable=no">
   
6.   <title>websocket</title>
   , Q0 H8 h. d6 s0 p2 H
7. </head>
   8 W) b. p7 L  Y
8. <body>
   
9. <input id="text" value="">
   
10. <input type="submit" value="send" onclick="start()">
    + s2 B2 q' h3 h
11. <input type="submit" value="close" onclick="close()">
    
12. <div id="msg"></div>
    
13. <script>
    
14. /**
    / {; T# _! i& e. s
15. 0：未连接
    
16. 1：连接成功，可通讯
    
17. 2：正在关闭
      R. |' b$ Z; C  b$ X
18. 3：连接已关闭或无法打开
    . O; J) G' O/ Y( S
19. */
    7 I7 r4 P9 Y5 ~
20. 
    
21.     //创建一个webSocket 实例
    ' H7 D$ |: _# w6 s( R+ f4 S3 c
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    5 b! }, v6 D5 `. ]/ x5 s+ Z
23. 
    
24. 
    9 ?5 e  F' z1 H  T3 ]0 C( G
25.     webSocket.onerror = function (event){
    
26.         onError(event);
    
27.     };
    6 j! ]4 k+ `. x& |+ e$ V* t4 W( v) {
28. 
    
29.     // 打开websocket
    
30.     webSocket.onopen = function (event){
    ; i1 ~# c( G' n7 o+ Z4 R$ V: s
31.         onOpen(event);
    9 R0 I  `- i% _' [2 A
32.     };
    + O+ F% W6 `% r: S( u
33. 
    1 @) A/ w' O/ D: f, t; d' l
34.     //监听消息
    * m8 H" o. X/ \' f" u5 t6 A% H2 ^
35.     webSocket.onmessage = function (event){
    2 d/ V- ~, a* p3 o" E
36.         onMessage(event);
    
37.     };
    
38. 
    7 N1 t  `# s9 L& }. z# Y) b% r
39. 
    6 X8 m( m2 v/ I9 D2 G, J! n% {6 a
40.     webSocket.onclose = function (event){
    
41.         onClose(event);
    
42.     }
    9 P' `& @  E( {. w8 p( j5 w
43. 
    ( ?* Y% Z/ ^# v, X+ e
44.     //关闭监听websocket
    
45.     function onError(event){
    
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    ; D8 ?2 L( P  X9 g; L, `/ ]8 U
47.         console.log("error"+event.data);
    
48.     };
    
49. 
    3 C5 q! [$ o9 k8 j
50.     function onOpen(event){
    ( i. }0 Z/ x( j) _9 B
51.         console.log("open:"+sockState());
    2 p- |4 I4 {3 f8 e! x
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    ( H- T- v" v' L
53.     };
    
54.     function onMessage(event){
    
55.         console.log("onMessage");
    8 U: f7 u- G) l( U$ g' ]
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    1 \* I8 ~0 r2 x
57.     };
    
58. 
    # x7 T9 J: o" s- N$ A( Z
59.     function onClose(event){
    & d, W/ S) _, _
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
      E$ E/ G# |' g6 {0 Z# z+ l
61.         console.log("close:"+sockState());
    , k+ c6 U" h1 [, f" g8 j7 `+ f9 J
62.         webSocket.close();
    
63.     }
    ' ?$ K0 M$ x( Q: u6 E
64. 
    
65.     function sockState(){
      g! ^  J* c3 d, l! ~" r
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    
67.             return status[webSocket.readyState];
    * }0 z; t; E$ z
68.     }
    
69. 
    + B0 l" q  W% o7 i! t8 ?* h
70. 
    # @4 ^! g) q. }8 m
71. 
    
72. function start(event){
    
73.         console.log(webSocket);
    
74.         var msg = document.getElementById('text').value;
    
75.         document.getElementById('text').value = '';
    3 ?0 M, v; [1 v  G
76.         console.log("send:"+sockState());
    & t4 X. j4 B2 b/ Y+ o+ v7 j  D
77.         console.log("msg="+msg);
    - C+ T* R; N# a$ R" j7 j( z( F
78.         webSocket.send("msg="+msg);
    * b8 Z2 s& {% H. d. @
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    5 F0 v; ?# ?% T8 a; x3 V
80.     };
    0 F3 U9 d) o' j- \  C& n5 a
81. 
    " z7 L8 V' ~5 a8 A
82.     function close(event){
    
83.         webSocket.close();
    $ [7 M  J' q9 J$ c: y; l) h" F
84.     }
    % r9 m; O3 t, p. A. |1 E
85. </script>
      {2 O5 ]; j5 [2 E6 K
86. </body>
    7 T! A( {; q2 E* I
87. </html>

复制代码

: ^2 [! P) O* n2 Y$ H' Z: v