Swoole Tcp 学习

最近一直在学习Swoole,刚好有个老项目的一小部分(一个脚本)有用到了Tcp 协议,借此机会重构一下。

场景描述:

该脚本的作用用一句话就可以概述:将本地数据源推送给另外一台服务器。

原始的处理方式,不合理的地方有以下几点:

  1. 目标服务器需要开放指定端口,这会导致目标服务器向外暴露,不安全。
  2. 如果有多台目标服务器,这会导致频繁需要修改源码,脚本维护起来不方便。

重构

重构需要解决的问题有如下:

  1. 当客户端连接成功后,才会向该客户端推送数据。
  2. 当客户端断开连接时,停止向该客户端推送数据。
  3. 允许多个客户端同时连接。
  4. 因为数据源是不间断的,理论上只要客户端的连接不主动断开,服务端的数据推送就不会主动停止。

最终使用Swoole 的Tcp + Process 实现了以上需求,核心代码如下:

on("Connect", function ($server, $fd) {
    global $workers;
    
    // 创建子进程
    $process = new swoole_process(function (swoole_process $worker) use ($server, $fd) {
        echo "Client Connect" . PHP_EOL;

      // todo 业务逻辑
      ...
       
      // 向客户端推送消息   
      $server->send($fd, $str);
      
    }, true, 0, false);
    
    // 启动子进程
    $pid = $process->start();
    
    array_push($workers, ["pid" => $pid, "fd" => $fd]);
});

/**
 * 监听数据接收事件
 */
$server->on("Receive", function ($server, $fd, $from_id, $data){
    $server->send($fd, "Server: " . $data);
});

/**
 * 监听连接关闭事件
 */
$server->on("Close", function ($server, $fd) {
    global $workers;
    
    foreach ($workers as $worker) {
      if ($worker['fd'] === $fd){
        // 检查子进程是否存在
          if (Process::kill($worker['pid'], 0)){
              array_shift($worker);
              // 通过信号终止子进程
              Process::kill($worker['pid'], SIGKILL);
          }
      }
    }
    echo "Client Close" . PHP_EOL;
});

// 启动TCP 服务器
$server->start();

其实实现的原理很简单,利用Swoole 的基于事件的 Tcp 异步编程,当有客户端连接时,就创建一个子进程进行推送数据,但客户端连接断开时,就通过信号结束该客户端对应的子进程。

你可能感兴趣的