golang websocket绑定用户_websocket+php socket实现聊天室

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

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

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

ichat v3.0 版本正在和 layim 合作中，详情可以参看 http://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 Protocolrn" .
195 "Upgrade: websocketrn" .
196 "Connection: Upgradern" .
197 "Sec-WebSocket-Accept: " . $this->calcKey($key) . "rnrn"; //必须以两个回车结尾
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: (.*)rn/" , $req, $match)) { $h = $match[1]; }
210 if (preg_match("/Origin: (.*)rn/" , $req, $match)) { $o = $match[1]; }
211 if (preg_match("/Sec-WebSocket-Key: (.*)rn/", $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并发测试：

作者： NickBai

出处：https://www.cnblogs.com/nickbai/articles/6169745.html

版权：本文采用「署名-非商业性使用-相同方式共享 4.0 国际」知识共享许可协议进行许可。