社区分享〡Workerman中利用popen实现多进程Web SSH

<?php

require_once __DIR__ . '/vendor/autoload.php';

date_default_timezone_set('Asia/Shanghai');

//进程池类
class ProcessPool
{
    private array $processes = [];
    public int $maxProcesses;

    public function __construct($maxProcesses = 5)
    {
        $this->maxProcesses = $maxProcesses;
    }

    //添加进程 popen打开的进程执行完会退出，不做复用
    public function add(string $command)
    {
        $process           = popen($command, 'r');
        $this->processes[] = $process;
        return $process;
    }

    // 释放进程
    public function releaseProcess($process): void
    {
        $key = array_search($process, $this->processes);
        if ($key !== false) {
            unset($this->processes[$key]); // 从活动进程池移除
        }
    }

    // 获取池中的所有进程数量
    public function processCount(): int
    {
        return count($this->processes);
    }
}

// 创建一个websocket Worker
$task = new \Workerman\Worker("websocket://0.0.0.0:3232");

// 初始化进程池
$processPool = new ProcessPool(5);

// 添加定时器，每秒打印进程数
$task->onWorkerStart = function (\Workerman\Worker $worker) use ($processPool) {
    \Workerman\Timer::add(1, function () use ($processPool) {
        var_dump('[' . date('H:i:s') . '] 活动进程数：' . $processPool->processCount());
    });
};

$task->onMessage = function (Workerman\Connection\TcpConnection $connection, $str) use ($processPool) {
    $data = json_decode($str, JSON_OBJECT_AS_ARRAY);
    $command = $data['command'] ?? '';
    if (empty($command)) {
        return $connection->send('无效命令');
    }

    //超过最大进程数时阻塞等待退出
    if ($processPool->processCount() >= $processPool->maxProcesses) {
        return $connection->send('使用进程数已达最大数，等待中...');
    }

    // 获取一个进程资源
    $process = $processPool->add($command);
    if (is_resource($process)) {

        // 创建读事件
        \Workerman\Worker::$globalEvent->onReadable($process, function ($pipe) use ($connection, $process, $processPool) {
            // 读取进程输出
            $output = fread($pipe, 8192);
            if ($output === false || feof($pipe)) {
                // 进程结束，释放资源
                fclose($pipe);
                $processPool->releaseProcess($pipe);
                $connection->send("进程结束.");
                \Workerman\Worker::$globalEvent->offReadable($process);
            } else {
                // 将输出发送给客户端
                $connection->send($output);
            }
        });

        return true;

    } else {
        return $connection->send("服务繁忙，请稍后重试.");
    }
};

\Workerman\Worker::runAll();

<?php

for ($i = 0; $i < 5; $i++) {
    sleep(1);
    echo '测试' . $i . PHP_EOL;
}