php中使用帮助函数array_get()读取多维数组的值
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');
非常方便
相关文章