使用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);
};

登录后复制

更多workerman知识请关注workerman教程栏目。

以上就是使用workerman实现在线聊天的方法的详细内容，更多请关注Work网其它相关文章！