php实现websocket实时消息推送

admin

php实现websocket实时消息推送

! u" w; c& K0 b" i( D+ D$ d
SocketService.php

1. <?php
   ' j* y9 Y+ Q, Y4 k. L
2. /**
   
3. * Created by xwx
   * z" V5 ]2 |& M
4. * Date: 2017/10/18
   
5. * Time: 14:33
   
6. */
   0 m3 c) j9 ^0 i7 z- y  s
7. 
   # f6 z* z7 S. h2 O2 }
8. class SocketService
   1 X8 S# J/ n( f+ Y  e
9. {
   
10.     private $address  = '0.0.0.0';
    $ ]5 U6 ~& }9 |# Y  K
11.     private $port = 8083;
    
12.     private $_sockets;
    7 j7 D* ?3 L+ U1 i: c5 j1 `
13.     public function __construct($address = '', $port='')
    ' g4 n: F8 ]6 V8 [4 w
14.     {
    ) `7 Q! v- l3 R6 r7 F# w
15.             if(!empty($address)){
    
16.                 $this->address = $address;
    
17.             }
    
18.             if(!empty($port)) {
    
19.                 $this->port = $port;
    ! ~+ ~3 E7 c$ T% L( {
20.             }
    7 F7 w6 u. f8 ^5 v4 ~- K
21.     }
    
22. 
    4 b' ~! b% f" t8 A) Q
23.     public function service(){
    
24.         //获取tcp协议号码。
    
25.         $tcp = getprotobyname("tcp");
    " ?8 y. }: d) @* [- v. J) x9 K" N
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    1 b- i2 b1 a% J! R
28.         if($sock < 0)
    0 ]3 l! n& C. |( i2 _* j4 Q
29.         {
    ; c& {1 k/ N  g4 x, e5 q- X$ n2 W/ e
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    * m0 m  x& j' a! t3 F
31.         }
    
32.         socket_bind($sock, $this->address, $this->port);
    
33.         socket_listen($sock, $this->port);
    # }! K% G4 V* x& R1 A
34.         echo "listen on $this->address $this->port ... \n";
    
35.         $this->_sockets = $sock;
    ( \8 f3 {/ M( I2 [# |8 K
36.     }
    & d1 M5 [) `# v* i
37. 
    
38.     public function run(){
    
39.         $this->service();
    0 a! S; I% L- o# R+ v$ @* x
40.         $clients[] = $this->_sockets;
    : Y7 y* k1 e+ r6 x" I  L
41.         while (true){
    
42.             $changes = $clients;
    ; `3 ^! @9 R0 V, p6 J2 P/ \1 V+ Z  L& A
43.             $write = NULL;
    
44.             $except = NULL;
    
45.             socket_select($changes,  $write,  $except, NULL);
    
46.             foreach ($changes as $key => $_sock){
    4 |& A3 G: J9 Y# n4 `
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    
48.                     if(($newClient = socket_accept($_sock))  === false){
    / O$ n( B, f. h0 j" c8 R
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    
50.                     }
    7 y: W  `6 L& S
51.                     $line = trim(socket_read($newClient, 1024));
    
52.                     $this->handshaking($newClient, $line);
    ' |0 X8 W/ L1 x  ?
53.                     //获取client ip
    
54.                     socket_getpeername ($newClient, $ip);
    
55.                     $clients[$ip] = $newClient;
    
56.                     echo  "Client ip:{$ip}   \n";
    6 h# z& |, r" o0 }6 O. S; O6 B
57.                     echo "Client msg:{$line} \n";
    1 T' h6 @1 U! b' o% h7 _  U
58.                 } else {
    - {. q) _) M. x, A  B
59.                     socket_recv($_sock, $buffer,  2048, 0);
    
60.                     $msg = $this->message($buffer);
    
61.                     //在这里业务代码
    , y4 V; h7 _% k, H8 U) ^$ l# Y
62.                     echo "{$key} clinet msg:",$msg,"\n";
    
63.                     fwrite(STDOUT, 'Please input a argument:');
    & q7 R5 G: _$ R
64.                     $response = trim(fgets(STDIN));
    5 A' p, |, ?2 h# M' _0 r
65.                     $this->send($_sock, $response);
    
66.                     echo "{$key} response to Client:".$response,"\n";
    2 A9 F1 g$ X: f+ e
67.                 }
    $ I. \* L, S/ f
68.             }
    
69.         }
    
70.     }
    
71. 
    
72.     /**
    
73.      * 握手处理
    
74.      * @param $newClient socket
    
75.      * @return int  接收到的信息
    
76.      */
    
77.     public function handshaking($newClient, $line){
    
78. 
    
79.         $headers = array();
    0 |8 m7 {5 `2 S- ~" Y! d: s7 X" H# z
80.         $lines = preg_split("/\r\n/", $line);
    0 Y: ^6 ^. T9 z. B8 T) |5 [/ P
81.         foreach($lines as $line)
    % O# t/ D: |* f0 c3 @
82.         {
    / _  o$ ^2 M: p) t/ w' R+ i2 E) A
83.             $line = chop($line);
    
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    , x( F; f( f2 w; V
85.             {
    8 B$ k1 U6 L! A
86.                 $headers[$matches[1]] = $matches[2];
    " ?& ]; Z# I9 |
87.             }
    2 x: ~9 B1 e0 v  N
88.         }
    
89.         $secKey = $headers['Sec-WebSocket-Key'];
    
90.         $secAccept = base64_encode(pack('H*', sha1($secKey . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11')));
    / ^* c5 A# D1 f. Q1 X& b4 m; F9 H
91.         $upgrade  = "HTTP/1.1 101 Web Socket Protocol Handshake\r\n" .
    & Y$ N7 ~% O! G& J+ j8 A3 }
92.             "Upgrade: websocket\r\n" .
    
93.             "Connection: Upgrade\r\n" .
    
94.             "WebSocket-Origin: $this->address\r\n" .
    
95.             "WebSocket-Location: ws://$this->address:$this->port/websocket/websocket\r\n".
    ! M& y; z. f' b% e* J  a. d2 v
96.             "Sec-WebSocket-Accept:$secAccept\r\n\r\n";
    2 ]' U8 x, c1 u& O7 j, G
97.         return socket_write($newClient, $upgrade, strlen($upgrade));
    ) N, O2 _) [2 x- ?
98.     }
    # g* z: o9 Q4 a) w' D0 {) a
99. 
    . r& G) N* Y6 A
100.     /**
     ' ?: b4 O0 l, B# n( z, k
101.      * 解析接收数据
     
102.      * @param $buffer
     + Q6 M; m$ q/ M) h4 n) ~4 Q: ~7 A
103.      * @return null|string
     ! q7 x% @1 V# D
104.      */
     / {, J" U: v; {
105.     public function message($buffer){
     : P$ r2 j- ]8 x0 q7 n0 A& b
106.         $len = $masks = $data = $decoded = null;
     ( ]/ F9 z  }, @9 i
107.         $len = ord($buffer[1]) & 127;
     & o& F0 A/ ~, C( s
108.         if ($len === 126)  {
     
109.             $masks = substr($buffer, 4, 4);
     7 j4 P! J' t* z/ i! A
110.             $data = substr($buffer, 8);
     ( S- a$ u. f3 l6 f. w0 N3 Z" I( z* g
111.         } else if ($len === 127)  {
     . h, z# A( e' r9 o" D9 D# m: W
112.             $masks = substr($buffer, 10, 4);
     
113.             $data = substr($buffer, 14);
     
114.         } else  {
     3 p* D- p+ K5 f1 b
115.             $masks = substr($buffer, 2, 4);
     ) K# F2 x% [. @, z5 Y0 W
116.             $data = substr($buffer, 6);
     * G4 {% Y. x  ~( f) A& L- x6 G' [
117.         }
     & w5 |# n$ c0 V& n+ K! c* j
118.         for ($index = 0; $index < strlen($data); $index++) {
     
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     
120.         }
     ; e) u% j  M$ `" l9 P* R5 K
121.         return $decoded;
     4 F: |$ z0 Z+ k9 D3 g: V2 O
122.     }
     ! K: q0 ]# x! \2 W: b! O
123. 
     8 D" e8 T& v6 {
124.     /**
     
125.      * 发送数据
     2 r; z" B( _0 i9 g1 Q: ?* l
126.      * @param $newClinet 新接入的socket
     6 _6 y+ l/ ]1 z3 |  T
127.      * @param $msg   要发送的数据
     . E$ t% F$ e6 @& r0 Q9 u
128.      * @return int|string
     " n) j2 T( ~: p, a6 V
129.      */
     ! B3 F0 [/ |4 u5 z# d  n3 ?- K
130.     public function send($newClinet, $msg){
     + B3 ~4 N/ r& s
131.         $msg = $this->frame($msg);
     # t9 v: m% K8 w! W4 d' h
132.         socket_write($newClinet, $msg, strlen($msg));
     7 r& ^+ ~" C: E: [
133.     }
     
134. 
     
135.     public function frame($s) {
     : ^8 a" U: F2 B9 `: y! k2 |. v
136.         $a = str_split($s, 125);
     
137.         if (count($a) == 1) {
     3 u6 `2 D  e. F( d& {' z( G; D! m
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     
139.         }
     
140.         $ns = "";
     
141.         foreach ($a as $o) {
     2 `0 \; @& L% w' ]7 f4 k2 W8 ^
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     ! m' Y% D2 p4 L) _' u4 c- f) j
143.         }
     
144.         return $ns;
     " c) J1 b" y7 n; G, ~6 ^
145.     }
     ! r7 O8 M4 S) S
146. 
     
147.     /**
     
148.      * 关闭socket
     
149.      */
     8 g& Q6 S/ t2 j
150.     public function close(){
     + Y  \! p# i0 r; U
151.         return socket_close($this->_sockets);
     
152.     }
     8 P5 n, ?% o' J2 l0 F
153. }
     # e) C- v( C4 n: ^8 G
154. 
     
155. $sock = new SocketService();
     7 f" D3 j  }2 N! _' G, Y4 t
156. $sock->run();
     
157. 
     

复制代码

web.html
: V: O8 Y, Y2 T9 ?/ J- k: J% u

1. <!doctype html>
   , {$ w+ j; W/ {# B' O# y
2. <html lang="en">
   
3. <head>
   / N5 f( W3 X' c
4.   <meta charset="UTF-8">
   & s' q5 U# k6 ~+ @2 E7 m  F# l
5.   <meta name="viewport" content="width=device-width,initial-scale=1, maximum-scale=1, user-scalable=no">
   7 S5 w) B. G/ I# u8 O" O
6.   <title>websocket</title>
   9 G. S9 L; b* b  b: d0 K; [; [
7. </head>
   ' O: j' R2 F2 e) B
8. <body>
   , k/ e& A2 z3 B9 ^6 V
9. <input id="text" value="">
   
10. <input type="submit" value="send" onclick="start()">
    
11. <input type="submit" value="close" onclick="close()">
    1 X. F8 W0 C6 n1 ?7 |
12. <div id="msg"></div>
    
13. <script>
    , @! [% p/ w$ i
14. /**
    
15. 0：未连接
    ' r4 l& @; F* o
16. 1：连接成功，可通讯
    
17. 2：正在关闭
    $ y- X; B, x1 ^3 @. V
18. 3：连接已关闭或无法打开
    
19. */
    
20. 
    
21.     //创建一个webSocket 实例
    , h9 E; S$ `% B, F$ s0 z/ t
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    1 W# G& V' g  z' ]/ ^, ~4 f6 Q
23. 
    
24. 
    
25.     webSocket.onerror = function (event){
    
26.         onError(event);
    . Z- v' _- b# Q! b2 |
27.     };
    * [/ H5 l, c3 c* B* ]
28. 
    9 z/ z+ e& f+ I9 }. D: y6 B3 i
29.     // 打开websocket
    
30.     webSocket.onopen = function (event){
    
31.         onOpen(event);
    
32.     };
    
33. 
    4 A* A! K9 ~$ f4 d# O3 K
34.     //监听消息
    3 _3 n; Y6 V$ A$ H
35.     webSocket.onmessage = function (event){
    0 {( B, E$ s; V7 g
36.         onMessage(event);
    0 P' X/ \3 w( C& O# `+ r
37.     };
    
38. 
    
39. 
    ) L/ c: n0 M. K
40.     webSocket.onclose = function (event){
    
41.         onClose(event);
    
42.     }
    ; |1 Q) R3 w, z: d
43. 
    2 D, [; w% z1 k# A0 _$ |% \0 X* S
44.     //关闭监听websocket
    
45.     function onError(event){
    6 c6 G' I4 N: ~
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    : ^! I' i1 L8 O: @$ m$ }
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>";
    " n. ]3 N6 `1 ^. a4 }- u+ x
53.     };
    % x0 d7 W9 Q" C! p% I7 {
54.     function onMessage(event){
    5 E6 ?! m6 L0 X, b4 Q: ?% w
55.         console.log("onMessage");
    
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    3 v5 \; ^1 l- }2 i+ U
57.     };
    
58. 
    - V! m( {! P+ V( ]7 q
59.     function onClose(event){
    
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    ) E2 Q' t) I# k6 Z5 N
61.         console.log("close:"+sockState());
    3 U) `6 |- p- ?, d
62.         webSocket.close();
    ) m5 [, C- E+ x1 J, b  B% W
63.     }
    ; p+ R8 o0 o; v7 {5 B' A/ P+ l9 O
64. 
    : U/ `; O% B; }
65.     function sockState(){
    3 ^8 V9 b6 o) j9 Y) _, o
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    8 I* ^: r9 j  f0 i+ f* f, w) V
67.             return status[webSocket.readyState];
    
68.     }
    
69. 
    
70. 
    
71. 
      u7 C# N1 A6 j6 I/ Q- _2 u& F
72. function start(event){
    
73.         console.log(webSocket);
    ' v1 q  y& X  f7 _9 M
74.         var msg = document.getElementById('text').value;
    
75.         document.getElementById('text').value = '';
    ! t! s1 D0 L$ R3 e' a
76.         console.log("send:"+sockState());
    
77.         console.log("msg="+msg);
    
78.         webSocket.send("msg="+msg);
    # V6 B. m% k7 ]! A4 c6 l( f4 P# H
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    & J" H2 G5 i; k, B9 Q9 s( k
80.     };
    ) e% O( F  ^! N, m) \/ a
81. 
    , A8 g9 ~! _. b7 c. u" L0 t. y$ p1 {
82.     function close(event){
    
83.         webSocket.close();
    
84.     }
    , T: R( L+ r* m9 q
85. </script>
    
86. </body>
    
87. </html>

复制代码

: F2 C' n! \0 ~, X8 k# z5 F: V' ]3 w
/ k! h/ B/ u1 J1 N4 `
9 ]3 F/ d: y0 }, @