php实现websocket实时消息推送

admin

php实现websocket实时消息推送


SocketService.php
4 ^% c0 Q7 E- I) K% x+ @3 E/ `

1. <?php
   # E8 T" R1 m- G1 M7 U2 T6 w
2. /**
   6 f3 ^- j4 k/ Z. I
3. * Created by xwx
   
4. * Date: 2017/10/18
   
5. * Time: 14:33
   
6. */
   , g$ k4 k! x# a8 E7 q3 _: |
7. 
   " f! P- D! j/ B+ v1 r
8. class SocketService
   6 X, a, d$ ?/ v/ K6 k% f
9. {
   3 d, R1 v8 ~/ g/ N" }7 J
10.     private $address  = '0.0.0.0';
    ) K- S& ?1 h; S' R& x
11.     private $port = 8083;
    
12.     private $_sockets;
    
13.     public function __construct($address = '', $port='')
    
14.     {
    6 Y. @& r+ O  ]8 G) j
15.             if(!empty($address)){
    
16.                 $this->address = $address;
    
17.             }
    
18.             if(!empty($port)) {
    ( c3 Z7 [* @3 s& |3 Z+ @' R  f
19.                 $this->port = $port;
    
20.             }
    
21.     }
    0 M# _! a" s+ j2 {* c2 c) G
22. 
    
23.     public function service(){
    
24.         //获取tcp协议号码。
    ( }# s% l$ o$ [9 S' ?
25.         $tcp = getprotobyname("tcp");
    % K4 @# Y7 I8 g) z% Z, w
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    + u2 {$ F! ?. n
28.         if($sock < 0)
    
29.         {
    
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    
31.         }
    
32.         socket_bind($sock, $this->address, $this->port);
    + Z; V% V3 `; S8 w
33.         socket_listen($sock, $this->port);
    
34.         echo "listen on $this->address $this->port ... \n";
    6 o+ ^4 {/ y, U/ }- _- d6 h8 Z
35.         $this->_sockets = $sock;
    , f5 T( C0 Y% A) [" {, L
36.     }
    ; w& G$ T+ E2 L2 B1 g) Y7 f% t
37. 
    
38.     public function run(){
    1 O" ?% W, o& @; V3 `$ m
39.         $this->service();
      X& p6 G  l6 Y6 |( p, W! S
40.         $clients[] = $this->_sockets;
    
41.         while (true){
    
42.             $changes = $clients;
    ; a6 |3 k! n  d) }5 l
43.             $write = NULL;
    : Q! e6 x3 h7 ~6 H/ I" _
44.             $except = NULL;
    
45.             socket_select($changes,  $write,  $except, NULL);
    5 d6 v! _* G# R
46.             foreach ($changes as $key => $_sock){
    
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    
48.                     if(($newClient = socket_accept($_sock))  === false){
    3 O6 N/ M. l; S$ T4 K
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    
50.                     }
    ! b5 x  w2 h- z* A/ Z0 z$ M
51.                     $line = trim(socket_read($newClient, 1024));
    
52.                     $this->handshaking($newClient, $line);
    
53.                     //获取client ip
    8 f7 s3 h; ?+ L, l
54.                     socket_getpeername ($newClient, $ip);
    
55.                     $clients[$ip] = $newClient;
    + `! z  b" {. V7 v$ b: F
56.                     echo  "Client ip:{$ip}   \n";
    . k/ s, A7 c- D- m5 B  B2 e
57.                     echo "Client msg:{$line} \n";
    
58.                 } else {
    
59.                     socket_recv($_sock, $buffer,  2048, 0);
    
60.                     $msg = $this->message($buffer);
    
61.                     //在这里业务代码
    9 U  q% e: s5 z5 \# w8 w9 h- i* u
62.                     echo "{$key} clinet msg:",$msg,"\n";
    
63.                     fwrite(STDOUT, 'Please input a argument:');
    
64.                     $response = trim(fgets(STDIN));
    " w5 Z! n1 N4 ]
65.                     $this->send($_sock, $response);
    
66.                     echo "{$key} response to Client:".$response,"\n";
    1 B0 g4 g$ `/ P1 g4 ^4 [( c
67.                 }
    
68.             }
    
69.         }
    ( x3 V, s2 n8 b' ]! R$ K; {& E
70.     }
    
71. 
    
72.     /**
    
73.      * 握手处理
    
74.      * @param $newClient socket
    
75.      * @return int  接收到的信息
    : t1 C! O6 ^. s
76.      */
    3 ^+ e, S# ^; n" }
77.     public function handshaking($newClient, $line){
    
78. 
    . z4 |$ }/ k2 r3 q2 P
79.         $headers = array();
    
80.         $lines = preg_split("/\r\n/", $line);
    
81.         foreach($lines as $line)
    9 y) h2 A0 p+ t: D
82.         {
    % C/ o4 }# t4 x2 e( y
83.             $line = chop($line);
    % Y. j1 z& E3 D2 h) Z
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    
85.             {
    
86.                 $headers[$matches[1]] = $matches[2];
    
87.             }
    
88.         }
    . P7 r+ P( W/ ]+ Q" p
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" .
    7 l: W( J) ?# d4 y4 ~5 Q7 {
92.             "Upgrade: websocket\r\n" .
    
93.             "Connection: Upgrade\r\n" .
    8 R9 c# B1 ^; ]
94.             "WebSocket-Origin: $this->address\r\n" .
    : a: ]7 u; n, @" k" H, Z# o; C
95.             "WebSocket-Location: ws://$this->address:$this->port/websocket/websocket\r\n".
    
96.             "Sec-WebSocket-Accept:$secAccept\r\n\r\n";
    ) K1 n$ Z) `# H3 H
97.         return socket_write($newClient, $upgrade, strlen($upgrade));
    # V2 O# }% P3 D  h$ H: {: T: {/ @
98.     }
    
99. 
    ) P+ w, U; t6 c1 u8 l& ~! ?0 F' d6 {
100.     /**
     ( Y$ i, X) p" O7 {' r, U7 o0 j3 W$ R
101.      * 解析接收数据
     
102.      * @param $buffer
     
103.      * @return null|string
     
104.      */
     
105.     public function message($buffer){
     / f' l; k9 N; V: E; b" X
106.         $len = $masks = $data = $decoded = null;
     & V, E/ ^. g6 N) l
107.         $len = ord($buffer[1]) & 127;
     + O" b, e% ~# r5 C
108.         if ($len === 126)  {
     2 D# A& H& {- h- q0 h, x$ `
109.             $masks = substr($buffer, 4, 4);
     3 t( f3 S, A' @+ G6 Z
110.             $data = substr($buffer, 8);
     1 L* i! ?4 v! N* C. J! x
111.         } else if ($len === 127)  {
     ) W  Q6 ]( _3 t
112.             $masks = substr($buffer, 10, 4);
     
113.             $data = substr($buffer, 14);
     
114.         } else  {
     
115.             $masks = substr($buffer, 2, 4);
     
116.             $data = substr($buffer, 6);
     4 q& @0 J/ P# b% _
117.         }
     
118.         for ($index = 0; $index < strlen($data); $index++) {
     
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     7 b: _7 G/ L& m" t, L; U
120.         }
     1 U1 K( i$ I+ g: P3 e& ^9 }
121.         return $decoded;
     ( {; F; b, c) |6 A" x: u  j! F
122.     }
     4 S( g2 A' d0 n( V4 O2 P% I- ?& F
123. 
     
124.     /**
     2 }- F7 L" P0 X+ p, l( H, z, f. p& |
125.      * 发送数据
     
126.      * @param $newClinet 新接入的socket
     
127.      * @param $msg   要发送的数据
     , |  o0 W7 |; q. b6 T3 p
128.      * @return int|string
     
129.      */
     
130.     public function send($newClinet, $msg){
     # X9 l0 c+ ~( z0 w9 e% b, R! l7 |( W+ x
131.         $msg = $this->frame($msg);
     
132.         socket_write($newClinet, $msg, strlen($msg));
     . [+ Q4 o2 l2 q1 x& O3 u
133.     }
     ) t& y; b% A0 h' Z! r0 B: f
134. 
     & z% |  v& P+ p: ^* F
135.     public function frame($s) {
     & d; Z! R" E0 y! C+ Z6 {
136.         $a = str_split($s, 125);
     
137.         if (count($a) == 1) {
     # O0 Q+ `- F6 W; J7 u( D
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     
139.         }
     
140.         $ns = "";
     0 a8 a% G3 Y( S
141.         foreach ($a as $o) {
     " L& G0 m& |: |9 }( ^
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     
143.         }
     ! e, c7 Z9 Z1 n' `. q
144.         return $ns;
     
145.     }
     
146. 
     8 g4 P% p4 L: Y6 U: v: |5 N. y
147.     /**
     
148.      * 关闭socket
     
149.      */
     ) U3 U/ i3 `# {, w
150.     public function close(){
     
151.         return socket_close($this->_sockets);
     
152.     }
     3 i; f7 n7 I4 O* ^7 F/ i5 q% M
153. }
     + p8 T/ Q0 q% m
154. 
     0 `9 [: z( E2 |- w8 H% q* p; m! y% h
155. $sock = new SocketService();
     
156. $sock->run();
     : R- Z5 C) O7 A
157. 
     

复制代码

web.html
# t* n: Y( s, i; `7 U! c5 S$ f

1. <!doctype html>
   3 e7 D8 b: d& e0 P2 {! F6 c
2. <html lang="en">
   # }2 s/ e$ e9 ?4 {( X
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>
   
7. </head>
   
8. <body>
   
9. <input id="text" value="">
   1 h1 F; `9 v/ L5 U. A$ Y+ O( R5 v# T
10. <input type="submit" value="send" onclick="start()">
    
11. <input type="submit" value="close" onclick="close()">
    
12. <div id="msg"></div>
    
13. <script>
    
14. /**
    5 V3 H, e3 B8 P$ [. T8 Y
15. 0：未连接
    
16. 1：连接成功，可通讯
    
17. 2：正在关闭
    
18. 3：连接已关闭或无法打开
    4 a! \  i% b$ ]
19. */
    0 n8 j/ o2 K# n& S+ u2 A$ [/ D0 j
20. 
    
21.     //创建一个webSocket 实例
    
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    & q/ r0 z- T$ G1 M
23. 
    
24. 
    
25.     webSocket.onerror = function (event){
    
26.         onError(event);
      D: O* Y* s7 _& v4 z" _, n
27.     };
    + p0 y' O$ ]; D: d9 e1 q/ {
28. 
    ( s6 P  G7 `' a+ f
29.     // 打开websocket
    
30.     webSocket.onopen = function (event){
    
31.         onOpen(event);
    7 R8 C6 E2 N7 J" i; E
32.     };
    
33. 
    
34.     //监听消息
    " C6 O9 m5 M  i3 z8 e
35.     webSocket.onmessage = function (event){
    
36.         onMessage(event);
    
37.     };
    7 F2 m% d5 Q3 ~6 |0 i
38. 
    7 y! ?" u( K* e* v, K" [8 }% i5 d3 C
39. 
    9 J. _9 ~: U3 Z, N8 f
40.     webSocket.onclose = function (event){
    
41.         onClose(event);
    
42.     }
    ' @, d' U8 w. A2 g+ K
43. 
    
44.     //关闭监听websocket
    
45.     function onError(event){
    $ T. C4 p: P. u8 i
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    5 \7 c* Q1 j3 r1 B8 s
47.         console.log("error"+event.data);
    9 D/ @1 G8 }, S! [1 [# B) _" a
48.     };
    
49. 
    
50.     function onOpen(event){
    ( O" I3 z( ?4 V! b
51.         console.log("open:"+sockState());
    
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    
53.     };
    
54.     function onMessage(event){
    : p; N: l3 j# _8 J
55.         console.log("onMessage");
    
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    
57.     };
    
58. 
    
59.     function onClose(event){
    % L1 j; \# `* B$ X3 t
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    4 X# f! e  s" t" {- l. d
61.         console.log("close:"+sockState());
    
62.         webSocket.close();
    6 u: i% P/ Q3 F2 @4 r- h: ]
63.     }
    
64. 
    - G- ?  l& C! L. n/ X" }
65.     function sockState(){
    # I  }  Z! l: e2 f1 B5 j0 z8 m
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    * f* _, r: e2 l
67.             return status[webSocket.readyState];
    
68.     }
    
69. 
    , W6 s7 A& E1 S6 E7 @( s
70. 
    
71. 
    
72. function start(event){
    , _; M# g; m, }0 G( t$ X5 r
73.         console.log(webSocket);
    ' N) \1 j% Y2 ~* J2 P' c
74.         var msg = document.getElementById('text').value;
    
75.         document.getElementById('text').value = '';
    
76.         console.log("send:"+sockState());
    2 r7 W% y9 }/ U4 e  |
77.         console.log("msg="+msg);
    4 H  E% b+ N& P2 \
78.         webSocket.send("msg="+msg);
    
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    + x9 r3 }9 h9 N3 Y. l
80.     };
    
81. 
    " X, q: ?" b# f5 v9 s) }6 l: h, B* r
82.     function close(event){
    * s; M3 p7 [& U/ y7 D
83.         webSocket.close();
    
84.     }
    + t. V. p# A# u5 t! {$ I- t
85. </script>
    
86. </body>
    
87. </html>

复制代码

& H+ h  l  F' O4 M$ ^# L' z' o4 v5 `0 l
6 f# Q. a& l$ s8 Y: v' y, c