php实现websocket实时消息推送

admin

php实现websocket实时消息推送
0 T; u: J( o4 `, P7 i( O$ M4 X
! B* ?5 U2 z" x" U: v+ w+ T
& {2 z5 P: j3 _% k1 q, N; ^SocketService.php

1. <?php
     x" E  ?5 g/ P+ d( G
2. /**
   8 K: \# t9 P* J6 @: ^; J2 m8 k
3. * Created by xwx
   
4. * Date: 2017/10/18
   & Y2 x- A; y+ k% G( g
5. * Time: 14:33
   
6. */
   
7. 
   ( R! h: {3 p' Q' s6 l) j
8. class SocketService
   ; T& M' y4 I3 u) X- l
9. {
   ! m& v2 \# W2 S' n* K$ J5 f$ g) W
10.     private $address  = '0.0.0.0';
    
11.     private $port = 8083;
    : Z/ V+ r+ f5 a) X3 R
12.     private $_sockets;
    
13.     public function __construct($address = '', $port='')
    3 }" f% b9 a* g# S
14.     {
    6 m  Z/ r* b) I4 i3 O, ~$ W" Y2 G7 o; q
15.             if(!empty($address)){
    
16.                 $this->address = $address;
    
17.             }
    % r7 u8 c% d* V- H9 w4 H' m
18.             if(!empty($port)) {
    
19.                 $this->port = $port;
    % ]% ?, d7 F" S& ?; T/ K
20.             }
    * a) a  d9 |  k6 x2 w' C* r, J
21.     }
    
22. 
    4 V% Q/ \4 @* D8 J- z! e' c- {/ m
23.     public function service(){
    
24.         //获取tcp协议号码。
    
25.         $tcp = getprotobyname("tcp");
    
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    , n2 z  O& Z/ p. E- {( L7 _  R; y
28.         if($sock < 0)
    
29.         {
    
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    
31.         }
    2 v0 M' k+ I6 M3 X* U% x  ^$ e
32.         socket_bind($sock, $this->address, $this->port);
    0 `8 t  h$ s- j8 u  E( O0 w
33.         socket_listen($sock, $this->port);
    
34.         echo "listen on $this->address $this->port ... \n";
    
35.         $this->_sockets = $sock;
    
36.     }
    1 R: K  E5 D8 U  N8 j: Q
37. 
    
38.     public function run(){
    ; u- }, v( E, q: O6 d
39.         $this->service();
    
40.         $clients[] = $this->_sockets;
    
41.         while (true){
    
42.             $changes = $clients;
    ) W1 f5 r- d7 Y: y: h& N! j0 m8 K
43.             $write = NULL;
    
44.             $except = NULL;
    
45.             socket_select($changes,  $write,  $except, NULL);
    
46.             foreach ($changes as $key => $_sock){
    1 T6 ?, Q/ l; N1 ^
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    
48.                     if(($newClient = socket_accept($_sock))  === false){
    
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    . b- V9 X. f2 A. _
50.                     }
    
51.                     $line = trim(socket_read($newClient, 1024));
    
52.                     $this->handshaking($newClient, $line);
    : U, O6 i6 _) Z& W+ c! u
53.                     //获取client ip
    / O5 `% `' [. T  N
54.                     socket_getpeername ($newClient, $ip);
    
55.                     $clients[$ip] = $newClient;
    1 p( y- v1 o4 N1 H& H: i
56.                     echo  "Client ip:{$ip}   \n";
    
57.                     echo "Client msg:{$line} \n";
    * E2 ]# r6 u- K
58.                 } else {
    , x8 c  K9 j) p7 k
59.                     socket_recv($_sock, $buffer,  2048, 0);
    
60.                     $msg = $this->message($buffer);
    
61.                     //在这里业务代码
      k  U4 r3 e( e
62.                     echo "{$key} clinet msg:",$msg,"\n";
    
63.                     fwrite(STDOUT, 'Please input a argument:');
    ' q0 p, L/ t& A0 M4 Y, |! G  a. I
64.                     $response = trim(fgets(STDIN));
    ' t) x2 Z2 M( o, H! h, K4 L
65.                     $this->send($_sock, $response);
    
66.                     echo "{$key} response to Client:".$response,"\n";
    + z8 V5 B6 Y/ ^" A. ]# A
67.                 }
    7 C/ ]; u; _  P6 e
68.             }
    
69.         }
    
70.     }
    & e- a' ~2 `4 h" S
71. 
    
72.     /**
    
73.      * 握手处理
    
74.      * @param $newClient socket
    $ M2 |4 O' ~1 q$ d6 @# C' X2 i
75.      * @return int  接收到的信息
      |/ l- G! ^# D% s4 L1 Z9 v
76.      */
    & H( l6 Z4 x2 U
77.     public function handshaking($newClient, $line){
    
78. 
    ! K- }. y) q% m+ v! T# y3 b
79.         $headers = array();
    # A) K% a2 [2 R& _8 N+ M7 {
80.         $lines = preg_split("/\r\n/", $line);
    + H0 _* s/ I. k; p' ?
81.         foreach($lines as $line)
    ' p3 n) u& _6 x/ [6 L: V
82.         {
    
83.             $line = chop($line);
    2 {+ s% |+ A' H' F$ t
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    
85.             {
    ' W- I# N9 m1 K
86.                 $headers[$matches[1]] = $matches[2];
    
87.             }
    " X# s2 h+ x' B. I2 k5 R
88.         }
    
89.         $secKey = $headers['Sec-WebSocket-Key'];
    . _4 ]9 ]& r) I
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" .
    & L, c3 y6 j4 q  d  {( [
93.             "Connection: Upgrade\r\n" .
    9 c7 L7 ~; s( A  Y( y
94.             "WebSocket-Origin: $this->address\r\n" .
    
95.             "WebSocket-Location: ws://$this->address:$this->port/websocket/websocket\r\n".
    . [* \! ]8 h# b# w
96.             "Sec-WebSocket-Accept:$secAccept\r\n\r\n";
    
97.         return socket_write($newClient, $upgrade, strlen($upgrade));
    1 V) o- E2 I) A9 Q/ G
98.     }
    ) \/ O/ @) q0 T
99. 
    8 b3 E1 L2 G$ k2 B7 t3 L
100.     /**
     
101.      * 解析接收数据
     : D2 W) `& i6 @( e  y! K3 _# k
102.      * @param $buffer
     
103.      * @return null|string
     4 k/ F5 N0 @/ d
104.      */
     
105.     public function message($buffer){
     / V7 n8 |& l- R0 j" p, ]$ C
106.         $len = $masks = $data = $decoded = null;
     9 D: W0 j; P* B* N
107.         $len = ord($buffer[1]) & 127;
     
108.         if ($len === 126)  {
     
109.             $masks = substr($buffer, 4, 4);
     
110.             $data = substr($buffer, 8);
     
111.         } else if ($len === 127)  {
     . W4 v1 ^, N% J( ^
112.             $masks = substr($buffer, 10, 4);
     ; N9 a8 g2 C) N0 f
113.             $data = substr($buffer, 14);
     
114.         } else  {
     , |5 O. D- I2 t) z; e
115.             $masks = substr($buffer, 2, 4);
     3 _& h2 d& n) b; K# m2 ?  I
116.             $data = substr($buffer, 6);
     , G5 c/ D/ F  k. P. u# a6 {
117.         }
     ! V1 l6 s$ b# p6 ^9 P$ j. M
118.         for ($index = 0; $index < strlen($data); $index++) {
     
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     + S" t( U( s( w/ I# j
120.         }
     
121.         return $decoded;
     3 d& W: M% N- |% T$ u: C
122.     }
     . N$ N+ l. \' v/ A* l' J4 o) d
123. 
     
124.     /**
     
125.      * 发送数据
     
126.      * @param $newClinet 新接入的socket
     
127.      * @param $msg   要发送的数据
       U' C  z$ A/ p# o5 Z% a. `
128.      * @return int|string
     
129.      */
     4 ~- \; T& e+ X0 ^# f$ j$ }
130.     public function send($newClinet, $msg){
     1 f5 b* \: c* m
131.         $msg = $this->frame($msg);
     
132.         socket_write($newClinet, $msg, strlen($msg));
     6 y8 Y; I4 B( k+ v9 r
133.     }
     
134. 
     % n# E! T. N9 }' i
135.     public function frame($s) {
     
136.         $a = str_split($s, 125);
     2 c1 f( O- A/ @7 Y  x3 r
137.         if (count($a) == 1) {
     ! a% a" |$ u' g+ {
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     6 d8 |  i: u* w( v$ E; ]" O8 J$ D
139.         }
     + X) u+ W3 G6 h0 _% S# T
140.         $ns = "";
     
141.         foreach ($a as $o) {
     
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     2 G; ]& ?; r# g- p
143.         }
     8 {3 y- z& T. X
144.         return $ns;
     ) S9 Y' `) a& V
145.     }
     1 B) G4 R8 c& ~5 ^
146. 
     ; g. j' k! A, K6 {& v0 U2 S
147.     /**
     
148.      * 关闭socket
     
149.      */
     8 f6 v0 z8 P1 R3 G7 J& T, F" l9 M
150.     public function close(){
     . ]- O% r9 z# B5 {- s
151.         return socket_close($this->_sockets);
     1 q# S- K! i4 {, m1 q/ A7 |# g
152.     }
     1 c) N0 h0 d6 c; x
153. }
     
154. 
     7 G( Z* c  @5 C7 W
155. $sock = new SocketService();
     
156. $sock->run();
     , Y. O) j0 D' N
157. 
     

复制代码

web.html

1. <!doctype html>
   7 v& R: y/ {% s4 i" P
2. <html lang="en">
   
3. <head>
   
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>
   " e8 u+ Z& _! J" m* K
7. </head>
   
8. <body>
   
9. <input id="text" value="">
   
10. <input type="submit" value="send" onclick="start()">
    ) Q. g% M9 S2 _+ P3 T
11. <input type="submit" value="close" onclick="close()">
    
12. <div id="msg"></div>
    
13. <script>
    ' S4 P6 o7 w/ `7 q
14. /**
    & H, ~  \" v! t% M$ R7 c
15. 0：未连接
    
16. 1：连接成功，可通讯
    
17. 2：正在关闭
    
18. 3：连接已关闭或无法打开
    % o5 c! {. I9 v
19. */
    0 N) M* A9 M" }' {& D; D/ D/ ]2 ^2 e
20. 
    
21.     //创建一个webSocket 实例
    3 z( O- ^1 u3 g# Z
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    
23. 
    3 `1 j9 x+ W$ E& p$ y
24. 
    % H" Q- ~  @  n0 Q4 |' ?% p$ A2 x# g
25.     webSocket.onerror = function (event){
    + {0 J: T3 f/ q+ d3 H
26.         onError(event);
    
27.     };
    
28. 
    ( ~# ^! h$ }+ v; Q; V8 T
29.     // 打开websocket
    & w$ L4 |2 L4 |4 H! E: F
30.     webSocket.onopen = function (event){
    
31.         onOpen(event);
    $ f5 }  Q; S, {4 h
32.     };
    
33. 
    
34.     //监听消息
    
35.     webSocket.onmessage = function (event){
    % t0 {" o# \& g5 p0 ]
36.         onMessage(event);
    # H3 K- o/ ?: V7 A; C6 c, P
37.     };
    6 T- z- n! D% @7 w
38. 
    7 C* w; h1 |" n5 D2 |& N9 E1 v+ {' t
39. 
    
40.     webSocket.onclose = function (event){
    
41.         onClose(event);
    
42.     }
    
43. 
    
44.     //关闭监听websocket
    
45.     function onError(event){
    9 P& l& L! u# K! \, t' H2 D
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
47.         console.log("error"+event.data);
    , F( j9 ^: t& m  r# V
48.     };
    
49. 
    / S# ^* d  K0 T! l* u
50.     function onOpen(event){
    
51.         console.log("open:"+sockState());
    
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    9 O) C7 S2 ^- w3 K5 ^4 g9 |
53.     };
    # a! j" J5 L6 c; l/ }. s
54.     function onMessage(event){
    
55.         console.log("onMessage");
    
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    / O3 z6 p8 @8 @9 M* j: o
57.     };
    
58. 
    0 G5 }( }7 i1 U0 {0 s5 X
59.     function onClose(event){
    3 [2 j+ t% T- T- i
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    7 H7 D# ^; S- Q- \4 s6 Z
61.         console.log("close:"+sockState());
    4 d& d7 l( S1 g9 v5 K) g/ p& X% }
62.         webSocket.close();
    8 b# D& F1 ]$ _% u. c. g
63.     }
    ; t" Z; H& k5 q: h" V2 v
64. 
    
65.     function sockState(){
    
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    
67.             return status[webSocket.readyState];
    
68.     }
    
69. 
    
70. 
    
71. 
    & t' `& j( ]) o. G6 }
72. function start(event){
    
73.         console.log(webSocket);
    0 \5 {4 ?; x. x) a
74.         var msg = document.getElementById('text').value;
    
75.         document.getElementById('text').value = '';
    
76.         console.log("send:"+sockState());
    * x2 C: N$ T( T& J
77.         console.log("msg="+msg);
    / f# h5 R. L! Q  ]/ f; I' `
78.         webSocket.send("msg="+msg);
    
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    $ O2 Q$ o+ y( @; `$ n
80.     };
    
81. 
    # T3 Z! W, }6 S
82.     function close(event){
    2 ?0 |) a" N. a
83.         webSocket.close();
    
84.     }
    - E) X* c7 @+ ^* @: E. F3 L2 z
85. </script>
    # a' p6 v: G9 C
86. </body>
    
87. </html>

复制代码

) n0 P, n9 J! T& S. A  j) ^+ K0 t