php实现websocket实时消息推送

admin

php实现websocket实时消息推送
, w! R- ~* I% y, {. @3 J& D

% i7 N8 B9 |: c* oSocketService.php

1. <?php
   
2. /**
   1 a$ N2 N0 @5 ^3 k& r1 u
3. * Created by xwx
   . a1 |6 e3 k9 I
4. * Date: 2017/10/18
   
5. * Time: 14:33
   
6. */
   
7. 
   - h2 f' w2 x- n; e
8. class SocketService
   
9. {
   ! F5 y4 [4 {& f4 q
10.     private $address  = '0.0.0.0';
    
11.     private $port = 8083;
    % C0 o: c5 A) L% Z( i+ D
12.     private $_sockets;
    9 b6 O3 g1 S( x( M
13.     public function __construct($address = '', $port='')
    
14.     {
    
15.             if(!empty($address)){
    ) u: z- k1 L" e
16.                 $this->address = $address;
    1 E1 p8 s* h" a8 v. m
17.             }
    : ]2 f8 d/ {6 w) z6 n
18.             if(!empty($port)) {
    7 R8 i4 e; C0 f+ o9 U. o" V
19.                 $this->port = $port;
    $ Q; N0 P1 Q6 C& P0 b1 F
20.             }
    
21.     }
    7 T, d( K2 C* `) J$ J
22. 
    ( p9 X2 {. t) X  g% b
23.     public function service(){
    
24.         //获取tcp协议号码。
    
25.         $tcp = getprotobyname("tcp");
    
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    4 ]/ ^) _% U" k
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    
28.         if($sock < 0)
    # U0 z; g6 t( z
29.         {
    $ W7 o! O# p, B3 v( g: g7 T
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    
31.         }
    & `: R$ ^, O* c  J1 n  h& j
32.         socket_bind($sock, $this->address, $this->port);
    0 M9 y1 S7 S+ \+ I* r
33.         socket_listen($sock, $this->port);
    7 l3 G' c) P6 z6 d! D/ T
34.         echo "listen on $this->address $this->port ... \n";
    7 }% X) b) r# ?5 F% L
35.         $this->_sockets = $sock;
    
36.     }
    
37. 
    
38.     public function run(){
    : I& J7 x7 U% f$ |+ F) P
39.         $this->service();
    
40.         $clients[] = $this->_sockets;
    
41.         while (true){
    
42.             $changes = $clients;
    
43.             $write = NULL;
    + u" Z- Y8 y, w4 s. j' w
44.             $except = NULL;
    
45.             socket_select($changes,  $write,  $except, NULL);
    
46.             foreach ($changes as $key => $_sock){
    
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    ! P9 \* i. U/ x: T0 {, g8 Y5 @
48.                     if(($newClient = socket_accept($_sock))  === false){
    
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    " A. T$ \& P0 `% X& }" F
50.                     }
    " s& P6 K+ \* q5 g; i9 {4 q
51.                     $line = trim(socket_read($newClient, 1024));
    0 [+ A7 D' c: O. H) |* P6 J
52.                     $this->handshaking($newClient, $line);
    . ?2 k; S( s) _  ?
53.                     //获取client ip
    / j/ v! z3 _+ g! n0 M2 p$ C8 u' q. }1 R
54.                     socket_getpeername ($newClient, $ip);
    
55.                     $clients[$ip] = $newClient;
    9 P, x1 M" d% m  X! b  F7 I0 {
56.                     echo  "Client ip:{$ip}   \n";
    
57.                     echo "Client msg:{$line} \n";
    % `$ Y4 @6 b+ N/ ?5 }
58.                 } else {
    3 o' v% u- f' [( R$ m
59.                     socket_recv($_sock, $buffer,  2048, 0);
    7 R* P/ p' K4 ?
60.                     $msg = $this->message($buffer);
    
61.                     //在这里业务代码
    ' E" L' w1 x7 B1 {) n: R
62.                     echo "{$key} clinet msg:",$msg,"\n";
    . y. q  A8 y' H# H) D1 c9 D
63.                     fwrite(STDOUT, 'Please input a argument:');
    , f( i9 l9 ~8 v) a2 J
64.                     $response = trim(fgets(STDIN));
    
65.                     $this->send($_sock, $response);
    ! x( P/ p3 a  j" Z9 T* @
66.                     echo "{$key} response to Client:".$response,"\n";
    
67.                 }
    % ]( D% f) i; I. r  {
68.             }
    
69.         }
    
70.     }
    * r3 Z3 C; p5 O$ e3 Z: k) b+ Y
71. 
    . B( b3 ^  G  k: b& O- p7 w# F
72.     /**
    
73.      * 握手处理
    
74.      * @param $newClient socket
    
75.      * @return int  接收到的信息
    
76.      */
    % ~' d$ i; c" w& l3 q+ ]
77.     public function handshaking($newClient, $line){
    ! A' ~6 M; [9 W( h1 H0 l7 S) V
78. 
    
79.         $headers = array();
    & j! y0 [; @2 A* R: v
80.         $lines = preg_split("/\r\n/", $line);
    
81.         foreach($lines as $line)
    
82.         {
    
83.             $line = chop($line);
    ' |6 \  W/ t7 d1 a  e0 \
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    ( S7 n3 ]5 z3 v3 m
85.             {
    
86.                 $headers[$matches[1]] = $matches[2];
    & I4 e+ z. m+ c) K! r0 W
87.             }
    
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" .
    2 D/ P1 o* v1 }' a- A
92.             "Upgrade: websocket\r\n" .
    
93.             "Connection: Upgrade\r\n" .
    7 s$ z9 `+ n$ I2 J$ |$ P3 u- _
94.             "WebSocket-Origin: $this->address\r\n" .
    ; ^1 ~3 ~* A* H4 w3 F& ~
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.     }
    
99. 
    
100.     /**
     
101.      * 解析接收数据
     & u2 G/ m4 I8 S/ t; Q7 L& b/ z/ }
102.      * @param $buffer
     0 z% e7 w, M% z0 m5 h
103.      * @return null|string
     
104.      */
     - t7 M, z; Q& G2 }: A( v
105.     public function message($buffer){
     
106.         $len = $masks = $data = $decoded = null;
     
107.         $len = ord($buffer[1]) & 127;
     
108.         if ($len === 126)  {
     " m( ~/ k$ N9 ]2 W! u3 y, ^6 S
109.             $masks = substr($buffer, 4, 4);
     
110.             $data = substr($buffer, 8);
     
111.         } else if ($len === 127)  {
     
112.             $masks = substr($buffer, 10, 4);
     
113.             $data = substr($buffer, 14);
     8 A2 h' X9 e9 H0 E1 O& W
114.         } else  {
     
115.             $masks = substr($buffer, 2, 4);
     / E1 H4 U5 W; ]
116.             $data = substr($buffer, 6);
     
117.         }
     5 j! j8 N5 c0 ?
118.         for ($index = 0; $index < strlen($data); $index++) {
     
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     
120.         }
     
121.         return $decoded;
     
122.     }
     " M) L) L5 D+ _) W
123. 
     ( J5 X* @- V; A3 [4 t# D
124.     /**
     
125.      * 发送数据
     8 E  J  Z# T. e' I
126.      * @param $newClinet 新接入的socket
     
127.      * @param $msg   要发送的数据
     
128.      * @return int|string
     
129.      */
     
130.     public function send($newClinet, $msg){
     
131.         $msg = $this->frame($msg);
     # s* C+ O: P+ S2 U- s' q' ?4 C
132.         socket_write($newClinet, $msg, strlen($msg));
     
133.     }
     
134. 
     
135.     public function frame($s) {
     : Y, G1 j$ U1 F+ ?0 N$ c6 m
136.         $a = str_split($s, 125);
     ( v3 y7 F. m( _3 d* y
137.         if (count($a) == 1) {
     " f: _6 r& d$ h/ x& s: _
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     ! y9 s0 k. {9 c' L% C  T6 h2 X
139.         }
     
140.         $ns = "";
     2 u0 s6 j, Q% [% v* F0 ~: k& N1 {
141.         foreach ($a as $o) {
     
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     * Q% b/ I) e( D4 Y
143.         }
     
144.         return $ns;
     $ w/ B1 S: \/ A# M
145.     }
     
146. 
     
147.     /**
     7 c9 V6 |1 R- z& a# T9 {2 i7 {: N
148.      * 关闭socket
     2 L, U0 {  e; ^4 R* E9 w$ i- w  l
149.      */
     0 M, b6 e5 b% o$ p
150.     public function close(){
     
151.         return socket_close($this->_sockets);
     
152.     }
     0 w+ _% G! P9 ]6 u. F6 L
153. }
     
154. 
     
155. $sock = new SocketService();
     
156. $sock->run();
     
157. 
     ' k1 y# k* R. y6 ?5 N. ^* m+ [

复制代码

web.html

1. <!doctype html>
   
2. <html lang="en">
   / y+ H0 H0 ~, @0 _  o& Z" Q/ d
3. <head>
   
4.   <meta charset="UTF-8">
   
5.   <meta name="viewport" content="width=device-width,initial-scale=1, maximum-scale=1, user-scalable=no">
   9 V' k9 C# i) ~8 N, a! n
6.   <title>websocket</title>
   
7. </head>
   5 w7 ^7 t" |/ O4 S/ d
8. <body>
     K; G" s1 K) `8 Y# D8 \1 L; A7 p
9. <input id="text" value="">
   
10. <input type="submit" value="send" onclick="start()">
    
11. <input type="submit" value="close" onclick="close()">
    
12. <div id="msg"></div>
    2 E7 _$ d# D# X- v: x  i, Z: z: k
13. <script>
    " b6 a+ n, {1 L
14. /**
    
15. 0：未连接
    ) |5 I* P1 B* w& g9 O  m- n
16. 1：连接成功，可通讯
    
17. 2：正在关闭
    
18. 3：连接已关闭或无法打开
    
19. */
    
20. 
    2 b: @( j2 E* |/ J7 f! D. L
21.     //创建一个webSocket 实例
    
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    
23. 
    + L0 N- @6 q# v! m' f. G7 J
24. 
    - J# Z# m/ p& t  N+ }. a8 j9 g5 K
25.     webSocket.onerror = function (event){
    ) p1 A: U7 S7 e0 E
26.         onError(event);
    
27.     };
    
28. 
    # G# X# K! {) |
29.     // 打开websocket
    
30.     webSocket.onopen = function (event){
    
31.         onOpen(event);
    
32.     };
    
33. 
    
34.     //监听消息
    
35.     webSocket.onmessage = function (event){
    & I; ?) _: X1 G6 \
36.         onMessage(event);
    
37.     };
    
38. 
      J1 W+ ?8 [) s* m% b
39. 
    8 l# W7 @$ ^, s8 Q: F
40.     webSocket.onclose = function (event){
    
41.         onClose(event);
    # Y% b- J( A& ~0 S5 b! G7 O
42.     }
    
43. 
    % w+ x" z) n+ i$ z
44.     //关闭监听websocket
    8 v7 ?" f1 s8 C% _' Z
45.     function onError(event){
    
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    ' G; i$ H# ?2 ?& b. y6 H1 `
47.         console.log("error"+event.data);
    2 ~, E( Y5 n1 Q6 E$ w8 E, v" U
48.     };
    
49. 
    
50.     function onOpen(event){
    % v; B1 E. q, G6 z
51.         console.log("open:"+sockState());
    
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    / K* Q5 E& a% L7 {
53.     };
    / y1 B& q$ r6 w" C. a. F  s8 D1 Z
54.     function onMessage(event){
    ) h$ \0 A  _" n$ S
55.         console.log("onMessage");
    / N- _' G- P+ P8 W- \- J
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
      D# g. ~! g  {4 P. g3 q- h
57.     };
    8 m' K% C  L) p+ p' X! I
58. 
    7 E) p; e0 ^  a
59.     function onClose(event){
    3 ^0 e) g# W- M
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    - {1 @& M$ W3 S+ ?2 l
61.         console.log("close:"+sockState());
    * {, W. u  O& B, r9 }
62.         webSocket.close();
    * ?. h' A' E- I5 g# R  v! T
63.     }
    
64. 
    
65.     function sockState(){
    + V* |) L% r, W/ l1 f; ]
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    
67.             return status[webSocket.readyState];
    ( Z' B$ C  o. Y7 s( r, ]6 {( `$ T
68.     }
    5 G$ _+ O  ~! q" i. n
69. 
    ! a+ a3 \) M) C4 w2 h3 O: J; G
70. 
    
71. 
    
72. function start(event){
    
73.         console.log(webSocket);
    
74.         var msg = document.getElementById('text').value;
    : h* C. t. `( c3 t: a
75.         document.getElementById('text').value = '';
    
76.         console.log("send:"+sockState());
    7 v! ~4 o% i# c9 R
77.         console.log("msg="+msg);
    & @* k3 R9 I: r4 ~" k. z+ n, c
78.         webSocket.send("msg="+msg);
    
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    ( Y7 d, `8 {4 [$ I) d. ~8 K
80.     };
    
81. 
    3 ], _  w$ `# Y  h, O, t2 i
82.     function close(event){
    1 @: |1 \5 A6 o% f" s
83.         webSocket.close();
    5 h) d. s0 O+ ]- s6 K  R3 E
84.     }
    # D' {2 m% c4 z# A5 k& Q) I
85. </script>
    
86. </body>
    6 z7 T7 U5 ?4 K0 g% @
87. </html>

复制代码

& ~& `! t' @  T; a! S
4 ~" h2 W, o' M) Y& f4 y' Q2 {: Q! T