PHP多进程用例--swoole和pcntl

<?php

class MyMulProcess {
/**
* 进程列表
* @var array
*/
private $_works = array();

/**
* 对应CPU核心
* @var int
*/
private $_cpuNums = 24;

/**
* 进程名字
* @var string
*/
private $_name = "MyProcess";

/**
* php 命令路径
* @var string
*/
private $_phpbin = "/usr/bin/php";

/**
* 实例对象
* @var null
*/
private static $_instance = null;

/**
* 默认是否存在swoole模块
* @var bool
*/
private static $_isSwoole = false;

/**
* 私有构造函数
* MyMulProcess constructor.
*/
private function __construct() {
}

/**
* 返回实例对象
* @return MyMulProcess|null
*/
public static function getInst() {
if ( self::$_instance ) {
return self::$_instance;
}

//模块载入
if ( extension_loaded("swoole") ) {
self::$_isSwoole = true;
}

self::$_instance = new self();
return self::$_instance;
}

/**
* 设置CPU核数
* @param $nums
* @return $this
*/
public function setCpuNums($nums) {
$this->_cpuNums = $nums;
return $this;
}

/**
* 设置主进程
* @param $name
* @return $this
*/
public function setProcessName($name) {
$this->_name = $name;

return $this;
}

/**
* 设置php可执行的路径
* @param $phpbin
* @return $this
*/
public function setPhpBin($phpbin) {
$this->_phpbin = $phpbin;

return $this;
}

public function start ($total, $pageSize, $phpfile, $extras = array()) {
if (self::$_isSwoole) {
swoole_set_process_name($this->_name);
} else if (extension_loaded("cli_set_process_title")) {
cli_set_process_title($this->_name);
}

$name = $this->_name;
$phpbin = $this->_phpbin;
$count = 0;
$pages = ceil($total/$pageSize);
while (true) {
for ($i = $count; $i<$pages; $i++) {
if ( count($this->_works) >= $this->_cpuNums ) {
break;
}

if (self::$_isSwoole) {
$process = new swoole_process(function (swoole_process $work) use
($name, $phpbin, $phpfile, $i, $pageSize,$pages, $extras) {
array_unshift($extras, $pages);
array_unshift($extras, $pageSize);
array_unshift($extras, $i);
array_unshift($extras, $name);
array_unshift($extras, $phpfile);
$work->exec($phpfile, $extras);

$work->exit(0);
}, false, true);

$pid = $process->start();

if (empty($pid)) {
echo swoole_errno().''.swoole_strerror();
continue;
}

$this->_works[$pid] = $process;
} else {
$pid = pcntl_fork();
if ($pid == 0) { //子进程
array_unshift($extras, $pages);
array_unshift($extras, $pageSize);
array_unshift($extras, $i);
array_unshift($extras, $name);
array_unshift($extras, $phpfile);
pcntl_exec($phpbin, $extras);

exit(0);
} else if ($pid > 0) {
$this->_works[$pid] = $pid;
} else {
continue;
}
}
$count++;
}

foreach ($this->_works as $work) {
if (self::$_isSwoole) {
if ($ret = $work->wait(false)) {
unset($this->_works[$work->pid]);
}
} else {
$status = 0;
$pid = pcntl_waitpid($work, $status, WNOHANG);
if($pid == -1 || $pid > 0) {
unset($this->_works[$pid]);
}
}
}
if (empty($this->_works) && $count >= $pages) {
echo "empty works" . "\n";
break;
}
// 防止该进程抢占CPU
sleep(1);
}
}
}

//index.php 接受参数
/**
* 命令行 接收参数定义说明
* -------------------------------
* 第零个 参数 phpdiamante文件完整路径
* 第一个 参数 进程名字 $name
* 第二个 参数 页数从0开始$i, 从数据库拿数据的话,需要[$i*$limit]位置开始
* 第三个 参数 每页大小$limit
* 第四个 参数 总页数$pages,如果需要判断($i+1 >= $pages)最后一个可以跑所有的数据(要不然会丢失一些数据)
* 第五个 参数 自己传入
* ........
* 第N个 参数 这个N不能太大,要不然命令直接挂掉
*/
$name = $_SERVER['argv'][1];
$i = $_SERVER['argv'][2];
$limit = $_SERVER['argv'][3];
$pages = $_SERVER['argv'][4];

//设置进程名字,主要是好管理而已
if (extension_loaded("swoole")) {
swoole_set_process_name($name.'-child-'.$i);
} else if(extension_loaded("cli_set_process_title")) {
cli_set_process_title($name.'-child-'.$i);
}

//测试代码
// 测试
MyMulProcess::getInst()
->setCpuNums(5) // CPU数
->setProcessName("TTTT") // 进程名称
->setPhpBin("/usr/bin/php7") // PHP 可执行路径
->start($total, $pageSize, "/index.php", array('bbb', 'ccc'));