workerman is a communication service written in PHP. Previous projects used it as a data interface service

This time I used it to make a simple online chat room~

1. Download the latest version of workerman

You can go http://www.workerman.net Go to download

I have separated service and client into two folders to facilitate management

The general project structure is as follows.

Client:

The client is simple. A simple html code. Embedded a websocket monitoring service

var ws, name, client_list={};
function connect() {
       // 創(chuàng)建websocket
       ws = new WebSocket("ws://192.168.0.88:2345");
       // 當(dāng)socket連接打開時(shí)，輸入用戶名
       ws.onopen = onopen;
       // 當(dāng)有消息時(shí)根據(jù)消息類型顯示不同信息
       ws.onmessage = onmessage;
       ws.onclose = function() {
          console.log("連接關(guān)閉，定時(shí)重連");
          connect();
       };
       ws.onerror = function() {
          console.log("出現(xiàn)錯(cuò)誤");
       };
    }

realizes the opening of websocket, monitoring of messages, and close

1. When a client is opened, a dialog box for entering a name will pop up immediately

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

and push the data to the server. type =1 represents login.

2. When receiving a message, determine the message type, whether it is a group message or a private message. And then process.

In addition, every time a new user logs in, a user list will be pushed to each client. Rendering

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)
        {
            // 在線用戶列表 個(gè)人信息
            name=data.data.userinfo;
            //console.log(data.data);
        }
    }

Then there is the code for each user to send messages. It can be a private chat or a group message

$('#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",'');
    });

The client is almost like this.

The client has several pitfalls,

Pit 1. If the variable name is name, it will not be reset when the web page is refreshed, otherwise it will be reset. (After checking the information, I found that the name variable is window.name. Therefore, the value will not be refreshed when the web page is refreshed)

Pit 2. JS group array, the variable must use "" and the outermost layer is '' For example: data='{"type":"1","user":"' name '"}'; Otherwise, there will be problems with parsing. Can't do it the other way around!

Server:

The server is mainly the worker component and uses the Channel distributed communication component to implement subscription, cluster push, group push and private chat.

First, of course, is monitoring. Enable a worker's websocket monitoring.

// 創(chuàng)建一個(gè)Worker監(jiān)聽2346端口，使用websocket協(xié)議通訊
$ws_worker = new Worker("websocket://0.0.0.0:2345");
$channel_server = new Channel\Server('0.0.0.0', 2206);
// 啟動4個(gè)進(jìn)程對外提供服務(wù)
$ws_worker->count = 4;
$ws_worker->name="kinmoschat";

When the workerman monitoring is enabled, register the channel communication.

$ws_worker->onWorkerStart=function($ws_worker)
{
    // channel 客戶端鏈接上 服務(wù)器
    Channel\Client::connect('127.0.0.1',2206);
    $event_name='私聊';
    // 訂閱 worker-<id 事件，并注冊事件處理函數(shù)
    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;;
    // 收到廣播 向所有客戶端發(fā)送消息
    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);
        }
    });
};

Register two events, a broadcast event, a private chat event, a broadcast for online notification, and a group message. Private chat is private chat. . Here, you can also do group sending. However, this version has not been implemented yet.

Then there is the callback for the client link.

$ws_worker->onConnect=function($connection){
    $connection->id = md5($connection->id."_".time()."_".rand(10000,99999));
};

Here, when the client calls back, I will modify the client's connectid. A simple md5 is mainly to prevent serial IDs from being exploited too easily. .

Then, the main body of the entire project, the processing callback of the server-side message.

For each incoming client, assign a unique ID

Maintain a relationship table with connectid=>user

Because multiple processes are opened, it is saved in the session Invalid, so I plan to save it in the database

When the link is disconnected, delete the data

$ws_worker->onMessage = function($connection, $data)
{
    $res=array('code'=>200, 'msg'=>'ok', 'data'=>null,'type'=>1);
    // 向客戶端發(fā)送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'=>'消息包格式錯(cuò)誤', 'data'=>null);
    }else{
        switch ($data['type']) {
            case '1': // 客戶端上線消息
                //print_r($connection->id);
                 
                if(!isset($data['user'])||empty($data['user']))
                {
                    $res=array('code'=>301, 'msg'=>'消息包格式錯(cuò)誤', 'data'=>null);
                    break;
                }
                // 維護(hù)一個(gè)數(shù)組 保存 用戶 connection_id => user
 
                $dsn='mysql:host=127.0.0.1;dbname=kinmoschat;';
                $pdo=new PDO($dsn,'root','123456');
                //準(zhǔn)備SQL語句
                $sql = "INSERT INTO `user`(`connect_id`,`username`) VALUES (:connect_id,:username)";
 
                //調(diào)用prepare方法準(zhǔn)備查詢
                $stmt = $pdo->prepare($sql);
 
                //傳遞一個(gè)數(shù)組為預(yù)處理查詢中的命名參數(shù)綁定值，并執(zhí)行SQL
                $stmt->execute(array(':connect_id' => $connection->id,':username' => $data['user']));
                //獲取最后一個(gè)插入數(shù)據(jù)的ID值
                //echo $pdo->lastInsertId() . &#39;<br />&#39;;
 
                // 向自己推送一條消息
                $res2[&#39;type&#39;]=3;// 系統(tǒng)信息
                $res2[&#39;data&#39;]=array(&#39;userinfo&#39; =>$data[&#39;user&#39;]);// 系統(tǒng)信息
                $connection->send(json_encode($res2));
 
                $msg="用戶 ".$data[&#39;user&#39;]." 上線了~~";
                $res[&#39;data&#39;]=$msg;
                break;
            case &#39;2&#39;: // 客戶端群發(fā)送消息
                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;消息包格式錯(cuò)誤&#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;消息包格式錯(cuò)誤&#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);
                // 推送給單個(gè)用戶
                $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
    ));
};

The connectid=>name of each user will be stored in the database. .

When an online message is received, it is broadcast to all users.

Received a group message. . Broadcast to all clients.

Received a private message. Then push it individually to yourself and the person who sent it.

Listen to the client closing event. When the client is closed, delete the relevant records in the user table

// 關(guān)閉鏈接 將數(shù)據(jù)庫中的該數(shù)據(jù)刪除
$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);
};

For more workerman knowledge, please pay attention to the workerman tutorial column.

The above is the detailed content of How to implement online chat using Workerman. For more information, please follow other related articles on the PHP Chinese website!