php实现websocket实时消息推送

admin

php实现websocket实时消息推送

- I3 n( D8 A1 e' C* S0 h6 G) w, J. @
SocketService.php
3 {- K: y5 H$ R5 u9 S

1. <?php
   7 o: B# @8 |# p. c) a/ X
2. /**
   4 M. I$ v0 I$ z/ C6 O8 N  J& g9 b
3. * Created by xwx
   1 @  A6 y+ t- T3 w  g. u. h
4. * Date: 2017/10/18
   , Q3 Y! N- j4 `3 ?( Q
5. * Time: 14:33
   - O! `0 E. o+ k, t, W
6. */
   4 M8 L- b, U: s
7. 
   
8. class SocketService
   3 v" D2 W3 Z" O! s
9. {
   , P$ u# @) ^1 _* I8 v3 H
10.     private $address  = '0.0.0.0';
    
11.     private $port = 8083;
    
12.     private $_sockets;
    0 m* T7 A4 }/ j: L, k
13.     public function __construct($address = '', $port='')
    
14.     {
    ( Y) d) U7 f6 L9 l" Y7 y/ R, v
15.             if(!empty($address)){
    
16.                 $this->address = $address;
    
17.             }
    
18.             if(!empty($port)) {
    5 i4 B, r# ^' B2 e- B
19.                 $this->port = $port;
    
20.             }
    
21.     }
    ' I' B# [  i1 l" b2 a
22. 
    
23.     public function service(){
    
24.         //获取tcp协议号码。
    
25.         $tcp = getprotobyname("tcp");
    ! Y2 e) z. e0 @8 E% V6 l: X
26.         $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
    
27.         socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
    
28.         if($sock < 0)
    
29.         {
    6 M' z4 B5 h2 r% v# b
30.             throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
    9 h# p" f' {4 q0 b
31.         }
    . Z0 N  ]) r' Z* t( z( i- A
32.         socket_bind($sock, $this->address, $this->port);
    
33.         socket_listen($sock, $this->port);
    
34.         echo "listen on $this->address $this->port ... \n";
    ( c. M5 U/ w& R
35.         $this->_sockets = $sock;
    5 h% ~& @+ _) Y8 n
36.     }
    2 m, N. u, t3 T& D
37. 
    % R: ]2 c3 \9 u2 u
38.     public function run(){
    ) P4 k1 L" U# z/ o% i( M
39.         $this->service();
    
40.         $clients[] = $this->_sockets;
    ' V5 S) C# T% _5 u" \
41.         while (true){
    # I$ p( A1 U& q, m9 n
42.             $changes = $clients;
    
43.             $write = NULL;
    ; h9 {3 E# y4 E2 C
44.             $except = NULL;
    
45.             socket_select($changes,  $write,  $except, NULL);
    
46.             foreach ($changes as $key => $_sock){
    1 r/ @+ D, K( H/ ]
47.                 if($this->_sockets == $_sock){ //判断是不是新接入的socket
    
48.                     if(($newClient = socket_accept($_sock))  === false){
    " ?2 R+ Z; a. L0 r. ^
49.                         die('failed to accept socket: '.socket_strerror($_sock)."\n");
    * E# O# P$ H( ^
50.                     }
      }- k6 f" A' M- r* E" m
51.                     $line = trim(socket_read($newClient, 1024));
    
52.                     $this->handshaking($newClient, $line);
    4 X) L) d) ^& g! g% {6 ~  P
53.                     //获取client ip
    
54.                     socket_getpeername ($newClient, $ip);
    7 z9 @4 m& j6 B% ^1 @
55.                     $clients[$ip] = $newClient;
    
56.                     echo  "Client ip:{$ip}   \n";
    
57.                     echo "Client msg:{$line} \n";
    - ~0 w- q" ~0 W) z  n0 u, K. G
58.                 } else {
    
59.                     socket_recv($_sock, $buffer,  2048, 0);
    
60.                     $msg = $this->message($buffer);
    ) l  v" h9 O* Y$ [5 M2 Z
61.                     //在这里业务代码
    
62.                     echo "{$key} clinet msg:",$msg,"\n";
    6 i& v# F2 Y% }% H
63.                     fwrite(STDOUT, 'Please input a argument:');
    " z% D/ }4 a. e3 ?6 J: k0 T5 }3 n0 b
64.                     $response = trim(fgets(STDIN));
    
65.                     $this->send($_sock, $response);
    
66.                     echo "{$key} response to Client:".$response,"\n";
    
67.                 }
    
68.             }
    
69.         }
    
70.     }
    1 d: V* F! ~. g: n
71. 
    
72.     /**
    3 Z6 _8 X' Z- @
73.      * 握手处理
    
74.      * @param $newClient socket
    - [+ R/ r! @, w/ r1 r$ B
75.      * @return int  接收到的信息
    
76.      */
    
77.     public function handshaking($newClient, $line){
    9 s  }' [( j! r( o
78. 
    
79.         $headers = array();
    
80.         $lines = preg_split("/\r\n/", $line);
    7 e% N3 u% j) ^  O& u% \
81.         foreach($lines as $line)
    
82.         {
    
83.             $line = chop($line);
    
84.             if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
    
85.             {
    
86.                 $headers[$matches[1]] = $matches[2];
    2 m& r$ @* L# ?
87.             }
    
88.         }
    
89.         $secKey = $headers['Sec-WebSocket-Key'];
    
90.         $secAccept = base64_encode(pack('H*', sha1($secKey . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11')));
    $ r) M* @4 U8 O9 m2 U
91.         $upgrade  = "HTTP/1.1 101 Web Socket Protocol Handshake\r\n" .
    ( v' l3 S0 e# L3 b
92.             "Upgrade: websocket\r\n" .
    9 c. G# S, l- J* n- Y- U
93.             "Connection: Upgrade\r\n" .
    * O7 b4 K- D+ p) n4 Z& V
94.             "WebSocket-Origin: $this->address\r\n" .
    9 t' t; b, O8 u
95.             "WebSocket-Location: ws://$this->address:$this->port/websocket/websocket\r\n".
    : e/ j! e; R% g3 B- N1 ], k
96.             "Sec-WebSocket-Accept:$secAccept\r\n\r\n";
    1 o' q2 V& W, P. M8 z
97.         return socket_write($newClient, $upgrade, strlen($upgrade));
    , `& T& ^' q% q4 U
98.     }
    ! x/ y: x+ F5 g: W6 u
99. 
    
100.     /**
     % m, m& [2 \! \! B' C5 k7 |: r
101.      * 解析接收数据
     6 @& e% P& L( c2 q3 X  J
102.      * @param $buffer
     ' H% ]0 r# \% z, n4 n: E9 W$ e
103.      * @return null|string
     
104.      */
     ; Z- Y3 F2 K) w( D/ U: ]! _) N3 S1 }
105.     public function message($buffer){
     
106.         $len = $masks = $data = $decoded = null;
     
107.         $len = ord($buffer[1]) & 127;
     
108.         if ($len === 126)  {
     
109.             $masks = substr($buffer, 4, 4);
     
110.             $data = substr($buffer, 8);
     
111.         } else if ($len === 127)  {
     5 J) N# \2 y+ U, Y" A
112.             $masks = substr($buffer, 10, 4);
     4 E/ Y. \8 D: W: S
113.             $data = substr($buffer, 14);
     5 B7 i& ~4 t. t  @
114.         } else  {
     ' S9 h1 b$ k( q! ]& v# d9 a
115.             $masks = substr($buffer, 2, 4);
     
116.             $data = substr($buffer, 6);
     # m0 L& u& Z  i( G7 a0 v1 ]
117.         }
     7 R- w$ X/ D) `/ Q: O) Z
118.         for ($index = 0; $index < strlen($data); $index++) {
     0 H! {* C, f/ @+ T$ z
119.             $decoded .= $data[$index] ^ $masks[$index % 4];
     + [# k6 V3 }$ d6 @6 x% S- z
120.         }
     0 ^( r% R. @* r4 B7 m" q5 c" n9 p
121.         return $decoded;
     - A% i* {. y2 C7 C( L5 l* c
122.     }
     
123. 
     
124.     /**
     6 b; R1 F: b7 h7 m9 S7 e; P
125.      * 发送数据
     $ N! y2 D7 K2 H8 P
126.      * @param $newClinet 新接入的socket
     , u9 E7 K9 Q$ X5 ?: t4 W0 J+ }7 j
127.      * @param $msg   要发送的数据
     8 _& `/ C2 F; W. w
128.      * @return int|string
     0 r) X7 i+ W- Y4 I5 C
129.      */
     
130.     public function send($newClinet, $msg){
     
131.         $msg = $this->frame($msg);
     + e5 o: }. q9 `& g/ M( \7 X
132.         socket_write($newClinet, $msg, strlen($msg));
     
133.     }
     1 s2 ~4 s8 B/ Z# ?' o
134. 
     & {2 V) \/ e  c1 Z' t/ u3 E, w
135.     public function frame($s) {
     
136.         $a = str_split($s, 125);
     9 Y8 Y, K4 C% ~: J
137.         if (count($a) == 1) {
     # H1 x. F6 ]) M% q4 s$ ]- z
138.             return "\x81" . chr(strlen($a[0])) . $a[0];
     
139.         }
     
140.         $ns = "";
     2 h( ~% v2 e. T, r
141.         foreach ($a as $o) {
     
142.             $ns .= "\x81" . chr(strlen($o)) . $o;
     7 W3 @1 o3 d+ [! C
143.         }
     7 b6 S4 g7 @. {) e
144.         return $ns;
     / B4 a" u- i  T' R
145.     }
     # a5 E/ S7 ]+ P' S8 q2 O3 A* E
146. 
     
147.     /**
     - k8 }1 h6 Z) z5 M1 {" M
148.      * 关闭socket
     9 Q* q" @9 I' c5 q$ V$ g
149.      */
       \+ U. v# n9 C$ ^0 F
150.     public function close(){
     
151.         return socket_close($this->_sockets);
     
152.     }
     $ _" S$ w, h+ u7 d
153. }
     0 I+ X% B4 F/ ~5 ?. @; \5 I& |
154. 
     
155. $sock = new SocketService();
     
156. $sock->run();
     % J" |4 z% b8 {( p* ]
157. 
     , J% H, j7 f- s

复制代码

web.html
. `9 I0 `) Y4 \

1. <!doctype html>
   
2. <html lang="en">
   
3. <head>
   
4.   <meta charset="UTF-8">
   ! B( z6 }+ ?- j  r# U
5.   <meta name="viewport" content="width=device-width,initial-scale=1, maximum-scale=1, user-scalable=no">
   5 [4 C5 f! M: W3 k' {3 t$ A
6.   <title>websocket</title>
   / j* }: t6 d( n- C  U: O
7. </head>
   
8. <body>
   : T+ X8 L7 G& C# |# a% r; @- I
9. <input id="text" value="">
   
10. <input type="submit" value="send" onclick="start()">
    
11. <input type="submit" value="close" onclick="close()">
    
12. <div id="msg"></div>
    ! D8 ~0 |0 ?/ ~; B
13. <script>
    
14. /**
    4 N" e3 V8 @  A: J6 _
15. 0：未连接
    7 n* ]: m# w9 I2 P& T
16. 1：连接成功，可通讯
    / `5 O1 ~# l) g
17. 2：正在关闭
    8 R% e4 R6 W$ R% r  X( t( X9 l  B
18. 3：连接已关闭或无法打开
    
19. */
    
20. 
    
21.     //创建一个webSocket 实例
    
22.     var webSocket  = new  WebSocket("ws://192.168.31.152:8083");
    ' V0 X4 z: F& {3 e  Q- z
23. 
    # H( u: l. U+ y- l  y3 F
24. 
    & @6 |  U+ W3 @2 _0 k% U0 K
25.     webSocket.onerror = function (event){
    
26.         onError(event);
    
27.     };
    
28. 
    8 p! x; J' \8 m6 e+ i
29.     // 打开websocket
    7 N" L' _) m) e3 {8 L
30.     webSocket.onopen = function (event){
    + T8 K- B/ d; D" o- A9 T
31.         onOpen(event);
    
32.     };
    
33. 
    
34.     //监听消息
    
35.     webSocket.onmessage = function (event){
    + T5 b' A& D% O2 W& _! g, v
36.         onMessage(event);
    ( U; w/ Q( ?7 Y9 F# @0 t5 C' M: D
37.     };
    
38. 
    
39. 
    6 c" w% e3 G( u! C
40.     webSocket.onclose = function (event){
    , e. Q3 u$ Q( z% o
41.         onClose(event);
    + U, c3 Q; l% `; i3 r* j* F
42.     }
    + z# n: F2 S7 [2 Y
43. 
    * h2 O5 X" y% U  J2 ^# r+ b& B
44.     //关闭监听websocket
    
45.     function onError(event){
    * P) V# o5 T1 u. ~: G
46.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
47.         console.log("error"+event.data);
    " o: k% O. V' i$ \. }
48.     };
    
49. 
    
50.     function onOpen(event){
    ' y, \% R* S) w- {- k4 |) L
51.         console.log("open:"+sockState());
    
52.         document.getElementById("msg").innerHTML = "<p>Connect to Service</p>";
    / \+ O: x: Q/ K7 m* ~
53.     };
      ~8 p) U0 U3 q7 M4 g* K
54.     function onMessage(event){
    ) ?7 h/ z! e0 O  G0 B. t6 T
55.         console.log("onMessage");
      j: I# ~) i; T2 c& [
56.         document.getElementById("msg").innerHTML += "<p>response:"+event.data+"</p>"
    
57.     };
    
58. 
    
59.     function onClose(event){
    
60.         document.getElementById("msg").innerHTML = "<p>close</p>";
    
61.         console.log("close:"+sockState());
    
62.         webSocket.close();
    
63.     }
    
64. 
    % g$ _4 t8 F$ J( c
65.     function sockState(){
    ' U; ]5 r4 e. w. Z8 |' g( c
66.         var status = ['未连接','连接成功，可通讯','正在关闭','连接已关闭或无法打开'];
    / G- @8 L: y" k/ L5 J; e; }# H
67.             return status[webSocket.readyState];
    / B- A8 l  s- ]# c- x% Y
68.     }
    
69. 
    
70. 
    
71. 
    
72. function start(event){
    4 Z) w& I$ o1 A% x, f7 S
73.         console.log(webSocket);
    
74.         var msg = document.getElementById('text').value;
    
75.         document.getElementById('text').value = '';
    
76.         console.log("send:"+sockState());
    
77.         console.log("msg="+msg);
    : J6 \; z6 @& u' s% r* w
78.         webSocket.send("msg="+msg);
    
79.         document.getElementById("msg").innerHTML += "<p>request"+msg+"</p>"
    
80.     };
    ; C! i2 E4 n- S8 h
81. 
    1 z7 l& ]5 `! `9 t: v$ \; E
82.     function close(event){
    3 J, L* B0 T: n# K- L
83.         webSocket.close();
    ' S: d5 f9 t1 n4 c" O" {
84.     }
    % e4 }, t  R0 e+ Q1 b- X: i2 l
85. </script>
    : I; c& I2 T8 y
86. </body>
    & c9 I" u% b, z9 x/ i. ?; r
87. </html>

复制代码

5 y, [" l, z! `, p& H
; c0 q$ c8 q) ~4 O
3 i6 p  C5 C7 V9 k% a