php实现websocket实时消息推送

admin

php实现websocket实时消息推送
+ S: W! v6 |% O$ M! C5 n7 _2 p

SocketService.php
* \: ]. X& _: u6 t2 A

1. <?php
   
2. /**
   . p4 d' W3 v, `6 O* z# `- I1 c  j
3. * Created by xwx
   $ n* ^* Q6 r6 A
4. * Date: 2017/10/18
   
5. * Time: 14:33
   2 y! [& Y; S" a+ n% @& ?* B/ k
6. */
   6 }  J- W3 O0 i. r. J+ C/ t
7. 
   
8. class SocketService
   ' v5 p9 U, X3 _& ~
9. {
   1 X" |! b- g- g3 X1 Z7 u& x
10.     private $address  = '0.0.0.0';
    
11.     private $port = 8083;
    2 r) Q4 V2 H& m; m  H, m) j
12.     private $_sockets;
    
13.     public function __construct($address = '', $port='')
    ; s& u. Q, e$ s) ]1 ]
14.     {
    
15.             if(!empty($address)){
    , \' m$ M0 p. \2 U1 ~
16.                 $this->address = $address;
    
17.             }
    , i$ h( N7 f8 j9 `0 f. T
18.             if(!empty($port)) {
    
19.                 $this->port = $port;
    4 x! o) |4 o  q- f3 I( O$ z
20.             }
    
21.     }
    1 y, L8 G  R6 Q$ ^, K8 U
22. 
    
23.     public function service(){
    ! E' R4 s$ @" f' A/ R0 O) q3 n7 o
24.         //获取tcp协议号码。
    " Y  i" h; B7 M# S, D# N
25.         $tcp = getprotobyname("tcp");
    
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    - h' \" O; Y1 t  E6 ~$ m' E
28.         if($sock < 0)
    % R% N, _) ~0 e1 v+ v7 Q9 U! ?7 L
29.         {
    5 O! U6 j. {& s* [
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    2 x# X& L/ o9 b: A8 X
31.         }
    
32.         socket_bind($sock, $this->address, $this->port);
    1 Z& v& U( H7 u4 T% j1 E5 j+ K3 _
33.         socket_listen($sock, $this->port);
    : p# H9 ]: z! }) T. g3 ]! e
34.         echo "listen on $this->address $this->port ... \n";
    9 c: _1 B( p6 G3 w4 ^0 A9 P. Q9 C9 }
35.         $this->_sockets = $sock;
    
36.     }
    + u8 r2 ]1 i9 K" @3 `
37. 
    - t. m& x6 M2 x
38.     public function run(){
    
39.         $this->service();
    
40.         $clients[] = $this->_sockets;
    
41.         while (true){
    
42.             $changes = $clients;
    
43.             $write = NULL;
    2 G8 ~* v* C/ Q9 m* r
44.             $except = NULL;
    
45.             socket_select($changes,  $write,  $except, NULL);
    
46.             foreach ($changes as $key => $_sock){
    9 b* `3 {, Y5 N0 H' ~" \' U
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    
48.                     if(($newClient = socket_accept($_sock))  === false){
    
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    
50.                     }
    / r+ J$ q% q3 H8 K8 I. K
51.                     $line = trim(socket_read($newClient, 1024));
    
52.                     $this->handshaking($newClient, $line);
    
53.                     //获取client ip
    7 o( t9 Y' q9 L0 o! i! I' f( j
54.                     socket_getpeername ($newClient, $ip);
    / o0 q4 |9 P0 c& j$ P  U4 p2 G
55.                     $clients[$ip] = $newClient;
    
56.                     echo  "Client ip:{$ip}   \n";
    
57.                     echo "Client msg:{$line} \n";
    
58.                 } else {
    
59.                     socket_recv($_sock, $buffer,  2048, 0);
    
60.                     $msg = $this->message($buffer);
    / K5 x0 M. p+ r0 m& }7 [8 N- y
61.                     //在这里业务代码
    4 A$ r. o2 Y1 o: G
62.                     echo "{$key} clinet msg:",$msg,"\n";
    9 k  c1 F4 T8 P
63.                     fwrite(STDOUT, 'Please input a argument:');
    8 d) h- w$ M1 P, y
64.                     $response = trim(fgets(STDIN));
    
65.                     $this->send($_sock, $response);
    
66.                     echo "{$key} response to Client:".$response,"\n";
    
67.                 }
    
68.             }
    ( e8 K* B: \# U+ ?2 D
69.         }
    4 ~- k% h; f' t8 i" Q! I
70.     }
    " x5 u- b. y* P/ m. i
71. 
    
72.     /**
    
73.      * 握手处理
    
74.      * @param $newClient socket
    # Q. M0 L) W, o
75.      * @return int  接收到的信息
    + j5 k8 q: U3 W/ q- x$ }) S) {- \
76.      */
    
77.     public function handshaking($newClient, $line){
    
78. 
    & I5 L3 o0 G/ j% }6 Q
79.         $headers = array();
    
80.         $lines = preg_split("/\r\n/", $line);
    
81.         foreach($lines as $line)
    
82.         {
    0 }& }. h4 `/ u! V8 m) I
83.             $line = chop($line);
    
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    , {" m2 `1 h8 c: K' \& @+ E
85.             {
    
86.                 $headers[$matches[1]] = $matches[2];
    ! H5 p: g' u# i4 U; |( t
87.             }
    $ t* v" @  w& u+ K$ `7 G' I+ 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" .
    & H3 c& q. b* P; O
92.             "Upgrade: websocket\r\n" .
    
93.             "Connection: Upgrade\r\n" .
    : X5 L6 W. V9 f/ ?% t$ _( ?
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";
    ' m6 y( p  j$ x% M5 l+ {( l7 `' U
97.         return socket_write($newClient, $upgrade, strlen($upgrade));
    
98.     }
    
99. 
    
100.     /**
     + t$ r& _% m$ m# G; U! X
101.      * 解析接收数据
     : r- Y- E' G  C! H' O% W
102.      * @param $buffer
     
103.      * @return null|string
     
104.      */
     ( E+ k) X/ X4 @; k" u% b
105.     public function message($buffer){
     0 E* ]4 W; L3 R' _
106.         $len = $masks = $data = $decoded = null;
     4 D  z7 V( _% |1 K
107.         $len = ord($buffer[1]) & 127;
     + g( ]/ X$ U& q: Z2 d4 h
108.         if ($len === 126)  {
     1 z" r. I1 ^# {9 o* N; ~
109.             $masks = substr($buffer, 4, 4);
     - I; I  l# w4 i) X! U/ R
110.             $data = substr($buffer, 8);
     
111.         } else if ($len === 127)  {
     
112.             $masks = substr($buffer, 10, 4);
     
113.             $data = substr($buffer, 14);
     8 @7 c4 c# ^4 P& o- O/ i1 A
114.         } else  {
     
115.             $masks = substr($buffer, 2, 4);
     ' T# @5 |; |) ?' X0 I- N' v  B) T; e
116.             $data = substr($buffer, 6);
     
117.         }
     1 V* ?: \5 Y+ \: _+ }4 a* K+ i" W
118.         for ($index = 0; $index < strlen($data); $index++) {
     ( I6 d; g+ J/ W+ ?9 h
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     
120.         }
     
121.         return $decoded;
     - \. o3 ^- C$ m! F9 }& N
122.     }
     % `, g* n# O0 R6 d
123. 
     $ f* v% ^( V) q6 C& U6 e
124.     /**
     * o& L  K" f2 g
125.      * 发送数据
     ( Q/ n0 p  M# d" S. J7 {! Z
126.      * @param $newClinet 新接入的socket
     
127.      * @param $msg   要发送的数据
     
128.      * @return int|string
     
129.      */
     
130.     public function send($newClinet, $msg){
     
131.         $msg = $this->frame($msg);
     ! [; Q# y. n& }3 N$ t
132.         socket_write($newClinet, $msg, strlen($msg));
     ; v8 J4 Y' e1 w! {4 Y+ a8 M
133.     }
     
134. 
     / f) R+ g" [/ K
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];
     2 E) g( p$ t3 }: Z% O5 k+ C
139.         }
     " }5 n3 t: Q! b" s/ S
140.         $ns = "";
     ; f+ s" u% t, S" l+ A
141.         foreach ($a as $o) {
     , D) C! [$ E2 E, x: s( c
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     5 n; D# ~* V* L6 v  u) _9 b& t2 A
143.         }
     , m. g+ `% e' @2 k% d! g9 X
144.         return $ns;
     6 U9 _! M( Q& d6 f2 q2 ]. w
145.     }
     9 Q3 O- ~- j; {  S3 N" e
146. 
     3 o" }) I' I4 \$ \, L
147.     /**
     2 U# U1 C+ I& [7 K
148.      * 关闭socket
     
149.      */
     
150.     public function close(){
     
151.         return socket_close($this->_sockets);
     & V% Q: L" G, j% Y
152.     }
     ) l- w# E$ c& N5 I- X
153. }
     
154. 
     
155. $sock = new SocketService();
     : T( K( N: A& N, e+ L" a, v
156. $sock->run();
     
157. 
     

复制代码

web.html
; T: d4 C& k$ Z4 E

1. <!doctype html>
   ; r0 I: g7 b$ ~$ E* r
2. <html lang="en">
   - p, ~( u0 r& _$ s
3. <head>
   ) c& J( B; v% W. z' `% V
4.   <meta charset="UTF-8">
   - u% v' W8 }& g% L6 [; P
5.   <meta name="viewport" content="width=device-width,initial-scale=1, maximum-scale=1, user-scalable=no">
   3 ^0 X9 g. ^) @. X3 X# v
6.   <title>websocket</title>
   ! X$ ?1 L) O9 L/ E
7. </head>
   
8. <body>
   
9. <input id="text" value="">
   ( s. b) `/ p/ i4 Y1 U; b" u
10. <input type="submit" value="send" onclick="start()">
    
11. <input type="submit" value="close" onclick="close()">
    1 l" ?) Q# V: i! C1 \) u: f
12. <div id="msg"></div>
    
13. <script>
    # r& S. Z/ l9 j. D; x0 i
14. /**
    
15. 0：未连接
    
16. 1：连接成功，可通讯
    
17. 2：正在关闭
    
18. 3：连接已关闭或无法打开
    
19. */
    
20. 
    9 Q: e6 E  Q% |2 N
21.     //创建一个webSocket 实例
    
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    
23. 
    
24. 
    
25.     webSocket.onerror = function (event){
    2 d: ?" B' Q$ s' Y8 v; a
26.         onError(event);
    3 C) F# }- O  U3 ?
27.     };
    
28. 
    
29.     // 打开websocket
    ) G! _7 V7 k6 q- _0 {4 d, l  j7 r
30.     webSocket.onopen = function (event){
    
31.         onOpen(event);
    * o, V  ^$ K2 h( }: d
32.     };
    
33. 
    
34.     //监听消息
    
35.     webSocket.onmessage = function (event){
    
36.         onMessage(event);
    
37.     };
    : F- X8 y9 ?. k. v  K$ R" W. G
38. 
    
39. 
    
40.     webSocket.onclose = function (event){
    1 _/ O" A! R* d. U
41.         onClose(event);
    
42.     }
    * g0 @. l0 [2 E1 {
43. 
    
44.     //关闭监听websocket
    
45.     function onError(event){
    
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
47.         console.log("error"+event.data);
    * I6 V! I  q: H; ?$ _
48.     };
    
49. 
    ) a4 _, M& h1 q& a- x6 q
50.     function onOpen(event){
    : o: P* P/ @' Q! m# K3 A& n
51.         console.log("open:"+sockState());
    8 O9 U- f- f2 D: w, x
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    6 C. E& `) e, g& k) q5 d
53.     };
    
54.     function onMessage(event){
    / B" f2 K! y8 d! i; R( m; ^. e
55.         console.log("onMessage");
    
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    4 N! A6 F0 _3 V; [# v0 y" o
57.     };
    
58. 
    
59.     function onClose(event){
    1 a7 j, K6 |9 k
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
61.         console.log("close:"+sockState());
    
62.         webSocket.close();
    
63.     }
    1 W3 Q$ ?8 {) ]1 R+ F
64. 
    ' v# v1 J& ]0 m
65.     function sockState(){
    
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    
67.             return status[webSocket.readyState];
    
68.     }
    
69. 
    
70. 
    
71. 
    
72. function start(event){
    ! |$ |, c/ z6 H4 V3 H# V9 l
73.         console.log(webSocket);
    
74.         var msg = document.getElementById('text').value;
    $ T: n4 x# K3 f- l# S, i
75.         document.getElementById('text').value = '';
    
76.         console.log("send:"+sockState());
    9 s3 M: c" n" _5 w; K3 q: I+ a3 Y/ j- i
77.         console.log("msg="+msg);
    
78.         webSocket.send("msg="+msg);
    
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    
80.     };
      }+ w1 o$ i3 x9 t" P. U
81. 
    
82.     function close(event){
    
83.         webSocket.close();
    
84.     }
    
85. </script>
    * i/ E  M4 O/ h5 w
86. </body>
    9 P8 T& u& W; K. q
87. </html>

复制代码

) M0 ^+ X3 I1 V8 ^( W* Z+ J5 r* e
* f$ A: y; G" M% X% ]$ a