swoole_process源码分析之设置CPU亲和性

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;
}