自定义WEB SOCKET 命令解析

EasySwoole为了方便用户使用WEB_SOCKET进行开发，同样封装了对Swoole的sock操作。

定义命令解析

  1. namespace App\Model\WebSock;
  4. use Core\Component\Socket\AbstractInterface\AbstractClient;
  5. use Core\Component\Socket\AbstractInterface\AbstractCommandParser;
  6. use Core\Component\Socket\Common\Command;
  8. class Parser extends AbstractCommandParser
  9. {
  11.     function parser(Command $result, AbstractClient $client, $rawData)
  12.     {
  13.         // TODO: Implement parser() method.
  14.         //这里的解析规则是与客户端匹配的，等会请看客户端代码
  15.         $js = json_decode($rawData,1);
  16.         if(is_array($js)){
  17.             if(isset($js['action'])){
  18.                 $result->setCommand($js['action']);
  19.             }
  20.             if(isset($js['content'])){
  21.                 $result->setMessage($js['content']);
  22.             }
  23.         }
  24.     }
  25. }

定义命令注册类

  2. namespace App\Model\WebSock;
  5. use Core\Component\Socket\AbstractInterface\AbstractCommandRegister;
  6. use Core\Component\Socket\Client\TcpClient;
  7. use Core\Component\Socket\Common\Command;
  8. use Core\Component\Socket\Common\CommandList;
  9. use Core\Swoole\AsyncTaskManager;
  10. use Core\Swoole\Server;
  12. class Register extends AbstractCommandRegister
  13. {
  15.     function register(CommandList $commandList)
  16.     {
  17.         // TODO: Implement register() method.
  18.         $commandList->addCommandHandler('who',function (Command $command,TcpClient $client){
  19.             return 'your fd is '.$client->getFd();
  20.         });
  21.         $commandList->addCommandHandler('sendTo',function (Command $command,TcpClient $client){
  22.             $dest = intval($command->getMessage());
  23.             $info =  Server::getInstance()->getServer()->connection_info($dest);
  24.             if($info['websocket_status']){
  25.                 Server::getInstance()->getServer()->push($dest,'you receive a message from '.$client->getFd());
  26.                 return 'send success';
  27.             }else{
  28.                 return 'fd error';
  29.             }
  30.         });
  31.         $commandList->addCommandHandler('broadcast',function (Command $command){
  32.             /*
  33.                * 注意   本example未引入redis来做fd信息记录，因此每次采用遍历的形式来获取结果，
  34.                * 仅供思路参考，不建议在生产环节使用
  35.              */
  36.             $message = $command->getMessage();
  37.             $list = array();
  38.             foreach (Server::getInstance()->getServer()->connections as $fd){
  39.                 $info =  Server::getInstance()->getServer()->connection_info($fd);
  40.                 if($info['websocket_status']){
  41.                     $list[] = $fd;
  42.                 }
  43.             }
  44.             //广播属于重任务，交给Task执行
  45.             AsyncTaskManager::getInstance()->add(function ()use ($list,$message){
  46.                 foreach ( $list as $fd) {
  47.                     Server::getInstance()->getServer()->push($fd,"this is broadcast :{$message}");
  48.                 }
  49.             });
  50.         });
  51.     }
  52. }

添加定时广播

  1. function onWorkerStart(\swoole_server $server, $workerId)
  2.     {
  3.         // TODO: Implement onWorkerStart() method.
  4.         //如何避免定时器因为进程重启而丢失
  5.         //例如，我第一个进程，添加一个10秒的定时器
  6.         if($workerId == 0){
  7.             Timer::loop(3*1000,function (){
  8.                 /*
  9.                 * 注意   本example未引入redis来做fd信息记录，因此每次采用遍历的形式来获取结果，
  10.                 * 仅供思路参考，不建议在生产环节使用
  11.                  */
  12.                 $list = array();
  13.                 foreach (Server::getInstance()->getServer()->connections as $fd){
  14.                     $info =  Server::getInstance()->getServer()->connection_info($fd);
  15.                     if($info['websocket_status']){
  16.                         $list[] = $fd;
  17.                     }
  18.                 }
  19.                 //广播属于重任务，交给Task执行
  20.                 AsyncTaskManager::getInstance()->add(function ()use ($list){
  21.                     foreach ( $list as $fd) {
  22.                         Server::getInstance()->getServer()->push($fd,"this is tick broadcast ");
  23.                     }
  24.                 });
  25.             });
  26.         }
  27.     }

添加监听

  1. function beforeWorkerStart(\swoole_server $server)
  2.     {
  3.         // TODO: Implement beforeWorkerStart() method.
  4.         $server->on("message",function (\swoole_websocket_server $server, \swoole_websocket_frame $frame){
  5.             Dispatcher::getInstance(Register::class,Parser::class)->dispatchWEBSOCK($frame);
  6.         });
  7.     }

客户端代码

  1. <html>
  2. <head>
  3.     <meta http-equiv="Content-Type" content="text/html; charset=utf-8" />
  4. </head>
  5. <body>
  6. <div>
  7.     <div>
  8.         <p>info below</p>
  9.         <ul  id="line">
  11.         </ul>
  12.     </div>
  13.     <div>
  14.         <select id="action">
  15.             <option value="who">当前fd</option>
  16.             <option value="sendTo">发给指定fd</option>
  17.             <option value="broadcast">广播</option>
  18.         </select>
  19.         <input type="text" id="says">
  20.         <button onclick="say()">发送</button>
  21.     </div>
  22. </div>
  23. </body>
  24. <script src="https://cdnjs.cloudflare.com/ajax/libs/jquery/3.2.1/jquery.js"></script>
  25. <script>
  26.     var wsServer = 'ws://127.0.0.1:9501';
  27.     var websocket = new WebSocket(wsServer);
  28.     window.onload = function () {
  29.         websocket.onopen = function (evt) {
  30.             addLine("Connected to WebSocket server.");
  31.         };
  33.         websocket.onclose = function (evt) {
  34.             addLine("Disconnected");
  35.         };
  37.         websocket.onmessage = function (evt) {
  38.             addLine('Retrieved data from server: ' + evt.data);
  39.         };
  41.         websocket.onerror = function (evt, e) {
  42.             addLine('Error occured: ' + evt.data);
  43.         };
  44.     };
  45.     function addLine(data) {
  46.         $("#line").append("<li>"+data+"</li>");
  47.     }
  48.     function say() {
  49.         var content = $("#says").val();
  50.         var action = $("#action").val();
  51.         $("#says").val('');
  52.         websocket.send(JSON.stringify({
  53.             action:action,
  54.             content:content
  55.         }));
  56.     }
  57. </script>
  58. </html>

建立客户端控制器

  1. namespace App\Controller;
  4. use Core\AbstractInterface\AbstractController;
  6. class Index extends AbstractController
  7. {
  9.     function index()
  10.     {
  11.         // TODO: Implement index() method.
  12.         $this->response()->write(file_get_contents(ROOT."/App/Static/Template/client.html"));
  13.     }
  15.     function onRequest($actionName)
  16.     {
  17.         // TODO: Implement onRequest() method.
  18.     }
  20.     function actionNotFound($actionName = null, $arguments = null)
  21.     {
  22.         // TODO: Implement actionNotFound() method.
  23.         $this->response()->withStatus(404);
  24.     }
  26.     function afterAction()
  27.     {
  28.         // TODO: Implement afterAction() method.
  29.     }
  30. }

注意，请修改配置文件，使得EasySwoole为SERVER_TYPE_WEB_SOCKET模式