workerman 从服务器端向用户端推送消息代码

àì夳堔傛蜴生んèń 2022-04-22 13:34 389阅读 0赞

数据推送

<?php
    
    $client = stream_socket_client('tcp://127.0.0.1:5678', $errno, $errmsg, 1);
    // 推送的数据，包含用户，表示是给这个用户推送
    $data = array('uid'=>'','group'=>'admin', 'message'=>'发送成功啦');
    // 发送数据，注意5678端口是Text协议的端口，Text协议需要在数据末尾加上换行符
    fwrite($client, json_encode($data)."\n");
    // 读取推送结果
    echo fread($client, 8192);

接收服务器端推送的数据并将数据发送到客户端

<?php
    use Workerman\Worker;
    require_once 'Workerman\Autoloader.php';
    require_once 'Workerman\Lib\Timer.php';
    
    
    $worker = new Worker('websocket://127.0.0.1:1234');
    $worker->count = 1;
    // worker进程启动后建立一个内部通讯端口
    $worker->onWorkerStart = function($worker)
    {
        // 开启一个内部端口，方便内部系统推送数据，Text协议格式 文本+换行符
        $inner_text_worker = new Worker('Text://127.0.0.1:5678');
        $inner_text_worker->onMessage = function($connection, $buffer)
        {
            global $worker;
            // $data数组格式，里面有uid，表示向那个uid的页面推送数据
            $data = json_decode($buffer, true);
            $uid = $data['uid'];
            // 通过workerman，向uid的页面推送数据
            $ret = sendMessageByUid($uid, $buffer);
            // 返回推送结果
            $connection->send($ret ? 'ok' : 'fail');
        };
        $inner_text_worker->listen();
    };
    // 新增加一个属性，用来保存uid到connection的映射
    $worker->uidConnections = array();
    // 当有客户端发来消息时执行的回调函数
    $worker->onMessage = function($connection, $data)use($worker)
    {
        $connection->lastMessageTime = time();
        // 判断当前客户端是否已经验证,既是否设置了uid
        if(!isset($connection->uid))
        {
            // 没验证的话把第一个包当做uid（这里为了方便演示，没做真正的验证）
            $connection->uid = $data;
            /* 保存uid到connection的映射，这样可以方便的通过uid查找connection，
             * 实现针对特定uid推送数据
             */
            $worker->uidConnections[$connection->uid] = $connection;
            return;
        }
    };
    
    // 当有客户端连接断开时
    $worker->onClose = function($connection)use($worker)
    {
        global $worker;
        if(isset($connection->uid))
        {
            // 连接断开时删除映射
            unset($worker->uidConnections[$connection->uid]);
        }
    };
    
    // 向所有验证的用户推送数据
    function broadcast($message)
    {
        global $worker;
        foreach($worker->uidConnections as $connection)
        {
            $connection->send($message);
        }
    }
    
    // 针对用户推送数据
    function sendMessageByUid($uid, $message)
    {
        global $worker;
        var_dump($worker->uidConnections);
        if(isset($worker->uidConnections[$uid]))
        {
            $connection = $worker->uidConnections[$uid];
            $connection->send($message);
            return true;
        }
        return false;
    }
    
    // 运行所有的worker
    Worker::runAll();

客户端

<!DOCTYPE html>
    <html lang="en">
    <head>
        <meta charset="UTF-8">
        <title>Title</title>
    </head>
    <body>
    <script>
        var ws = new WebSocket('ws://127.0.0.1:1234');
        ws.onopen = function(){
            var uid = (new Date()).getTime();
            var data = {'group':'admin','uid':'uid2'};
            ws.send(JSON.stringify(data));
        };
        ws.onmessage = function(e){
            if (e.data!=''){
                alert(JSON.parse(e.data));
            }
        };
    </script>
    </body>
    </html>