php: 幫助函數 array_get 的擴展實現方式

項目組整合了一些laravel的內容，其中有個函數array_get這個函數使用最多,先來看看用法：

array_get()
array_get方法使用”.“號從嵌套數組中獲取值：

    $array = ['products' => ['desk' => ['price' => 100]]];
    $value = array_get($array, 'products.desk');
    // ['price' => 100]

array_get函數還接收一個默認值，如果指定鍵不存在的話則返回該默認值：

  $value = array_get($array, 'names.john', 'default');

不適用的場合：

      數組key中含有.的內容。

我們先來看看它的php實現方式：

function array_get($array, $key, $default = null)
    {
        if (is_null($key)) {
            return $array; 
        }

        if (isset($array[$key])) {
            return $array[$key];
        }

        foreach (explode('.', $key) as $segment) {
            if (! is_array($array) || ! array_key_exists($segment, $array)) {
                return value($default);
            }

            $array = $array[$segment];
        }
        return $array;
    }

當key為空的時候，上述函數直接返回了整個array的內容, 在用C實現的時候我選擇返回了空, 代碼如下：

PHP_FUNCTION(array_get)
{
    zval *arr; // array
    zend_string* strkey; // key
    zval *defaultval = NULL; // default value
    zval *retval;
    HashTable *arrHashTable;
    zval *dest_entry;
    if (zend_parse_parameters(ZEND_NUM_ARGS(), "zS|z", &arr, 
        &strkey, &defaultval) == FAILURE) {
        return;
    }
    if ((retval = zend_hash_find(Z_ARRVAL_P(arr), strkey)) != NULL){
        RETURN_ZVAL(retval, 1, 0);
    } 
    // foreach
    if (zend_memrchr(ZSTR_VAL(strkey), '.', ZSTR_LEN(strkey))) {
        char *entry, *ptr, *seg;
        HashTable *target = Z_ARRVAL_P(arr);
        entry = estrndup(ZSTR_VAL(strkey), ZSTR_LEN(strkey));
        if ((seg = php_strtok_r(entry, ".", &ptr))) {
            do {
                if (target == NULL || (retval = zend_symtable_str_find(target, seg, strlen(seg))) == NULL) {
                    break;
                }

                if (Z_TYPE_P(retval) == IS_ARRAY) {
                    target = Z_ARRVAL_P(retval);
                } else {
                    target = NULL;
                }
            } while ((seg = php_strtok_r(NULL, ".", &ptr)));
        }
        efree(entry);
        if (retval) {
            RETURN_ZVAL(retval, 1, 0);
        }
    }
    // end foreach
    if (defaultval) {
        RETURN_ZVAL(defaultval, 1, 0);
    } else {
        RETURN_NULL();
    }
}