php实现websocket实时消息推送

admin

php实现websocket实时消息推送
# B9 e  k3 }! g$ Z4 S: y. f+ `
/ ~3 H- o/ ~, Y& n: [) r
SocketService.php

1. <?php
   
2. /**
   ( n. W' f$ }& F0 {3 o
3. * Created by xwx
   
4. * Date: 2017/10/18
   
5. * Time: 14:33
   $ z" `# M; b1 Q
6. */
   , [* H" T7 F: }( e4 A4 O% h& ^
7. 
   
8. class SocketService
   0 U0 F9 I  t& [- m% J6 y! s
9. {
   
10.     private $address  = '0.0.0.0';
    
11.     private $port = 8083;
    0 W! L, p2 D) j
12.     private $_sockets;
    
13.     public function __construct($address = '', $port='')
    : F5 T- c% W! G
14.     {
    : r( k# o1 P% V& l
15.             if(!empty($address)){
    * r, p5 X* i' f7 u" P
16.                 $this->address = $address;
    8 z- X1 Q4 X" u( C: u  ^
17.             }
    , K  I# S. C. v/ m$ K) p- O
18.             if(!empty($port)) {
    8 ^" E7 `' W% c0 a/ o- q
19.                 $this->port = $port;
    
20.             }
    . `! r( `4 j( ?* L6 [7 h* W+ l6 s
21.     }
    6 b/ Y! M- O6 I$ ]* N$ @
22. 
    6 k2 q, ?9 e. Q0 V
23.     public function service(){
    ! ^# `' V6 X; A
24.         //获取tcp协议号码。
    
25.         $tcp = getprotobyname("tcp");
    
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    
28.         if($sock < 0)
    
29.         {
    7 P* s% |& V. a: f
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    7 W3 X; T& d  n5 I
31.         }
    
32.         socket_bind($sock, $this->address, $this->port);
    
33.         socket_listen($sock, $this->port);
    
34.         echo "listen on $this->address $this->port ... \n";
    . T7 u% n9 ?' i  T/ J2 ~  E
35.         $this->_sockets = $sock;
    . Z# ]- |$ T" h9 o5 q
36.     }
    . B2 _& z) s  U2 H% a! `  z
37. 
    ( E3 G. j5 T5 z# J7 e
38.     public function run(){
    
39.         $this->service();
    
40.         $clients[] = $this->_sockets;
    ! K. V# d5 k- L; n3 l3 i( o
41.         while (true){
    
42.             $changes = $clients;
    : [9 i7 n- o4 g& |  \+ \+ J) X
43.             $write = NULL;
    
44.             $except = NULL;
    4 U/ c. g" Y. O. B' r, R
45.             socket_select($changes,  $write,  $except, NULL);
    
46.             foreach ($changes as $key => $_sock){
    : U: y! h6 B% c1 \6 Z; _
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    
48.                     if(($newClient = socket_accept($_sock))  === false){
    
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    7 e  S; o% P0 G7 o- G2 l: ^% C9 I# @
50.                     }
    
51.                     $line = trim(socket_read($newClient, 1024));
    
52.                     $this->handshaking($newClient, $line);
    2 E! Y- X9 z4 m) N
53.                     //获取client ip
    % C+ R5 N2 K) K5 j5 X1 N
54.                     socket_getpeername ($newClient, $ip);
    ! N9 t. r8 U* Z, |& e& q
55.                     $clients[$ip] = $newClient;
    
56.                     echo  "Client ip:{$ip}   \n";
    
57.                     echo "Client msg:{$line} \n";
    
58.                 } else {
    $ l9 C; |2 b8 q: d: T, F% R
59.                     socket_recv($_sock, $buffer,  2048, 0);
    
60.                     $msg = $this->message($buffer);
    
61.                     //在这里业务代码
    $ r" }6 b  R) F3 S
62.                     echo "{$key} clinet msg:",$msg,"\n";
    0 _* ~- a0 ~0 |4 K7 a* z
63.                     fwrite(STDOUT, 'Please input a argument:');
    1 |& x$ S! l1 F" P
64.                     $response = trim(fgets(STDIN));
    ; Y1 M9 \# n, r$ m
65.                     $this->send($_sock, $response);
    
66.                     echo "{$key} response to Client:".$response,"\n";
    $ h# q! u! A$ Z9 X
67.                 }
    
68.             }
    
69.         }
    / l; {1 p$ u. v' Q
70.     }
    
71. 
    
72.     /**
    + y$ y5 M; p% O) |1 q4 b  y, I
73.      * 握手处理
    ( ]9 ]* _6 E! X$ d3 P  f
74.      * @param $newClient socket
    , f/ ~/ B* [( e* R" E: P! o
75.      * @return int  接收到的信息
    
76.      */
    " h  g4 I: M& N- d# v: t
77.     public function handshaking($newClient, $line){
    
78. 
    
79.         $headers = array();
    * w# v- _/ F  Z/ U! w
80.         $lines = preg_split("/\r\n/", $line);
    & U1 w- ]+ l. m
81.         foreach($lines as $line)
    
82.         {
    
83.             $line = chop($line);
    
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    
85.             {
    : T! e5 {, _# l) l
86.                 $headers[$matches[1]] = $matches[2];
    
87.             }
    1 G/ X. i! P% V4 G3 G8 {3 C" a
88.         }
    
89.         $secKey = $headers['Sec-WebSocket-Key'];
    
90.         $secAccept = base64_encode(pack('H*', sha1($secKey . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11')));
    3 K+ O) u  V7 {0 A+ A
91.         $upgrade  = "HTTP/1.1 101 Web Socket Protocol Handshake\r\n" .
    
92.             "Upgrade: websocket\r\n" .
    6 }  B* t: [! J" B
93.             "Connection: Upgrade\r\n" .
    
94.             "WebSocket-Origin: $this->address\r\n" .
    
95.             "WebSocket-Location: ws://$this->address:$this->port/websocket/websocket\r\n".
    ' a; U  B/ ~& X9 m& E
96.             "Sec-WebSocket-Accept:$secAccept\r\n\r\n";
    $ T: A8 {9 R& t* m6 s
97.         return socket_write($newClient, $upgrade, strlen($upgrade));
    ' {) @" R% S# P7 A9 X
98.     }
    
99. 
    
100.     /**
     7 n7 C3 A: i# V" r1 p, t" b* \
101.      * 解析接收数据
     8 k: B: ~. E5 H5 m0 H* X2 o* G
102.      * @param $buffer
     
103.      * @return null|string
     & F3 n* Z' Y4 ^
104.      */
       H# a, @8 W  L4 s- E
105.     public function message($buffer){
     
106.         $len = $masks = $data = $decoded = null;
     
107.         $len = ord($buffer[1]) & 127;
     ) ]! v, M' [# h9 N* P
108.         if ($len === 126)  {
     ; ~' Q+ Y% [5 \& b$ a% o
109.             $masks = substr($buffer, 4, 4);
     / h+ ^$ V  S/ x6 K( X0 u' [9 ~
110.             $data = substr($buffer, 8);
     + E1 U* P) U3 H; H  l0 V5 q
111.         } else if ($len === 127)  {
     6 [) U# L! H" b, H& ^2 r* Y
112.             $masks = substr($buffer, 10, 4);
     - ~. K5 Q/ i9 e' X# T
113.             $data = substr($buffer, 14);
     
114.         } else  {
     
115.             $masks = substr($buffer, 2, 4);
     2 m0 s, h/ F3 t4 O
116.             $data = substr($buffer, 6);
     
117.         }
     * Y2 g+ G; y3 t: Z& k. `0 z
118.         for ($index = 0; $index < strlen($data); $index++) {
     8 s7 Y9 D& H' u" r& k
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     
120.         }
     
121.         return $decoded;
     
122.     }
     ! K2 M* V) @  _7 h8 q  M1 ?
123. 
     ' O1 }% a5 g1 ~4 k) `0 @, m7 l
124.     /**
     5 b% t9 K9 }8 P& \6 f
125.      * 发送数据
     
126.      * @param $newClinet 新接入的socket
     
127.      * @param $msg   要发送的数据
     ! R3 m- v+ J0 r6 ?
128.      * @return int|string
     # o1 }. _1 A3 P# M
129.      */
     - H3 M& c& o% Q. j
130.     public function send($newClinet, $msg){
     3 \5 K3 a7 ^0 J: R0 J1 P8 e
131.         $msg = $this->frame($msg);
     * p% i% h( A/ _6 I5 a
132.         socket_write($newClinet, $msg, strlen($msg));
     
133.     }
     , m' d7 a* v) v( b& s/ t! h! }
134. 
     
135.     public function frame($s) {
     ' V* P! @( i4 Q4 \5 j
136.         $a = str_split($s, 125);
     3 m- O. H8 a. {9 [2 A
137.         if (count($a) == 1) {
     
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     
139.         }
     
140.         $ns = "";
     % U* y$ M, w& p/ W
141.         foreach ($a as $o) {
     * i9 A0 S8 S7 |+ k) g
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     
143.         }
     
144.         return $ns;
     5 F' x% B1 f2 Q
145.     }
     . K2 n1 w* c+ Q6 W* m7 i/ Q
146. 
     
147.     /**
     
148.      * 关闭socket
     % Q+ j) H! F6 G' f  p* h
149.      */
     
150.     public function close(){
     
151.         return socket_close($this->_sockets);
     
152.     }
     ( @6 E# g. X: |1 r
153. }
     ) r2 [4 H! Z& h" l( K; f
154. 
     1 ^- o! v+ m3 i2 G# y7 D; Y
155. $sock = new SocketService();
     ; T( N- K% [7 N  R/ f' t
156. $sock->run();
     
157. 
     - H, ?" n9 B  F

复制代码

web.html
; z! a. ?) D/ M0 C% s( `$ E

1. <!doctype html>
   1 \" r. d3 T; E$ y
2. <html lang="en">
   
3. <head>
   
4.   <meta charset="UTF-8">
   6 f! J" N# S) n3 ~" V6 n) Z% C& p
5.   <meta name="viewport" content="width=device-width,initial-scale=1, maximum-scale=1, user-scalable=no">
   8 W3 v8 F  q: i, f$ B
6.   <title>websocket</title>
   . I  `7 c8 f) n9 ]# p" \
7. </head>
   ) E/ U$ [9 ]' q) x" J
8. <body>
   / d2 C& ~0 X# x7 x0 L" j8 m& a
9. <input id="text" value="">
   0 X2 Q: e1 Q: i$ k) s5 Z: r
10. <input type="submit" value="send" onclick="start()">
    
11. <input type="submit" value="close" onclick="close()">
    + ~5 `$ }$ f) h- r) ^0 F$ l
12. <div id="msg"></div>
    
13. <script>
    
14. /**
    
15. 0：未连接
    
16. 1：连接成功，可通讯
    " a; _# u/ L: A1 I7 H) K
17. 2：正在关闭
    4 g7 E0 d+ O- v2 D, ?& K6 h
18. 3：连接已关闭或无法打开
    
19. */
    2 N9 y  _4 O* k9 S# Z. D
20. 
    . j' y. z* c8 G7 U+ k
21.     //创建一个webSocket 实例
    
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    : d. v' l2 s7 M% N4 n  K) Y0 G0 i5 R1 t
23. 
    
24. 
    
25.     webSocket.onerror = function (event){
    
26.         onError(event);
    
27.     };
    ( V& k' X+ O( \$ J5 a* V
28. 
    
29.     // 打开websocket
    
30.     webSocket.onopen = function (event){
    6 _6 M8 B8 ~# T7 v+ J
31.         onOpen(event);
    3 L, l# l1 p/ d, P  r7 n: x  @
32.     };
    
33. 
    % ^, |' T  V& P2 {9 G$ J$ u8 k6 r
34.     //监听消息
    3 O( |( r) E- i% f7 ?( A
35.     webSocket.onmessage = function (event){
    
36.         onMessage(event);
    
37.     };
    
38. 
    0 B& S- `1 r7 ?' K. o: s' m
39. 
    0 e/ o+ Z/ p4 z7 l
40.     webSocket.onclose = function (event){
    ; V& {! K- j( O& t4 z
41.         onClose(event);
    
42.     }
    * y: w& w( p/ j% J5 R* R
43. 
    3 C; x& z# L# {; X
44.     //关闭监听websocket
    
45.     function onError(event){
    + E& D  @1 w) H, ?8 h7 v
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
47.         console.log("error"+event.data);
    - V3 D# A6 f/ |' r" F3 D9 Z7 G; e
48.     };
    : {* d6 k. j. `
49. 
    0 u  V: B, d% S+ |5 ?- C' v; j
50.     function onOpen(event){
    , G6 X) D$ Y# F0 G7 T
51.         console.log("open:"+sockState());
    ( s+ b. A5 t# i- x% Y2 t
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    , m$ K: V# ]6 X
53.     };
    
54.     function onMessage(event){
    % Q6 d( P  C3 i0 C, P
55.         console.log("onMessage");
    
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    & [* D  G5 a8 Z  Y
57.     };
    
58. 
    2 c; z+ x) E) h
59.     function onClose(event){
    
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    9 P2 H( _$ y' D. a9 W  h, ]
61.         console.log("close:"+sockState());
      R/ M8 E4 S2 S
62.         webSocket.close();
    
63.     }
    
64. 
    ! Q6 r, N) y) A
65.     function sockState(){
    
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    
67.             return status[webSocket.readyState];
    # S3 A& C( n7 R( j5 F- I& p
68.     }
    
69. 
    
70. 
    
71. 
    & Z" X  F; q2 f1 z
72. function start(event){
    
73.         console.log(webSocket);
    ) O8 U, G* \, @: F. y# j
74.         var msg = document.getElementById('text').value;
    
75.         document.getElementById('text').value = '';
    
76.         console.log("send:"+sockState());
    $ `9 V5 Z1 f. y, d
77.         console.log("msg="+msg);
    
78.         webSocket.send("msg="+msg);
    
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    
80.     };
    
81. 
    
82.     function close(event){
    
83.         webSocket.close();
    2 b8 O0 u# I, E4 i" x
84.     }
    9 O. k) `: d- N: e
85. </script>
    
86. </body>
    
87. </html>

复制代码

$ m7 ]% S2 }. H8 [) \