php实现websocket实时消息推送

admin

php实现websocket实时消息推送

& V% `5 _& l* _/ k) n& |- L: i
( v1 Q  f) h$ l  dSocketService.php

1. <?php
   1 V! L  I8 t- s  c
2. /**
   
3. * Created by xwx
   
4. * Date: 2017/10/18
   % T% H8 V* F; ^1 z( X
5. * Time: 14:33
   
6. */
   $ N0 h  F; {5 Q! l
7. 
   
8. class SocketService
   ; R7 e/ i* l1 g9 f1 ?
9. {
   
10.     private $address  = '0.0.0.0';
    
11.     private $port = 8083;
    # ?9 M! j% E# g# Q
12.     private $_sockets;
    : h8 j, r' |+ ]
13.     public function __construct($address = '', $port='')
      A& a7 l; u+ \4 F; D# H. p
14.     {
    1 i' \7 p# x" I/ x; b% }
15.             if(!empty($address)){
    
16.                 $this->address = $address;
    ) H( ^& M$ V3 y% t4 V
17.             }
    
18.             if(!empty($port)) {
    8 L0 \  Y8 [9 B" f, ]! Z
19.                 $this->port = $port;
    
20.             }
    
21.     }
    , }7 B5 `" ?7 b! I9 g
22. 
    
23.     public function service(){
    
24.         //获取tcp协议号码。
    
25.         $tcp = getprotobyname("tcp");
    
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    
28.         if($sock < 0)
    3 a$ q  F" I4 s. ^( B
29.         {
    
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    
31.         }
    / |, c3 }# ]/ ~0 ^
32.         socket_bind($sock, $this->address, $this->port);
    * E# V1 \+ y; B9 i
33.         socket_listen($sock, $this->port);
    ( f( \  H  s( G; q
34.         echo "listen on $this->address $this->port ... \n";
    
35.         $this->_sockets = $sock;
    1 z$ V$ @3 v# M
36.     }
    
37. 
    / d8 E2 l5 x3 K  l( Z
38.     public function run(){
    2 N- c# s' X8 n, w) i, m1 y, \
39.         $this->service();
    
40.         $clients[] = $this->_sockets;
    ! V* d3 H9 q" E0 |
41.         while (true){
    9 ^) \6 M" b4 w% `! I
42.             $changes = $clients;
    % K) J9 H: i4 w" r& d
43.             $write = NULL;
    
44.             $except = NULL;
    
45.             socket_select($changes,  $write,  $except, NULL);
    
46.             foreach ($changes as $key => $_sock){
    0 s  b3 X' @0 W& M6 w/ a
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    ' g: J( Q1 k8 g" f# X6 F# h
48.                     if(($newClient = socket_accept($_sock))  === false){
    
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    
50.                     }
    
51.                     $line = trim(socket_read($newClient, 1024));
    & l" e; [/ o! R1 A2 C/ c* x
52.                     $this->handshaking($newClient, $line);
    
53.                     //获取client ip
    ; Z9 h- f! |/ b2 |  |" Y& |* L
54.                     socket_getpeername ($newClient, $ip);
    6 o$ P; \9 i7 C0 F* r
55.                     $clients[$ip] = $newClient;
    
56.                     echo  "Client ip:{$ip}   \n";
    3 f+ T( m4 j5 g6 G- O" j
57.                     echo "Client msg:{$line} \n";
    
58.                 } else {
    
59.                     socket_recv($_sock, $buffer,  2048, 0);
    ; F5 Y+ g& I, y7 U- i
60.                     $msg = $this->message($buffer);
    7 i0 b3 e( d$ V4 h, ~3 r0 g
61.                     //在这里业务代码
    
62.                     echo "{$key} clinet msg:",$msg,"\n";
    
63.                     fwrite(STDOUT, 'Please input a argument:');
    4 [- a+ M; s- v
64.                     $response = trim(fgets(STDIN));
    
65.                     $this->send($_sock, $response);
    
66.                     echo "{$key} response to Client:".$response,"\n";
    
67.                 }
    
68.             }
    * b! p1 R* S& G$ S. O& C
69.         }
    % v1 L0 _# m* c  g  J9 T
70.     }
    
71. 
    
72.     /**
    
73.      * 握手处理
    
74.      * @param $newClient socket
    
75.      * @return int  接收到的信息
    
76.      */
    
77.     public function handshaking($newClient, $line){
    6 ?& b8 ^4 Q! e8 j2 Y0 }, \- ?
78. 
    
79.         $headers = array();
    ) h. G2 ~2 U& k6 e* B' {, _, u
80.         $lines = preg_split("/\r\n/", $line);
      D  g% F7 t1 @8 `6 N3 G5 }; m
81.         foreach($lines as $line)
    
82.         {
    
83.             $line = chop($line);
    * s) Z" f" V! a0 ?5 _; k
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    3 b7 W! S) q9 k1 M2 U
85.             {
    7 s( h. e$ ?1 @7 V5 Y1 v
86.                 $headers[$matches[1]] = $matches[2];
    ' p0 Q) S9 M  L1 S1 H
87.             }
    6 C2 u6 W! |6 s
88.         }
    , I. N1 h: @4 m9 M* e
89.         $secKey = $headers['Sec-WebSocket-Key'];
    
90.         $secAccept = base64_encode(pack('H*', sha1($secKey . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11')));
    & |1 V$ H* }. B6 ?* U: b
91.         $upgrade  = "HTTP/1.1 101 Web Socket Protocol Handshake\r\n" .
    
92.             "Upgrade: websocket\r\n" .
    ! d# q; G; T3 m9 V$ ]! F
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* k8 b5 S6 T+ y: K, m8 G% k
96.             "Sec-WebSocket-Accept:$secAccept\r\n\r\n";
    
97.         return socket_write($newClient, $upgrade, strlen($upgrade));
    / j, q$ \: ?) B/ x, U7 v+ ^
98.     }
    
99. 
    6 R3 V0 Q' A" c' ]) j
100.     /**
     
101.      * 解析接收数据
     
102.      * @param $buffer
     
103.      * @return null|string
     
104.      */
     
105.     public function message($buffer){
     
106.         $len = $masks = $data = $decoded = null;
     * c7 a  X1 z4 L8 N0 |
107.         $len = ord($buffer[1]) & 127;
     
108.         if ($len === 126)  {
       m' `8 o$ `+ _7 L' |* ^, b8 b
109.             $masks = substr($buffer, 4, 4);
     
110.             $data = substr($buffer, 8);
     
111.         } else if ($len === 127)  {
     1 y: g) l* P& q, k& j7 K
112.             $masks = substr($buffer, 10, 4);
     
113.             $data = substr($buffer, 14);
     1 I& T$ Y1 d4 ?4 }% v0 T# t
114.         } else  {
     
115.             $masks = substr($buffer, 2, 4);
     
116.             $data = substr($buffer, 6);
     ' Q- y, p8 o/ f0 S+ W
117.         }
       w# ]: U4 n- F4 f+ e
118.         for ($index = 0; $index < strlen($data); $index++) {
     * ^3 y2 F0 u1 q* U# o( g
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     
120.         }
     4 j( M$ ~; H9 p  V
121.         return $decoded;
     
122.     }
     
123. 
     
124.     /**
     
125.      * 发送数据
     * b/ U! @6 \! U$ t6 {
126.      * @param $newClinet 新接入的socket
     + F& Q  ^& ?* P  I4 c/ l
127.      * @param $msg   要发送的数据
     
128.      * @return int|string
     0 y2 D' r4 l" j  o0 }& J
129.      */
     
130.     public function send($newClinet, $msg){
     
131.         $msg = $this->frame($msg);
     
132.         socket_write($newClinet, $msg, strlen($msg));
     . R; }0 s4 y, o2 H9 ~8 V( @7 b- X
133.     }
     
134. 
     
135.     public function frame($s) {
     " [. T4 F+ @% y1 Z# x. v
136.         $a = str_split($s, 125);
     
137.         if (count($a) == 1) {
     
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
       T4 u& j- _" A, ^
139.         }
     , W; t4 @% w* I9 H
140.         $ns = "";
     1 n9 U* ~1 f- m. h: h4 ?
141.         foreach ($a as $o) {
     / E- V9 ^, Q, W; J9 b
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     , W; G; V3 O: S1 q& t
143.         }
     7 m7 a* J- ]; j6 p# D  D
144.         return $ns;
     
145.     }
     ) b  k# s. Q* @- M5 r9 k$ K& k/ I
146. 
     $ X2 t  K. W) d" E
147.     /**
     - E: x# l# c7 m7 Z# r) D2 I! p( r
148.      * 关闭socket
     
149.      */
     
150.     public function close(){
     7 E- Z6 L8 \. ]  [+ |
151.         return socket_close($this->_sockets);
     0 m6 i/ n2 Z$ i! m& L; z7 z
152.     }
     - Y) P! ?2 Q9 ~  ]6 N; F! @
153. }
     7 D4 f6 c* c2 V- F+ @/ @  u
154. 
     
155. $sock = new SocketService();
     ( c; u, W" t3 ]; V9 B2 J& v! M) w5 N
156. $sock->run();
     ' U, N" ]( W# `% \' c! y% q
157. 
     $ B  U$ E5 a; A- h

复制代码

web.html

1. <!doctype html>
   # @1 Z0 ~5 o0 \; x  I
2. <html lang="en">
   6 c' w+ g# Y2 `: f
3. <head>
   ; w8 e7 `" z; |0 G
4.   <meta charset="UTF-8">
   8 y& M( R% `. B: D" s$ P
5.   <meta name="viewport" content="width=device-width,initial-scale=1, maximum-scale=1, user-scalable=no">
   
6.   <title>websocket</title>
   # e1 S; i- U, p* N% H
7. </head>
   8 M2 I5 r6 }8 V0 j2 I  {
8. <body>
   8 H8 y3 K$ S: e1 y/ ]
9. <input id="text" value="">
   ! n! b! A; J' u/ m6 f5 E
10. <input type="submit" value="send" onclick="start()">
    
11. <input type="submit" value="close" onclick="close()">
    ! ]- S, ]0 ?4 J  ?) G
12. <div id="msg"></div>
      a: U) R# Q; b# Q. H7 C, c% h/ y4 l
13. <script>
    
14. /**
    
15. 0：未连接
    5 D1 t) K9 o! x- W( B% ]
16. 1：连接成功，可通讯
    - I8 H: ]* w2 d/ ]& j& ^4 I
17. 2：正在关闭
    , e/ @- {1 G! \+ q
18. 3：连接已关闭或无法打开
    + d5 D9 o* I6 G# |2 S
19. */
    
20. 
    
21.     //创建一个webSocket 实例
    
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    7 i' v. c" T+ ?1 I) |6 A# w
23. 
    , ~4 G$ V8 }1 H
24. 
    
25.     webSocket.onerror = function (event){
    
26.         onError(event);
    6 h. g5 G2 F% M$ K0 E7 {) o
27.     };
    2 ?* S" {1 s/ r' V
28. 
    $ ?( A# T* \' p5 n6 A
29.     // 打开websocket
    3 m8 f  @! s* H" I, L. g
30.     webSocket.onopen = function (event){
    9 _# p. E/ t# w' t* j: y
31.         onOpen(event);
    ' Z; h5 J: b. l$ U7 E
32.     };
    
33. 
    
34.     //监听消息
    
35.     webSocket.onmessage = function (event){
    
36.         onMessage(event);
    ! X4 U5 C- p" v2 u) U* ~
37.     };
    
38. 
    & j6 Z5 W) b, V4 K3 T
39. 
    
40.     webSocket.onclose = function (event){
    
41.         onClose(event);
    
42.     }
    ) s5 k( H0 ^- Z' _+ E2 L
43. 
    
44.     //关闭监听websocket
    
45.     function onError(event){
    9 M' ^5 [6 V3 v) p6 c' g
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
47.         console.log("error"+event.data);
    
48.     };
    
49. 
    
50.     function onOpen(event){
    5 q& w- r# x  u: b) U
51.         console.log("open:"+sockState());
    # a% F! b3 f( ?# c0 _$ O
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    
53.     };
    6 T$ j2 e' ]% v  H$ Z0 m, x1 u
54.     function onMessage(event){
    0 k6 q# [0 c- ], Z: Z% R; [6 B: Z3 o
55.         console.log("onMessage");
    
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    
57.     };
    
58. 
    2 i. s0 p; }3 i; P* Q% p
59.     function onClose(event){
    
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
61.         console.log("close:"+sockState());
    
62.         webSocket.close();
    
63.     }
    5 o8 |9 `3 n: k" ~8 j* a
64. 
    ' C/ I: `7 w5 W% z; X
65.     function sockState(){
    . {3 \3 A4 S5 S3 e- N
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    5 v+ H& B; m- ^* U$ A) d
67.             return status[webSocket.readyState];
    ) Q$ i9 ^0 x8 M: \8 b6 p
68.     }
    ! Z, I: g; S5 L. ^
69. 
    8 U( t" v' F  j# a. a: K1 ~
70. 
    8 p  p$ g# E9 r' h
71. 
    3 ~/ H! L4 q0 n8 Y/ s( Q
72. function start(event){
    1 q- u2 h/ B! A. F2 o/ X
73.         console.log(webSocket);
    # e" }* E! g" |2 d; S0 X2 [7 a
74.         var msg = document.getElementById('text').value;
      I1 \/ _3 L! o, H
75.         document.getElementById('text').value = '';
    ( d# S1 v* ~5 P  _
76.         console.log("send:"+sockState());
    
77.         console.log("msg="+msg);
    
78.         webSocket.send("msg="+msg);
    3 W- U# |' b# j- o& r/ r
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    8 @0 I) O4 |3 p, R+ }: v4 f: ~
80.     };
    ( r) k7 u6 @% B7 O3 U
81. 
    ' u" w- V4 Q  u
82.     function close(event){
    3 J' M7 n' a. I7 R) ~( p, [1 ~
83.         webSocket.close();
    7 T3 ^  [( ^$ N  ~" V4 }' {  @" F
84.     }
    
85. </script>
    9 s1 B- M6 m  f, y$ u0 e* Q4 {) _
86. </body>
    
87. </html>

复制代码

5 u: e+ N! _  {& S) n8 I) Y, E8 v6 z

# a5 q# N* M0 d; U