版权声明:转载请注明来源 https://blog.csdn.net/u013702678/article/details/82936145
swoole_process提供的setAffinity用于设置CPU亲和性,可以将进程绑定到特定的CPU核上。
function swoole_process::setAffinity(array $cpu_set);
- 接受一个数组参数表示绑定哪些
CPU
核,如array(0,2,3)
表示绑定CPU0/CPU2/CPU3
- 成功返回
true
,失败返回false
下面我们分析下其流程。
static PHP_METHOD(swoole_process, setaffinity)
{
zval *array;
//解析输入参数信息,输入参数用于控制当前进程在哪些CPU上执行
if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "a", &array) == FAILURE)
{
RETURN_FALSE;
}
if (Z_ARRVAL_P(array)->nNumOfElements == 0)//数组长度为0,属于异常情况
{
RETURN_FALSE;
}
if (Z_ARRVAL_P(array)->nNumOfElements > SW_CPU_NUM)//数组长度大于CPU个数,属于异常情况
{
swoole_php_fatal_error(E_WARNING, "More than the number of CPU");
RETURN_FALSE;
}
zval *value = NULL;
cpu_set_t cpu_set;//定义后续系统调用的数据结构
CPU_ZERO(&cpu_set);//初始化
//下述逻辑将用户输入按要求转存到cpu_set结构中
SW_HASHTABLE_FOREACH_START(Z_ARRVAL_P(array), value)
convert_to_long(value);
if (Z_LVAL_P(value) >= SW_CPU_NUM)
{
swoole_php_fatal_error(E_WARNING, "invalid cpu id [%d]", (int) Z_LVAL_P(value));
RETURN_FALSE;
}
CPU_SET(Z_LVAL_P(value), &cpu_set);
SW_HASHTABLE_FOREACH_END();
#ifdef __FreeBSD__
if (cpuset_setaffinity(CPU_LEVEL_WHICH, CPU_WHICH_PID, -1,
sizeof(cpu_set), &cpu_set) < 0)//调用系统调用
#else
if (sched_setaffinity(getpid(), sizeof(cpu_set), &cpu_set) < 0)//调用系统调用
#endif
{
swoole_php_sys_error(E_WARNING, "sched_setaffinity() failed.");
RETURN_FALSE;
}
RETURN_TRUE;
}