php实现websocket实时消息推送

admin

php实现websocket实时消息推送

7 _9 \; C. y$ _5 k1 c
0 g. c9 t/ J+ ]7 K$ iSocketService.php
0 [% l, ?; e0 a6 u7 ]

1. <?php
   . ], D  i% I) y4 ?0 u2 }
2. /**
   4 P7 ~# g1 O* ^8 w$ e
3. * Created by xwx
   , x# b/ [* R) i5 ^
4. * Date: 2017/10/18
   6 D- s  `- C' l$ K$ p
5. * Time: 14:33
   $ F- ^$ f* Y7 [+ g+ a7 ~. A
6. */
   
7. 
   3 r7 z: \% s7 r( C  J4 j: [
8. class SocketService
   6 `; _  p5 s- U5 V* A* s8 @
9. {
   
10.     private $address  = '0.0.0.0';
    
11.     private $port = 8083;
    ' j. \* F0 |" p
12.     private $_sockets;
    4 ]; @6 q! S% r. h
13.     public function __construct($address = '', $port='')
    
14.     {
    
15.             if(!empty($address)){
    . O9 A& W8 ]. T. Q& N& w* F
16.                 $this->address = $address;
    8 o5 R0 Y( A% R& c/ u* f" W- l
17.             }
    & P) C. k/ ?9 }. v. @
18.             if(!empty($port)) {
    7 H8 o1 p* Q8 T. i$ B% L$ ?
19.                 $this->port = $port;
    / f8 ?- a9 c' {
20.             }
    & ^  H6 t8 E9 ~! t5 n: B% H
21.     }
    ) {/ ], R7 R1 v) N2 c7 S
22. 
    / w5 W7 E3 x3 w
23.     public function service(){
    + @, q' Q: ^, p, z5 Z3 p0 {* D( O
24.         //获取tcp协议号码。
    + n, m, m) p0 I% u* r
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)
    - U6 _2 T6 ]3 o9 d: y
29.         {
    : g) Y: Z/ S9 Y7 x) G
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    
31.         }
    : U# ]4 u5 }0 a# S( S% H& g
32.         socket_bind($sock, $this->address, $this->port);
    " d4 g% c; l5 E4 R
33.         socket_listen($sock, $this->port);
    & y/ b* n& w: S$ t! Y
34.         echo "listen on $this->address $this->port ... \n";
    
35.         $this->_sockets = $sock;
    
36.     }
    0 H9 u1 e. d) n* X6 K
37. 
    
38.     public function run(){
    5 a8 d8 O, U* E$ h3 a' B
39.         $this->service();
    
40.         $clients[] = $this->_sockets;
    
41.         while (true){
    
42.             $changes = $clients;
    
43.             $write = NULL;
    ! A2 o2 E% p, j6 ]' z1 W) `
44.             $except = NULL;
    3 {; x- |5 j# R; A
45.             socket_select($changes,  $write,  $except, NULL);
    
46.             foreach ($changes as $key => $_sock){
    4 O% b( z! d" @
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    
48.                     if(($newClient = socket_accept($_sock))  === false){
    ( B& ^% i7 b( s; s8 ]2 d* y
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    . v, F$ @# Y- l. F
50.                     }
    5 i' l# B" X: R. X2 {3 P
51.                     $line = trim(socket_read($newClient, 1024));
    
52.                     $this->handshaking($newClient, $line);
    6 `* A2 ~& C, o* d
53.                     //获取client ip
    
54.                     socket_getpeername ($newClient, $ip);
    1 F( j! t( w6 P. }+ c/ G. v5 X
55.                     $clients[$ip] = $newClient;
    3 i7 c2 a: g8 K# \2 ~7 F! k
56.                     echo  "Client ip:{$ip}   \n";
    * ^# Z! p; }: H
57.                     echo "Client msg:{$line} \n";
    
58.                 } else {
    ' H* R" t  t9 K2 h+ J) D7 U# p0 ?
59.                     socket_recv($_sock, $buffer,  2048, 0);
    
60.                     $msg = $this->message($buffer);
      n& ]% G9 E' B5 M# h
61.                     //在这里业务代码
    
62.                     echo "{$key} clinet msg:",$msg,"\n";
    
63.                     fwrite(STDOUT, 'Please input a argument:');
    ( I5 u5 e7 x, _/ Z
64.                     $response = trim(fgets(STDIN));
    ; c2 j9 U/ b! _! o6 t3 x
65.                     $this->send($_sock, $response);
    
66.                     echo "{$key} response to Client:".$response,"\n";
    ( {0 N. w6 E6 }; m
67.                 }
    
68.             }
    
69.         }
    
70.     }
    
71. 
    5 J3 y* t! @  n2 X9 w
72.     /**
    : S! T0 V* O6 \/ v
73.      * 握手处理
    0 A) w1 T% b$ `
74.      * @param $newClient socket
    
75.      * @return int  接收到的信息
    
76.      */
    8 `3 o9 [; ^1 r; V" V
77.     public function handshaking($newClient, $line){
    
78. 
    8 R% g; U; N4 U& [) ~
79.         $headers = array();
    $ f6 ~: E4 z$ B1 M0 g) `
80.         $lines = preg_split("/\r\n/", $line);
    
81.         foreach($lines as $line)
    
82.         {
    
83.             $line = chop($line);
    
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    0 M5 Z" }7 Y  `& K
85.             {
    ! h; X; ^9 |7 R7 ^. |
86.                 $headers[$matches[1]] = $matches[2];
    . t% \4 ]/ |5 y) ?" v3 v
87.             }
    $ I0 [) u# ~( e% G0 `, S# l! l
88.         }
    9 V: J  `: k- \, ]# R5 R
89.         $secKey = $headers['Sec-WebSocket-Key'];
    ' V& G7 ]- _* u. Q. W+ O" D8 E! {
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" .
    
93.             "Connection: Upgrade\r\n" .
    ; a# q  f( F2 R6 R% `
94.             "WebSocket-Origin: $this->address\r\n" .
    # Y) u2 W; a. S
95.             "WebSocket-Location: ws://$this->address:$this->port/websocket/websocket\r\n".
    6 V/ |; n, ^+ S; K2 V, F  f
96.             "Sec-WebSocket-Accept:$secAccept\r\n\r\n";
    
97.         return socket_write($newClient, $upgrade, strlen($upgrade));
    . P/ ]0 `- `4 k' m
98.     }
    ! M4 }/ o+ ^9 t$ x/ x
99. 
    / n6 t9 j  U; U& k, f+ Q
100.     /**
     
101.      * 解析接收数据
     ( ?1 D( L. R/ g5 A# E# u  P
102.      * @param $buffer
     : z* @9 ^3 V& e) X& b+ t
103.      * @return null|string
     
104.      */
     
105.     public function message($buffer){
     + O2 n! r2 U; U8 F/ g
106.         $len = $masks = $data = $decoded = null;
     
107.         $len = ord($buffer[1]) & 127;
     
108.         if ($len === 126)  {
     " @$ q6 ~& j$ O5 j  T
109.             $masks = substr($buffer, 4, 4);
     
110.             $data = substr($buffer, 8);
     ; H  W2 Y* }& E  @- q0 s; X6 M
111.         } else if ($len === 127)  {
     
112.             $masks = substr($buffer, 10, 4);
     
113.             $data = substr($buffer, 14);
     
114.         } else  {
     ' v$ `% C( `. g) U- |
115.             $masks = substr($buffer, 2, 4);
     ( g* X' m! K2 Y- G! D. A
116.             $data = substr($buffer, 6);
     * b  ]2 d2 W9 Y. o- {) J/ w  N) `
117.         }
     
118.         for ($index = 0; $index < strlen($data); $index++) {
     
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     
120.         }
     
121.         return $decoded;
     ! \1 |6 t% G, O1 b; ~2 [
122.     }
     " j' S7 {9 `/ x  G* v
123. 
     
124.     /**
     
125.      * 发送数据
     % J2 y8 T  h: b5 p7 S8 y
126.      * @param $newClinet 新接入的socket
     
127.      * @param $msg   要发送的数据
     
128.      * @return int|string
     4 X5 G$ Z. t. B0 A9 @! |
129.      */
     
130.     public function send($newClinet, $msg){
     
131.         $msg = $this->frame($msg);
     
132.         socket_write($newClinet, $msg, strlen($msg));
     
133.     }
     
134. 
     
135.     public function frame($s) {
     0 V, _2 K1 @& h
136.         $a = str_split($s, 125);
     ; P$ p( u" ]) f- t; ]
137.         if (count($a) == 1) {
     
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     
139.         }
     0 z/ p2 c5 w  F$ o; S4 K& q+ @
140.         $ns = "";
     - d0 L) Y5 o) @" M2 j" Z  ?
141.         foreach ($a as $o) {
     
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     
143.         }
     3 D# [2 a0 `, x) @, v' q
144.         return $ns;
     % i$ l0 d+ \: D% |& F3 w' o: M* S* Y
145.     }
     
146. 
     . H& x3 d& x$ l1 N: {, d
147.     /**
     ( \  K" P9 J  R5 W  ^3 F
148.      * 关闭socket
     
149.      */
     
150.     public function close(){
     
151.         return socket_close($this->_sockets);
     
152.     }
     
153. }
     9 n1 S9 y/ p* U6 \
154. 
     # }( k9 ~$ X- s5 S) y) B" x
155. $sock = new SocketService();
     
156. $sock->run();
     , o3 q% F6 ?% ?- d5 i; T
157. 
     ) @1 i5 T7 G. `

复制代码

web.html
5 z; L4 `( {* }+ q

1. <!doctype html>
   
2. <html lang="en">
   + ~5 g' m5 E# m2 ?/ q: S' o4 P
3. <head>
   
4.   <meta charset="UTF-8">
   - x/ T( R7 D4 t9 W
5.   <meta name="viewport" content="width=device-width,initial-scale=1, maximum-scale=1, user-scalable=no">
   
6.   <title>websocket</title>
   ! c+ j) n/ I' X5 p# }# _) E8 j
7. </head>
   7 h+ p7 x, `+ A% A6 m
8. <body>
   
9. <input id="text" value="">
   
10. <input type="submit" value="send" onclick="start()">
    6 p. }" M( F& Q, y. T4 P
11. <input type="submit" value="close" onclick="close()">
    4 {9 q6 ], h, I( [( p9 c' B) ]9 Y# y
12. <div id="msg"></div>
    ( i! d% u) F4 R
13. <script>
    
14. /**
    
15. 0：未连接
    * E# O. U- x" W( L. j6 V
16. 1：连接成功，可通讯
    
17. 2：正在关闭
    
18. 3：连接已关闭或无法打开
    
19. */
    
20. 
    & N( G2 Z9 K9 F. m
21.     //创建一个webSocket 实例
    & }) `8 [2 \; w2 v. G
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    * ~7 [- t! T, J8 F0 I$ M1 ]0 O: X
23. 
    ! S( j; t. w# z- q; T
24. 
    6 y( d8 y" u+ R& Z: H) Q9 z
25.     webSocket.onerror = function (event){
    % f, L- n. b/ V$ ~
26.         onError(event);
    4 o1 _6 I& d' P
27.     };
      z9 B; ~% g, H: ?
28. 
    
29.     // 打开websocket
    7 z' A, W$ N8 j) m3 v7 k8 B- {
30.     webSocket.onopen = function (event){
    + M6 o+ ~/ b6 H: b) Y
31.         onOpen(event);
    , B1 k! ?6 f# A7 i# u+ k  @
32.     };
    
33. 
    7 S  f( x$ A3 `- g% ~
34.     //监听消息
    # I1 j4 w: r0 `
35.     webSocket.onmessage = function (event){
    ' Y' f& b4 n2 y+ ~
36.         onMessage(event);
    / `3 w7 D3 ^. G' C2 f# h
37.     };
    7 V. j- Y5 T  k6 \6 ^* `( s
38. 
    5 o7 w9 z) t1 C  S
39. 
    
40.     webSocket.onclose = function (event){
    
41.         onClose(event);
    3 ~! F( Y( z/ x5 z4 T0 L
42.     }
    
43. 
    
44.     //关闭监听websocket
    
45.     function onError(event){
    
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
47.         console.log("error"+event.data);
    5 @9 L) B4 E9 d& Y! P5 X1 _
48.     };
    % B1 J" G7 _2 c( R) q+ s
49. 
    
50.     function onOpen(event){
    
51.         console.log("open:"+sockState());
    
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    
53.     };
    
54.     function onMessage(event){
    ) g& ~( y7 {) y; \2 u
55.         console.log("onMessage");
    
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    
57.     };
    7 U4 s5 F( y( V1 _
58. 
    . A6 c& S' M0 L- p/ X2 f
59.     function onClose(event){
    
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    : {0 r) }8 x: f& b/ I7 u$ N
61.         console.log("close:"+sockState());
    & g( V  g8 F! m9 X* ?
62.         webSocket.close();
    
63.     }
    ) P6 d5 A# T/ Z. I8 c1 c$ s9 _
64. 
    
65.     function sockState(){
    
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
      ~/ V" N4 h0 K) M
67.             return status[webSocket.readyState];
    & x+ \; c0 f; E* z; q$ \. `
68.     }
    
69. 
    
70. 
    4 a- n" ?$ {2 B
71. 
    
72. function start(event){
    
73.         console.log(webSocket);
    
74.         var msg = document.getElementById('text').value;
    % E0 o3 P3 E* l1 v- ?
75.         document.getElementById('text').value = '';
    ; {8 R5 t0 q7 m* a4 D
76.         console.log("send:"+sockState());
    5 Q) B  @: b1 h& H2 W; Q% U
77.         console.log("msg="+msg);
    
78.         webSocket.send("msg="+msg);
    6 m5 u$ s0 Y* f# c; |2 U: M
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    
80.     };
    
81. 
    
82.     function close(event){
    
83.         webSocket.close();
    ; z* f9 B4 ?7 b; o4 f6 r
84.     }
    
85. </script>
    , U/ j, c& E5 ?, q( `
86. </body>
    , X6 p5 b  \) U, Q8 M& i9 n+ H8 @6 O
87. </html>

复制代码

+ O3 X4 y& ?* |2 P" r2 V: W