php实现websocket实时消息推送

admin

php实现websocket实时消息推送


SocketService.php
- \0 p# f- D# w

1. <?php
   8 ^( i# @, e) \" k) y% R* `5 I. l; X$ _
2. /**
   ' X& }6 Q# r) s% O) C# E8 t7 C: Q
3. * Created by xwx
   4 j. B! w) S3 {, q, w
4. * Date: 2017/10/18
   2 V/ B" R5 n  }( W! `% e- W7 T' K: Y
5. * Time: 14:33
   ' ?2 x+ x) u: j8 S( }4 Z, p
6. */
   9 r  E" b5 c: L+ I! [& Y) s
7. 
   - i$ N7 b6 R5 _
8. class SocketService
   
9. {
   4 p+ t+ I8 K/ t+ X) Q
10.     private $address  = '0.0.0.0';
    / X9 `7 c2 A' G3 d4 c; {) k+ `
11.     private $port = 8083;
    
12.     private $_sockets;
    ; ^- \  _2 \/ [8 X# z4 h
13.     public function __construct($address = '', $port='')
    $ R- s" t8 Z5 M- M
14.     {
    / V$ y1 H- g6 t; s
15.             if(!empty($address)){
    
16.                 $this->address = $address;
    
17.             }
    
18.             if(!empty($port)) {
    ( X8 x* Z# o5 j  i' [
19.                 $this->port = $port;
    
20.             }
    
21.     }
    : D2 m% m: X4 l$ a3 A8 A8 s
22. 
    
23.     public function service(){
    ' y, @- O, w  y/ Q; G4 C
24.         //获取tcp协议号码。
    
25.         $tcp = getprotobyname("tcp");
    2 x3 ^1 `- [9 {& E+ R
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    * ^  I6 V% |  N. U4 o$ r- l' |  F& `) N
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    
28.         if($sock < 0)
    : o. |4 D  A' e7 }7 j7 D9 }" n
29.         {
    ! W1 l; c3 u- H4 t0 K' i& y1 ]3 E
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    
31.         }
    8 t" E1 m; W, ^2 P
32.         socket_bind($sock, $this->address, $this->port);
    : ]+ h+ A- B! x; v8 i: [9 d
33.         socket_listen($sock, $this->port);
    6 S: `+ C+ n; p) _! }: V: f  U% F
34.         echo "listen on $this->address $this->port ... \n";
    / f% o8 R: L& m6 @/ x0 A; g. b
35.         $this->_sockets = $sock;
    5 a5 ^. g$ q- o! f1 g
36.     }
    
37. 
    8 v" k2 ^4 \; h* f- ^
38.     public function run(){
    / \7 _. p+ _' O7 H5 E
39.         $this->service();
    / q  @9 M3 b. r3 ?+ ~) k+ t7 y
40.         $clients[] = $this->_sockets;
    
41.         while (true){
    
42.             $changes = $clients;
    5 c* N7 Q2 A5 v+ [! O
43.             $write = NULL;
    9 r' s  v2 H9 Y6 T; D
44.             $except = NULL;
    0 t! I* k" |; z0 ]
45.             socket_select($changes,  $write,  $except, NULL);
    
46.             foreach ($changes as $key => $_sock){
    : k* K" M4 f( I5 y4 a9 d/ O9 X+ j) |& ^
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    
48.                     if(($newClient = socket_accept($_sock))  === false){
    
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    
50.                     }
    . x* b+ C, B) B
51.                     $line = trim(socket_read($newClient, 1024));
    
52.                     $this->handshaking($newClient, $line);
    
53.                     //获取client ip
    
54.                     socket_getpeername ($newClient, $ip);
    
55.                     $clients[$ip] = $newClient;
    - R# P( q- c" _. ?# L. J
56.                     echo  "Client ip:{$ip}   \n";
    
57.                     echo "Client msg:{$line} \n";
    ! m  Y' Y+ ~8 C. r- u
58.                 } else {
    
59.                     socket_recv($_sock, $buffer,  2048, 0);
    
60.                     $msg = $this->message($buffer);
    
61.                     //在这里业务代码
    
62.                     echo "{$key} clinet msg:",$msg,"\n";
    & [7 V+ ^6 Q+ p) [* c
63.                     fwrite(STDOUT, 'Please input a argument:');
    9 Z0 c% `8 u8 j5 d+ H
64.                     $response = trim(fgets(STDIN));
    
65.                     $this->send($_sock, $response);
    
66.                     echo "{$key} response to Client:".$response,"\n";
    2 f2 {. o& R+ g7 L
67.                 }
    0 g) h/ M  Z  h
68.             }
    
69.         }
    
70.     }
    
71. 
    6 ~# j  t& J  H' t7 m
72.     /**
    * J5 u7 W6 K8 }! d  l; Z3 Y/ t! a
73.      * 握手处理
    
74.      * @param $newClient socket
    
75.      * @return int  接收到的信息
    
76.      */
    7 F3 E+ W7 c+ M# [2 t3 W
77.     public function handshaking($newClient, $line){
    
78. 
    1 b$ q) C% f8 C7 D
79.         $headers = array();
    ; Q0 k! Z, `0 h5 A3 X# W; N
80.         $lines = preg_split("/\r\n/", $line);
    
81.         foreach($lines as $line)
    7 k! S) H) v4 @6 L0 o5 L
82.         {
    , C( V0 ^: F9 f! P% s" ?) k+ Z
83.             $line = chop($line);
    
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    2 E* V# o& \% @0 L* L( b
85.             {
    
86.                 $headers[$matches[1]] = $matches[2];
    7 }+ h( U3 x% a9 }% V# J
87.             }
    
88.         }
    
89.         $secKey = $headers['Sec-WebSocket-Key'];
    ! V! }# g" [; }2 M" g( ^- e( h
90.         $secAccept = base64_encode(pack('H*', sha1($secKey . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11')));
    0 C* ^+ a* }- g  j( V; P
91.         $upgrade  = "HTTP/1.1 101 Web Socket Protocol Handshake\r\n" .
    
92.             "Upgrade: websocket\r\n" .
    ! S( {: F( s4 ?- }/ {9 J9 _, _# P
93.             "Connection: Upgrade\r\n" .
    0 t! f6 l" _% p) i+ R
94.             "WebSocket-Origin: $this->address\r\n" .
    & h  _- ^7 ^1 J2 q
95.             "WebSocket-Location: ws://$this->address:$this->port/websocket/websocket\r\n".
    7 W) V2 ^6 g- k
96.             "Sec-WebSocket-Accept:$secAccept\r\n\r\n";
    5 U7 ~; w, Q9 H( l# C0 m8 e
97.         return socket_write($newClient, $upgrade, strlen($upgrade));
    + D, p" p; i) j1 v) n& H. ~$ K
98.     }
    
99. 
    
100.     /**
     
101.      * 解析接收数据
     $ x  r! y( Z; V5 D
102.      * @param $buffer
     . T% S3 M2 J6 Y6 v
103.      * @return null|string
     
104.      */
     
105.     public function message($buffer){
     
106.         $len = $masks = $data = $decoded = null;
     3 m5 ~/ _. f  s) c! f, r. a1 G* r
107.         $len = ord($buffer[1]) & 127;
     * G2 k4 P5 O. F$ i. [2 Y5 z* V
108.         if ($len === 126)  {
     
109.             $masks = substr($buffer, 4, 4);
     * k( D  }: r/ L- B" F0 H
110.             $data = substr($buffer, 8);
     
111.         } else if ($len === 127)  {
     $ N# A% a6 n: H1 X) V# C4 T
112.             $masks = substr($buffer, 10, 4);
     
113.             $data = substr($buffer, 14);
     
114.         } else  {
     , y- q; ~, {8 ^2 f! p
115.             $masks = substr($buffer, 2, 4);
     
116.             $data = substr($buffer, 6);
     
117.         }
     
118.         for ($index = 0; $index < strlen($data); $index++) {
     0 w+ f0 i3 b+ e1 p. J
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     ( u7 p! v7 A1 L. m7 ?- x% i
120.         }
     
121.         return $decoded;
     3 w; ^3 l7 V* K2 L, H  y! J0 e, g. Q
122.     }
     0 n" F, o1 u+ f" \/ N) P) M
123. 
     
124.     /**
     
125.      * 发送数据
     ' s$ B8 M* x5 h7 b0 Q6 K
126.      * @param $newClinet 新接入的socket
     , J. g5 \* R2 Z( X% Z4 ?; o$ k
127.      * @param $msg   要发送的数据
     
128.      * @return int|string
     ( v5 |5 B6 h+ x8 H# f; h- z
129.      */
     $ S) D9 h" K9 m
130.     public function send($newClinet, $msg){
     * H0 |8 S2 h' Z/ f* M
131.         $msg = $this->frame($msg);
     8 [) I- ?- \; r1 E7 U8 S7 \8 O
132.         socket_write($newClinet, $msg, strlen($msg));
     
133.     }
     # B. X' p# N1 Q. z% G2 K5 P8 r  P) _
134. 
     , x' Z- f7 G! \5 w4 h% w* Y) x# o
135.     public function frame($s) {
     
136.         $a = str_split($s, 125);
     
137.         if (count($a) == 1) {
     ) @( W; \( Y, Q3 f: }
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     ; b" a8 K: y& I+ j3 e( s
139.         }
     - ~: C* t' e4 u; Q9 u  U& t
140.         $ns = "";
     # x! X/ {0 _* M
141.         foreach ($a as $o) {
     ' P! Y* q' e+ x: {& Z
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     
143.         }
     
144.         return $ns;
     
145.     }
     % A9 s$ j, ~* h0 f+ ?( z
146. 
     
147.     /**
     
148.      * 关闭socket
     
149.      */
     
150.     public function close(){
     9 W' ]1 k, N4 d; d
151.         return socket_close($this->_sockets);
     
152.     }
     6 E% M5 ]- G/ y- C  J. r3 Z" W
153. }
     
154. 
     
155. $sock = new SocketService();
     " I9 d& q  R' h* j7 y
156. $sock->run();
     : m0 g, m  Y3 ~! q8 z
157. 
     

复制代码

web.html
: t7 {  o' m% c5 z- R% F6 v

1. <!doctype html>
   
2. <html lang="en">
   : f9 L" u# ?9 Z' j5 m+ `( i; P* Z
3. <head>
   
4.   <meta charset="UTF-8">
   
5.   <meta name="viewport" content="width=device-width,initial-scale=1, maximum-scale=1, user-scalable=no">
   
6.   <title>websocket</title>
   6 W  p" d7 C* k- ^6 z
7. </head>
   6 [0 m/ c/ C/ B" t/ a
8. <body>
   & U5 h$ m8 G4 i& g2 `7 `
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>
    % ]9 P8 t; q( q: Z* j) l
13. <script>
    
14. /**
    
15. 0：未连接
    
16. 1：连接成功，可通讯
    
17. 2：正在关闭
    6 S/ w! u- _3 e  A8 b" o
18. 3：连接已关闭或无法打开
    + K  F$ ]# e( b
19. */
    
20. 
    
21.     //创建一个webSocket 实例
    % v: m: x& o% ]1 j* p
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    
23. 
    
24. 
    
25.     webSocket.onerror = function (event){
    
26.         onError(event);
    
27.     };
    
28. 
    
29.     // 打开websocket
    
30.     webSocket.onopen = function (event){
    ) j4 j* k( O& w2 V6 w3 F  l* D
31.         onOpen(event);
    
32.     };
    
33. 
    
34.     //监听消息
    $ v  y9 L1 S' z& T/ o  s4 C0 n4 S4 y; J
35.     webSocket.onmessage = function (event){
    
36.         onMessage(event);
    ! B5 U% v. \! y6 c
37.     };
    # G- T* |" G# R8 ]7 n, Y5 c4 f
38. 
    ) u4 x+ X% @' ~! j6 o8 x
39. 
    ' R  V& s7 b8 G; s$ |
40.     webSocket.onclose = function (event){
    
41.         onClose(event);
    
42.     }
    ' b1 q( p+ u9 d% S9 R
43. 
      O) {, S0 ^3 i( [. m8 s& _
44.     //关闭监听websocket
    
45.     function onError(event){
    3 h1 K6 ]+ m- _' B, s6 T5 |; m2 U
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    4 E5 R0 l% c+ n
47.         console.log("error"+event.data);
    
48.     };
    7 _- D4 A' m6 _- D9 l, K
49. 
    
50.     function onOpen(event){
    
51.         console.log("open:"+sockState());
    0 W4 Q8 b* q& N
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    2 O# k) `7 j4 e& g* q8 Q
53.     };
    
54.     function onMessage(event){
    
55.         console.log("onMessage");
    8 F( ~: L! Q5 W0 v; a
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    - `* S5 s8 ^  r1 e5 k' e4 M
57.     };
    7 g$ q" \; X  M8 \& b. j
58. 
    
59.     function onClose(event){
    
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    9 F4 Y$ Y/ B3 d4 B3 |  _
61.         console.log("close:"+sockState());
    5 n$ C. F# a6 V  \  q" }- M- z
62.         webSocket.close();
    
63.     }
    
64. 
    
65.     function sockState(){
    2 b9 P" U9 Y' k& M! p2 M2 c
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    
67.             return status[webSocket.readyState];
    
68.     }
    
69. 
    : x& I# h; a' D
70. 
    8 `' ~8 Q6 m0 E$ D4 z$ d3 d# `8 F
71. 
    , }* M2 B" G- b9 J
72. function start(event){
    
73.         console.log(webSocket);
    
74.         var msg = document.getElementById('text').value;
    . H  \+ {& m0 \! T( r- E7 W. D
75.         document.getElementById('text').value = '';
    : ~7 _0 M8 P/ }4 ]5 a
76.         console.log("send:"+sockState());
    
77.         console.log("msg="+msg);
    
78.         webSocket.send("msg="+msg);
    1 X% K! s. o& a* Z& z" E% {
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    0 W9 Z+ e' p, X' l% s: d
80.     };
    / A' T: X7 n/ F
81. 
    , O8 F  D) S; f6 J- Q$ z. p
82.     function close(event){
    
83.         webSocket.close();
    
84.     }
    $ V! r8 \% W; `5 S
85. </script>
    
86. </body>
    
87. </html>

复制代码

: n3 W8 s+ |5 F' V! g; I! t
& F4 b- N6 Z( J! u! S