php实现websocket实时消息推送

admin

php实现websocket实时消息推送
9 @- a% g. X$ F$ h, J" C7 O4 f
6 K' \1 Q' r' J$ X
0 E0 P; v; X% vSocketService.php

1. <?php
   , V/ I$ t5 m$ ]
2. /**
   ! W0 z& K. v# E1 E
3. * Created by xwx
   
4. * Date: 2017/10/18
   2 p7 y9 ^* u" e" o1 Z8 f5 R" L
5. * Time: 14:33
   
6. */
   ( w+ u& z3 u0 F' _
7. 
   5 H+ o% t3 A. {; y( D$ N' F
8. class SocketService
   
9. {
   " w5 A. N4 `' |7 F
10.     private $address  = '0.0.0.0';
    : _7 |3 Y2 s" L. x( C# H5 F5 I
11.     private $port = 8083;
    $ W! j/ P$ ]5 V" \5 c  w) j
12.     private $_sockets;
    
13.     public function __construct($address = '', $port='')
    ' L; [) c5 R7 _1 G$ S7 c1 Z
14.     {
    
15.             if(!empty($address)){
    
16.                 $this->address = $address;
    $ j5 f$ o5 n3 p. O5 |8 u4 _
17.             }
    
18.             if(!empty($port)) {
    
19.                 $this->port = $port;
    
20.             }
    
21.     }
    
22. 
    / k- M9 T) x) N0 K+ [  V
23.     public function service(){
    
24.         //获取tcp协议号码。
    
25.         $tcp = getprotobyname("tcp");
    # I: v) v; @+ z$ v8 h9 P$ C
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    7 P- O; W" e9 V2 y" O
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    
28.         if($sock < 0)
    9 q- b5 b6 s+ g! w5 R
29.         {
    ; @& t! y. r$ i0 I
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    & W+ u4 V" m. z  Q6 _& N# p; X2 {: C
31.         }
    " z0 P& s. o- k6 ]  X" K
32.         socket_bind($sock, $this->address, $this->port);
    
33.         socket_listen($sock, $this->port);
    " T' z% H- r- |% e3 Y/ r5 D' N
34.         echo "listen on $this->address $this->port ... \n";
    4 Y! Z4 e5 ]/ p; p+ x% a
35.         $this->_sockets = $sock;
    , f. F) M5 d1 M; M: B
36.     }
    
37. 
    5 E. g" i- h8 z, `" y. {! ^5 e; s
38.     public function run(){
    . V+ s- J' }1 l4 U! Z: f: q
39.         $this->service();
    : Y* Q8 G- u( \) I- M- h
40.         $clients[] = $this->_sockets;
    
41.         while (true){
    
42.             $changes = $clients;
    
43.             $write = NULL;
    
44.             $except = NULL;
    : E# t* Q6 d0 o, {' ?2 y
45.             socket_select($changes,  $write,  $except, NULL);
    & E3 f1 U/ _+ d  U% {( Y
46.             foreach ($changes as $key => $_sock){
    ' s! t, q9 o: C# Y' \
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    
48.                     if(($newClient = socket_accept($_sock))  === false){
    # v# c1 j9 \) p" T$ i2 x
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    
50.                     }
    ' x$ Q, e% R5 w  Q
51.                     $line = trim(socket_read($newClient, 1024));
    / `' y! w7 N& `7 Q2 d; c
52.                     $this->handshaking($newClient, $line);
    
53.                     //获取client ip
    
54.                     socket_getpeername ($newClient, $ip);
    
55.                     $clients[$ip] = $newClient;
    
56.                     echo  "Client ip:{$ip}   \n";
    
57.                     echo "Client msg:{$line} \n";
    9 A6 f0 x: l3 {# @% T. ]4 c
58.                 } else {
    
59.                     socket_recv($_sock, $buffer,  2048, 0);
    
60.                     $msg = $this->message($buffer);
    
61.                     //在这里业务代码
    * {' q3 M9 _. k, m+ P
62.                     echo "{$key} clinet msg:",$msg,"\n";
    . z! @- P' B, t3 b  U8 K
63.                     fwrite(STDOUT, 'Please input a argument:');
    
64.                     $response = trim(fgets(STDIN));
    ' C5 F8 _% _. C. N2 Z" V& B
65.                     $this->send($_sock, $response);
      c4 z) @6 Y1 I4 P
66.                     echo "{$key} response to Client:".$response,"\n";
    
67.                 }
    
68.             }
    
69.         }
    4 W$ G5 e7 a; [( l
70.     }
    % x4 D3 G% _; v
71. 
    ) \/ D2 c9 C  \' m* j$ S; K
72.     /**
    
73.      * 握手处理
    $ p. N5 c6 o: N" D9 O
74.      * @param $newClient socket
    
75.      * @return int  接收到的信息
    
76.      */
    
77.     public function handshaking($newClient, $line){
    8 ?* L5 ?; x3 I  ~
78. 
    
79.         $headers = array();
    ! Y& P2 S; |% Z. u! x2 f  h3 i
80.         $lines = preg_split("/\r\n/", $line);
      q8 E2 p* o, |& P
81.         foreach($lines as $line)
    
82.         {
    ! h9 X: ^5 G" h7 h9 e# S) v0 j2 H; R
83.             $line = chop($line);
    
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    ! ~' B( Z; v% l. i: W
85.             {
    
86.                 $headers[$matches[1]] = $matches[2];
    
87.             }
    * h' h/ [9 t1 O. e& _! l
88.         }
    
89.         $secKey = $headers['Sec-WebSocket-Key'];
    
90.         $secAccept = base64_encode(pack('H*', sha1($secKey . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11')));
    0 r' A. x2 O  o, R( ^1 g: a
91.         $upgrade  = "HTTP/1.1 101 Web Socket Protocol Handshake\r\n" .
    ' E( K5 V- f. e/ @
92.             "Upgrade: websocket\r\n" .
    
93.             "Connection: Upgrade\r\n" .
    
94.             "WebSocket-Origin: $this->address\r\n" .
    
95.             "WebSocket-Location: ws://$this->address:$this->port/websocket/websocket\r\n".
    2 W9 |; P4 R* q0 v' h
96.             "Sec-WebSocket-Accept:$secAccept\r\n\r\n";
    3 Q3 _) q6 |# [* G- z( _1 ]8 O
97.         return socket_write($newClient, $upgrade, strlen($upgrade));
    # ~( ~" D( m* O, P# |+ \
98.     }
    
99. 
    
100.     /**
     . v/ h# E9 ~5 F, G/ p/ j4 v
101.      * 解析接收数据
     
102.      * @param $buffer
     7 G. P- F8 {8 R, T
103.      * @return null|string
     ! f; C8 j& {3 y' l3 G
104.      */
     9 t2 ?- D0 P  F! X( V2 K9 e& O
105.     public function message($buffer){
     5 |& K" s- K( W
106.         $len = $masks = $data = $decoded = null;
     
107.         $len = ord($buffer[1]) & 127;
     , c8 A, Q) n  u  x4 t5 P) x
108.         if ($len === 126)  {
     
109.             $masks = substr($buffer, 4, 4);
     
110.             $data = substr($buffer, 8);
     
111.         } else if ($len === 127)  {
     1 I5 V- R/ f; X5 [: f6 z0 y
112.             $masks = substr($buffer, 10, 4);
     
113.             $data = substr($buffer, 14);
     
114.         } else  {
     
115.             $masks = substr($buffer, 2, 4);
     - M9 h" N8 o! l% }5 b
116.             $data = substr($buffer, 6);
     
117.         }
     3 U( E- @8 @( W8 d
118.         for ($index = 0; $index < strlen($data); $index++) {
     
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     
120.         }
     2 U0 M+ b2 n# y: }4 h0 A
121.         return $decoded;
     
122.     }
     
123. 
     2 R: T2 P* j0 i) a, F
124.     /**
     
125.      * 发送数据
     
126.      * @param $newClinet 新接入的socket
     2 e4 k4 J8 R/ ]& R( E" V7 m
127.      * @param $msg   要发送的数据
     
128.      * @return int|string
     
129.      */
     & b. J0 H3 F3 d" c, F
130.     public function send($newClinet, $msg){
     7 o3 A. g& {( k$ c+ F5 U: Z
131.         $msg = $this->frame($msg);
     
132.         socket_write($newClinet, $msg, strlen($msg));
     . u9 r- J% {6 p2 T' R2 r; z. ]7 t% t
133.     }
     4 X% Q8 R+ S/ V5 K. s
134. 
     # \5 m& ~+ a# o0 H% W" {. e, ^4 S# ^
135.     public function frame($s) {
     
136.         $a = str_split($s, 125);
     
137.         if (count($a) == 1) {
     ( Z) J9 [3 Q* Q* D6 M* f
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     
139.         }
     / d8 U# O  N2 g- p2 `
140.         $ns = "";
     
141.         foreach ($a as $o) {
     " t4 {. H- N3 p& A
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     
143.         }
     
144.         return $ns;
     
145.     }
     5 R: v- o/ N6 t# O/ b8 h
146. 
     
147.     /**
       Y9 f0 ^! L) y
148.      * 关闭socket
     5 U+ A* `- A! n) Y5 q- I/ F- J
149.      */
     
150.     public function close(){
     ! l' s2 t- G+ \+ g4 {" ^) x' J
151.         return socket_close($this->_sockets);
     
152.     }
     
153. }
     4 d/ L5 i# ]" Z9 h* ?
154. 
     . o9 e. G. }. B% ]/ ]7 G7 h3 L7 X
155. $sock = new SocketService();
     " F, c/ a' i9 i5 X; D* _$ e
156. $sock->run();
     
157. 
     

复制代码

web.html

1. <!doctype html>
   , ?7 g0 i9 r' X
2. <html lang="en">
   
3. <head>
   4 B8 S( U5 ]6 k* P" A7 m
4.   <meta charset="UTF-8">
   / _+ h% `$ L: S
5.   <meta name="viewport" content="width=device-width,initial-scale=1, maximum-scale=1, user-scalable=no">
   . \% a( c( B6 \; X
6.   <title>websocket</title>
   ! ~+ K5 n, [$ ~
7. </head>
   
8. <body>
   
9. <input id="text" value="">
   
10. <input type="submit" value="send" onclick="start()">
    
11. <input type="submit" value="close" onclick="close()">
    # l5 \  X1 Y# H7 W
12. <div id="msg"></div>
    
13. <script>
    
14. /**
    
15. 0：未连接
    
16. 1：连接成功，可通讯
    
17. 2：正在关闭
    
18. 3：连接已关闭或无法打开
    
19. */
    % Y0 [3 m$ r8 y# E& ?" b6 g) h  v
20. 
    5 @1 M5 L# T2 M+ z" ^
21.     //创建一个webSocket 实例
    
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    + O6 l3 x. `* e6 w% w( b
23. 
    
24. 
    
25.     webSocket.onerror = function (event){
    
26.         onError(event);
    
27.     };
    % W; T$ q: r* u+ `' y
28. 
    
29.     // 打开websocket
    7 [, V5 [) F$ e' b3 u* u; a
30.     webSocket.onopen = function (event){
    : W/ {9 {( c' F/ S5 }% e2 f
31.         onOpen(event);
    9 T' g0 j' t7 j& u5 M
32.     };
    4 X" R9 J8 z" p  c2 r
33. 
    
34.     //监听消息
    $ i) `! o& G2 G; t0 J7 L
35.     webSocket.onmessage = function (event){
    
36.         onMessage(event);
    
37.     };
    
38. 
    7 ^7 k9 H/ P, m! R8 Z8 z
39. 
    6 O+ e* K4 k9 `; x$ Q
40.     webSocket.onclose = function (event){
    / I; V, p0 D! i! L  K! X
41.         onClose(event);
    
42.     }
    ! Z9 `# P0 N: Y! _( C0 S* A( ~
43. 
    , m0 D  a/ M; M) C9 i' J; W
44.     //关闭监听websocket
    
45.     function onError(event){
    
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    9 }2 L, M2 O* C; x6 K- n
47.         console.log("error"+event.data);
    
48.     };
    
49. 
    & ?! M1 d. l7 E
50.     function onOpen(event){
    - I. I% o5 M* G$ k# l0 t
51.         console.log("open:"+sockState());
    $ j& l9 Z6 X5 I& \# l  Q7 r
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    7 w0 O( o( i# h& Q; @' p8 I
53.     };
    
54.     function onMessage(event){
    0 _. v; o# F7 \- e2 Q
55.         console.log("onMessage");
    0 E; x! T3 E. G  V" B1 O
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    
57.     };
    
58. 
    2 Y- O4 ?+ p# Q/ \
59.     function onClose(event){
    . s2 `; i( k- ~- Z  l! x! k
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    * ~$ H) n7 I+ }
61.         console.log("close:"+sockState());
    
62.         webSocket.close();
    1 ^  L7 ?) C" C' M7 v7 l/ B
63.     }
    
64. 
    
65.     function sockState(){
    
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    
67.             return status[webSocket.readyState];
    
68.     }
    
69. 
    & q* j  G9 ?5 O, I  c
70. 
    * a* L+ y9 F" R; c& ^
71. 
    * O4 d! t7 l; V. k/ B( b6 o$ z
72. function start(event){
    2 w5 A2 x  C. y
73.         console.log(webSocket);
    
74.         var msg = document.getElementById('text').value;
    / u' y6 w; R- t9 T2 y4 \
75.         document.getElementById('text').value = '';
    + G  [$ y3 [' H, j
76.         console.log("send:"+sockState());
    
77.         console.log("msg="+msg);
    
78.         webSocket.send("msg="+msg);
    . }% h6 v$ C/ W
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    
80.     };
    
81. 
    
82.     function close(event){
    
83.         webSocket.close();
    9 s3 H+ W' R' T2 Y# g8 q. [5 @
84.     }
    
85. </script>
    
86. </body>
    
87. </html>

复制代码

' V! j4 N) e+ m! @' J. @6 A