websocket+php socket实现聊天室

　　这两天用了点时间，研究了一下，用php socket+ websocket实现了一个小型的聊天室。我采用的是 select/poll 的同步模型，虽然扛不住很大的并发，但是理论上 维持 几百人在线还是可以的。

目前完成了第一版。这一版的由于采用的是 select/poll 和单进程，所以在win下面就可以运行。不需要额外的其他扩展支持。

 

我最近在 看云 发表了 ThinkPHP5+workerman+layIM打造聊天系统 教程，感兴趣的可以去看看。传送门：ThinkPHP5+workerman+layIM打造聊天系统

ichat v3.0 版本正在和 layim 合作中，详情可以参看 layim.layui.com。现开通了 ichat 线上预览功能。地址 ：ichat v3预览 

 

　　项目的依赖：

　　php版本大于 5.4，浏览器支持 websocket和localstorage。

　　看一下核心的服务端代码吧：

  1 <?php
  2 /**
  3  * author: NickBai
  4  * createTime: 2016/12/9 0009 下午 4:17
  5  */
  6 namespace NickBai;
  7 
  8 class SocketChat
  9 {
 10     private $timeout = 60;  //超时时间
 11     private $handShake = False; //默认未牵手
 12     private $master = 1;  //主进程
 13     private $port = 2000;  //监听端口
 14     private static $connectPool = [];  //连接池
 15     private static $maxConnectNum = 1024; //最大连接数
 16     private static $chatUser = [];  //参与聊天的用户
 17 
 18 
 19     public function __construct( $port = 0 )
 20     {
 21         !empty( $port ) && $this->port = $port;
 22         $this->startServer();
 23     }
 24 
 25     //开始服务器
 26     public function startServer()
 27     {
 28         $this->master = socket_create_listen( $this->port );
 29         if( !$this->master ) throw new \Exception('listen $this->port fail !');
 30 
 31         $this->runLog("Server Started : ".date('Y-m-d H:i:s'));
 32         $this->runLog("Listening on   : 127.0.0.1 port " . $this->port);
 33         $this->runLog("Master socket  : ".$this->master."\n");
 34 
 35         self::$connectPool[] = $this->master;
 36 
 37         while( true ){
 38             $readFds = self::$connectPool;
 39             //阻塞接收客户端链接
 40             @socket_select( $readFds, $writeFds, $e = null, $this->timeout );
 41 
 42             foreach( $readFds as $socket ){
 43                 //当前链接 是主进程
 44                 if( $this->master == $socket ){
 45 
 46                     $client = socket_accept( $this->master );  //接收新的链接
 47                     $this->handShake = False;
 48 
 49                     if ($client < 0){
 50                         $this->log('clinet connect false!');
 51                         continue;
 52                     } else{
 53                         //超过最大连接数
 54                         if( count( self::$connectPool ) > self::$maxConnectNum )
 55                             continue;
 56 
 57                         //加入连接池
 58                         $this->connect( $client );
 59                     }
 60 
 61                 }else{
 62                     //不是主进程,开始接收数据
 63                     $bytes = @socket_recv($socket, $buffer, 2048, 0);
 64                     //未读取到数据
 65                     if( $bytes == 0 ){
 66                         $this->disConnect( $socket );
 67                     }else{
 68                         //未握手 先握手
 69                         if( !$this->handShake ){
 70 
 71                             $this->doHandShake( $socket, $buffer );
 72                         }else{
 73 
 74                             //如果是已经握完手的数据，广播其发送的消息
 75                             $buffer = $this->decode( $buffer );
 76                             $this->parseMessage( $buffer, $socket );
 77                         }
 78                     }
 79 
 80                 }
 81             }
 82 
 83         }
 84     }
 85 
 86     //解析发送的数据
 87     public function parseMessage( $message, $socket )
 88     {
 89         //msg type  1 初始化  2 通知  3 一般聊天  4 断开链接  5 获取在线用户 6 通知下线
 90         $message = json_decode( $message, true );
 91         switch( $message['type'] ){
 92 
 93             case 1:
 94                 $this->bind( $socket, $message );
 95                 //通知其他客户端,当前用户上线
 96                 $msg = [
 97                     'type' => "2",
 98                     'msg' => 'online',
 99                     'avar' => $message['avar']
100                 ];
101                 $this->sendToAll( $socket,  $msg );
102                 //更新在线用户
103                 $this->freshOnlineUser();
104 
105                 break;
106             case 3:
107                 $this->sendToAll( $socket, $message );
108                 break;
109             case 4:
110                 //通知用户离线
111                 $msgOutline = [
112                     'type' => '6',
113                     'user' => self::$chatUser[(int)$socket]['user']
114                 ];
115                 $this->tellOnlineInfo( $msgOutline );
116                 //断开 要离线的用户
117                 $this->disConnect( $socket );
118                 //更新在线用户
119                 $this->freshOnlineUser();
120 
121                 break;
122             default:
123                 break;
124         }
125     }
126 
127     //用户--链接 绑定
128     public function bind( $socket, $user )
129     {
130         self::$chatUser[(int) $socket] = [
131             'user' => $user['user'],
132             'avar' => $user['avar']
133         ];
134     }
135 
136     //用户--链接 解绑
137     public function unBind( $socket )
138     {
139         unset( self::$chatUser[(int) $socket] );
140     }
141 
142     //获取在线用户
143     public function getOnlineUser()
144     {
145         return self::$chatUser;
146     }
147 
148     //更新在线用户
149     public function freshOnlineUser()
150     {
151         $msgOnlie = [
152             'type' => "5",
153             'msg' => 'online user',
154             'info' => self::$chatUser
155         ];
156         $this->tellOnlineInfo( $msgOnlie );
157     }
158 
159     //广播所有的客户端(排除自己和master)
160     public function sendToAll( $client, $mess )
161     {
162         //拼装发送者的名称
163         $mess['user'] = self::$chatUser[(int) $client]['user'];
164         $mess['stime'] = date('Y-m-d H:i:s');
165 
166         foreach( self::$connectPool as $socket ){
167             if( $socket != $this->master && $socket != $client  ){
168                 $this->send( $socket, $mess );
169             }
170         }
171     }
172 
173     //广播客户端在线用户信息
174     public function tellOnlineInfo( $mess )
175     {
176         foreach( self::$connectPool as $socket ){
177             if( $socket != $this->master ){
178                 $this->send( $socket, $mess );
179             }
180         }
181     }
182 
183     //处理发送信息
184    public function send( $client, $msg )
185     {
186         $msg = $this->frame( json_encode( $msg ) );
187         socket_write( $client, $msg, strlen($msg) );
188     }
189 
190     //握手协议
191     function doHandShake($socket, $buffer)
192     {
193         list($resource, $host, $origin, $key) = $this->getHeaders($buffer);
194         $upgrade  = "HTTP/1.1 101 Switching Protocol\r\n" .
195             "Upgrade: websocket\r\n" .
196             "Connection: Upgrade\r\n" .
197             "Sec-WebSocket-Accept: " . $this->calcKey($key) . "\r\n\r\n";  //必须以两个回车结尾
198 
199         socket_write($socket, $upgrade, strlen($upgrade));
200         $this->handShake = true;
201         return true;
202     }
203 
204     //获取请求头
205     function getHeaders( $req )
206     {
207         $r = $h = $o = $key = null;
208         if (preg_match("/GET (.*) HTTP/"              , $req, $match)) { $r = $match[1]; }
209         if (preg_match("/Host: (.*)\r\n/"             , $req, $match)) { $h = $match[1]; }
210         if (preg_match("/Origin: (.*)\r\n/"           , $req, $match)) { $o = $match[1]; }
211         if (preg_match("/Sec-WebSocket-Key: (.*)\r\n/", $req, $match)) { $key = $match[1]; }
212         return [$r, $h, $o, $key];
213     }
214 
215     //验证socket
216     function calcKey( $key )
217     {
218         //基于websocket version 13
219         $accept = base64_encode(sha1($key . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', true));
220         return $accept;
221     }
222 
223 
224     //打包函数 返回帧处理
225     public function frame( $buffer )
226     {
227         $len = strlen($buffer);
228         if ($len <= 125) {
229 
230             return "\x81" . chr($len) . $buffer;
231         } else if ($len <= 65535) {
232 
233             return "\x81" . chr(126) . pack("n", $len) . $buffer;
234         } else {
235 
236             return "\x81" . char(127) . pack("xxxxN", $len) . $buffer;
237         }
238     }
239 
240     //解码 解析数据帧
241     function decode( $buffer )
242     {
243         $len = $masks = $data = $decoded = null;
244         $len = ord($buffer[1]) & 127;
245 
246         if ($len === 126) {
247             $masks = substr($buffer, 4, 4);
248             $data = substr($buffer, 8);
249         }
250         else if ($len === 127) {
251             $masks = substr($buffer, 10, 4);
252             $data = substr($buffer, 14);
253         }
254         else {
255             $masks = substr($buffer, 2, 4);
256             $data = substr($buffer, 6);
257         }
258         for ($index = 0; $index < strlen($data); $index++) {
259             $decoded .= $data[$index] ^ $masks[$index % 4];
260         }
261         return $decoded;
262     }
263 
264     //客户端链接处理函数
265     function connect( $socket )
266     {
267         array_push( self::$connectPool, $socket );
268         $this->runLog("\n" . $socket . " CONNECTED!");
269         $this->runLog(date("Y-n-d H:i:s"));
270     }
271 
272     //客户端断开链接函数
273     function disConnect( $socket )
274     {
275         $index = array_search( $socket, self::$connectPool );
276         socket_close( $socket );
277 
278         $this->unBind( $socket );
279         $this->runLog( $socket . " DISCONNECTED!" );
280         if ($index >= 0){
281             array_splice( self::$connectPool, $index, 1 );
282         }
283     }
284 
285     //打印运行信息
286     public function runLog( $mess = '' )
287     {
288         echo $mess . PHP_EOL;
289     }
290 
291     //系统日志
292     public function log( $mess = '' )
293     {
294         @file_put_contents( './' . date("Y-m-d") . ".log", date('Y-m-d H:i:s') . "  " . $mess . PHP_EOL, FILE_APPEND );
295     }
296 }

　　客户端的代码，篇幅有限，我就不放出了。项目已经放入 本人github，需要了解的请 关注 ：https://github.com/nick-bai/HappyChat

　　看一下页面效果吧：

 

ab并发测试：

 

参考文章：

http://blog.csdn.net/shagoo/article/details/6396089

http://www.cnblogs.com/hustskyking/p/websocket-with-php.html

https://www.web-tinker.com/article/20306.html

posted @ 2016-12-13 14:17  NickBai  阅读(22941)  评论(4)  编辑  收藏  举报