php实现websocket实时消息推送

admin

php实现websocket实时消息推送
; q7 y, W4 t2 F2 Q8 O+ X: S

/ v6 I5 Y* b6 `5 M1 A! y5 W! DSocketService.php
( i9 B) R2 ?3 ?" c1 n

1. <?php
   
2. /**
   
3. * Created by xwx
   
4. * Date: 2017/10/18
   6 T( c1 r8 r- c: u' p
5. * Time: 14:33
   + {: q9 W3 B$ g# H* N+ j' m. F; q, q
6. */
   2 W3 i4 }0 w: f
7. 
   
8. class SocketService
   
9. {
   
10.     private $address  = '0.0.0.0';
    
11.     private $port = 8083;
    7 `, N' g( D  k  `
12.     private $_sockets;
    + F3 Y, w  h( w. }7 w* \# [$ {
13.     public function __construct($address = '', $port='')
    ; Y% X! p& t- I0 k5 y
14.     {
    
15.             if(!empty($address)){
    
16.                 $this->address = $address;
    
17.             }
    " H* @$ F8 t% Y, v  t4 i
18.             if(!empty($port)) {
      o, k8 {: y- V
19.                 $this->port = $port;
    2 C+ x+ |) S7 R4 h) f; X
20.             }
    + y- K0 o; E' O3 I( f$ Z" L* ]
21.     }
    * r3 f0 ]5 R5 L* h5 W  A
22. 
    
23.     public function service(){
    4 h$ O  j8 @( a9 x. o
24.         //获取tcp协议号码。
    
25.         $tcp = getprotobyname("tcp");
    ( B/ U1 r: ]2 h9 ~
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    % ~0 }" D; h% e7 [0 y7 k5 F
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    
28.         if($sock < 0)
    + t, n, N& Q5 T& S+ J0 V; |) o
29.         {
    ; x# m: B' n1 J4 f6 ?$ o, V' K+ X
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    
31.         }
    
32.         socket_bind($sock, $this->address, $this->port);
    + H3 X9 |! n) Z! I5 D
33.         socket_listen($sock, $this->port);
    
34.         echo "listen on $this->address $this->port ... \n";
    
35.         $this->_sockets = $sock;
    * y7 d2 ^) v) [+ L+ u2 _; `
36.     }
    9 J- n  D+ E/ S& b. r1 x
37. 
    
38.     public function run(){
    1 J/ f5 a$ i! `) u" ]9 M$ J
39.         $this->service();
    
40.         $clients[] = $this->_sockets;
    6 D. \' Z; A& g3 d) m6 m( U
41.         while (true){
    8 I2 i0 T) b; p4 R  i$ A
42.             $changes = $clients;
    
43.             $write = NULL;
    
44.             $except = NULL;
    $ ~" O3 J# t$ D- C+ D. Z8 H) \2 J
45.             socket_select($changes,  $write,  $except, NULL);
    ! K$ M7 `. M2 K  S# O
46.             foreach ($changes as $key => $_sock){
    & b, t: Y( O. K
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    
48.                     if(($newClient = socket_accept($_sock))  === false){
    
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    
50.                     }
    6 A, ?8 k, E: {1 X. c8 H
51.                     $line = trim(socket_read($newClient, 1024));
    
52.                     $this->handshaking($newClient, $line);
    
53.                     //获取client ip
    : ]6 i% k  a- w0 o1 D
54.                     socket_getpeername ($newClient, $ip);
    4 J9 G9 V# R& g% O; z
55.                     $clients[$ip] = $newClient;
    
56.                     echo  "Client ip:{$ip}   \n";
    ( z0 \  h2 R. k+ v+ V8 t
57.                     echo "Client msg:{$line} \n";
    ! x, p) J* {' [7 t8 F2 R# D
58.                 } else {
    . q+ i4 K# F1 Y5 s
59.                     socket_recv($_sock, $buffer,  2048, 0);
    - g( F3 v6 u( g7 |/ m5 m, @
60.                     $msg = $this->message($buffer);
    8 |. P$ r$ @8 X) K) l
61.                     //在这里业务代码
    5 p# ]" i7 M1 D3 h1 l+ Q, C4 b
62.                     echo "{$key} clinet msg:",$msg,"\n";
    
63.                     fwrite(STDOUT, 'Please input a argument:');
    
64.                     $response = trim(fgets(STDIN));
    
65.                     $this->send($_sock, $response);
    
66.                     echo "{$key} response to Client:".$response,"\n";
    
67.                 }
    
68.             }
    ) K  N  {) Y0 r$ H8 l8 n8 ?- k, r+ P
69.         }
    5 |' h: a2 Y% v) @
70.     }
    8 I' _5 L# M7 {% K$ b7 d
71. 
    0 h' l$ N0 r, I$ a9 w/ p8 d
72.     /**
    
73.      * 握手处理
    ( L1 g$ }0 e$ V; a3 t% C
74.      * @param $newClient socket
    + [) Y5 b. g  W9 a8 a) |* O! E
75.      * @return int  接收到的信息
    - H7 Z$ J) s7 d
76.      */
    ' ?4 m; s" s8 N+ ~( _1 \7 `
77.     public function handshaking($newClient, $line){
    
78. 
    6 B7 }; j) i3 o) b
79.         $headers = array();
    
80.         $lines = preg_split("/\r\n/", $line);
    ! w* D! ?+ ?: e6 H  Z) j; t
81.         foreach($lines as $line)
    2 n2 [. z! K# M6 Y1 R+ |2 v1 j
82.         {
    ) K4 L! k) O0 e6 O5 P
83.             $line = chop($line);
    
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    
85.             {
    
86.                 $headers[$matches[1]] = $matches[2];
    / C, S% C$ ~( f. L% ], q
87.             }
    7 |# w( O4 A" ^6 r% v
88.         }
    
89.         $secKey = $headers['Sec-WebSocket-Key'];
    : U5 O3 ^9 f, m- P3 @; e- [  ^* ^# p
90.         $secAccept = base64_encode(pack('H*', sha1($secKey . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11')));
    ! ?9 _! j0 C: K: B6 [, s
91.         $upgrade  = "HTTP/1.1 101 Web Socket Protocol Handshake\r\n" .
    ' O% p. |/ x4 O: C* L/ \) p
92.             "Upgrade: websocket\r\n" .
    * e6 y& R0 E5 U9 l
93.             "Connection: Upgrade\r\n" .
    
94.             "WebSocket-Origin: $this->address\r\n" .
    
95.             "WebSocket-Location: ws://$this->address:$this->port/websocket/websocket\r\n".
    5 C2 Q; \) |9 A8 x' y. I* ^3 e
96.             "Sec-WebSocket-Accept:$secAccept\r\n\r\n";
    
97.         return socket_write($newClient, $upgrade, strlen($upgrade));
    
98.     }
    
99. 
    & y0 L! j3 e+ T7 L* H! @
100.     /**
     - w8 L0 L3 n3 @* f0 d$ l
101.      * 解析接收数据
     2 I! M, D( J* i, g" B! X% H
102.      * @param $buffer
     
103.      * @return null|string
     8 |+ T" x- g6 {4 `; j/ K+ G5 D
104.      */
     
105.     public function message($buffer){
     7 M0 x8 ^8 m8 V& v4 U+ M" B" O% f! p( H
106.         $len = $masks = $data = $decoded = null;
     
107.         $len = ord($buffer[1]) & 127;
     
108.         if ($len === 126)  {
     : v1 n5 C1 P6 {
109.             $masks = substr($buffer, 4, 4);
     
110.             $data = substr($buffer, 8);
     3 L; m+ b2 Q& }6 Q5 h# z2 r
111.         } else if ($len === 127)  {
     
112.             $masks = substr($buffer, 10, 4);
     
113.             $data = substr($buffer, 14);
     $ M7 o# Q5 p* O0 ]5 O! e  U( j
114.         } else  {
     4 F' i0 e! q0 P- W$ y" A5 A; ]2 z
115.             $masks = substr($buffer, 2, 4);
     
116.             $data = substr($buffer, 6);
     
117.         }
     % Q3 Y) p. f6 V: m
118.         for ($index = 0; $index < strlen($data); $index++) {
     
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     
120.         }
     & }4 d2 s) k) j1 }7 N& v8 {; I
121.         return $decoded;
     
122.     }
     
123. 
     
124.     /**
     ! X) i9 ?8 I) K  T" W' v* h
125.      * 发送数据
     / J' [: W* S: {9 y4 r! G7 W6 f
126.      * @param $newClinet 新接入的socket
     / R9 Q% ^9 A9 }" ]; c8 P3 M% u
127.      * @param $msg   要发送的数据
     1 v+ m) g/ _7 n3 o% j& {; w
128.      * @return int|string
     
129.      */
     : E( e  Q$ |/ |7 ~
130.     public function send($newClinet, $msg){
     
131.         $msg = $this->frame($msg);
     
132.         socket_write($newClinet, $msg, strlen($msg));
     
133.     }
     
134. 
     8 E& D; a9 d1 M: ]: s; \" ?( {1 A
135.     public function frame($s) {
     
136.         $a = str_split($s, 125);
     ; \, d, C7 z# n( N4 t
137.         if (count($a) == 1) {
     
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     ' U* d6 M4 E! z: s, Y
139.         }
     ' w0 O# a2 m$ `
140.         $ns = "";
     ( E- E0 {9 P7 ]% F
141.         foreach ($a as $o) {
     
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     
143.         }
     + i9 W* f! f# w; L
144.         return $ns;
     + {% K9 Z7 S: @8 O9 }7 v, N; g
145.     }
     8 |- S: G% c( C% l$ B1 T% F
146. 
     
147.     /**
     
148.      * 关闭socket
     ! I5 @. k/ w3 z3 `2 }( D; y
149.      */
     - `2 w6 ?# @: j* O
150.     public function close(){
     ; C" d+ o) o$ h: e1 _5 G- }
151.         return socket_close($this->_sockets);
     9 ?2 P2 u( i' u/ H1 A$ K
152.     }
     ! a/ E1 S& y+ W- I9 G- o
153. }
     
154. 
     
155. $sock = new SocketService();
     4 T- D. v2 L: M9 ]: |: b  {
156. $sock->run();
     
157. 
     

复制代码

web.html
4 R; L+ \+ q" C( T( C; s6 l! @; N

1. <!doctype html>
   1 A0 R/ k) o, W, U4 \* H
2. <html lang="en">
   
3. <head>
   , W" o. {7 d2 H: J! \' ], Y
4.   <meta charset="UTF-8">
   
5.   <meta name="viewport" content="width=device-width,initial-scale=1, maximum-scale=1, user-scalable=no">
   
6.   <title>websocket</title>
   0 S. V# B6 s3 H3 `
7. </head>
   0 }5 h3 w$ Q  V' D
8. <body>
   $ S9 @  ]/ p% o$ P- W
9. <input id="text" value="">
   * u$ K- b/ H( N- K
10. <input type="submit" value="send" onclick="start()">
    1 [% W$ M3 A! j$ s+ G  \( e
11. <input type="submit" value="close" onclick="close()">
    
12. <div id="msg"></div>
    
13. <script>
    
14. /**
    
15. 0：未连接
    ' D! U. t7 b& h2 x; Q3 a
16. 1：连接成功，可通讯
    
17. 2：正在关闭
    1 Q* j) ^4 D/ P
18. 3：连接已关闭或无法打开
      q1 c, n5 R3 f% t7 s# A
19. */
    7 R, d4 h) V0 K: X
20. 
    
21.     //创建一个webSocket 实例
    6 m7 J  o- _  n5 P* o) v
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    4 d( h2 G/ v# R" H, _6 E4 d8 ?: G8 S
23. 
    
24. 
    $ C6 {) v, g) G$ h- x; H
25.     webSocket.onerror = function (event){
    5 l0 E6 p1 n+ o; p0 D* Y/ W1 I
26.         onError(event);
    0 Q# ]0 K  ?, e$ c  t5 B+ |8 e
27.     };
      \2 b5 U: M7 N/ L  X, ~! u" A
28. 
    & J' [  k9 ?' E6 S
29.     // 打开websocket
    
30.     webSocket.onopen = function (event){
    
31.         onOpen(event);
    ( \8 U5 m; W2 H7 g5 _% F; J8 r6 ?3 e
32.     };
    7 @* E: ]) ~! p5 V
33. 
    
34.     //监听消息
    $ U9 K7 e7 u( U, c8 K7 l  m$ t
35.     webSocket.onmessage = function (event){
    
36.         onMessage(event);
    
37.     };
    
38. 
    
39. 
    
40.     webSocket.onclose = function (event){
    . n# Z8 s. k2 G) @' ~3 ~6 m" I
41.         onClose(event);
    
42.     }
    5 S( Z- p4 ]7 ~4 ~7 J4 ^
43. 
    % S2 M1 Y7 @, ]( t2 B
44.     //关闭监听websocket
    
45.     function onError(event){
      I9 ~# h% s. [  S8 N
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
47.         console.log("error"+event.data);
    6 P7 }9 ]& c9 r5 E
48.     };
    
49. 
    $ R: o+ {+ a) ^( J1 R0 \( i
50.     function onOpen(event){
    3 c5 `8 E" m" t( m
51.         console.log("open:"+sockState());
    , J- J6 D* W3 g/ f: i% |
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    
53.     };
    7 m1 u6 b) P; W" f9 V; ]
54.     function onMessage(event){
    
55.         console.log("onMessage");
    
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    
57.     };
    1 C. ?1 l: e& F  q
58. 
    0 }/ i5 }: t1 m# d
59.     function onClose(event){
    # t" v; H% n' G  M
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
61.         console.log("close:"+sockState());
    
62.         webSocket.close();
    
63.     }
    
64. 
    
65.     function sockState(){
    
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    
67.             return status[webSocket.readyState];
    
68.     }
    
69. 
    0 U7 i9 ~  P5 ]2 |1 U. w0 M
70. 
    . G2 m0 r3 B  K$ L4 Z1 K! {- i
71. 
    
72. function start(event){
    # n- B$ ^$ Z% Q+ G2 K
73.         console.log(webSocket);
    # g7 [5 e: n6 P& s! b  c
74.         var msg = document.getElementById('text').value;
    
75.         document.getElementById('text').value = '';
    + T% P* R: x2 J7 D  D& E
76.         console.log("send:"+sockState());
    # u. R0 h* j: i2 x/ s2 w# [; @
77.         console.log("msg="+msg);
    - F# V2 t( D3 ~0 w7 j4 u
78.         webSocket.send("msg="+msg);
    
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    
80.     };
    
81. 
    9 b" l  U) A# l) o( c3 J/ q! s& X
82.     function close(event){
    
83.         webSocket.close();
    
84.     }
    
85. </script>
    + H1 k* @' U4 Z
86. </body>
    
87. </html>

复制代码

# x; S1 H" x3 R. \1 R0 c' s& u
2 V3 G- G) {( P7 k
4 p! k; O1 s8 B% b/ Z4 a* x