使用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)
            $(&#39;#chatinfo&#39;).html(info+&#39;<br/>&#39;+data.data);
        else if(data.type==2)
        {
            // 在线用户列表 userinfo
            $(&#39;#userinfo&#39;).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[&#39;to_connection_id&#39;];
        $message=$event_data[&#39;content&#39;];
 
        foreach ($ws_worker->connections as $connection) {
 
            if($connection->id==$to_connect_id)
            {
                $connection->send($message);
            }
                 
        }
 
        // if(!isset($ws_worker->connections[$to_connect_id]))
        // {
        //     echo &#39;connect is not exist\n&#39;;
        //     return;
        // }
        // $to_connection=$ws_worker->connections[$to_connect_id];
        // $to_connection->send($message);
    });
 
    // 订阅广播事件
    $event_name = &#39;广播&#39;;
    // 收到广播 向所有客户端发送消息
    Channel\Client::on($event_name,function($event_data)use($ws_worker){
        //print_r($event_data);
        $message=$event_data[&#39;content&#39;];
        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() . &#39;<br />&#39;;
 
                // 向自己推送一条消息
                $res2[&#39;type&#39;]=3;// 系统信息
                $res2[&#39;data&#39;]=array(&#39;userinfo&#39; =>$data[&#39;user&#39;]);// 系统信息
                $connection->send(json_encode($res2));
 
                $msg="用户 ".$data[&#39;user&#39;]." 上线了~~";
                $res[&#39;data&#39;]=$msg;
                break;
            case &#39;2&#39;: // 客户端群发送消息
                if(!isset($data[&#39;user&#39;])||empty($data[&#39;user&#39;])||!isset($data[&#39;msg&#39;])||empty($data[&#39;msg&#39;]))
                {
                    $res=array(&#39;code&#39;=>301, &#39;msg&#39;=>&#39;消息包格式错误&#39;, &#39;data&#39;=>null);
                    break;
                }
                $msg="用户 ".$data[&#39;user&#39;]."说：".$data[&#39;msg&#39;];
                $res[&#39;data&#39;]=$msg;
                break;
            case &#39;3&#39;: // 客户端私聊
                if(!isset($data[&#39;user&#39;])||empty($data[&#39;user&#39;])||!isset($data[&#39;msg&#39;])||empty($data[&#39;msg&#39;])||!isset($data[&#39;friend_id&#39;])||empty($data[&#39;friend_id&#39;]))
                {
                    $res=array(&#39;code&#39;=>301, &#39;msg&#39;=>&#39;消息包格式错误&#39;, &#39;data&#39;=>null);
                    break;
                }
                $msg="用户 ".$data[&#39;user&#39;]."对您说：".$data[&#39;msg&#39;];
                $res[&#39;data&#39;]=$msg;
                $res[&#39;type&#39;]=1;// 聊天消息
                $res1=json_encode($res);
                // 推送给单个用户
                $event_name = &#39;私聊&#39;;
                Channel\Client::publish($event_name, array(
                    &#39;content&#39;          => $res1,
                    &#39;to_connection_id&#39; =>$data[&#39;friend_id&#39;]
                ));
                // 另外还要给自己推条消息
                $msg="您对 ".$data[&#39;friendname&#39;]."说：".$data[&#39;msg&#39;];
                $res[&#39;data&#39;]=$msg;
                $res[&#39;type&#39;]=1;// 聊天消息
                $res2=json_encode($res);
                Channel\Client::publish($event_name, array(
                    &#39;content&#39;          => $res2,
                    &#39;to_connection_id&#39; =>$connection->id
                ));
                return;
                break;
             
            default:
                # code...
                break;
        }
    }
    $res[&#39;type&#39;]=1;// 聊天消息
    $res=json_encode($res);
    // 广播给所有客户端
    $event_name = &#39;广播&#39;;
    Channel\Client::publish($event_name, array(
        &#39;content&#39;          => $res
    ));
 
    $dsn=&#39;mysql:host=127.0.0.1;dbname=kinmoschat;&#39;;
    $dbh=new PDO($dsn,&#39;root&#39;,&#39;123456&#39;);
    $stmt=$dbh->query(&#39;SELECT connect_id,username FROM user&#39;);
    $row=$stmt->fetchAll();
    $uerHtml="";
    foreach ($row as $key => $value) {
 
        $uerHtml.=&#39;<a class="user" onclick="userclick(\&#39;&#39;.$value[&#39;username&#39;].&#39;\&#39;,\&#39;&#39;.$value[&#39;connect_id&#39;].&#39;\&#39;);" value="&#39;.$value[&#39;connect_id&#39;].&#39;" href="javascript:void(0);">&#39;.$value[&#39;username&#39;].&#39;</a><br/>&#39;;
    }
    //print_r($row);
    $res1[&#39;type&#39;]=2;// 用户消息
    $res1[&#39;data&#39;]=$uerHtml;
    $res1=json_encode($res1);
     
 
    $event_name = &#39;广播&#39;;
    Channel\Client::publish($event_name, array(
        &#39;content&#39;          => $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教程栏目。

Atas ialah kandungan terperinci 使用workerman实现在线聊天的方法. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!