php中使用帮助函数array_get()读取多维数组的值

2023-06-01 00:00:00 php 函数 Array

array_get函数用法:

使用'.'号从嵌套数组中获取值;

函数还接收一个默认值,如果指定键不存在的话则返回该默认值;


源码实现代码:

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



demo测试代码:

$arr =  [
     'User'=>[
        'avatar'=> '123'
     ]
 ]


使用帮助函数array_get():

array_get($arr, 'User.avatar');


非常方便




相关文章