php实现websocket实时消息推送

admin

php实现websocket实时消息推送


SocketService.php

1. <?php
   
2. /**
   , i8 k! s3 `. t3 ?8 V3 j9 e
3. * Created by xwx
   , U% i* s! E, Z, O# D
4. * Date: 2017/10/18
   , g) b, Y+ C4 D- g
5. * Time: 14:33
   ( F2 L4 m0 X2 |) c2 l
6. */
   1 A7 M' {  _2 c* m* D+ }
7. 
   , [8 E1 _4 u( ?( g% o8 J  _! D
8. class SocketService
   
9. {
   , u0 {: G4 F; T# P. h1 ?
10.     private $address  = '0.0.0.0';
    
11.     private $port = 8083;
    & c+ X: F0 L) F  q+ C
12.     private $_sockets;
    , C7 n5 B2 b; w; p0 S4 O) `+ Z3 A
13.     public function __construct($address = '', $port='')
    ' Z  a0 b& f) h" }
14.     {
    " U9 G4 g3 ^. Z; W' I
15.             if(!empty($address)){
    % n; }& r& x5 p  Y
16.                 $this->address = $address;
    
17.             }
    & D1 f0 s. k" g1 R
18.             if(!empty($port)) {
    
19.                 $this->port = $port;
    9 `  w! Y6 Y; X9 A9 Q
20.             }
    2 V3 I( X8 F1 D3 U( v! ^
21.     }
    ! S: J! A! [2 f$ m6 j
22. 
    ) c2 U6 g( \% S, s' I+ k+ U/ S
23.     public function service(){
      ~- x+ }, _3 e4 h; ^9 L$ N5 ^
24.         //获取tcp协议号码。
    
25.         $tcp = getprotobyname("tcp");
    # \7 ]+ `/ E" o8 y" P- T6 x
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    # N* {! X: x9 M8 o$ t
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    % a0 m$ j& H2 l/ U- c3 g& ^
28.         if($sock < 0)
    ; z9 x( y0 P6 h
29.         {
    4 W8 R% q4 \' _! D* J: B
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    : C  l. t+ I- L' K6 Z' D, [
31.         }
    4 i2 K' L9 k, E
32.         socket_bind($sock, $this->address, $this->port);
    
33.         socket_listen($sock, $this->port);
    * \! J* y7 _3 J' K2 [7 d, K
34.         echo "listen on $this->address $this->port ... \n";
    
35.         $this->_sockets = $sock;
    
36.     }
    
37. 
    7 C2 _$ g6 v% r% G5 v. g( q
38.     public function run(){
    7 `9 {8 Q( H+ u9 Q+ I
39.         $this->service();
    ) l) T+ Q3 z( s
40.         $clients[] = $this->_sockets;
    
41.         while (true){
    
42.             $changes = $clients;
    1 r4 S4 p2 U0 K
43.             $write = NULL;
    
44.             $except = NULL;
    
45.             socket_select($changes,  $write,  $except, NULL);
    : n* P# O7 }0 `3 H5 a* x$ `" F2 Q: O
46.             foreach ($changes as $key => $_sock){
    & P& |2 `% U+ U& y6 ?+ r. \
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    , @9 k$ A2 `8 z% u& [
48.                     if(($newClient = socket_accept($_sock))  === false){
    
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    . }& \7 g+ m" Y
50.                     }
    
51.                     $line = trim(socket_read($newClient, 1024));
    
52.                     $this->handshaking($newClient, $line);
    " Y$ E! q8 |; f& ?- }; k
53.                     //获取client ip
    4 M4 V5 ~! T+ d! b/ Y  ~; I" |, G
54.                     socket_getpeername ($newClient, $ip);
    + z2 r6 X9 g$ a) R
55.                     $clients[$ip] = $newClient;
    
56.                     echo  "Client ip:{$ip}   \n";
    
57.                     echo "Client msg:{$line} \n";
    
58.                 } else {
    0 P7 u! ^- N; U* n% q
59.                     socket_recv($_sock, $buffer,  2048, 0);
    1 }5 _5 f( b6 c" k# B
60.                     $msg = $this->message($buffer);
    7 D: B* A; [" v7 i& M; ^2 f  x& ~& B
61.                     //在这里业务代码
    
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";
    8 g% T6 ]+ U, k5 o
67.                 }
    
68.             }
    + l! v2 D1 l, B1 f& |# Z" d6 _
69.         }
    / M6 Q! w- S( x) m6 ?9 p
70.     }
    
71. 
    
72.     /**
    + y0 |* Z3 l* |9 e& L$ n! H
73.      * 握手处理
    
74.      * @param $newClient socket
    9 Z3 N" k$ I3 I$ n7 R6 k  n
75.      * @return int  接收到的信息
    
76.      */
    $ p" W4 Y/ k4 D  H& T4 M2 y
77.     public function handshaking($newClient, $line){
    ) J9 q) G& s/ b+ C% X, ~' k& @" {
78. 
    
79.         $headers = array();
    4 v( W6 f! k& {. D- v# l
80.         $lines = preg_split("/\r\n/", $line);
    , e: ?+ ^+ }6 R" x; `
81.         foreach($lines as $line)
    ( N4 S. I0 g- s" u& B- g3 O. d
82.         {
    
83.             $line = chop($line);
    
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    
85.             {
    ( k* t+ ?6 a- d/ M8 {. `
86.                 $headers[$matches[1]] = $matches[2];
    3 M$ S: k" f  B% E6 ^' P% h5 T( T
87.             }
    2 a. l" g$ K  p9 ?' b) t! _  L) ]- u
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" .
    ! {6 @5 z3 o' e5 u% S: r; b5 L
92.             "Upgrade: websocket\r\n" .
    
93.             "Connection: Upgrade\r\n" .
    3 L9 T) m6 n2 H/ f+ T6 J
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";
    
97.         return socket_write($newClient, $upgrade, strlen($upgrade));
    
98.     }
    & o1 G- P3 i! a' B+ V
99. 
    / j& c& v+ n, @, I- D! C, g
100.     /**
     $ Q2 x( B2 ?0 H: B
101.      * 解析接收数据
     
102.      * @param $buffer
     # I3 K2 Q# i6 z- |' ^. |" J  E$ M
103.      * @return null|string
     
104.      */
     
105.     public function message($buffer){
     4 W" M& {! Y+ V. V, R) h
106.         $len = $masks = $data = $decoded = null;
     
107.         $len = ord($buffer[1]) & 127;
     
108.         if ($len === 126)  {
     
109.             $masks = substr($buffer, 4, 4);
     
110.             $data = substr($buffer, 8);
     , ^$ Y- V$ m3 |+ X- q8 \
111.         } else if ($len === 127)  {
     . V/ ]$ C0 \5 R$ s+ l4 t9 w/ E
112.             $masks = substr($buffer, 10, 4);
       K0 S3 D# v' o% E9 r, A4 d
113.             $data = substr($buffer, 14);
     
114.         } else  {
     9 v$ V9 T. B% g% e  @5 H- @
115.             $masks = substr($buffer, 2, 4);
     , X- U3 R: V: x. ]! h( [; m
116.             $data = substr($buffer, 6);
     
117.         }
     ( l, {, @! E6 v5 b( A0 x5 e
118.         for ($index = 0; $index < strlen($data); $index++) {
     3 [; `) s9 i6 T7 o' P
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     . r6 Z6 h) |2 ^6 C! t
120.         }
     
121.         return $decoded;
     
122.     }
     
123. 
     
124.     /**
     5 b1 ~3 t2 x, s& M9 i+ ~* p
125.      * 发送数据
     9 m$ w3 B0 x7 l8 K6 ?
126.      * @param $newClinet 新接入的socket
     
127.      * @param $msg   要发送的数据
     + g& \" }0 H) y. y' x$ K" H! j  C
128.      * @return int|string
     
129.      */
     
130.     public function send($newClinet, $msg){
     ! b$ J+ d( O; s
131.         $msg = $this->frame($msg);
     8 I4 [$ ]6 Z* C* ]( V! b
132.         socket_write($newClinet, $msg, strlen($msg));
     ) Z; b+ G9 H  g* ^
133.     }
     
134. 
     9 U7 L  I/ i. C3 V; x0 x- B
135.     public function frame($s) {
     
136.         $a = str_split($s, 125);
     
137.         if (count($a) == 1) {
     
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     
139.         }
     
140.         $ns = "";
     
141.         foreach ($a as $o) {
     
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     
143.         }
     
144.         return $ns;
     
145.     }
     
146. 
     & {' D; b' u# i, O
147.     /**
     
148.      * 关闭socket
     5 T; V/ e, e$ P6 M# d: l  T
149.      */
     ( v6 V7 p/ k, p7 T2 I
150.     public function close(){
     & x, l8 a# E0 W8 ?4 n8 ?
151.         return socket_close($this->_sockets);
     9 w0 `5 U: N5 Q/ H3 j# B- I
152.     }
     . e. N4 j0 C, T9 G
153. }
     4 `  e9 R9 ?! m6 m4 p0 x8 s
154. 
     
155. $sock = new SocketService();
     9 n1 s8 g, w0 V0 a6 y
156. $sock->run();
     2 |# R) U1 n5 \5 |& V7 U. y6 H
157. 
     

复制代码

web.html

1. <!doctype html>
   4 \) r" R) t4 S* b% N$ P( S7 q
2. <html lang="en">
   
3. <head>
   . Z* f3 B2 y- x3 ~" e7 }  h/ N: G) j$ v
4.   <meta charset="UTF-8">
   ( y& I8 ]: z4 L$ r0 e# T1 B
5.   <meta name="viewport" content="width=device-width,initial-scale=1, maximum-scale=1, user-scalable=no">
   . k  q9 m2 t/ ~5 X; a
6.   <title>websocket</title>
   + c* h- c" h7 n2 R4 q) z; ]0 v2 i' R
7. </head>
   
8. <body>
   0 ?/ \1 G8 K! ~% s* S7 E. |
9. <input id="text" value="">
   
10. <input type="submit" value="send" onclick="start()">
    & I, u2 a: `% @; `
11. <input type="submit" value="close" onclick="close()">
    
12. <div id="msg"></div>
    5 V( O( d: H9 k5 p2 @( N
13. <script>
    8 [& _/ d5 o2 @- T  ]/ X
14. /**
    
15. 0：未连接
    2 E9 O9 \, c, u; H
16. 1：连接成功，可通讯
    
17. 2：正在关闭
    
18. 3：连接已关闭或无法打开
    , x- i) ?" v3 I& [6 c. B  U
19. */
    
20. 
    ! i4 ^  _2 p2 i& T% r  V" m
21.     //创建一个webSocket 实例
    ! h% F' o3 g1 h1 V: \% ?/ C
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    ) `' K8 W4 h8 k) R( H$ Y# a. o
23. 
    + B+ L% z4 K: f, ]! x& {
24. 
    ( T2 F! [0 f; [) Y( R$ t
25.     webSocket.onerror = function (event){
    
26.         onError(event);
    
27.     };
    & r! N, e% v3 m5 t. z. ~# i
28. 
    % M$ p9 n" j" P6 j% ~$ ~0 J; x
29.     // 打开websocket
    ) H4 C" S0 F. v( d; k9 W" D1 B) H
30.     webSocket.onopen = function (event){
    4 F1 O( u) X/ \* N+ O
31.         onOpen(event);
    5 h4 K6 \- f* [
32.     };
    , w- w- Z% ?8 b8 H4 V* o# P
33. 
    ) V. f5 |% t" G0 Y! V
34.     //监听消息
    
35.     webSocket.onmessage = function (event){
    
36.         onMessage(event);
    % b+ W. m- g1 e! M3 T
37.     };
    
38. 
    ( ^) x0 Z3 a; d* E8 O+ ?+ Y( B# D& J
39. 
    
40.     webSocket.onclose = function (event){
    . w7 R5 y! L$ X4 l4 r
41.         onClose(event);
    ( B) t  T5 {! p. p% |
42.     }
    , G$ @9 S. M  E1 F! j
43. 
    . r+ W2 V! `' X0 N! J
44.     //关闭监听websocket
    
45.     function onError(event){
    ! \# p+ X: f1 e% V
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
47.         console.log("error"+event.data);
    ( `) Y$ P( R5 q
48.     };
    
49. 
    3 A: R# ^( v1 N# r/ F
50.     function onOpen(event){
    
51.         console.log("open:"+sockState());
    , }5 b4 @' }8 m9 M
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    
53.     };
      G) i- i9 A+ J$ w! r
54.     function onMessage(event){
    
55.         console.log("onMessage");
    : }( F( V* t: i6 b1 F
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    
57.     };
    
58. 
    ; I7 O% w0 B0 n% C0 M8 `- K
59.     function onClose(event){
    2 v6 z2 K) H2 X
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    2 ~; j" A% |  H, c- q# b- ^
61.         console.log("close:"+sockState());
    0 ^6 A, U  M1 _3 P" K5 y
62.         webSocket.close();
    7 p" c6 C+ k, _
63.     }
    
64. 
    $ d+ K( Y- D4 \' o9 f
65.     function sockState(){
    
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    
67.             return status[webSocket.readyState];
    9 j" b- W" m# S% v4 x) G7 k5 q
68.     }
    
69. 
    
70. 
    " s7 V5 K" l$ B
71. 
    
72. function start(event){
    
73.         console.log(webSocket);
    : }2 z9 }, h  g/ U; ^  f
74.         var msg = document.getElementById('text').value;
    
75.         document.getElementById('text').value = '';
    % t7 D0 _7 z0 t2 r
76.         console.log("send:"+sockState());
    
77.         console.log("msg="+msg);
    $ Y) c5 A+ A( C# X# x+ U; d
78.         webSocket.send("msg="+msg);
    $ u/ I  [5 q# W& W3 ^- J/ X
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    0 G7 M$ i0 e* f3 l2 Z& X' A: U/ g
80.     };
    9 i1 g8 D* i, @( v! V7 z8 ~
81. 
    : R7 l; o1 D0 v8 t
82.     function close(event){
    
83.         webSocket.close();
    
84.     }
    
85. </script>
    + I! ~+ l* I# I" Y9 P; F$ f1 c
86. </body>
    
87. </html>

复制代码

/ ~$ p% R- J, s' S: h& l
, L$ V5 T& J: {9 ~* L/ e5 h