使用workerman实现在线聊天的方法

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

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

1.下载最新版本的workerman

我这里将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);

};

以上就是使用workerman实现在线聊天的方法的详细内容

以上内容希望帮助到大家，很多PHPer在进阶的时候总会遇到一些问题和瓶颈，业务代码写多了没有方向感，不知道该从那里入手去提升，对此我整理了一些资料，包括但不限于：分布式架构、高可扩展、高性能、高并发、服务器性能调优、TP6，laravel，YII2，Redis，Swoole、Swoft、Kafka、Mysql优化、shell脚本、Docker、微服务、Nginx等多个知识点高级进阶干货需要的可以免费分享给大家，需要的可以加入我的官方群点击此处。

八重樱。

发布了189 篇原创文章 · 获赞 12 · 访问量 3774

私信关注

使用workerman实现在线聊天的方法

猜你喜欢