php实现websocket实时消息推送

admin

php实现websocket实时消息推送


SocketService.php

1. <?php
   - P# Y- m- y9 V
2. /**
   / k  i0 H& u! h- ]/ z* ~* n- R
3. * Created by xwx
   
4. * Date: 2017/10/18
   * u3 a, L, \0 J7 q% t, H
5. * Time: 14:33
   * K  K4 b' ]0 _! W2 Z
6. */
   
7. 
   * y3 v* P3 r: N; m6 J
8. class SocketService
   
9. {
   
10.     private $address  = '0.0.0.0';
    # U/ M- R, O  Y7 e
11.     private $port = 8083;
    
12.     private $_sockets;
    
13.     public function __construct($address = '', $port='')
    
14.     {
    8 Q9 N' D; I+ I! F. x9 U0 O9 ^* x
15.             if(!empty($address)){
    
16.                 $this->address = $address;
    
17.             }
    : [3 M0 j. w; f7 B. W& |
18.             if(!empty($port)) {
    & w- {! X% Z# r
19.                 $this->port = $port;
    
20.             }
    
21.     }
    . [* v' {4 t0 {* B, F! E
22. 
    
23.     public function service(){
    
24.         //获取tcp协议号码。
    7 o" w4 X( v, a3 ^; v
25.         $tcp = getprotobyname("tcp");
    
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    8 c2 w. k3 x' V
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    4 g8 o) a: Y$ y- I
28.         if($sock < 0)
    ! R& T( }; M3 r4 X( q
29.         {
    . P% t: @$ J+ [. V8 W
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    
31.         }
    
32.         socket_bind($sock, $this->address, $this->port);
    4 \0 @0 l. g+ \1 U8 L5 c
33.         socket_listen($sock, $this->port);
    ' ^$ K! |( ^( U( M7 k# g1 `
34.         echo "listen on $this->address $this->port ... \n";
    4 b4 W+ F; o! E- s2 K" F
35.         $this->_sockets = $sock;
    
36.     }
    ( o4 s# Y  q- ~! V9 h( F3 P
37. 
    * \9 n) l4 ?1 _
38.     public function run(){
    . ]6 c- x4 |1 V; Q
39.         $this->service();
    1 I) R% z, \6 s0 a
40.         $clients[] = $this->_sockets;
    
41.         while (true){
    8 ~: O& ~0 Y. h, s" J8 @1 X
42.             $changes = $clients;
    ' H7 {, x9 N# d1 U3 P  ?% x
43.             $write = NULL;
    
44.             $except = NULL;
    
45.             socket_select($changes,  $write,  $except, NULL);
    
46.             foreach ($changes as $key => $_sock){
    
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    5 {' T( c9 t/ D
48.                     if(($newClient = socket_accept($_sock))  === false){
    9 q* S2 \2 @( Z% k8 u. x" ~
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    
50.                     }
    2 I5 c" [  A& T9 S2 C2 \: Y
51.                     $line = trim(socket_read($newClient, 1024));
    7 ]1 f% d7 K( [' e  w0 O
52.                     $this->handshaking($newClient, $line);
    7 f# ^0 p( Q6 g8 I4 c7 W8 y: X; |
53.                     //获取client ip
    
54.                     socket_getpeername ($newClient, $ip);
    " |5 g0 s) n8 B7 l" L' S+ `
55.                     $clients[$ip] = $newClient;
    
56.                     echo  "Client ip:{$ip}   \n";
    * `9 w3 K. w; [; a/ p; w
57.                     echo "Client msg:{$line} \n";
    
58.                 } else {
    ( `) z4 W7 B8 Z+ z5 v
59.                     socket_recv($_sock, $buffer,  2048, 0);
    
60.                     $msg = $this->message($buffer);
    
61.                     //在这里业务代码
    
62.                     echo "{$key} clinet msg:",$msg,"\n";
    " F2 q' R' A0 }: ^+ B7 I
63.                     fwrite(STDOUT, 'Please input a argument:');
    
64.                     $response = trim(fgets(STDIN));
    : r2 m' ~* l. R
65.                     $this->send($_sock, $response);
    1 k: E9 q% R' e) e3 C( M7 i/ `
66.                     echo "{$key} response to Client:".$response,"\n";
    6 K, f9 @( Q6 N( Y2 D0 N- T
67.                 }
    3 b: P% O/ t! Z
68.             }
    
69.         }
    
70.     }
    
71. 
    9 m4 X3 C# T5 m/ E
72.     /**
    3 H1 J6 `0 I0 u8 K$ c+ @/ C
73.      * 握手处理
    
74.      * @param $newClient socket
    
75.      * @return int  接收到的信息
    
76.      */
    : X. V) b6 Y' C' Z) Z( P, y. M2 c
77.     public function handshaking($newClient, $line){
    : b1 w& j6 G) q) L. \+ Z5 W* B) {
78. 
    
79.         $headers = array();
    
80.         $lines = preg_split("/\r\n/", $line);
    : Z1 i3 c1 z0 K3 n/ n* l' w+ ?
81.         foreach($lines as $line)
    
82.         {
    
83.             $line = chop($line);
    
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    - K5 n* `" r& N7 \, t7 @" \
85.             {
    
86.                 $headers[$matches[1]] = $matches[2];
    
87.             }
    / D% k6 t: E# K7 Z0 P+ x: \& Y
88.         }
    ! w+ a* {. D  `, e+ A
89.         $secKey = $headers['Sec-WebSocket-Key'];
    6 ], F' Q3 K4 X1 Q4 }" j
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" .
    # L6 i+ T3 p+ v0 ^' `3 }; g$ q
92.             "Upgrade: websocket\r\n" .
    . ^. w! ?, r2 Z6 B9 [. S; X
93.             "Connection: Upgrade\r\n" .
    ' _. H' W1 x! n: i
94.             "WebSocket-Origin: $this->address\r\n" .
    ' v+ T2 I! m2 q  A% V
95.             "WebSocket-Location: ws://$this->address:$this->port/websocket/websocket\r\n".
    - @! H' w, p& j: L) ]% X+ k# S
96.             "Sec-WebSocket-Accept:$secAccept\r\n\r\n";
    
97.         return socket_write($newClient, $upgrade, strlen($upgrade));
    : b' I2 y0 b3 m4 J# V
98.     }
    
99. 
    + S2 t# k1 Z0 m, p; l5 L2 Y
100.     /**
     
101.      * 解析接收数据
     5 Z0 c9 A2 b& V  b0 j
102.      * @param $buffer
     : _9 ?, @0 Q1 Y
103.      * @return null|string
       N: R! R! u! U8 d3 o) d( x1 p
104.      */
     
105.     public function message($buffer){
     & x: X- N; k, v2 L; G
106.         $len = $masks = $data = $decoded = null;
     0 @3 r/ l2 k; b; Z
107.         $len = ord($buffer[1]) & 127;
     
108.         if ($len === 126)  {
     + e+ y- V( o2 K( u/ N- ~4 c. g
109.             $masks = substr($buffer, 4, 4);
     
110.             $data = substr($buffer, 8);
     
111.         } else if ($len === 127)  {
     - p# L+ {' Z/ F
112.             $masks = substr($buffer, 10, 4);
     , Z6 O  U8 a& p  _
113.             $data = substr($buffer, 14);
     
114.         } else  {
     
115.             $masks = substr($buffer, 2, 4);
     ! j: w. V, s% e! T' W" W
116.             $data = substr($buffer, 6);
     
117.         }
     ) M' e2 E: q* T0 t
118.         for ($index = 0; $index < strlen($data); $index++) {
     # v" h/ j- h' Z/ r4 M) B1 h# h# g
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     ' |2 F0 |- i6 Y) Q( |6 @
120.         }
     + [, G5 b7 W2 @% d# e4 b4 M8 [6 m
121.         return $decoded;
     : j/ t+ r' Q3 f) \4 J
122.     }
     : ?* K( V" m& ~5 S
123. 
     
124.     /**
     . ?3 A- O6 M3 R  X3 V/ l# g3 O
125.      * 发送数据
     ) }' P# D8 a% Y
126.      * @param $newClinet 新接入的socket
     % ?( N3 M8 V- W, I
127.      * @param $msg   要发送的数据
     
128.      * @return int|string
     
129.      */
     
130.     public function send($newClinet, $msg){
     
131.         $msg = $this->frame($msg);
     
132.         socket_write($newClinet, $msg, strlen($msg));
     ( T! {5 F0 v( c+ o4 W9 X; s  k
133.     }
     
134. 
     7 a. b! G/ v& x# @# Q1 g
135.     public function frame($s) {
     
136.         $a = str_split($s, 125);
     . @( n' T% O2 y' u. L* F# U" `4 e
137.         if (count($a) == 1) {
     % T' L  v6 e' r2 D4 d
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     
139.         }
     5 D* `: R2 ^5 o0 @( W6 [, s$ @
140.         $ns = "";
     6 f" z8 C: `, `
141.         foreach ($a as $o) {
     
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     
143.         }
     
144.         return $ns;
     
145.     }
     ) A; a8 S- E) ^5 l/ z
146. 
     
147.     /**
     ! J9 p2 A) ]2 r8 n4 T. m
148.      * 关闭socket
     ' f9 H& \7 R) N
149.      */
     8 H0 t5 f$ e: X5 _$ ]: ^
150.     public function close(){
     , S8 A8 }3 z  Z
151.         return socket_close($this->_sockets);
     6 h  Z6 c% |- S% F. G1 C
152.     }
     
153. }
     
154. 
     
155. $sock = new SocketService();
     
156. $sock->run();
     & G4 ?9 ~# `( ?
157. 
     

复制代码

web.html

1. <!doctype html>
   8 s2 b+ o" |0 U* N
2. <html lang="en">
   
3. <head>
   
4.   <meta charset="UTF-8">
   , T5 T4 m2 j* @/ T& x
5.   <meta name="viewport" content="width=device-width,initial-scale=1, maximum-scale=1, user-scalable=no">
   
6.   <title>websocket</title>
   
7. </head>
   $ _0 J: q6 n$ u$ q$ s0 v. U1 E
8. <body>
   
9. <input id="text" value="">
   6 o' u% @  l9 f3 ]0 Y! e; S; c0 r5 T
10. <input type="submit" value="send" onclick="start()">
    
11. <input type="submit" value="close" onclick="close()">
      r& W/ p' l  h
12. <div id="msg"></div>
    8 A$ n6 E5 G+ o' v& U8 i" @2 p1 i
13. <script>
    9 r* z# t: l. H  U# J7 q, C3 j4 c
14. /**
    ( G7 I. ^& b& ]) ~, Q
15. 0：未连接
    4 q# W; t, {# r) G, m; {
16. 1：连接成功，可通讯
    
17. 2：正在关闭
    
18. 3：连接已关闭或无法打开
    + p! _1 r+ e5 ]
19. */
    
20. 
    
21.     //创建一个webSocket 实例
    
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    % w8 d) [. D! w  ~9 C8 v; J
23. 
    
24. 
    9 \& }( Y- Y" ^2 ^5 f3 T8 \0 B
25.     webSocket.onerror = function (event){
    % x; y# c1 G/ U' }( {  I- s
26.         onError(event);
    
27.     };
    
28. 
    2 t0 D# Q' k6 t0 s) U7 {9 n
29.     // 打开websocket
    , Q% \3 s3 d1 l- Y: R( d7 t; M. L
30.     webSocket.onopen = function (event){
    6 M& b1 k; t/ S/ H
31.         onOpen(event);
    
32.     };
    
33. 
    ! P2 D1 T# L2 \- e! V
34.     //监听消息
    
35.     webSocket.onmessage = function (event){
    
36.         onMessage(event);
    " `% \# n; b  |4 g- u& S
37.     };
    
38. 
    
39. 
    # j1 y( S- f& v! C
40.     webSocket.onclose = function (event){
    
41.         onClose(event);
      s4 `/ q, X) l. ^. m
42.     }
    
43. 
    
44.     //关闭监听websocket
    0 _/ f! T. b6 ]+ \# h
45.     function onError(event){
    , [8 \* _: Q, [- A4 i; s
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    5 G. a: M) z; a# j5 k4 i
47.         console.log("error"+event.data);
    
48.     };
    / S( i  h6 E: `
49. 
    
50.     function onOpen(event){
    
51.         console.log("open:"+sockState());
    / l. P% G! x0 s7 k' b9 J6 E6 Z3 V
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    % [7 c! E# I& [* n
53.     };
    . `- b) l' A1 J9 s$ _
54.     function onMessage(event){
    - f3 U5 m5 Q) p5 e( K
55.         console.log("onMessage");
    ! t% C! `/ u* U# I; T
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    + d( Y# p+ m5 u* i
57.     };
    
58. 
    
59.     function onClose(event){
    
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    % b* W, F1 ]7 u9 f3 b0 u
61.         console.log("close:"+sockState());
    
62.         webSocket.close();
    8 n) a* j, R5 |1 o5 M2 g$ n
63.     }
    
64. 
    8 O# ^8 _& g. A' c$ m/ R, ]: n
65.     function sockState(){
    
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    6 ~: q/ j, S  K4 O. C5 q* Z
67.             return status[webSocket.readyState];
    
68.     }
    $ {6 A6 `$ C- H5 H
69. 
      Z! O& ~' j3 h6 k) h
70. 
    
71. 
    + i. f9 T, G6 T) o
72. function start(event){
    
73.         console.log(webSocket);
    ' S( [% f0 l) z, y" G# }/ G
74.         var msg = document.getElementById('text').value;
    
75.         document.getElementById('text').value = '';
    
76.         console.log("send:"+sockState());
    3 ]; R3 K6 k3 C8 @% ]: M
77.         console.log("msg="+msg);
    
78.         webSocket.send("msg="+msg);
    0 l2 d9 ~% W. M  }" F; _0 ?
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    
80.     };
    
81. 
    
82.     function close(event){
    : f4 X; _; j5 }; v& P* V% P# C
83.         webSocket.close();
    ( Q" i; B2 c  a, f: F4 A, F( L
84.     }
    1 `, Z, u0 H% M- ^1 r
85. </script>
    
86. </body>
      h- l( |" D. R/ e% A0 k
87. </html>

复制代码

% f* e. y- Z: k( a3 _6 S