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

客户端：

客户端就简单了。一个简单的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==&＃39;null&＃39;){ 
                name = &＃39;咕哒子&＃39;;
            }
        }
 
        $(&＃39;#curuser&＃39;).text(name);
 
         data=&＃39;{"type":"1","user":"&＃39;+name+&＃39;"}&＃39;;
         
        ws.send(data);
    }

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

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

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

function onmessage(e){
        //console.log(e.data);
        var data = eval("("+e.data+")");
        var info=$(&＃39;#chatinfo&＃39;).html();
        if(data.type==1)
            $(&＃39;#chatinfo&＃39;).html(info+&＃39;
&＃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);
        }
    }

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

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

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

客户端，有几个坑 ，

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

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

服务端：

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

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

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

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

$ws_worker->OnWorkerStart=function($ws_worker)
{
    // channel 客户端链接上 服务器
    Channel\Client::connect(&＃39;127.0.0.1&＃39;,2206);
    $event_name=&＃39;私聊&＃39;;
    // 订阅 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(&＃39;code&＃39;=>200, &＃39;msg&＃39;=>&＃39;ok&＃39;, &＃39;data&＃39;=>null,&＃39;type&＃39;=>1);
    // 向客户端发送hello $data
    //print_r($data);
    $data=json_decode($data,true);
    //print_r($data);
    if(!isset($data[&＃39;type&＃39;])||empty($data[&＃39;type&＃39;]))// type 1  2
    {
        $res=array(&＃39;code&＃39;=>301, &＃39;msg&＃39;=>&＃39;消息包格式错误&＃39;, &＃39;data&＃39;=>null);
    }else{
        switch ($data[&＃39;type&＃39;]) {
            case &＃39;1&＃39;: // 客户端上线消息
                //print_r($connection->id);
                 
                if(!isset($data[&＃39;user&＃39;])||empty($data[&＃39;user&＃39;]))
                {
                    $res=array(&＃39;code&＃39;=>301, &＃39;msg&＃39;=>&＃39;消息包格式错误&＃39;, &＃39;data&＃39;=>null);
                    break;
                }
                // 维护一个数组 保存 用户 connection_id => user
 
                $dsn=&＃39;mysql:host=127.0.0.1;dbname=kinmoschat;&＃39;;
                $pdo=new PDO($dsn,&＃39;root&＃39;,&＃39;123456&＃39;);
                //准备SQL语句
                $sql = "INSERT INTO `user`(`connect_id`,`username`) VALUES (:connect_id,:username)";
 
                //调用prepare方法准备查询
                $stmt = $pdo->prepare($sql);
 
                //传递一个数组为预处理查询中的命名参数绑定值，并执行SQL
                $stmt->execute(array(&＃39;:connect_id&＃39; => $connection->id,&＃39;:username&＃39; => $data[&＃39;user&＃39;]));
                //获取最后一个插入数据的ID值
                //echo $pdo->lastInsertId() . &＃39;
&＃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;&＃39;.$value[&＃39;username&＃39;].&＃39;
&＃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=&＃39;mysql:host=127.0.0.1;dbname=kinmoschat;&＃39;;
    $pdo=new PDO($dsn,&＃39;root&＃39;,&＃39;123456&＃39;);
    $sql="delete from user where connect_id=&＃39;".$connection->id."&＃39;";
    //print_r($sql);
    $pdo->exec($sql);
};

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

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