Workerman创建聊天室实例

// 标记是全局启动
define('GLOBAL_START', 1);

require_once __DIR__ . '/Workerman/Connection.php';
require_once __DIR__ . '/Workerman/Autoloader.php';

use Workerman\Worker;
use Workerman\Lib\Timer;

// 心跳间隔25秒
define('HEARTBEAT_TIME', 25);

// 设置时区
date_default_timezone_set('PRC');

// 以websocket协议为例
// Create a Websocket server
$ws = new Worker("websocket://118.xxx.xxx.xx:4980");

// 启动4个进程对外提供服务
$ws->count = 4;

// 已连接客户端 ，便于统计在线用户
$ws->hasConnections = array();

// 当新客户端连上来时分配uid，并保存连接，并通知所有客户端
$ws->onConnect = function($connection)
{
    global $ws;
    // 为这个链接分配一个uid
    $connection->uid = ++$global_uid;

    foreach ($ws->connections as $conn) {
        $conn->send("用户{$connection->uid} 已上线");
    }
    $connection->close(); // 关闭socket连接
    // echo "New connection\n";
 };

// 当客户端发送消息过来时，转发给所有人 (聊天主要使用的功能)
$ws->onMessage = function($connection, $message)
{
    global $ws,$db;
    $data = json_decode($message, true);

    $data['time'] = date('Y-m-d H:i:s');
    $insert = array(
        'uid'         =>$data['uid'],
        'avatar'      =>$data['avatar'],
        'name'        =>$data['name'],
        'type'        =>$data['type'],
        'content'     =>$data['content'],
        'img_path'    =>$data['img_path'],
        'voice_path'  =>$data['voice_path'],
        'topic_id'    =>$data['topic_id'],
        'is_question' =>$data['is_question'],
        'status'      =>$data['status'],
        'time'        =>$data['time']
    );

    switch ($data['type']) {
        case 'save':
            // 把房间和用户信息保存下来
            $ws->hasConnections[$connection->id] = array('name' => $data['name'], 'uid' => $data['uid'], 'topic_id' => $data['topic_id'], 'avatar' => $data['avatar']);
            sendMessage($data, $data['topic_id']);
            // $back_data = array('content' => $content, 'client_id' => $connection->id, 'client_name' => $data['name'], 'type' => 'login', 'clients' => $ws->hasConnections, 'time' => date('Y-m-d H:i:s'));

            break;

        case 'text':
            sendMessage($data, $data['topic_id']);
            $insert_id = $db->insert('chat_record')->cols($insert)->query();  // 存数据库 save data
            break;

        case 'img':
            sendMessage($data, $data['topic_id']);
            $insert_id = $db->insert('chat_record')->cols($insert)->query();  // 存数据库 save data
            break;

        case 'voice':
            sendMessage($data, $data['topic_id']);
            $insert_id = $db->insert('chat_record')->cols($insert)->query();  // 存数据库 save data
            break;
        default:
            break;
    }

    // $connection->close(); // 关闭socket连接
};

// 当客户端断开时，广播给所有客户端
// $ws->onClose = function($connection)
// {
//     global $ws;
//     foreach ($ws->connections as $conn) {
//         $conn->send("用户[{$connection->uid}] 消消走了");
//     }
//     $connection->close(); // 关闭socket连接
// };

// 给房间的每个人发消息
function sendMessage($data, $topic_id){
    global $ws;
    $data = json_encode($data);
    foreach ($ws->connections as $id => $conn) {
        if ($ws->hasConnections[$id]['topic_id'] == $topic_id) {
            $conn->send($data);
            // $conn->send($ws->hasConnections[$id]);
        }
    }
}

// 进程启动后设置一个每秒运行一次的定时器(心跳, 保持长连接)
$ws->onWorkerStart = function($ws) {
    global $db;
    // ip, 端口号, 账号, 密码, 库名
    $db = new \Workerman\MySQL\Connection('127.0.0.1', '3306', 'username', 'password', 'database');

    Timer::add(1, function()use($ws){
    // global $ws;
        $arr = array('type'=>'keep','content'=>'ping');
        $arr = json_encode($arr);
        foreach($ws->connections as $connection) {
            $connection->send($arr);
        }
    });
};

// Run worker
Worker::runAll();