php实现websocket实时消息推送

admin

php实现websocket实时消息推送
, w7 I5 K+ l! z4 s" R5 P
. `% J. S! C, o, E
/ W6 V6 E. y; i! @5 V8 H9 USocketService.php
) K( r7 [' Z0 ]

1. <?php
   
2. /**
   * n. p. V8 G" B. V; T
3. * Created by xwx
   
4. * Date: 2017/10/18
   
5. * Time: 14:33
   
6. */
   6 g+ x8 z& _. W' H( ^3 }. L
7. 
   % M- ^9 V9 U7 L/ c8 E$ m+ y# W
8. class SocketService
   
9. {
   6 z  p' N) w# T6 l! r) K
10.     private $address  = '0.0.0.0';
    
11.     private $port = 8083;
      ?9 R  ?$ B' t7 W% G
12.     private $_sockets;
    ; k4 ^8 g  ~7 G4 Y. a  u. q' S
13.     public function __construct($address = '', $port='')
    7 y# H9 ]* G8 k% v
14.     {
    & }4 c; c7 @, a8 V6 e% \$ Z
15.             if(!empty($address)){
    
16.                 $this->address = $address;
    
17.             }
    + P+ i8 I- J( ?* q% p# \/ o
18.             if(!empty($port)) {
    
19.                 $this->port = $port;
    
20.             }
    
21.     }
    
22. 
    % m0 k7 i6 r. B) L7 }! H) ^& W2 D9 Q
23.     public function service(){
    
24.         //获取tcp协议号码。
    . ?5 O% Z" ?6 E) d. w2 n, @; b2 u2 ?7 F
25.         $tcp = getprotobyname("tcp");
    
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    5 g0 N$ x' c8 {
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    
28.         if($sock < 0)
    2 \% |3 l7 m. I1 i& [
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);
    5 E  q/ Q8 ~; i
34.         echo "listen on $this->address $this->port ... \n";
    6 ~& P0 `; h# d* x
35.         $this->_sockets = $sock;
    
36.     }
    8 }( n: K1 B; T% _1 O* X- R
37. 
    4 L! m( W( R& \% ?$ r! u
38.     public function run(){
    9 @( X. ]6 ]! `, C5 y% ^* p+ ~
39.         $this->service();
    ' X& Z6 F7 P$ t. r. W- |
40.         $clients[] = $this->_sockets;
    * b; Z: l+ S/ T
41.         while (true){
    7 v6 r' [( T* m% u* z2 e
42.             $changes = $clients;
    7 @; {3 y. B2 s6 ]3 U; o5 q, |. b
43.             $write = NULL;
    6 V( g1 ^! c- n8 l8 u: f/ c- _& R4 ~
44.             $except = NULL;
    
45.             socket_select($changes,  $write,  $except, NULL);
    & K: T( V: V( }
46.             foreach ($changes as $key => $_sock){
    - Z/ V. u: R. M! I
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
      ]7 Q; T! Q2 k& ~/ v
48.                     if(($newClient = socket_accept($_sock))  === false){
    ! @: Z9 Q+ @; U; |% S
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    % e- T+ d8 ]1 Z; J& ~+ K& B
50.                     }
    , a5 j+ \* p5 v0 A  T8 ?( d" K
51.                     $line = trim(socket_read($newClient, 1024));
    
52.                     $this->handshaking($newClient, $line);
    . Q! L4 }; I1 S1 P' n* A
53.                     //获取client ip
    
54.                     socket_getpeername ($newClient, $ip);
    6 X# }+ d) O3 |+ v( k
55.                     $clients[$ip] = $newClient;
    9 f) R- }# E: f' ]
56.                     echo  "Client ip:{$ip}   \n";
    
57.                     echo "Client msg:{$line} \n";
    
58.                 } else {
    0 Y. z- Z/ Z& }8 B
59.                     socket_recv($_sock, $buffer,  2048, 0);
    ; R1 W6 `7 N. v( o3 M8 A4 E
60.                     $msg = $this->message($buffer);
    
61.                     //在这里业务代码
    % K' E* \# F! A/ m# K7 c
62.                     echo "{$key} clinet msg:",$msg,"\n";
    
63.                     fwrite(STDOUT, 'Please input a argument:');
    
64.                     $response = trim(fgets(STDIN));
    & a6 H7 Q; [& W8 \
65.                     $this->send($_sock, $response);
    $ |. b) i- T6 S) ]* u# F4 h$ i
66.                     echo "{$key} response to Client:".$response,"\n";
    
67.                 }
    
68.             }
    $ F3 ], U( L; ?$ }! e2 w- E9 s
69.         }
    6 n7 F) [, L- }4 a+ d8 c9 Y
70.     }
    0 O% J2 G0 T) M( X' i! u
71. 
    
72.     /**
    
73.      * 握手处理
    
74.      * @param $newClient socket
    6 i* Q% ^+ \& w
75.      * @return int  接收到的信息
    
76.      */
    
77.     public function handshaking($newClient, $line){
    0 [# B! D2 w4 s2 F
78. 
    
79.         $headers = array();
    # |! M. B3 q0 V# j
80.         $lines = preg_split("/\r\n/", $line);
    
81.         foreach($lines as $line)
    0 m7 M' Q6 x% d) l' u# b
82.         {
    
83.             $line = chop($line);
    7 M$ `& x9 X3 B' a
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    " t% l# }  v$ `
85.             {
    / }1 K# u( j9 x. U% m7 \
86.                 $headers[$matches[1]] = $matches[2];
    + D0 v/ l$ s9 t3 F* x9 ^
87.             }
    
88.         }
    " n8 m: ], w# K: b' n( h. [
89.         $secKey = $headers['Sec-WebSocket-Key'];
    9 T1 I' Z7 Z" Q3 P5 ^; c' y. w
90.         $secAccept = base64_encode(pack('H*', sha1($secKey . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11')));
    2 U" I- L, V$ Q3 g5 E' E
91.         $upgrade  = "HTTP/1.1 101 Web Socket Protocol Handshake\r\n" .
    
92.             "Upgrade: websocket\r\n" .
    
93.             "Connection: Upgrade\r\n" .
    1 G2 L5 c  B3 l! ]5 ?
94.             "WebSocket-Origin: $this->address\r\n" .
    " F7 w. e1 M, Q2 G
95.             "WebSocket-Location: ws://$this->address:$this->port/websocket/websocket\r\n".
    5 z5 t( {2 b! x" ]) K9 R: z
96.             "Sec-WebSocket-Accept:$secAccept\r\n\r\n";
    6 o, H9 \( z& H. `" U
97.         return socket_write($newClient, $upgrade, strlen($upgrade));
    * B/ @) _- @2 Q/ A) P2 T# q: F- L3 L
98.     }
    , f, X3 V" j: u1 s% n
99. 
    
100.     /**
     
101.      * 解析接收数据
     4 d+ }$ g: m- k# F9 a
102.      * @param $buffer
     : ]6 D$ \2 k  {, r1 B' i( e" t
103.      * @return null|string
     - i0 O7 y6 y$ l) e' ]0 K5 \( N
104.      */
     2 Q6 b, Q3 Z  j! i4 m) A4 X' Y
105.     public function message($buffer){
     
106.         $len = $masks = $data = $decoded = null;
     
107.         $len = ord($buffer[1]) & 127;
     
108.         if ($len === 126)  {
     + K3 Q# {1 z2 \: }- f
109.             $masks = substr($buffer, 4, 4);
     9 @: T: ]% v% a7 j; t
110.             $data = substr($buffer, 8);
     
111.         } else if ($len === 127)  {
     0 l, H, n" t& ?8 q
112.             $masks = substr($buffer, 10, 4);
     ) Q0 v9 e% \" |
113.             $data = substr($buffer, 14);
     
114.         } else  {
     
115.             $masks = substr($buffer, 2, 4);
     
116.             $data = substr($buffer, 6);
     % O- X3 e5 Z# X3 @$ I" p
117.         }
     
118.         for ($index = 0; $index < strlen($data); $index++) {
     1 r0 I7 Y! Y* A( R  p1 _2 G& s# B8 h
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     ) o( ~/ o  R  u8 n& d5 K: P2 g2 ^
120.         }
     
121.         return $decoded;
     
122.     }
     ) ~- \9 u% c$ @# n+ @, }
123. 
     
124.     /**
     
125.      * 发送数据
     
126.      * @param $newClinet 新接入的socket
     : V0 A5 _/ B4 \9 X
127.      * @param $msg   要发送的数据
     
128.      * @return int|string
     
129.      */
     / J1 z1 m1 j: P0 F) Z; F& A
130.     public function send($newClinet, $msg){
     3 F" [' G4 u+ h% G" s
131.         $msg = $this->frame($msg);
     / y" @6 Z  r& T$ l3 q" g. H
132.         socket_write($newClinet, $msg, strlen($msg));
     
133.     }
     
134. 
     " J+ f% z$ m- ~: z5 q
135.     public function frame($s) {
     : w+ `+ h! ?. p
136.         $a = str_split($s, 125);
     
137.         if (count($a) == 1) {
     
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
       d  B7 g/ l3 c
139.         }
     
140.         $ns = "";
     
141.         foreach ($a as $o) {
     
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     , f$ O* e. O# T# R4 |
143.         }
     
144.         return $ns;
     
145.     }
     
146. 
     ; {/ _( s3 a( K3 X* |+ ~0 t
147.     /**
     
148.      * 关闭socket
     % V1 R0 @$ C7 V0 Z) i2 ?
149.      */
     + ^" y' W0 T' R* S: P, ^) y7 O
150.     public function close(){
     
151.         return socket_close($this->_sockets);
     
152.     }
     1 t/ V% c: n- ~  x; W$ q1 m
153. }
     
154. 
     ! l! I5 x% @& ]6 V( `. ?( T/ V1 P
155. $sock = new SocketService();
     * }' M+ f6 P' W
156. $sock->run();
     # O) a, b. J: ?$ |' u* Z3 d' A3 t
157. 
     - l0 ^9 X" }/ U% B

复制代码

web.html
" _5 `' l; C' j6 \) p

1. <!doctype html>
   
2. <html lang="en">
   ) |/ u! b( ?8 c! W5 V, [8 W
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>
   0 \* r9 d# p! o5 C% N
8. <body>
   
9. <input id="text" value="">
   
10. <input type="submit" value="send" onclick="start()">
    
11. <input type="submit" value="close" onclick="close()">
    & D: a5 m* ^- s9 T8 V
12. <div id="msg"></div>
    
13. <script>
    . V# |# f% S0 w  }, ?
14. /**
    6 i1 p! a3 n$ z+ r
15. 0：未连接
    
16. 1：连接成功，可通讯
    
17. 2：正在关闭
    
18. 3：连接已关闭或无法打开
    0 g% u8 d% A+ j: e9 ]8 D- U
19. */
    
20. 
    
21.     //创建一个webSocket 实例
    ! B3 I  D: Y& ~
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    
23. 
    
24. 
    
25.     webSocket.onerror = function (event){
    
26.         onError(event);
    
27.     };
    ) c0 D: o* s" g* V5 `( \4 f: u
28. 
    7 Q) _$ u1 _, p% z. |) }; j
29.     // 打开websocket
    " V- U" C( ]) l9 u
30.     webSocket.onopen = function (event){
    
31.         onOpen(event);
    
32.     };
    6 t7 Y/ p& U( Q1 G( R
33. 
    
34.     //监听消息
    
35.     webSocket.onmessage = function (event){
    
36.         onMessage(event);
    # l: r7 K. j. V, d  T7 b# R
37.     };
    
38. 
    
39. 
    
40.     webSocket.onclose = function (event){
    , s' X/ n" m1 v  `* ~# k6 M
41.         onClose(event);
    
42.     }
    1 b1 h0 d, q5 `3 W. b
43. 
    
44.     //关闭监听websocket
    
45.     function onError(event){
    
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    + T/ i1 w  T1 u6 e! n
47.         console.log("error"+event.data);
    
48.     };
    
49. 
    
50.     function onOpen(event){
    
51.         console.log("open:"+sockState());
    
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    
53.     };
    : o* v! B7 t. L; V  ]6 F0 \/ c
54.     function onMessage(event){
    8 g3 x$ a- J% l2 J
55.         console.log("onMessage");
    . ?! x% @" p) Q8 ^2 a/ m; V3 a% |. @
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    ' C& M1 ^! X! a5 o
57.     };
    1 H$ t9 x$ X, e: b+ i5 H5 s
58. 
    7 {1 u2 r  g7 k  z6 @% w
59.     function onClose(event){
    
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    / b% w6 r5 O3 b# Z! V# J6 I( h, a
61.         console.log("close:"+sockState());
    9 j3 B, g. z2 r7 M$ F
62.         webSocket.close();
    
63.     }
    * l) \6 ^7 W1 p! ], s2 j* o( B7 Z
64. 
    * S+ I8 z7 C1 L; }
65.     function sockState(){
    " {+ m% F9 g2 Q, s- Z/ g6 K
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    ' _2 e# \; B1 Z
67.             return status[webSocket.readyState];
    
68.     }
    3 D, Y; a8 p+ }1 B: G
69. 
    ) B6 A% m* D8 s7 y) m$ P+ |
70. 
    
71. 
    , d# }% N  ~8 F
72. function start(event){
    
73.         console.log(webSocket);
    2 y+ c4 W3 B5 z: e8 t6 V
74.         var msg = document.getElementById('text').value;
    , _0 I* w3 M2 \) P
75.         document.getElementById('text').value = '';
    $ J: t( c" d7 W- P0 O
76.         console.log("send:"+sockState());
    5 T4 i/ w$ ^2 y$ l% l& _! ^- }4 }
77.         console.log("msg="+msg);
    
78.         webSocket.send("msg="+msg);
      f- F% D$ `$ F. b
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    7 o# Y2 o$ ?7 u& ^# `* s* s
80.     };
    3 Z. A! l1 ~  m
81. 
    + p; o( {8 s- c$ p
82.     function close(event){
    ) d$ J" y* ]- m) u
83.         webSocket.close();
    
84.     }
    0 T; m5 D# F# k3 y
85. </script>
    
86. </body>
    
87. </html>

复制代码

$ U' C7 {9 }' I' l8 g1 t4 M