php实现websocket实时消息推送

admin

php实现websocket实时消息推送
& j2 t# G" M; N- Y: ]/ g6 I

7 D; Z+ W& T: i* n6 b& BSocketService.php

1. <?php
   6 c0 E' W& P& w# I; R; O
2. /**
   
3. * Created by xwx
   
4. * Date: 2017/10/18
   ( y& z2 M3 g  G& q" t+ p
5. * Time: 14:33
   4 x8 r  r% G! _' n
6. */
   5 g7 y8 y" W0 O+ q3 T' x" a
7. 
   
8. class SocketService
   
9. {
   
10.     private $address  = '0.0.0.0';
    5 ]# [: S4 W" U/ b7 w7 x/ k
11.     private $port = 8083;
    ! x& m5 _, O7 y9 i, i" l3 \. x" H3 x9 M
12.     private $_sockets;
    
13.     public function __construct($address = '', $port='')
    , x% @$ y3 O5 p9 G; D
14.     {
    0 V9 a7 R0 @/ y# c$ Q! O2 ?& K
15.             if(!empty($address)){
    
16.                 $this->address = $address;
    
17.             }
    4 V; P5 h' @  Y: e& m  u8 u
18.             if(!empty($port)) {
    
19.                 $this->port = $port;
    / h/ D3 }1 ~- N& K* f
20.             }
    
21.     }
    
22. 
    ( z- I8 t0 l# w8 W3 E
23.     public function service(){
    
24.         //获取tcp协议号码。
    8 U3 ~( }+ [+ w; {9 g- k, b9 {; B
25.         $tcp = getprotobyname("tcp");
    
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    / g* h- {& B+ S' f* |% Q
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    9 |% I0 v$ ^& q( x# I# V2 B4 e" |3 z9 Z4 M
28.         if($sock < 0)
    
29.         {
    
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    
31.         }
    ; E# s, t( |1 b6 t' F7 H) E
32.         socket_bind($sock, $this->address, $this->port);
    
33.         socket_listen($sock, $this->port);
    5 {& a5 k" d( s5 U$ o- n' f9 r
34.         echo "listen on $this->address $this->port ... \n";
    
35.         $this->_sockets = $sock;
    
36.     }
    
37. 
    ; t7 r- c4 K5 J6 ]
38.     public function run(){
    
39.         $this->service();
    6 b  o+ c# ~6 O' A
40.         $clients[] = $this->_sockets;
    ! H+ l% v: l2 Y6 p
41.         while (true){
    
42.             $changes = $clients;
    
43.             $write = NULL;
    
44.             $except = NULL;
    7 p. E" Q$ j! Z- J
45.             socket_select($changes,  $write,  $except, NULL);
    # e7 ]8 y" J6 t4 ^' Y- @
46.             foreach ($changes as $key => $_sock){
    : b4 @' o+ G: d0 N0 }4 D5 W
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    3 L2 S0 A9 v: s7 e; e' L
48.                     if(($newClient = socket_accept($_sock))  === false){
    ) H, q2 }3 _$ M; p, I' I
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    
50.                     }
    2 v, c0 d  J( e" `3 e- G% i
51.                     $line = trim(socket_read($newClient, 1024));
    
52.                     $this->handshaking($newClient, $line);
    
53.                     //获取client ip
    
54.                     socket_getpeername ($newClient, $ip);
    
55.                     $clients[$ip] = $newClient;
    
56.                     echo  "Client ip:{$ip}   \n";
    
57.                     echo "Client msg:{$line} \n";
    
58.                 } else {
    
59.                     socket_recv($_sock, $buffer,  2048, 0);
    0 P8 C' D+ _" n6 A$ h2 d6 H! ?/ `5 H
60.                     $msg = $this->message($buffer);
    
61.                     //在这里业务代码
    6 t4 _/ N9 u' P' S% s) ^, H& w
62.                     echo "{$key} clinet msg:",$msg,"\n";
    2 `8 o* B. _; ^# j! K
63.                     fwrite(STDOUT, 'Please input a argument:');
    - m0 g% r+ `' P+ s5 H
64.                     $response = trim(fgets(STDIN));
    
65.                     $this->send($_sock, $response);
    
66.                     echo "{$key} response to Client:".$response,"\n";
    - R* h  T& ]1 B# I
67.                 }
    
68.             }
    
69.         }
    
70.     }
    8 O( t. p+ v' ?9 E$ p
71. 
    
72.     /**
    
73.      * 握手处理
    ! |1 E# b$ k% d8 }
74.      * @param $newClient socket
    
75.      * @return int  接收到的信息
    4 B( z% @+ r( @6 ^7 {8 N, z( M6 L
76.      */
    . H9 v' v& p: B# _/ _9 u
77.     public function handshaking($newClient, $line){
    
78. 
    
79.         $headers = array();
    
80.         $lines = preg_split("/\r\n/", $line);
    
81.         foreach($lines as $line)
    
82.         {
    $ n- H5 o. v1 y) E% q
83.             $line = chop($line);
    
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    
85.             {
    
86.                 $headers[$matches[1]] = $matches[2];
    9 B# J, h3 U, I5 D
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" .
    6 p0 s& I4 R9 q" |3 j5 v
92.             "Upgrade: websocket\r\n" .
    
93.             "Connection: Upgrade\r\n" .
    
94.             "WebSocket-Origin: $this->address\r\n" .
    * y4 t# P; |% S" M
95.             "WebSocket-Location: ws://$this->address:$this->port/websocket/websocket\r\n".
    1 U3 H  E: x& @% m
96.             "Sec-WebSocket-Accept:$secAccept\r\n\r\n";
    4 |* x. c4 \8 d+ Q, q( m
97.         return socket_write($newClient, $upgrade, strlen($upgrade));
    . e5 B! K. h# k) P
98.     }
    
99. 
    
100.     /**
     # |- O7 Z8 R- z# F- }8 w
101.      * 解析接收数据
     ! q7 d1 P2 b3 I# \9 ?3 ?: u
102.      * @param $buffer
     ) N/ g/ U  \0 c4 H; q# a
103.      * @return null|string
     8 w) y* q; G; B( {6 M$ U
104.      */
     
105.     public function message($buffer){
     - a4 a8 y# L) M2 a. V. }' e$ F
106.         $len = $masks = $data = $decoded = null;
     
107.         $len = ord($buffer[1]) & 127;
     
108.         if ($len === 126)  {
     
109.             $masks = substr($buffer, 4, 4);
     ; z/ M5 Q8 A! {$ i/ t" E# r9 _
110.             $data = substr($buffer, 8);
     " H& G1 g4 i0 }& D- {: l
111.         } else if ($len === 127)  {
     
112.             $masks = substr($buffer, 10, 4);
     ' D- u$ k) U& L3 @1 Q4 D  [
113.             $data = substr($buffer, 14);
     
114.         } else  {
     ' J8 Z! t# B9 t: t' a0 m3 v9 m! M
115.             $masks = substr($buffer, 2, 4);
     3 U& r! ?3 g" c, l1 r) ]
116.             $data = substr($buffer, 6);
     8 _, W  d: \# Z4 {
117.         }
     
118.         for ($index = 0; $index < strlen($data); $index++) {
     
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     2 ~' J/ _# D, |: s; f: e
120.         }
     : l6 ~" M7 ]. C5 M3 S+ j" n2 `
121.         return $decoded;
       F" d6 S; |. }; ^# @
122.     }
     
123. 
     
124.     /**
     
125.      * 发送数据
     % e4 k8 X+ K) e% w+ h
126.      * @param $newClinet 新接入的socket
     / ~3 n' L* B6 P: n& p& s
127.      * @param $msg   要发送的数据
     2 o% U! b  O& E# Y: o2 C' j
128.      * @return int|string
     
129.      */
     
130.     public function send($newClinet, $msg){
     
131.         $msg = $this->frame($msg);
     9 \% E: i+ p, u* \& o2 j
132.         socket_write($newClinet, $msg, strlen($msg));
     3 h- O3 Y7 X( t' U( X
133.     }
     - U. M3 j8 m  Q5 q
134. 
     
135.     public function frame($s) {
     
136.         $a = str_split($s, 125);
     0 F; Q4 {! s% n- T
137.         if (count($a) == 1) {
     ) s. v- S! R4 e. B. a* f, B. g- S( H
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     3 B7 m7 @$ e# [3 }) z
139.         }
     
140.         $ns = "";
     
141.         foreach ($a as $o) {
     + A, u0 _2 {& f* H8 ]7 k- P. R
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     
143.         }
     ) c' Z6 Y' t( P0 y' H# f
144.         return $ns;
     
145.     }
     # n( Z5 b- w- \5 P8 v& z: h
146. 
     " s3 M  J; \! M/ o
147.     /**
     
148.      * 关闭socket
     
149.      */
     
150.     public function close(){
     9 X1 J3 f" f* Q- `. _
151.         return socket_close($this->_sockets);
     ( m- c6 f( s* `0 u+ q* A
152.     }
     & V3 g3 X4 \# T4 y) q/ G$ I
153. }
     
154. 
     
155. $sock = new SocketService();
     % F; e3 w! C7 T6 b5 S& M* V
156. $sock->run();
     - k& ?% W$ T; o) v( h) g# y, l
157. 
     1 D' V5 [( P2 i4 g

复制代码

web.html
0 i& [) [6 x7 J3 m6 k3 A

1. <!doctype html>
   
2. <html lang="en">
   
3. <head>
   , @3 G' h+ R* \
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>
   ' Y+ \/ J, ]# C# V
7. </head>
   
8. <body>
   
9. <input id="text" value="">
   7 b+ T3 G$ V$ W% G
10. <input type="submit" value="send" onclick="start()">
    
11. <input type="submit" value="close" onclick="close()">
    
12. <div id="msg"></div>
    9 l, A3 l% N! g/ {1 H. O1 w/ g
13. <script>
    
14. /**
    ' \3 L2 D: |0 U4 u1 l$ k, O
15. 0：未连接
    ! ^" m4 `+ y) d- y% U
16. 1：连接成功，可通讯
    
17. 2：正在关闭
    
18. 3：连接已关闭或无法打开
    
19. */
    
20. 
      M  T8 v" `0 [( T1 o3 r: w
21.     //创建一个webSocket 实例
    , i3 Y1 V3 N( N% ^3 c* H
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    
23. 
    % T( N3 A: ]6 D. c4 S
24. 
    
25.     webSocket.onerror = function (event){
    ) G3 n$ N3 s9 N2 j
26.         onError(event);
    ! v, M9 P0 ^1 U1 W, h& y
27.     };
    
28. 
    ) U' V7 E* x. R) L# \
29.     // 打开websocket
    
30.     webSocket.onopen = function (event){
    / M# j2 N7 n& O5 W) Q
31.         onOpen(event);
    
32.     };
    8 X3 [) A8 e3 i1 a& j( l
33. 
    2 r/ n$ r* T! f6 W$ D
34.     //监听消息
    
35.     webSocket.onmessage = function (event){
    9 @$ `3 y. ~- Z
36.         onMessage(event);
    
37.     };
    . F1 S( b/ B6 f6 k% r0 B7 ]+ @
38. 
    
39. 
    ' `: g- y& [$ o
40.     webSocket.onclose = function (event){
    
41.         onClose(event);
    
42.     }
    , ]6 n7 W* X5 R7 d7 I; M
43. 
    
44.     //关闭监听websocket
      a1 t0 Z. j& K2 g
45.     function onError(event){
    
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
47.         console.log("error"+event.data);
    " a; a) V' M, Q
48.     };
    . R  I6 i. p& p6 n
49. 
    
50.     function onOpen(event){
    
51.         console.log("open:"+sockState());
    
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    
53.     };
    7 D: b& G4 P+ I  V8 Z
54.     function onMessage(event){
    
55.         console.log("onMessage");
    
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    & T& q0 s/ ^$ d
57.     };
    # Z6 q* _3 u* ~. U$ g1 j9 j9 v' n) _
58. 
    / n0 G( T: a+ L: E) d1 G
59.     function onClose(event){
    9 E2 \& t1 _- n8 j- b
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
61.         console.log("close:"+sockState());
    
62.         webSocket.close();
    
63.     }
    ( x" q) p6 s" ?2 {# R
64. 
    
65.     function sockState(){
    
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    
67.             return status[webSocket.readyState];
    
68.     }
    $ Z$ z4 j. M( W; b6 k; G8 Q2 k- }
69. 
    $ \- r( U+ T& J4 o
70. 
    ! @3 S; K. u# z1 X% ~- \
71. 
    
72. function start(event){
    + U' @( m5 g* c" o; h3 r
73.         console.log(webSocket);
    / v) M0 M, n+ F  g8 v# ^/ q/ N
74.         var msg = document.getElementById('text').value;
    
75.         document.getElementById('text').value = '';
    
76.         console.log("send:"+sockState());
    , R. b7 f3 m( q1 r  h0 u- c5 T: h
77.         console.log("msg="+msg);
    
78.         webSocket.send("msg="+msg);
    ' T3 E0 c# g$ x2 F$ {- q
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    
80.     };
    
81. 
    6 z  c7 S; c+ R) V% _& |
82.     function close(event){
    
83.         webSocket.close();
    
84.     }
    6 b5 C7 Z5 |$ W  l6 m; H
85. </script>
    
86. </body>
    0 Z- U" z  x. s! m% ]( Q
87. </html>

复制代码

' x3 E# B& j6 ~# I