php实现websocket实时消息推送

admin

php实现websocket实时消息推送
1 p5 M* p! J+ L" Y. i
' C. P4 X, M0 C, v7 x
9 _! |  d/ F3 q5 e) C7 X$ j% [SocketService.php
$ g1 H9 j. Q/ B. n- v

1. <?php
   ( U) t# ?& b1 }$ v) O
2. /**
   . ~# _* {9 K) T6 v2 w3 e1 a! f1 w
3. * Created by xwx
   % w' ]' P5 S  |
4. * Date: 2017/10/18
   ! A% _' N- y/ `) G; r
5. * Time: 14:33
   5 D8 `( K$ J7 Q
6. */
   / M! l9 A4 h( \5 t/ j  M/ ?
7. 
   
8. class SocketService
   1 |/ d: N6 f/ R
9. {
   
10.     private $address  = '0.0.0.0';
    
11.     private $port = 8083;
    # k. r" x6 x! m8 J
12.     private $_sockets;
    
13.     public function __construct($address = '', $port='')
    
14.     {
    
15.             if(!empty($address)){
    
16.                 $this->address = $address;
    7 q1 S+ f8 p- P% L% @) M5 G
17.             }
    ' J, h3 z, ~3 ?, L/ g
18.             if(!empty($port)) {
    : I+ R; O9 ^, ]$ v
19.                 $this->port = $port;
    
20.             }
    
21.     }
    0 k  v- A- P" \3 r
22. 
    
23.     public function service(){
    
24.         //获取tcp协议号码。
    
25.         $tcp = getprotobyname("tcp");
    ' _- L% l! K  C
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    
28.         if($sock < 0)
    ( |' f/ y" q. A
29.         {
    
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    
31.         }
    
32.         socket_bind($sock, $this->address, $this->port);
    
33.         socket_listen($sock, $this->port);
    
34.         echo "listen on $this->address $this->port ... \n";
    9 b, I* j  w8 ^" H/ v9 ~7 p4 D. r2 i' V  ^+ B
35.         $this->_sockets = $sock;
    
36.     }
    
37. 
    
38.     public function run(){
    
39.         $this->service();
    ' e" p8 V# i2 D1 V6 R! {9 o% X
40.         $clients[] = $this->_sockets;
    3 |! U; ]- Z, \
41.         while (true){
    
42.             $changes = $clients;
    
43.             $write = NULL;
    
44.             $except = NULL;
    
45.             socket_select($changes,  $write,  $except, NULL);
    
46.             foreach ($changes as $key => $_sock){
    
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    0 ~8 R; \6 i4 l
48.                     if(($newClient = socket_accept($_sock))  === false){
    : X; N* \$ V: O. P' e( ^6 _' c
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    4 I( p1 ?( s4 l9 _6 i
50.                     }
    
51.                     $line = trim(socket_read($newClient, 1024));
    
52.                     $this->handshaking($newClient, $line);
    ; b4 @7 a! I# q8 D% ?
53.                     //获取client ip
    
54.                     socket_getpeername ($newClient, $ip);
    
55.                     $clients[$ip] = $newClient;
    
56.                     echo  "Client ip:{$ip}   \n";
    
57.                     echo "Client msg:{$line} \n";
    : W" O3 R. l  z# J
58.                 } else {
    0 C" T: B1 T8 k! h
59.                     socket_recv($_sock, $buffer,  2048, 0);
    3 C- @7 H3 {. u8 ]5 R; o
60.                     $msg = $this->message($buffer);
    * H& D4 s0 a! N4 W
61.                     //在这里业务代码
    9 V9 }  Y$ L+ i: f6 C
62.                     echo "{$key} clinet msg:",$msg,"\n";
    1 F( ~6 a: Z% Y: K/ r
63.                     fwrite(STDOUT, 'Please input a argument:');
    ) o* f! j1 e$ |  H4 x% f
64.                     $response = trim(fgets(STDIN));
    
65.                     $this->send($_sock, $response);
    $ Y$ F8 h5 r6 x6 s
66.                     echo "{$key} response to Client:".$response,"\n";
    2 g. t) U: n2 y- E9 \! H- W
67.                 }
    8 I0 w: V$ M2 y* h
68.             }
    
69.         }
    
70.     }
    9 [* [* E( j' e& Q) g9 R, r
71. 
      p% H  R: {: ]. A2 P# J
72.     /**
    
73.      * 握手处理
    2 G3 m9 H) v2 P( j2 F- Y
74.      * @param $newClient socket
    
75.      * @return int  接收到的信息
    ! d$ y  `0 }& U1 Q; L
76.      */
    
77.     public function handshaking($newClient, $line){
    + X* J: q$ a5 E( [, H; k9 j4 A4 i
78. 
    
79.         $headers = array();
    
80.         $lines = preg_split("/\r\n/", $line);
    - @1 G+ S: f) F1 Y- F
81.         foreach($lines as $line)
    
82.         {
    
83.             $line = chop($line);
    
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    . |, D# U6 {. k
85.             {
    
86.                 $headers[$matches[1]] = $matches[2];
    5 p5 d5 I! g3 s5 U. L$ b/ z3 Z
87.             }
    
88.         }
    ' O/ i! @  V# c- ]
89.         $secKey = $headers['Sec-WebSocket-Key'];
    
90.         $secAccept = base64_encode(pack('H*', sha1($secKey . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11')));
    8 g4 j6 ]' f3 P5 ~8 E, ]& P% M; U7 s
91.         $upgrade  = "HTTP/1.1 101 Web Socket Protocol Handshake\r\n" .
    
92.             "Upgrade: websocket\r\n" .
    
93.             "Connection: Upgrade\r\n" .
    
94.             "WebSocket-Origin: $this->address\r\n" .
    8 w8 f$ s* b" N% `  p/ ]( _" w. Q
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));
    " q: a  n( n$ }! H
98.     }
    
99. 
    , e+ [4 B5 Y4 H0 \9 s
100.     /**
     
101.      * 解析接收数据
     
102.      * @param $buffer
     # W+ [5 X  ]9 e7 Z2 \, L
103.      * @return null|string
     ( |2 O, [, `/ f8 P  C  @, {: N1 V" M
104.      */
     
105.     public function message($buffer){
     ! c+ V' R/ i  G
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);
     
111.         } else if ($len === 127)  {
     ' o2 s! G7 e% \4 P3 g: o6 y3 D0 w
112.             $masks = substr($buffer, 10, 4);
     
113.             $data = substr($buffer, 14);
     
114.         } else  {
     - G" ]0 W0 Y) I
115.             $masks = substr($buffer, 2, 4);
     
116.             $data = substr($buffer, 6);
     " W$ \8 k& c$ p; o( M* z; a, v
117.         }
     
118.         for ($index = 0; $index < strlen($data); $index++) {
       j/ t9 l7 W+ o: Z% }( P
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     
120.         }
     
121.         return $decoded;
     1 C& c% R& O% p+ o
122.     }
     & _$ g7 I$ P' j$ A. h0 A7 n) I
123. 
     5 a# v8 R& s1 T6 k; g
124.     /**
     
125.      * 发送数据
     : T) `  i8 G: K; q
126.      * @param $newClinet 新接入的socket
     * y, Z7 x6 R3 `- [
127.      * @param $msg   要发送的数据
     
128.      * @return int|string
     
129.      */
     
130.     public function send($newClinet, $msg){
     
131.         $msg = $this->frame($msg);
     
132.         socket_write($newClinet, $msg, strlen($msg));
     , P. n. ^- X, G! }% C9 j
133.     }
     * S& S/ Y* `5 ?2 e3 d! a6 a. n" }0 X5 `
134. 
     
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) {
     % Q2 G, X3 G$ U* G/ E$ K" d
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     
143.         }
     ' ]7 O' ~& B  x& u$ p& s8 A
144.         return $ns;
     
145.     }
     
146. 
     
147.     /**
     $ i. w/ i" q5 P# U. X: A
148.      * 关闭socket
     
149.      */
     
150.     public function close(){
     ' c. X4 p$ N9 K7 T) N+ W
151.         return socket_close($this->_sockets);
     6 T# k' \3 p" C( E% j
152.     }
     . T; a( g. P8 C5 t9 R" h+ \, t5 i. W
153. }
     ( [# m: \7 j1 F% b
154. 
     - O2 L8 H6 o: [# n2 Q' m
155. $sock = new SocketService();
     
156. $sock->run();
     / v( K8 p# p0 l, t3 [3 O9 W2 a
157. 
       D* a% m" z3 [- x6 e

复制代码

web.html

1. <!doctype html>
   , V1 S4 n9 F0 r, i& L7 T& l; {$ b
2. <html lang="en">
   * {5 D! H8 D0 O
3. <head>
     n+ u  V- X* F0 |/ W" e% e3 P2 D
4.   <meta charset="UTF-8">
   + C$ U# c- l/ T" }: W
5.   <meta name="viewport" content="width=device-width,initial-scale=1, maximum-scale=1, user-scalable=no">
   ! Y% D* g8 a! \1 j7 p
6.   <title>websocket</title>
   7 o& Q' _: ~% Y) k" a8 J
7. </head>
   9 _' o  E2 F  X
8. <body>
   
9. <input id="text" value="">
   : N  ?: ~8 h  Y- C+ z
10. <input type="submit" value="send" onclick="start()">
    ' J/ v0 @, a6 q: j4 u8 X
11. <input type="submit" value="close" onclick="close()">
      l  I0 ?1 Z+ O/ S- N
12. <div id="msg"></div>
    
13. <script>
    6 C: m1 w; b) O
14. /**
    & \6 p4 a5 \" `+ _6 t5 f
15. 0：未连接
    
16. 1：连接成功，可通讯
    
17. 2：正在关闭
    & V' u, A1 {% m, |/ b; o
18. 3：连接已关闭或无法打开
    2 U: n' j/ d& z
19. */
    
20. 
    1 Q: u) l( y; L
21.     //创建一个webSocket 实例
    & h. L' y' t% X0 C
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    
23. 
    
24. 
    : S) ?& {3 C; I8 Q$ X
25.     webSocket.onerror = function (event){
    
26.         onError(event);
    ) W9 @, [8 {6 c
27.     };
    
28. 
    + ^; e# W. N+ f8 i
29.     // 打开websocket
    
30.     webSocket.onopen = function (event){
    . |# `- p, ~8 Z  z& r
31.         onOpen(event);
    
32.     };
    
33. 
    - W% S7 W& ?6 V. E
34.     //监听消息
    
35.     webSocket.onmessage = function (event){
    
36.         onMessage(event);
    
37.     };
    
38. 
    
39. 
    7 m" \9 j& _8 `% X9 a
40.     webSocket.onclose = function (event){
    $ h# t7 x' w3 j" t6 X( _
41.         onClose(event);
    
42.     }
    ; f7 [+ @# j0 j$ H! _4 O8 g0 M
43. 
    
44.     //关闭监听websocket
    
45.     function onError(event){
    ! [- ?! `5 q/ Q
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    9 [5 m* c8 F% l
47.         console.log("error"+event.data);
    
48.     };
    $ W2 l) O& e: g2 J/ M  u. W
49. 
    5 N' h! {, P) d
50.     function onOpen(event){
    
51.         console.log("open:"+sockState());
    ( `. r4 ~% [7 B* u$ P0 g, b/ b
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    
53.     };
    
54.     function onMessage(event){
    
55.         console.log("onMessage");
    3 a* E7 T4 j2 d2 Z6 z9 |- `" w
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    
57.     };
    $ t, u6 \" v! e0 f! J. U9 O
58. 
    0 t1 W7 X$ \) t+ a+ i; f3 b
59.     function onClose(event){
    0 b8 K0 E9 s4 d0 S' S3 i; G
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
61.         console.log("close:"+sockState());
    
62.         webSocket.close();
    
63.     }
    
64. 
    4 F) z( \) |6 U. f
65.     function sockState(){
    
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    
67.             return status[webSocket.readyState];
    ' I: a8 q$ m. |' w
68.     }
    ! N, O3 q9 P. F/ N, e2 o( D1 P2 `- p
69. 
    
70. 
    9 _8 N" C6 l( y7 P
71. 
    
72. function start(event){
    
73.         console.log(webSocket);
    6 A/ E' y1 k8 R/ ]4 y; R
74.         var msg = document.getElementById('text').value;
    
75.         document.getElementById('text').value = '';
    9 K# ]( S* A& d1 w; G
76.         console.log("send:"+sockState());
    
77.         console.log("msg="+msg);
    ( X3 U8 n5 s0 y) Z; b" M$ e
78.         webSocket.send("msg="+msg);
    
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    8 c3 s' u! @' Z$ W; v9 T
80.     };
    
81. 
    
82.     function close(event){
    : r! A/ l0 C; K% n2 z
83.         webSocket.close();
    
84.     }
    3 }: t$ T* ~0 E
85. </script>
    $ B* l% Z5 ?9 j( c
86. </body>
    # H& I; H$ P& o: y" v' I' i
87. </html>

复制代码

$ V# N5 d1 N! a+ z