用swoole和websocket开发简单聊天室
首先,我想说下写代码的一些习惯,第一,任何可配置的参数或变量都要写到一个config文件中。第二,代码中一定要有日志记录和完善的报错并记录报错。言归正传,swoole应该是每个phper必须要了解的,它号称重新定义了php。此聊天室利用了swoole高并发并且异步非阻塞的特点提高了程序的性能。
首先,定义一个 swoole_lock 和 swoole_websocket_server ,并且配置参数,具体参数详情可以去swoole官网查看。
class WebSocketServer {
private $server;
private $addr = '';
private $port = '';
private $users = array(); //保存连接的用户,fd=>nickname的形式保存
private $lock;
public $time;
public function __construct($addr, $port)
{
$this->addr = $addr;
$this->port = $port;
$this->time = 1;
}
public function start()
{
$this->lock = new swoole_lock(SWOOLE_MUTEX);
$this->server = new swoole_websocket_server($this->addr, $this->port);
$this->server->set(array(
'daemonize' => 0,
'worker_num' => 4,
'task_worker_num' => 10,
'max_request' => 1000,
'log_file' => ROOT_PATH . 'storage\\logs\\swoole.log'
));
$this->server->on('open', array($this, 'onOpen'));
$this->server->on('message', array($this, 'onMessage'));
$this->server->on('task', array($this, 'onTask'));
$this->server->on('finish', array($this, 'onFinish'));
$this->server->on('close', array($this, 'onClose'));
$this->server->start();
}
public function onOpen($server, $request)
{
$message = array(
'remote_addr' => $request->server['remote_addr'],
'request_time' => date('Y-m-d H:i:s', $request->server['request_time'])
);
write_log($message);
}
public function onMessage($server, $frame)
{
$data = json_decode($frame->data);
//这里进来用users把全部用户fd做key扔进数组里面后面返回信息,后面每次触发通过fd来做用户标识,每个进程都是独立的,直接不能相互获取,只能放到其他地方。例如文件,或者缓存
switch ($data->type) {
case 'init':
case 'INIT':
$this->users[$frame->fd] = $data->message;
$message = '欢迎' . $data->message . '加入了聊天室';
$response = array(
'type' => 1, // 1代表系统消息,2代表用户聊天
'message' => $message
);
break;
case 'chat':
case 'CHAT':
$this->time++;
echo $this->time; //每个进程都是独立的,直接不能相互获取,只能放到其他地方。例如文件,或者缓存
$message = $data->message;
$response = array(
'type' => 2, // 1代表系统消息,2代表用户聊天
'username' => $this->users[$frame->fd],
'message' => $message
);
break;
default:
return false;
}
$this->server->task($response);
}
public function onTask($server, $task_id, $from_id, $message)
{
foreach ($this->server->connections as $fd) {
$this->server->push($fd, json_encode($message));
}
$server->finish( 'Task' . $task_id . 'Finished' . PHP_EOL);
}
public function onFinish($server, $task_id, $data)
{
write_log( $data );
}
public function onClose($server, $fd)
{
$username = $this->users[$fd];
// 释放客户端,利用锁进行同步
$this->lock->lock();
unset($this->users[$fd]);
$this->lock->unlock();
if( $username ) {
$response = array(
'type' => 1, // 1代表系统消息,2代表用户聊天
'message' => $username . '离开了聊天室'
);
$this->server->task($response);
}
write_log( $fd . ' disconnected');
}
}
服务端完了,下面就是客户端,很简单,只需要用websocket链接就ok!
// websocket
let address = 'ws://<?php echo CLIENT_CONNECT_ADDR . ':' . CLIENT_CONNECT_PORT ?>';
address = 'ws://127.0.0.1:9601';
let webSocket = new WebSocket(address);
webSocket.onerror = function (event) {
alert('服务器连接错误,请稍后重试');
};
webSocket.onopen = function (event) {
if(!sessionStorage.getItem('username')) {
setName();
}else {
username = sessionStorage.getItem('username')
webSocket.send(JSON.stringify({
'message': username,
'type': 'init'
}));
}
};
webSocket.onmessage = function (event) {
console.log(event);
let data = JSON.parse(event.data);
if (data.type == 1) {
$('#chat-list2').append('<li class="ui-border-tb"><span class="username">系统消息:</span><span class="message">' + data.message + '</span></li>');
} else if (data.type == 2) {
$('#chat-list2').append('<li class="ui-border-tb"><span class="username">' + data.username + ':</span><span class="message">' + data.message + '</span></li>');
}
};
webSocket.onclose = function (event) {
alert('散了吧,服务器都关了');
};
详细代码可以去我的github下载