php实现websocket实时消息推送

admin

php实现websocket实时消息推送
6 D' _: q( J. R* {8 F0 K8 n+ C1 Y* ^

* N/ ]5 J, ^- A' l' s. s+ ?SocketService.php
2 b7 Z3 \% S5 j( e1 x; d- t. Y# A

1. <?php
   : ^$ |$ x4 A9 u+ t2 t) _
2. /**
   2 x- u0 Y  X" e; x# f
3. * Created by xwx
   
4. * Date: 2017/10/18
   1 J7 N) y5 Y. u" u# u
5. * Time: 14:33
   4 t! E/ b- f# ~7 y1 J" f; |
6. */
   
7. 
   
8. class SocketService
   
9. {
   
10.     private $address  = '0.0.0.0';
    
11.     private $port = 8083;
    
12.     private $_sockets;
    % }* w0 c- F0 G/ d: U
13.     public function __construct($address = '', $port='')
    % O# ^! o) Z3 c1 A
14.     {
    
15.             if(!empty($address)){
    , \% b& P* t; G8 Z- z* W2 }
16.                 $this->address = $address;
    ! R9 W0 N  f  m7 ~' @0 Q
17.             }
    
18.             if(!empty($port)) {
    
19.                 $this->port = $port;
    
20.             }
    7 z0 t5 q; y  s
21.     }
    
22. 
    
23.     public function service(){
    , \/ W0 w* z; }1 s- H7 R1 i9 M: g
24.         //获取tcp协议号码。
    5 B" m6 F+ C4 h$ F% N# a5 ^4 `# h
25.         $tcp = getprotobyname("tcp");
      J3 D' |0 f6 ]" c/ h0 o- w
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    ) W9 {! z5 R2 G3 r) v0 e/ s7 y( Z
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    $ U" @5 _6 @9 q) ^1 n
28.         if($sock < 0)
    
29.         {
    
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    
31.         }
    # Y8 F* \/ ]7 g
32.         socket_bind($sock, $this->address, $this->port);
    : f( Q7 v- V( a8 c0 I, T+ u
33.         socket_listen($sock, $this->port);
    
34.         echo "listen on $this->address $this->port ... \n";
    ! Y/ ~9 s' q- |) Y7 D
35.         $this->_sockets = $sock;
    
36.     }
    $ _; a1 L1 _% F8 z% x
37. 
    ! V  Y6 ^4 H& d
38.     public function run(){
    
39.         $this->service();
    
40.         $clients[] = $this->_sockets;
    
41.         while (true){
    , @( N4 U% _6 }' B
42.             $changes = $clients;
    ! O+ c0 w1 b, m7 c
43.             $write = NULL;
    
44.             $except = NULL;
    
45.             socket_select($changes,  $write,  $except, NULL);
    , Z% a& M. J0 A3 `
46.             foreach ($changes as $key => $_sock){
    , }/ N% R1 W9 @* i3 d' c
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    
48.                     if(($newClient = socket_accept($_sock))  === false){
    ' o" `* W9 U; C7 k
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    
50.                     }
    
51.                     $line = trim(socket_read($newClient, 1024));
    
52.                     $this->handshaking($newClient, $line);
    5 b! z) q% R$ J/ C: B1 H
53.                     //获取client ip
    
54.                     socket_getpeername ($newClient, $ip);
    % n' Q* K+ @1 n7 c$ X) ~
55.                     $clients[$ip] = $newClient;
    
56.                     echo  "Client ip:{$ip}   \n";
    * O6 e% t1 O0 W( C
57.                     echo "Client msg:{$line} \n";
    5 F2 e- K( b9 i% r- X
58.                 } else {
    
59.                     socket_recv($_sock, $buffer,  2048, 0);
    
60.                     $msg = $this->message($buffer);
    
61.                     //在这里业务代码
    & {) P% k! M3 b: |( p/ H! }3 e
62.                     echo "{$key} clinet msg:",$msg,"\n";
    $ e/ M0 S( t$ {0 E" C
63.                     fwrite(STDOUT, 'Please input a argument:');
    2 r. \. t# j0 e$ R0 {* p% d# S9 F
64.                     $response = trim(fgets(STDIN));
    
65.                     $this->send($_sock, $response);
    
66.                     echo "{$key} response to Client:".$response,"\n";
    
67.                 }
    * N* R7 N; N5 J' A% @
68.             }
    1 |. @, h6 e7 G# @* G
69.         }
    # A% U& _( _1 V
70.     }
    - V4 n2 J; X( p
71. 
    
72.     /**
    
73.      * 握手处理
    
74.      * @param $newClient socket
    
75.      * @return int  接收到的信息
    8 y) F& ~3 |) i! z
76.      */
    ' P3 w/ m$ D% \; A" }
77.     public function handshaking($newClient, $line){
    
78. 
    1 E; w" [7 o8 d2 C  K& R
79.         $headers = array();
    
80.         $lines = preg_split("/\r\n/", $line);
    
81.         foreach($lines as $line)
    / q1 B) N4 u- R0 g* }2 m
82.         {
    
83.             $line = chop($line);
    
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    
85.             {
    
86.                 $headers[$matches[1]] = $matches[2];
    
87.             }
    ( D) c! Y; `3 {) L3 m% K! {
88.         }
    
89.         $secKey = $headers['Sec-WebSocket-Key'];
    
90.         $secAccept = base64_encode(pack('H*', sha1($secKey . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11')));
    * Q9 D2 f! G. }+ ~
91.         $upgrade  = "HTTP/1.1 101 Web Socket Protocol Handshake\r\n" .
    9 _1 G+ Q4 s: t( p6 F3 n$ }5 y
92.             "Upgrade: websocket\r\n" .
    
93.             "Connection: Upgrade\r\n" .
    3 D5 B: `! r2 ~! }' Y' X; g, b( I
94.             "WebSocket-Origin: $this->address\r\n" .
    # V, `5 a0 Q  R3 X' d' t
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));
    / }5 V$ v# _/ V5 z0 i/ W
98.     }
    
99. 
    
100.     /**
     
101.      * 解析接收数据
     
102.      * @param $buffer
     # V" o% Y& j, ?$ Y- x
103.      * @return null|string
     ( Y: {/ W7 |1 v% H+ a6 L" B
104.      */
     
105.     public function message($buffer){
     6 h7 _6 K- K: A: Q
106.         $len = $masks = $data = $decoded = null;
     ! Y; ]2 X4 i4 U/ I  u1 d
107.         $len = ord($buffer[1]) & 127;
     
108.         if ($len === 126)  {
     6 P! `. x+ z1 w4 J% z# g" w  c
109.             $masks = substr($buffer, 4, 4);
     
110.             $data = substr($buffer, 8);
     
111.         } else if ($len === 127)  {
     % a7 I# ?) B( J) p
112.             $masks = substr($buffer, 10, 4);
     
113.             $data = substr($buffer, 14);
     ' J5 z2 V3 `: o' ?8 g
114.         } else  {
     
115.             $masks = substr($buffer, 2, 4);
     ; S: s5 R  b/ v7 R& J$ H
116.             $data = substr($buffer, 6);
     ! V2 m0 R- q" {: O0 ^7 I
117.         }
     
118.         for ($index = 0; $index < strlen($data); $index++) {
     . Y) }  t& E+ S7 x3 Q
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     3 L+ x+ u& i% h4 T, m2 e
120.         }
     
121.         return $decoded;
     
122.     }
     : v, D' W& k! ?) Q8 g
123. 
     ( w* Y5 G2 @! ~1 ~, Y
124.     /**
     
125.      * 发送数据
     
126.      * @param $newClinet 新接入的socket
     
127.      * @param $msg   要发送的数据
     2 ~% [- U0 ^$ q
128.      * @return int|string
     ! ]5 m1 _& Z7 @7 m6 G4 I& k* v
129.      */
     7 o( M& u3 p7 T. D8 z" b
130.     public function send($newClinet, $msg){
     
131.         $msg = $this->frame($msg);
     
132.         socket_write($newClinet, $msg, strlen($msg));
     : V3 n* A, M& ~8 c
133.     }
     & h) }* J2 t! r! g
134. 
     % X& _8 Z; {! f+ M: C
135.     public function frame($s) {
     + W; C& K" @) C: I+ n! z; J, a
136.         $a = str_split($s, 125);
     ) ]4 j" _- ]# i6 {; {: x, y7 s
137.         if (count($a) == 1) {
     1 c& y6 n" u6 i# {3 \5 E- B
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     
139.         }
     
140.         $ns = "";
     - K' R" K) e  J! c( ^
141.         foreach ($a as $o) {
     
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     
143.         }
     
144.         return $ns;
     
145.     }
     
146. 
     
147.     /**
     1 J$ x$ n: ]$ D  j* @
148.      * 关闭socket
     + {  Z% \. ?6 q$ }) o$ O8 K1 H: n
149.      */
       v! Z& U! t: @& M3 o- q
150.     public function close(){
     
151.         return socket_close($this->_sockets);
     & [; `6 l; k+ Z5 U, ]% b+ O. L7 z$ }4 U
152.     }
     
153. }
     
154. 
     7 G" N; ~3 w; a8 `% h! E
155. $sock = new SocketService();
     ) `) L8 V. ]/ i  c
156. $sock->run();
     3 z) f- B- M! r3 e8 l
157. 
     * N3 Z; F2 y. C# H

复制代码

web.html

1. <!doctype html>
   
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>
   
7. </head>
   9 \' J1 O) I$ y7 [) ]$ D
8. <body>
   $ S6 l. {$ q6 e
9. <input id="text" value="">
   
10. <input type="submit" value="send" onclick="start()">
    8 V8 D  J, @5 r  e6 @; J
11. <input type="submit" value="close" onclick="close()">
    
12. <div id="msg"></div>
    
13. <script>
    2 M; [5 h) Z  j
14. /**
    $ g5 w3 z& B0 j: R$ E! i
15. 0：未连接
    , Z: f2 @8 A9 }4 |9 H
16. 1：连接成功，可通讯
    ; `; c4 t' E! u% U" }
17. 2：正在关闭
    # C6 }- i% _- M1 j/ `
18. 3：连接已关闭或无法打开
    
19. */
    
20. 
    
21.     //创建一个webSocket 实例
    ' m( `3 ]0 G( ]6 E1 v
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    
23. 
    $ W' l6 P: K  f& A
24. 
    
25.     webSocket.onerror = function (event){
    2 `# _9 q: ?4 n  n$ w
26.         onError(event);
    
27.     };
    + N5 V" O: W5 \+ v$ H6 B
28. 
    
29.     // 打开websocket
    1 N: t* H. Z+ E1 i5 J- L0 m7 n
30.     webSocket.onopen = function (event){
    0 d4 G: g. ~8 o  F( O: d2 T  _
31.         onOpen(event);
    + P8 D1 i" o# G5 g& m' D
32.     };
    
33. 
    
34.     //监听消息
    2 I- a7 a. }7 ^" V7 U
35.     webSocket.onmessage = function (event){
    + [/ }  M: t% V  t& s
36.         onMessage(event);
    ' n/ ?, @& Y0 J" u' |
37.     };
    
38. 
    " [8 R8 s, `: S- g" x4 u4 V3 o
39. 
    $ @) K, G4 |: H) Q+ o' U: R7 w
40.     webSocket.onclose = function (event){
    & a8 O6 n% |/ g) F: J% G
41.         onClose(event);
    ; y+ i' x" C& y9 \- X+ ~1 U: ~" W6 E
42.     }
    
43. 
    
44.     //关闭监听websocket
    
45.     function onError(event){
    7 t: v# }$ ~# i7 v0 O
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
47.         console.log("error"+event.data);
    
48.     };
    ) u& a  s; s3 q8 E7 C
49. 
    
50.     function onOpen(event){
    2 }$ k  J2 I* E! }/ t0 [, Q& }; x- T
51.         console.log("open:"+sockState());
    1 X+ L& d* h8 D- C
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    9 ~# L. A4 E$ h" \. U3 b
53.     };
    
54.     function onMessage(event){
    ! y+ @1 F4 x1 s! l
55.         console.log("onMessage");
    
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    2 i# o8 a2 P7 v
57.     };
    
58. 
    7 o. x* G  e: y! _# N0 V
59.     function onClose(event){
    
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    # p9 g( U- u0 z
61.         console.log("close:"+sockState());
    0 d$ I0 x7 l" g/ p2 _
62.         webSocket.close();
    
63.     }
    . j. Y! T% }8 D
64. 
    
65.     function sockState(){
    " z7 Y* A. S3 a- |
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    # w7 H! b3 @; {: y% l
67.             return status[webSocket.readyState];
    ' \, b2 j% ~1 F! p# }: Z
68.     }
    2 j0 @1 n" H9 m! L
69. 
    
70. 
    
71. 
    
72. function start(event){
    
73.         console.log(webSocket);
    
74.         var msg = document.getElementById('text').value;
    
75.         document.getElementById('text').value = '';
    , |8 D5 X9 C# q
76.         console.log("send:"+sockState());
    
77.         console.log("msg="+msg);
    
78.         webSocket.send("msg="+msg);
    : O' u1 e! g3 w( C- o( l3 Q4 x* D
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    ( T; F5 r3 h2 T, d
80.     };
    
81. 
    * W8 h$ ^) Z3 [9 r6 H  o" T
82.     function close(event){
    
83.         webSocket.close();
    
84.     }
    . |* m! [* b0 ~/ ]4 C& u
85. </script>
      Z- t( q/ y- M5 W5 |, N
86. </body>
    
87. </html>

复制代码

5 F6 P. p  l# a+ Y7 X
- Y8 n# J6 N$ ?& B" h