php实现websocket实时消息推送

admin

php实现websocket实时消息推送

! J) |3 L5 e9 L, N5 G, \
SocketService.php

1. <?php
   $ i* ~# q. h8 A) \2 v8 @" Y
2. /**
   4 ]% I! ^( F: \- c' k
3. * Created by xwx
   & e# C- ]$ \6 c) M; q& D+ a
4. * Date: 2017/10/18
   
5. * Time: 14:33
   
6. */
   ) \( T) K7 m: R% A$ |! S
7. 
   
8. class SocketService
   % [  B. U5 ~5 `& e
9. {
   
10.     private $address  = '0.0.0.0';
      y0 s4 w  W& [1 y: Q5 c0 t6 `
11.     private $port = 8083;
    3 i6 t0 o+ f4 H8 c; x. `6 ]
12.     private $_sockets;
    3 w# d! W8 o' `3 ?/ X
13.     public function __construct($address = '', $port='')
    / w3 l$ _) d; ^6 q$ d
14.     {
    ) v: ^- C' C! J/ L: }
15.             if(!empty($address)){
    
16.                 $this->address = $address;
    ! o4 @9 F# U& ^. e; j" V2 A
17.             }
    
18.             if(!empty($port)) {
    
19.                 $this->port = $port;
    & \9 J' B; Y* G7 J2 r' _
20.             }
      Y6 \' c' f+ \3 q& }
21.     }
    
22. 
    
23.     public function service(){
    ' [0 B" m1 x( V( D8 w! s$ ^' M
24.         //获取tcp协议号码。
    
25.         $tcp = getprotobyname("tcp");
    ' k7 y$ K( E% b+ P6 _2 e6 d% K( N* |7 A
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    , D& a2 @% X1 y/ O" d3 V. {
28.         if($sock < 0)
    ; w( ?& j! q0 c( a0 m9 l
29.         {
    
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    " u1 e# ?& A  \
31.         }
    
32.         socket_bind($sock, $this->address, $this->port);
    
33.         socket_listen($sock, $this->port);
    8 b2 n4 V+ w% V$ h* B$ j9 C+ |
34.         echo "listen on $this->address $this->port ... \n";
    
35.         $this->_sockets = $sock;
    6 Q4 m4 a, g6 u3 c0 R9 E( @$ }
36.     }
    - X5 \8 D) }" m' i$ ]2 c
37. 
    3 y2 o+ f4 Y' a$ u7 _0 ?
38.     public function run(){
    % g3 I  ^; n$ }0 K0 t% Y1 g
39.         $this->service();
    ! i* R* J! c0 f
40.         $clients[] = $this->_sockets;
    
41.         while (true){
    
42.             $changes = $clients;
    # |" t7 S- V  Q# I+ D/ o6 _
43.             $write = NULL;
    # ~, N" Q/ q# r+ T/ L6 C2 w6 y2 F
44.             $except = NULL;
    
45.             socket_select($changes,  $write,  $except, NULL);
    
46.             foreach ($changes as $key => $_sock){
    
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    
48.                     if(($newClient = socket_accept($_sock))  === false){
    . V9 m  N/ n  k" n/ t
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    1 M1 P! w  @% D0 l  b: i
50.                     }
    $ W3 B3 E' E; J% }7 l$ N
51.                     $line = trim(socket_read($newClient, 1024));
    8 h% y( F7 @' q, u( R2 D
52.                     $this->handshaking($newClient, $line);
      c0 R! W; P. K. [! ]; [! S. \1 }
53.                     //获取client ip
    
54.                     socket_getpeername ($newClient, $ip);
    
55.                     $clients[$ip] = $newClient;
    5 R, r* Q* K& n- N7 X. G
56.                     echo  "Client ip:{$ip}   \n";
    
57.                     echo "Client msg:{$line} \n";
    9 j. t# P" s* m4 }& }2 t8 l
58.                 } else {
    0 g: ]  N7 w# d5 W& X6 h  W
59.                     socket_recv($_sock, $buffer,  2048, 0);
    ; N" l* T+ t) W# v- |& x. u6 ^  A
60.                     $msg = $this->message($buffer);
    ( l/ _% W. F3 D2 {; ~3 S' _8 B
61.                     //在这里业务代码
    
62.                     echo "{$key} clinet msg:",$msg,"\n";
    , a3 U- Q+ S" x. h, c: b5 m; r
63.                     fwrite(STDOUT, 'Please input a argument:');
    
64.                     $response = trim(fgets(STDIN));
    7 {" V6 r3 o! C! h& y: \
65.                     $this->send($_sock, $response);
    ; {- d+ H( v# [
66.                     echo "{$key} response to Client:".$response,"\n";
    
67.                 }
    0 d0 P/ v! L2 C: \$ s2 O+ p
68.             }
    0 C! `9 V' r- I
69.         }
    
70.     }
    5 q, A7 q) Y* h! K( y* f% n& U$ Q
71. 
    7 n9 g& K$ }" E" `* i/ Q
72.     /**
    % B2 b; |8 f; s0 w2 y2 g9 H
73.      * 握手处理
    " S* L' i* q3 |4 d
74.      * @param $newClient socket
    
75.      * @return int  接收到的信息
    
76.      */
    
77.     public function handshaking($newClient, $line){
    
78. 
    
79.         $headers = array();
    ' C7 T2 B) X' A- y' ^, q' w! [
80.         $lines = preg_split("/\r\n/", $line);
    . z: C8 I7 W; B6 u% t$ v1 |+ c
81.         foreach($lines as $line)
    + q7 q+ g, d+ H. h& L, V) `
82.         {
    
83.             $line = chop($line);
    
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    
85.             {
    
86.                 $headers[$matches[1]] = $matches[2];
    
87.             }
    $ p* @* U- ~# V$ q- |* R, {* u+ g1 _/ G
88.         }
    8 j4 A- X2 S8 v) n, q; n+ k
89.         $secKey = $headers['Sec-WebSocket-Key'];
    . G# |% G) J0 C6 M! P+ u4 l' _
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" .
    ) L$ ^6 M. N  U  v4 I& B' n1 U- j; G
92.             "Upgrade: websocket\r\n" .
    5 d: W& M; p% Z' h3 j7 M- _, K, G- f5 b
93.             "Connection: Upgrade\r\n" .
    ! e0 {2 O0 D6 ]8 h5 |
94.             "WebSocket-Origin: $this->address\r\n" .
    
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));
    3 i1 O$ |& i5 y
98.     }
    1 N( y! p) E: u  l
99. 
    
100.     /**
     1 {( L; {# m' A, h) s( X, g" p/ l
101.      * 解析接收数据
     6 s, [9 M, B3 ~5 @) C7 M1 ~( T1 ~) V
102.      * @param $buffer
     
103.      * @return null|string
     " M+ R, \9 i3 |! i# c, U7 C( k
104.      */
     3 _3 @/ c5 p- `7 U. D" f
105.     public function message($buffer){
     
106.         $len = $masks = $data = $decoded = null;
     
107.         $len = ord($buffer[1]) & 127;
       }& |* @7 W$ F3 o
108.         if ($len === 126)  {
     
109.             $masks = substr($buffer, 4, 4);
     . M6 a. {. J, q" \
110.             $data = substr($buffer, 8);
     8 X' b6 U8 W! ]! \
111.         } else if ($len === 127)  {
     
112.             $masks = substr($buffer, 10, 4);
     
113.             $data = substr($buffer, 14);
     
114.         } else  {
     
115.             $masks = substr($buffer, 2, 4);
     - Q% k/ c8 j$ K* N
116.             $data = substr($buffer, 6);
     
117.         }
     
118.         for ($index = 0; $index < strlen($data); $index++) {
     ( S( Y0 G5 Q/ ]
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
       W8 N/ z$ l4 H! J/ L
120.         }
     $ O/ S$ r0 [0 ~" `+ j* U
121.         return $decoded;
     
122.     }
     & H4 O5 I8 i5 n' F
123. 
     
124.     /**
     " p" p) f# N# k& \3 I7 ]
125.      * 发送数据
     : {5 s! j2 ?: n" x  s) C4 G; E" k
126.      * @param $newClinet 新接入的socket
     
127.      * @param $msg   要发送的数据
     
128.      * @return int|string
     
129.      */
     ; d1 \: h9 u3 u) Z/ u
130.     public function send($newClinet, $msg){
     
131.         $msg = $this->frame($msg);
     
132.         socket_write($newClinet, $msg, strlen($msg));
     
133.     }
     
134. 
     
135.     public function frame($s) {
     
136.         $a = str_split($s, 125);
     
137.         if (count($a) == 1) {
     $ V2 T2 w6 F; G! n/ ^! _) v# q7 k! P
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     4 q& t6 c7 q# l2 J9 W
139.         }
     5 B0 M: R9 {6 ^  E% p$ C1 W
140.         $ns = "";
     6 q$ y: I8 I4 M2 c' y( g
141.         foreach ($a as $o) {
     
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     
143.         }
     
144.         return $ns;
     ) `1 ^' w# l) s9 x$ Y! K
145.     }
     + w1 Y' e+ ?5 T/ Y8 T
146. 
     
147.     /**
     * A5 z& f) Z6 f  D* |( P
148.      * 关闭socket
     / s# ], ^* o. u. V1 x  K
149.      */
     " T- w# E. t- N
150.     public function close(){
     6 W2 E0 g( V6 Y$ b- i
151.         return socket_close($this->_sockets);
     ( [, P2 n+ F; K6 M* p" z$ e3 i
152.     }
     
153. }
     
154. 
     
155. $sock = new SocketService();
     & W' q8 K6 g6 |5 I
156. $sock->run();
     
157. 
     

复制代码

web.html
) C& D- O2 \3 h( B1 ]

1. <!doctype html>
   ! E/ }+ S6 q* B, H
2. <html lang="en">
   . z6 B& O6 m- D, J
3. <head>
   
4.   <meta charset="UTF-8">
   
5.   <meta name="viewport" content="width=device-width,initial-scale=1, maximum-scale=1, user-scalable=no">
     v5 b# R' Y$ d7 s' Q8 d5 \/ Z" j
6.   <title>websocket</title>
   : `$ m# `* }) u) ~4 i& z& \
7. </head>
     c, j2 f, D, u* g2 o. Q2 t! C  @+ l' t
8. <body>
   
9. <input id="text" value="">
   5 P1 v& H9 \1 p! ]4 ?
10. <input type="submit" value="send" onclick="start()">
    / O8 B4 i; E# X0 ^- m1 y0 C5 D
11. <input type="submit" value="close" onclick="close()">
    ; f0 w. u, P; y% @$ B4 P
12. <div id="msg"></div>
    
13. <script>
    3 V; X; q, }. f; m
14. /**
    # J: h* z' H6 f+ H) c
15. 0：未连接
    % |( g6 ?1 [' L4 Y9 S" i
16. 1：连接成功，可通讯
    
17. 2：正在关闭
    
18. 3：连接已关闭或无法打开
    
19. */
    8 C7 b% E# S5 h, k( M* u
20. 
    " Q2 f/ \$ z# R* q4 D
21.     //创建一个webSocket 实例
    
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    * V- w; H7 U. |) P( d7 ^
23. 
    ) T# z  F$ P* _+ m5 @1 {' O
24. 
    - Q8 q5 W; h; k9 s, E
25.     webSocket.onerror = function (event){
    7 b- \" j4 a! K" p; |0 q
26.         onError(event);
    + Q0 \- `0 V, P$ `3 [: ^3 ^; B$ A
27.     };
    
28. 
    " B  U0 Z& m3 J6 U5 n. y( s( ~9 K
29.     // 打开websocket
    9 l* i$ o/ k! K: m
30.     webSocket.onopen = function (event){
    
31.         onOpen(event);
    
32.     };
    9 u5 C8 K8 f' }. s! x0 v
33. 
    
34.     //监听消息
    & ~% U$ I5 q6 B% H* N. u
35.     webSocket.onmessage = function (event){
    
36.         onMessage(event);
    
37.     };
    
38. 
    + l3 k: r3 q( ~* P$ L# W
39. 
    2 b* C) w6 W6 E" I: W
40.     webSocket.onclose = function (event){
    
41.         onClose(event);
    8 c4 a  {3 _6 ]* r, Q3 u
42.     }
    
43. 
    8 }# W" p! a( }5 ]' f: x% ~
44.     //关闭监听websocket
    0 o4 b* N" b/ A. e# W1 f6 S
45.     function onError(event){
    
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
47.         console.log("error"+event.data);
    
48.     };
    
49. 
    ; {! d+ n& s9 A) t9 Y
50.     function onOpen(event){
    
51.         console.log("open:"+sockState());
    
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    
53.     };
    
54.     function onMessage(event){
    
55.         console.log("onMessage");
    - V! ~" a- b  G$ j
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    ; Z; q5 c) a' U2 w7 `# I3 |
57.     };
    8 m7 s4 H8 H& T. q% ]: R- I
58. 
    4 K$ G' Q; S  z7 i
59.     function onClose(event){
    5 H: X; Y+ D# y
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
61.         console.log("close:"+sockState());
    4 m( Y, v+ L% s/ I, V; f4 n5 b1 T5 K
62.         webSocket.close();
    ' o# U0 g' f: B5 i+ Q
63.     }
    
64. 
    
65.     function sockState(){
    
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    0 r' C2 j: e: _; p/ u
67.             return status[webSocket.readyState];
    
68.     }
    
69. 
    ) x/ b. V4 A* l5 [7 Q
70. 
    6 _6 {( _& \% J- `# f, ^
71. 
    , n+ l( f; j) B
72. function start(event){
    7 o$ @0 Q7 _' j) s
73.         console.log(webSocket);
    1 M* ?) Y5 C1 m. g5 H  s; r
74.         var msg = document.getElementById('text').value;
    / e; U4 U  ?" U2 ^  D  X
75.         document.getElementById('text').value = '';
    
76.         console.log("send:"+sockState());
    
77.         console.log("msg="+msg);
    
78.         webSocket.send("msg="+msg);
    
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    
80.     };
    6 p$ E# _2 a( w1 f
81. 
      N1 n: ?% c4 y9 I- D" ~
82.     function close(event){
    
83.         webSocket.close();
    
84.     }
    & k* {  E$ A2 j, e) `$ J3 o; x
85. </script>
    * y$ D4 F/ k# e4 z6 ~, k
86. </body>
    
87. </html>

复制代码

- y" @+ T! u- A" K5 d( A3 @) b