php实现websocket实时消息推送

admin

php实现websocket实时消息推送


SocketService.php

1. <?php
   
2. /**
   
3. * Created by xwx
   
4. * Date: 2017/10/18
   ' U1 q/ \& t: }/ x. s8 D0 k
5. * Time: 14:33
   
6. */
   ; \5 c, o: K7 ~
7. 
   7 P+ t6 E- i0 C( s) N4 u
8. class SocketService
   
9. {
   , o7 o4 ~/ g+ A% t0 D2 [
10.     private $address  = '0.0.0.0';
    
11.     private $port = 8083;
    9 S. u; Z0 n# ]% P0 v
12.     private $_sockets;
    1 U6 R+ V2 ?0 J$ @9 @
13.     public function __construct($address = '', $port='')
    : `( H/ `) ~( J! u3 |. S/ i
14.     {
    
15.             if(!empty($address)){
    5 }! [, O9 d# K; W9 D
16.                 $this->address = $address;
    + c# u7 h' p+ x( Q5 o( ], w4 x
17.             }
    8 Y7 F, _' \0 i: W$ B
18.             if(!empty($port)) {
    5 q& {9 [- X# G: _9 m3 W6 ]
19.                 $this->port = $port;
    
20.             }
    ; n' _7 p0 z% F' O) d1 y/ }
21.     }
    
22. 
    5 Q5 l6 l, T, [- Y& [3 I* k
23.     public function service(){
    ) |1 k' A. h. p- U' c1 S* F) Q
24.         //获取tcp协议号码。
    ! s5 w9 @' t5 H8 i) u& \9 P/ Y  {( R
25.         $tcp = getprotobyname("tcp");
    
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    ! K9 v- n( V4 z! y2 P0 y
28.         if($sock < 0)
    # L# I% f; N- \8 C+ V9 ?. A6 _. T
29.         {
    
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    
31.         }
    ) Z/ {8 N% z+ L9 u! b( U
32.         socket_bind($sock, $this->address, $this->port);
      h5 y( n2 F$ t( E, S( C3 @
33.         socket_listen($sock, $this->port);
    
34.         echo "listen on $this->address $this->port ... \n";
    8 b+ a6 ^' S' a- I) \1 V6 Y: Y
35.         $this->_sockets = $sock;
    
36.     }
    
37. 
    + I" t/ \; Z* O! F
38.     public function run(){
    : s4 I9 |/ o4 d2 U
39.         $this->service();
    : W9 ]: Q! X' B$ g
40.         $clients[] = $this->_sockets;
    ) L! v& _" B, \& `+ C9 `, X  ]/ ]
41.         while (true){
    3 z1 e  J3 a0 j# r9 g
42.             $changes = $clients;
    
43.             $write = NULL;
    
44.             $except = NULL;
    
45.             socket_select($changes,  $write,  $except, NULL);
    
46.             foreach ($changes as $key => $_sock){
    ; i: E- V" p9 G% Q8 m' U$ A% y1 I
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    
48.                     if(($newClient = socket_accept($_sock))  === false){
    & j4 W# P9 t# j) f8 _/ h$ m2 K8 _  |3 u
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    9 g$ O' v  F# d1 V6 H4 c$ a7 U& c
50.                     }
    , j+ n4 g! `+ A. \7 F6 ~6 \- E
51.                     $line = trim(socket_read($newClient, 1024));
    3 b4 Q/ j$ j4 F( x2 z, ~$ C4 v
52.                     $this->handshaking($newClient, $line);
    & A' C9 [$ ~/ R( c+ ^' t9 u
53.                     //获取client ip
    * f$ B& \* D% Q% i- G, e& P. d
54.                     socket_getpeername ($newClient, $ip);
    
55.                     $clients[$ip] = $newClient;
    ; r$ G8 a3 o; t4 [7 Q/ _- L$ I
56.                     echo  "Client ip:{$ip}   \n";
    
57.                     echo "Client msg:{$line} \n";
    7 |1 }1 m1 F  w
58.                 } else {
    7 H8 v# U" X, z. H5 W' ^4 m; s
59.                     socket_recv($_sock, $buffer,  2048, 0);
    1 G- [4 i; h5 `1 }
60.                     $msg = $this->message($buffer);
    8 b: P! ^+ F6 }" w& P
61.                     //在这里业务代码
    ! M  W' u( }- _  [" A
62.                     echo "{$key} clinet msg:",$msg,"\n";
    % U3 `! @6 H" U3 X* s
63.                     fwrite(STDOUT, 'Please input a argument:');
    
64.                     $response = trim(fgets(STDIN));
    3 Z2 x' |4 I, I8 m7 a) `6 l! b
65.                     $this->send($_sock, $response);
    1 M1 B, i6 x3 z' n% V( u( y
66.                     echo "{$key} response to Client:".$response,"\n";
    & y8 P! d& D* V
67.                 }
      c' m5 t  Y9 `4 l
68.             }
    % f) t0 }  N& i3 {& e8 ^& s% j* h2 B
69.         }
      [4 @, j/ E1 r# q2 P& C$ v
70.     }
    4 m  z: ]* h# O8 q9 l$ e& r7 b6 v
71. 
    5 y/ o: P2 ]/ @$ H( I, D0 w
72.     /**
    7 t1 I4 B) B  Z# K1 }# r1 n
73.      * 握手处理
    . {+ s8 G; x* O0 Q" `
74.      * @param $newClient socket
    3 @& x8 m$ ^& m! q
75.      * @return int  接收到的信息
    / O% b% Z; h% `& F8 \4 _
76.      */
    
77.     public function handshaking($newClient, $line){
    
78. 
    # @, y4 t# q7 S+ ?' {
79.         $headers = array();
    
80.         $lines = preg_split("/\r\n/", $line);
    
81.         foreach($lines as $line)
    
82.         {
    
83.             $line = chop($line);
    5 L8 x, G* `& q; p' A
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    7 _8 c- `, M/ ]1 U
85.             {
    
86.                 $headers[$matches[1]] = $matches[2];
    - \% d. ]8 x/ I6 ?
87.             }
    , ]0 F4 b/ U9 B% H# X  Z- k
88.         }
    
89.         $secKey = $headers['Sec-WebSocket-Key'];
    
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" .
    
92.             "Upgrade: websocket\r\n" .
    5 f- a) S, L0 E, p4 C) X! Y3 b9 _
93.             "Connection: Upgrade\r\n" .
    / V) h1 Y3 _7 ^8 ]" Y# C4 ]! @
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";
    3 h3 I. n  E( v2 |" v
97.         return socket_write($newClient, $upgrade, strlen($upgrade));
      `( P9 A1 B5 f+ M6 \) e
98.     }
    
99. 
    # |8 p. m3 V9 N6 j2 b/ f
100.     /**
     % D0 f" o# W( W9 k7 R. x
101.      * 解析接收数据
     ( ^- S& v, O4 I3 c
102.      * @param $buffer
     
103.      * @return null|string
     
104.      */
     
105.     public function message($buffer){
     ! w; l. j) e6 s% U- j
106.         $len = $masks = $data = $decoded = null;
     ; N( o' V. a' h4 v
107.         $len = ord($buffer[1]) & 127;
     
108.         if ($len === 126)  {
     & h" D3 G6 W8 B$ E
109.             $masks = substr($buffer, 4, 4);
     9 R  M# a" p# H7 ^2 f2 T$ @
110.             $data = substr($buffer, 8);
     
111.         } else if ($len === 127)  {
     ' o9 N4 v0 h. F' O
112.             $masks = substr($buffer, 10, 4);
     
113.             $data = substr($buffer, 14);
     ) K& _+ _4 ?& Y3 s$ h; F5 R
114.         } else  {
     
115.             $masks = substr($buffer, 2, 4);
     
116.             $data = substr($buffer, 6);
     5 ]; h$ ^; q1 l# U
117.         }
     
118.         for ($index = 0; $index < strlen($data); $index++) {
     . Q. U+ n1 P$ q2 b# P
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     + V- u" a" ~( i: Y: T; N  H  j- ^1 \
120.         }
     ) k" w+ ?' r7 h6 E
121.         return $decoded;
     5 Z1 N8 L0 P  W0 h+ W2 ]3 B
122.     }
     
123. 
     ) ]0 k% q) r7 G& J" e8 b
124.     /**
     
125.      * 发送数据
     
126.      * @param $newClinet 新接入的socket
     ( D- n6 {- H! U- Z# K' j
127.      * @param $msg   要发送的数据
     6 y! {: B5 W0 l9 G
128.      * @return int|string
     
129.      */
     
130.     public function send($newClinet, $msg){
     . e7 l" ?# s2 Z+ z
131.         $msg = $this->frame($msg);
     2 k5 M" g& a# j3 Q, }$ Z. F
132.         socket_write($newClinet, $msg, strlen($msg));
     , e9 m* z! j$ k5 c# K+ W2 J3 E  N/ U
133.     }
     ( @5 N% N6 m5 E* ^/ x
134. 
     . J* W# t8 H2 L9 }* P2 U/ A9 J% S
135.     public function frame($s) {
     ' L. F- D4 U$ W( g/ T$ s
136.         $a = str_split($s, 125);
     
137.         if (count($a) == 1) {
     
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     8 T/ d; k) }9 y; v
139.         }
     ; u6 U2 ~' S# }& U
140.         $ns = "";
     
141.         foreach ($a as $o) {
     
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     
143.         }
     
144.         return $ns;
     - J! b" }# O/ ?" Y( S- q3 F. r
145.     }
     ( r: y) T6 _* o! J8 g+ m9 `
146. 
     
147.     /**
       |1 O  G) @( v& ?
148.      * 关闭socket
     
149.      */
     - _3 c- {; H  c9 B/ S8 \
150.     public function close(){
     7 I1 \* o7 F% L. E# ?- r
151.         return socket_close($this->_sockets);
     4 [+ {- A. Q: `7 H6 t* f0 J
152.     }
     
153. }
     
154. 
     
155. $sock = new SocketService();
     % i# ^* f! _, U" \: D  A, Q
156. $sock->run();
     
157. 
     # }; h  T7 w& \; K0 k- b

复制代码

web.html

1. <!doctype html>
   
2. <html lang="en">
   3 ?# i: G$ P# a
3. <head>
   
4.   <meta charset="UTF-8">
   
5.   <meta name="viewport" content="width=device-width,initial-scale=1, maximum-scale=1, user-scalable=no">
   ) }! f, s. z# Z2 G
6.   <title>websocket</title>
   
7. </head>
   ' A8 y6 F* c# {/ e/ L( v; j" v4 N
8. <body>
   
9. <input id="text" value="">
   6 c/ y3 b' i2 m# r& r- p0 v3 y$ R
10. <input type="submit" value="send" onclick="start()">
    
11. <input type="submit" value="close" onclick="close()">
    
12. <div id="msg"></div>
    ' g1 ?$ A. _9 x" k
13. <script>
    * J% M" q; K7 f" J! g
14. /**
    
15. 0：未连接
    ' r) T) |/ B; s6 H: U3 L5 E
16. 1：连接成功，可通讯
    4 ^0 k; L! g5 x$ z& V! }( R
17. 2：正在关闭
    
18. 3：连接已关闭或无法打开
    
19. */
    
20. 
    ; E* }9 N, `' r4 k
21.     //创建一个webSocket 实例
    0 i. C5 `& n+ X8 q& p1 V- S5 q
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    ) ]4 s$ k& r* |0 K3 _- k, W* x% w
23. 
    - ?$ p  C/ Z8 F! W5 n6 Y. A  k* p
24. 
    
25.     webSocket.onerror = function (event){
    
26.         onError(event);
    
27.     };
    
28. 
    - O( h; t3 v$ m- y" C5 ~" `6 a
29.     // 打开websocket
    
30.     webSocket.onopen = function (event){
    
31.         onOpen(event);
    5 \7 P* z7 j  Y
32.     };
    
33. 
    : j2 J- T* ^* O; k% \. t
34.     //监听消息
    
35.     webSocket.onmessage = function (event){
    * a- k2 f9 ~0 Q3 E! E, W* L& c
36.         onMessage(event);
    
37.     };
    ! G) {* `, Y+ h/ Z* n7 u3 B
38. 
    1 J2 D* S; ?* w$ n8 K7 P/ ]
39. 
    
40.     webSocket.onclose = function (event){
    
41.         onClose(event);
    . C5 ~! {, ]  h1 n
42.     }
    
43. 
    
44.     //关闭监听websocket
    . e: q( l1 u  v/ D; t% C' Z) p
45.     function onError(event){
    # S6 M1 m! n+ Y, x7 G+ m
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
47.         console.log("error"+event.data);
    ; v7 `  Q% E; p$ E0 E
48.     };
    8 L( t1 w5 P: p& Z
49. 
    
50.     function onOpen(event){
    
51.         console.log("open:"+sockState());
    1 h) v: o# ?, J' |* I
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    
53.     };
    " I4 o& J8 p: s0 Y; @
54.     function onMessage(event){
    
55.         console.log("onMessage");
    
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    5 N7 v$ [$ ]3 x5 A
57.     };
    ) ]) R# J8 C1 ~9 U* V. j
58. 
    . u) v3 J4 N6 t2 `
59.     function onClose(event){
    - Q5 {) ~1 z6 [: _# ~. [
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
61.         console.log("close:"+sockState());
    
62.         webSocket.close();
    
63.     }
    
64. 
    
65.     function sockState(){
    9 ?' O# @& V/ S) x) {0 o
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    7 w- G' q" m( U: H; D" o
67.             return status[webSocket.readyState];
    8 J/ V2 G) v# Y1 C% m3 C7 v% Z( |
68.     }
    5 R1 Z; ~8 y$ f! @+ Y
69. 
    
70. 
    ' q) M* Q4 C7 h4 n3 N! c
71. 
    ! {3 M3 q1 w, w; i) \
72. function start(event){
    % R# a- C; S- }- R( |1 ?( v$ S. F% p
73.         console.log(webSocket);
    * o/ p  [: A; e
74.         var msg = document.getElementById('text').value;
    
75.         document.getElementById('text').value = '';
    
76.         console.log("send:"+sockState());
    
77.         console.log("msg="+msg);
    
78.         webSocket.send("msg="+msg);
    
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    
80.     };
    / h; B+ \9 G  X! g6 D) @" E$ R1 r
81. 
    
82.     function close(event){
    5 [2 k7 [" g* d- b3 `) R5 \
83.         webSocket.close();
    + v! i, j+ _4 y; y/ x' M
84.     }
    
85. </script>
    - X& Z/ [4 K7 t" X' G
86. </body>
    
87. </html>

复制代码

, U  ~: {8 w, |) r9 _