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

Posted 2020-09-28

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了使用workerman实现在线聊天-第一版相关的知识，希望对你有一定的参考价值。

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/

以上是关于使用workerman实现在线聊天-第一版的主要内容，如果未能解决你的问题，请参考以下文章

F4 workerman 长连接绑定用户id实现一对一客服聊天

使用workerman搭建即时聊天

试着用workerman开发一个在线聊天应用

PHP怎样开发聊天室

使用nssm部署windows服务启动应用

F8 workerman 长连接下聊天页面展示对方在线状态