使用workerman实现在线聊天-第一版

kinmos

workerman 是一个php编写的通讯服务。之前的项目都是用它做数据接口服务

这次用它做一个简单的在线聊天室~

 

1.下载最新版本的workerman

可以去  http://www.workerman.net 去下载

我这里将service 和 client 分开了两个文件夹，方便管理

大致的项目结构如下。

 

客户端：

客户端就简单了。一个简单的html代码。嵌入了一个 websocket 监听服务

 
var ws, name, client_list={};
function connect() {
       // 创建websocket
       ws = new WebSocket("ws://192.168.0.88:2345");
       // 当socket连接打开时，输入用户名
       ws.onopen = onopen;
       // 当有消息时根据消息类型显示不同信息
       ws.onmessage = onmessage; 
       ws.onclose = function() {
    	  console.log("连接关闭，定时重连");
          connect();
       };
       ws.onerror = function() {
     	  console.log("出现错误");
       };
    }

　　

实现websocket的 打开，message的监听，以及close 

1、当打开一个客户端，则立马弹出一个输入姓名的对话框

 

function onopen(){
        //console.log(name);
        //var username=connect_id="";
        if(!name)
        {
            name=prompt("请输入您的名字","");
            if(!name || name==\'null\'){  
                name = \'咕哒子\';
            }
        }

        $(\'#curuser\').text(name);

         data=\'{"type":"1","user":"\'+name+\'"}\';
        
        ws.send(data);
    }

 并将数据推送给服务端。type =1 代表登陆。

2、当收到消息时，判断消息类型，是群发消息 还是私聊消息。进而处理。

另外，每次用户有新用户登陆上来，都会 给各个客户端推送，用户列表。进行渲染

function onmessage(e){
        //console.log(e.data);
        var data = eval("("+e.data+")");
        var info=$(\'#chatinfo\').html();
        if(data.type==1)
            $(\'#chatinfo\').html(info+\'<br/>\'+data.data);
        else if(data.type==2)
        {
            // 在线用户列表 userinfo
            $(\'#userinfo\').html(data.data);
        }
        else if(data.type==3)
        {
            // 在线用户列表 个人信息
            name=data.data.userinfo;
            //console.log(data.data);
        }
    }

　　

 然后另外就是 每个用户发送消息的代码了。可以是私聊 ，也可以是群发

$(\'#send\').click(function(e){
        var msg=$(\'#msg\').val();
        var tofriend=$(\'#tofriend\').val();
        var tofriendname=$(\'#tofriendname\').val();
        if(tofriend!="")
        {
            data=\'{"type":"3","user":"\'+name+\'","msg":"\'+msg+\'","friend_id":"\'+tofriend+\'","friendname":"\'+tofriendname+\'"}\';
        }else{
            data=\'{"type":"2","user":"\'+name+\'","msg":"\'+msg+\'"}\';
        }
        ws.send(data);
        $(\'#msg\').attr("value",\'\');
    });

　　

客户端差不多就是这样的了。

客户端，有几个坑 ，

坑1、变量名是 name  则刷新网页不会被重置，否则就会被重置。（后面查资料发现，这个name变量  是 window.name 。所以刷新网页 该值也不会被刷新掉）

坑2、js组数组，变量要用"" 最外层为\'\' 如：data=\'{"type":"1","user":"\'+name+\'"}\';  否则解析出问题。不能倒过来！

 

服务端：

服务端主要是workerman 组件 以及 使用 Channel分布式通讯组件 实现订阅 和集群推送 分组推送 以及私聊。

 

首先，当然是监听，启用一个worker的websocket监听

// 创建一个Worker监听2346端口，使用websocket协议通讯
$ws_worker = new Worker("websocket://0.0.0.0:2345");
　　$channel_server = new Channel\Server(\'0.0.0.0\', 2206);
// 启动4个进程对外提供服务
$ws_worker->count = 4;

$ws_worker->name="kinmoschat";

　

在workerman 监听启用的时候，进行 channel通讯的注册。

$ws_worker->onWorkerStart=function($ws_worker)
{
    // channel 客户端链接上 服务器
    Channel\Client::connect(\'127.0.0.1\',2206);
    $event_name=\'私聊\';
    // 订阅 worker-<id 事件，并注册事件处理函数
    Channel\Client::on($event_name,function($event_data)use($ws_worker){

        //print_r($event_data);
        //print_r($ws_worker->connections);
        $to_connect_id=$event_data[\'to_connection_id\'];
        $message=$event_data[\'content\'];

        foreach ($ws_worker->connections as $connection) {

            if($connection->id==$to_connect_id)
            {
                $connection->send($message);
            }
                
        }

        // if(!isset($ws_worker->connections[$to_connect_id]))
        // {
        //     echo \'connect is not exist\n\';
        //     return;
        // }
        // $to_connection=$ws_worker->connections[$to_connect_id];
        // $to_connection->send($message);
    });

    // 订阅广播事件
    $event_name = \'广播\';
    // 收到广播 向所有客户端发送消息
    Channel\Client::on($event_name,function($event_data)use($ws_worker){
        //print_r($event_data);
        $message=$event_data[\'content\'];
        foreach ($ws_worker->connections as $connection) {
            $connection->send($message);
        }
    });
};

　　

注册两个事件，一个广播事件，一个私聊事件，用以上线通知的广播，以及群发消息。私聊 就是私聊了。。这里，还可以做 分组的群发。不过，这个版本还未实现。

 

然后是针对，客户端链接的回调。

$ws_worker->onConnect=function($connection){
    $connection->id = md5($connection->id."_".time()."_".rand(10000,99999));
};

　　这里，客户端回调，我会将客户端的  connectid修改掉。一个简单的md5 主要是为了防止 流水id太容易被利用吧。。

 

然后，整个项目的主体，服务端消息的处理回调。

针对每个进来的客户端，分配一个唯一 id
维护一个 connectid=>user 的关系表
由于开启了多个进程导致 存到 session中无效，故而 打算存到 数据库中
断开链接的时候，删除数据

$ws_worker->onMessage = function($connection, $data)
{
    $res=array(\'code\'=>200, \'msg\'=>\'ok\', \'data\'=>null,\'type\'=>1);
    // 向客户端发送hello $data
    //print_r($data);
    $data=json_decode($data,true);
    //print_r($data);
    if(!isset($data[\'type\'])||empty($data[\'type\']))// type 1  2
    {
        $res=array(\'code\'=>301, \'msg\'=>\'消息包格式错误\', \'data\'=>null);
    }else{
        switch ($data[\'type\']) {
            case \'1\': // 客户端上线消息
                //print_r($connection->id);
                
                if(!isset($data[\'user\'])||empty($data[\'user\']))
                {
                    $res=array(\'code\'=>301, \'msg\'=>\'消息包格式错误\', \'data\'=>null);
                    break;
                }
                // 维护一个数组 保存 用户 connection_id => user

                $dsn=\'mysql:host=127.0.0.1;dbname=kinmoschat;\';
                $pdo=new PDO($dsn,\'root\',\'123456\');
                //准备SQL语句
                $sql = "INSERT INTO `user`(`connect_id`,`username`) VALUES (:connect_id,:username)";

                //调用prepare方法准备查询
                $stmt = $pdo->prepare($sql);

                //传递一个数组为预处理查询中的命名参数绑定值，并执行SQL
                $stmt->execute(array(\':connect_id\' => $connection->id,\':username\' => $data[\'user\']));
                //获取最后一个插入数据的ID值
                //echo $pdo->lastInsertId() . \'<br />\';

                // 向自己推送一条消息
                $res2[\'type\']=3;// 系统信息
                $res2[\'data\']=array(\'userinfo\' =>$data[\'user\']);// 系统信息
                $connection->send(json_encode($res2));

                $msg="用户 ".$data[\'user\']." 上线了~~";
                $res[\'data\']=$msg;
                break;
            case \'2\': // 客户端群发送消息
                if(!isset($data[\'user\'])||empty($data[\'user\'])||!isset($data[\'msg\'])||empty($data[\'msg\']))
                {
                    $res=array(\'code\'=>301, \'msg\'=>\'消息包格式错误\', \'data\'=>null);
                    break;
                }
                $msg="用户 ".$data[\'user\']."说：".$data[\'msg\'];
                $res[\'data\']=$msg;
                break;
            case \'3\': // 客户端私聊
                if(!isset($data[\'user\'])||empty($data[\'user\'])||!isset($data[\'msg\'])||empty($data[\'msg\'])||!isset($data[\'friend_id\'])||empty($data[\'friend_id\']))
                {
                    $res=array(\'code\'=>301, \'msg\'=>\'消息包格式错误\', \'data\'=>null);
                    break;
                }
                $msg="用户 ".$data[\'user\']."对您说：".$data[\'msg\'];
                $res[\'data\']=$msg;
                $res[\'type\']=1;// 聊天消息
                $res1=json_encode($res);
                // 推送给单个用户
                $event_name = \'私聊\';
                Channel\Client::publish($event_name, array(
                    \'content\'          => $res1,
                    \'to_connection_id\' =>$data[\'friend_id\']
                ));
                // 另外还要给自己推条消息
                $msg="您对 ".$data[\'friendname\']."说：".$data[\'msg\'];
                $res[\'data\']=$msg;
                $res[\'type\']=1;// 聊天消息
                $res2=json_encode($res);
                Channel\Client::publish($event_name, array(
                    \'content\'          => $res2,
                    \'to_connection_id\' =>$connection->id
                ));
                return;
                break;
            
            default:
                # code...
                break;
        }
    }
    $res[\'type\']=1;// 聊天消息
    $res=json_encode($res);
    // 广播给所有客户端
    $event_name = \'广播\';
    Channel\Client::publish($event_name, array(
        \'content\'          => $res
    ));

    $dsn=\'mysql:host=127.0.0.1;dbname=kinmoschat;\';
    $dbh=new PDO($dsn,\'root\',\'123456\');
    $stmt=$dbh->query(\'SELECT connect_id,username FROM user\');
    $row=$stmt->fetchAll();
    $uerHtml="";
    foreach ($row as $key => $value) {

        $uerHtml.=\'<a class="user" onclick="userclick(\\'\'.$value[\'username\'].\'\\',\\'\'.$value[\'connect_id\'].\'\\');" value="\'.$value[\'connect_id\'].\'" href="javascript:void(0);">\'.$value[\'username\'].\'</a><br/>\';
    }
    //print_r($row);
    $res1[\'type\']=2;// 用户消息
    $res1[\'data\']=$uerHtml;
    $res1=json_encode($res1);
    

    $event_name = \'广播\';
    Channel\Client::publish($event_name, array(
        \'content\'          => $res1
    ));
};

　　

这里会将每个用户的  connectid=>name 存入数据库。。

当收到一条 上线消息的时候，广播给所有用户。

收到一条群发消息。。广播给所有客户端。

收到一条私聊消息。则单个推送给自己以及发送的人。

 

监听 客户端关闭事件，当客户端关闭，删除用户表相关记录

// 关闭链接 将数据库中的该数据删除
$ws_worker->onClose=function($connection)
{
    //echo 3233;
    $dsn=\'mysql:host=127.0.0.1;dbname=kinmoschat;\';
    $pdo=new PDO($dsn,\'root\',\'123456\');
    $sql="delete from user where connect_id=\'".$connection->id."\'";
    //print_r($sql);
    $pdo->exec($sql);
};

　　

简单版 demo  http://chat.kinmos.cn/

分类：

技术点：

相关文章：

  • 2021-10-20
  • 2021-11-21
  • 2021-11-07
  • 2019-12-05
  • 2021-10-20
  • 2021-04-26
猜你喜欢
  • 2021-06-23
  • 2021-12-02
  • 2021-12-30
  • 2021-05-14
  • 2021-10-25
  • 2021-11-07
  • 2021-10-20
相关资源
相似解决方案
热门标签
Java Python linux javascript Mysql C# Docker 算法 前端 SpringBoot Redis Vue spring 设计模式 .net core .net kubernetes c++ 数据库 数据结构 大数据 js 机器学习 微服务 Android Go 程序员 面试 JVM ASP.net core 云原生 人工智能 后端 PHP git CSS golang k8s Nginx Django mybatis 深度学习 多线程 React 架构 devops 爬虫 云计算 Spring Boot LeetCode