thinkphp5+workerman实现简单聊天室

440 阅读1分钟

本文已参与新人创作礼活动,一起开启掘金创作指路。

thinkphp5+workerman实现简单聊天室

1.服务器搭建。 项目根目录创建server.php。注:项目根目录,tp5需要加载文件,index/Worker (注意命名空间)

<?php
define('APP_PATH', __DIR__ . '/application/');
define('BIND_MODULE','index/Worker');
// 加载框架引导文件
require __DIR__ . '/thinkphp/start.php';

2.worker控制器代码 采用websocket模式,0.0.0.0代表所有人可以访问,如果只限本地访问可用127.0.0.1。端口号自行配置,记得安全组开放端口。

<?php

namespace app\index\controller;

use think\worker\Server;
use Workerman\Worker as Work;
class Worker extends Server
{
    protected $socket = 'websocket://0.0.0.0:2346';
    protected $processes = 1;
    protected $uidConnections = array();
    static $count  = 0;

/**
 * 收到信息
 * @param $connection
 * @param $data
 */
public function onMessage($connection, $data)
{
    // 判断当前客户端是否已经验证,既是否设置了uid,设置了直接发送消息,未设置第一次为设置uid
    if(!isset($connection->uid))
    {
        // 没验证的话把第一个包当做uid(这里为了方便演示,没做真正的验证)
        $connection->uid = $data;
        /* 保存uid到connection的映射,这样可以方便的通过uid查找connection,
         * 实现针对特定uid推送数据
         */
        $this->uidConnections[$connection->uid] = $connection;
        $data = '用户 '.$connection->uid.'加入房间 ||'.self::$count;
        $this->broadcast($data);
        return;
    }else{
        $data = '用户 '.$connection->uid.':'.$data.'||'.self::$count;
        $this->broadcast($data);
    }
}

/**
 * 当连接建立时触发的回调函数
 * @param $connection
 */
public function onConnect($connection)
{
        self::$count++;

    }

/**
 * 当连接断开时触发的回调函数
 * @param $connection
 */
public function onClose($connection)
{
    self::$count--;
    if(isset($connection->uid))
    {
        // 连接断开时删除映射
        $data = '用户 '.$connection->uid.'退出房间 ||'.self::$count;
        $this->broadcast($data);
        unset($this->uidConnections[$connection->uid]);
    }
}

/**
 * 当客户端的连接上发生错误时触发
 * @param $connection
 * @param $code
 * @param $msg
 */
public function onError($connection, $code, $msg)
{
    echo "error $code $msg\n";
}

/**
 * 每个进程启动
 * @param $worker
 */
public function onWorkerStart($socket)
{

}

// 向所有验证的用户推送数据
    function broadcast($message)
    {
        foreach($this->uidConnections as $connection)
        {
            $connection->send($message);
        }
    }

// 针对uid推送数据
    function sendMessageByUid($uid, $message)
    {
        if(isset($this->uidConnections[$uid]))
        {
            $connection = $this->uidConnections[$uid];
            $connection->send($message);
            return true;
        }
        return false;
    }
}

3.脚本文件运行 cd进入根目录使用cli模式 php Server.php start 运行脚本文件 在这里插入图片描述 这时脚本已经开始运行了。 4.前端执行 在这里插入图片描述 4.运行index文件 在这里插入图片描述