php实现websocket实时消息推送

admin

php实现websocket实时消息推送

" {, g- L' I2 b. l. Q% w' z
SocketService.php
" f( {/ F6 W2 L& Z

1. <?php
   
2. /**
   4 W# K) c( _0 s- _+ y, G3 S
3. * Created by xwx
   1 y3 t- |0 U, a7 _) t1 V$ g
4. * Date: 2017/10/18
   
5. * Time: 14:33
   ; N6 `+ g8 K/ L
6. */
   
7. 
   8 r. D% B. R6 @7 F
8. class SocketService
   5 N6 b* v' M, m% V8 V, G, |) F+ R
9. {
   
10.     private $address  = '0.0.0.0';
    * i8 [6 U3 Y/ R% {; s2 I5 v
11.     private $port = 8083;
    
12.     private $_sockets;
    3 ]4 e# J/ c! U- u$ T( }: Z
13.     public function __construct($address = '', $port='')
    
14.     {
    - r) w1 Z/ d8 [; Z! S* l* {
15.             if(!empty($address)){
    
16.                 $this->address = $address;
    
17.             }
    % s4 W: L4 q: l! W
18.             if(!empty($port)) {
    ( t1 Z% k7 x, y1 S: a% C# m2 y8 I
19.                 $this->port = $port;
    
20.             }
    
21.     }
    
22. 
    
23.     public function service(){
    
24.         //获取tcp协议号码。
    
25.         $tcp = getprotobyname("tcp");
    " h' q+ d; \' F/ n+ |/ n
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    0 j* T9 p. S5 M( P
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    3 W7 d- `$ c* L/ {2 [# }
28.         if($sock < 0)
    % B; b% d2 u) R5 X! J5 A( ]: a
29.         {
    0 N3 G" m( a! V$ v
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    
31.         }
    7 ?* G* K( g8 U( c3 e
32.         socket_bind($sock, $this->address, $this->port);
    
33.         socket_listen($sock, $this->port);
    : ?# M3 Y3 t" d* P9 {2 C
34.         echo "listen on $this->address $this->port ... \n";
    
35.         $this->_sockets = $sock;
    ! c: X9 b1 ^& Z" T) u7 V: Z- d
36.     }
    
37. 
    + C% l& Z! w: [# B9 ^% F
38.     public function run(){
    
39.         $this->service();
    
40.         $clients[] = $this->_sockets;
    * {( [7 E4 U; }% |6 X7 d
41.         while (true){
    
42.             $changes = $clients;
    " b5 s+ P* g7 @6 B6 I  V8 r% X# b
43.             $write = NULL;
    
44.             $except = NULL;
    
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){
    1 ]% e+ }% \8 z
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    
50.                     }
    6 I3 v0 n7 |' f6 t. x4 v
51.                     $line = trim(socket_read($newClient, 1024));
    
52.                     $this->handshaking($newClient, $line);
    
53.                     //获取client ip
    
54.                     socket_getpeername ($newClient, $ip);
    
55.                     $clients[$ip] = $newClient;
    
56.                     echo  "Client ip:{$ip}   \n";
    , K9 T6 B5 M/ L- X9 O9 p
57.                     echo "Client msg:{$line} \n";
    
58.                 } else {
    
59.                     socket_recv($_sock, $buffer,  2048, 0);
    / g6 X) b- K2 M. [3 x3 A$ C
60.                     $msg = $this->message($buffer);
    
61.                     //在这里业务代码
    * u* n* t! `6 c4 F+ B* l
62.                     echo "{$key} clinet msg:",$msg,"\n";
    2 ?7 d- F% P4 t0 Y$ i% [
63.                     fwrite(STDOUT, 'Please input a argument:');
    
64.                     $response = trim(fgets(STDIN));
    0 K; a6 L- }% O
65.                     $this->send($_sock, $response);
    
66.                     echo "{$key} response to Client:".$response,"\n";
    0 a5 g' w2 |+ O$ F/ H+ G( e
67.                 }
    + @2 s) Z  Q1 \' f3 L
68.             }
    
69.         }
    & d8 K$ {  D/ Y3 M$ _
70.     }
    
71. 
    : j( R8 ]3 D# o$ m9 w4 b# {
72.     /**
    
73.      * 握手处理
    
74.      * @param $newClient socket
    
75.      * @return int  接收到的信息
    # |2 V% ~* Z5 b& i; a
76.      */
    
77.     public function handshaking($newClient, $line){
    
78. 
    
79.         $headers = array();
    
80.         $lines = preg_split("/\r\n/", $line);
    * t' `+ B' t, J, P  H! x
81.         foreach($lines as $line)
    % P$ `: V: w9 v: h! M
82.         {
    3 t+ W; ~+ U' v4 ]9 b3 F
83.             $line = chop($line);
    1 i8 B* p) a5 v% a+ l% A2 T
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    0 o% `5 {0 }: h9 d+ A# G
85.             {
    
86.                 $headers[$matches[1]] = $matches[2];
    ; I: N0 [7 Q) ~: n, j6 h9 M# V
87.             }
    
88.         }
    
89.         $secKey = $headers['Sec-WebSocket-Key'];
    # d5 w. V3 ^1 ?' c* [
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" .
    ' s4 b  ~- n% K1 r; w2 _5 P- y" P
92.             "Upgrade: websocket\r\n" .
    0 U+ D& D3 p5 D  W' K% [- s
93.             "Connection: Upgrade\r\n" .
    
94.             "WebSocket-Origin: $this->address\r\n" .
    
95.             "WebSocket-Location: ws://$this->address:$this->port/websocket/websocket\r\n".
    5 e5 J: Q0 R* ^5 U1 [( u: p" ~
96.             "Sec-WebSocket-Accept:$secAccept\r\n\r\n";
    
97.         return socket_write($newClient, $upgrade, strlen($upgrade));
    
98.     }
    $ H+ J  m; A% T6 G6 d; t! g0 a
99. 
    
100.     /**
     
101.      * 解析接收数据
     
102.      * @param $buffer
     
103.      * @return null|string
     : g+ W# D; Z: r
104.      */
     
105.     public function message($buffer){
     
106.         $len = $masks = $data = $decoded = null;
     
107.         $len = ord($buffer[1]) & 127;
     
108.         if ($len === 126)  {
     & x/ F+ w  B/ _+ f
109.             $masks = substr($buffer, 4, 4);
     . u" B4 s/ a4 ^( T8 x2 u
110.             $data = substr($buffer, 8);
     
111.         } else if ($len === 127)  {
     
112.             $masks = substr($buffer, 10, 4);
     0 R9 A( Z9 i% s7 ]$ B( ]
113.             $data = substr($buffer, 14);
     & Q9 Y% K! T7 C2 J5 G
114.         } else  {
     + c" V4 [1 B5 i% C  m
115.             $masks = substr($buffer, 2, 4);
     
116.             $data = substr($buffer, 6);
     3 A' \$ k/ T1 y8 Q
117.         }
     3 j1 G3 d, |. [9 ?
118.         for ($index = 0; $index < strlen($data); $index++) {
     ! _$ l  h7 B' V& V; d: H
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     
120.         }
     ) h; K2 Q. }4 y* ~) p* l
121.         return $decoded;
     3 X& R8 R8 f4 K
122.     }
     
123. 
     
124.     /**
     ; u% ?/ ?; m" m0 H
125.      * 发送数据
     
126.      * @param $newClinet 新接入的socket
     
127.      * @param $msg   要发送的数据
     
128.      * @return int|string
     # C( X3 L  M+ z' L
129.      */
     6 s- t7 A/ G* I; `
130.     public function send($newClinet, $msg){
     
131.         $msg = $this->frame($msg);
     5 w8 X( m4 D( ]: b# z7 U* t
132.         socket_write($newClinet, $msg, strlen($msg));
     
133.     }
     
134. 
     
135.     public function frame($s) {
     
136.         $a = str_split($s, 125);
     
137.         if (count($a) == 1) {
     
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     . h$ q7 V5 A1 f6 W
139.         }
     
140.         $ns = "";
     3 C- X( ^7 G2 _' k
141.         foreach ($a as $o) {
     
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     ' t' e4 a) C& ^9 _/ W! c
143.         }
     
144.         return $ns;
     
145.     }
       ^- P' G! W# Q2 }
146. 
     : d) l- Q3 k+ v. z0 e( }, h
147.     /**
     
148.      * 关闭socket
     & e! i/ {" [0 H3 v: K4 A
149.      */
     
150.     public function close(){
     6 [( F8 B2 ?" C/ L0 I) |' G
151.         return socket_close($this->_sockets);
     ; }+ f, S$ L- o- S
152.     }
     
153. }
     , b# v) G) q% H4 m
154. 
     
155. $sock = new SocketService();
     - y1 T) q" v$ p6 k) }
156. $sock->run();
     
157. 
     

复制代码

web.html
# q! y3 E5 t$ w9 o/ R1 u2 f# z9 c

1. <!doctype html>
   ) M/ F4 ?0 x8 j( g
2. <html lang="en">
   9 E' F( q7 q( f) Q
3. <head>
   $ N# _- F# P1 a8 T: G7 s
4.   <meta charset="UTF-8">
   + g8 ^; g! c6 {9 ~- A6 L& Y$ J
5.   <meta name="viewport" content="width=device-width,initial-scale=1, maximum-scale=1, user-scalable=no">
   
6.   <title>websocket</title>
   
7. </head>
     Y; d. I9 e0 U9 `$ F  d# ?5 Q
8. <body>
   . Z, ]! K6 R8 y
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>
    
13. <script>
    
14. /**
    
15. 0：未连接
    : r' F7 B( g: C/ w, z
16. 1：连接成功，可通讯
    . H3 H5 d7 |, z6 @
17. 2：正在关闭
    
18. 3：连接已关闭或无法打开
    6 }' y3 o  w! j) I, ^; ^7 M3 H
19. */
    
20. 
    
21.     //创建一个webSocket 实例
    7 R) i3 K! t. `& `9 V3 P: w
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    * a$ J" B* \9 ~" N7 n' I
23. 
    0 w" ~* ^7 H0 [3 z
24. 
    
25.     webSocket.onerror = function (event){
    
26.         onError(event);
    
27.     };
    3 R6 F2 G4 a- Z! X, P
28. 
    # l  t7 {  h" K( E
29.     // 打开websocket
    
30.     webSocket.onopen = function (event){
    
31.         onOpen(event);
    6 {: a# U' _, k( D
32.     };
    , i! q6 @$ N& l& v' X
33. 
    % Q. y, F& f' w7 e9 t
34.     //监听消息
    
35.     webSocket.onmessage = function (event){
    ' e6 U" i& N) T/ c
36.         onMessage(event);
    
37.     };
    / [9 e: L8 I- |
38. 
    6 f  X$ a: V0 D; x; [
39. 
    
40.     webSocket.onclose = function (event){
    ' q+ O" P4 M& b5 V
41.         onClose(event);
    
42.     }
    
43. 
    # f5 U9 @" h% f$ L6 C" Y' Q  J
44.     //关闭监听websocket
    1 w7 _; l& z; |# A- b7 o
45.     function onError(event){
    
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
47.         console.log("error"+event.data);
    ( S; [( k1 \4 A1 m1 w
48.     };
    1 g5 D; K. z0 E' w
49. 
    
50.     function onOpen(event){
    
51.         console.log("open:"+sockState());
    
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    
53.     };
    ; p: R& ]2 U1 {- @& k
54.     function onMessage(event){
    
55.         console.log("onMessage");
    5 t  v$ h9 G' K5 p; U9 y
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    
57.     };
    
58. 
      ?  K# i0 d9 ~- a1 x/ @6 e
59.     function onClose(event){
    , c3 v# [& v# O, F  P
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    " z9 E% E; Q0 i
61.         console.log("close:"+sockState());
    
62.         webSocket.close();
    9 {8 y/ |- \& i& J( P) \/ S
63.     }
    4 v- W; T$ ^! Q8 O
64. 
    , y+ R0 s7 z, K8 i
65.     function sockState(){
    
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    
67.             return status[webSocket.readyState];
    
68.     }
    
69. 
    ) ~  b2 o0 G# x* w3 _% ~
70. 
    
71. 
    
72. function start(event){
    / w& ]1 \) S% D' W$ |9 t
73.         console.log(webSocket);
    
74.         var msg = document.getElementById('text').value;
    
75.         document.getElementById('text').value = '';
    
76.         console.log("send:"+sockState());
    * w2 W8 }2 `! n6 F9 d
77.         console.log("msg="+msg);
    " J! ?4 q5 ]- k2 X( Z$ t; J
78.         webSocket.send("msg="+msg);
    
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    
80.     };
    2 S) m7 Q) M  R2 L
81. 
    ; W3 p1 C; a9 J+ ^
82.     function close(event){
    * k& k8 _8 d' C7 L8 M/ C4 ?
83.         webSocket.close();
    ; `, j3 D$ n% f; W9 }$ l7 \3 V
84.     }
    
85. </script>
    9 L2 S3 R2 D. T; T4 w6 x7 v
86. </body>
    & n& C# v. F* O) _, U
87. </html>

复制代码

; ^. q; Z9 y: W
5 i) @- z8 P3 L" l