swoole源码分析之Buffer的expend操作

swoole_buffer提供的expend操作用于为缓存区扩容。

swoole_buffer->expand(int $new_size);

• $new_size 指定新的缓冲区尺寸，必须大于当前的尺寸

下面我们看下其实现过程。

static PHP_METHOD(swoole_buffer, expand)
{
    long size = -1;

    //解析输入参数信息，这里输入参数只有个size，表示将当前buffer扩容为size大小
    if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "l", &size) == FAILURE)
    {
        RETURN_FALSE;
    }

    swString *buffer = swoole_get_object(getThis());//获取swoole内部封装buffer对象

    if (size <= buffer->size)//新扩容的buffer大小需要大于当前buffer的size
    {
        php_error_docref(NULL TSRMLS_CC, E_WARNING, "new size must be more than %ld", buffer->size);
        RETURN_FALSE;
    }

    if (swString_extend(buffer, size) == SW_OK)//执行扩容操作，具体下面分析
    {
        zend_update_property_long(swoole_buffer_class_entry_ptr, getThis(), ZEND_STRL("capacity"), size TSRMLS_CC);//更新swoole_buffer的容量信息capacity
        RETURN_TRUE;
    }
    else
    {
        RETURN_FALSE;
    }
}

int swString_extend(swString *str, size_t new_size)
{
    assert(new_size > str->size);//断言判断size大小
    char *new_str = sw_realloc(str->str, new_size);//通过sw_realloc增加str->str对应的空间大小
    if (new_str == NULL)//申请空间失败
    {
        swSysError("realloc(%ld) failed.", new_size);
        return SW_ERR;
    }

    str->str = new_str;//str->str已经指向新申请的空间，感觉是多余的，哈哈
    str->size = new_size;//调整buffer的size大小，即调整为新的大小
    return SW_OK;
}