把两个数组,相同key的字符串值拼接。
代码奉上:
基础代码
这个扩展,我们将在say扩展上增加array_concat方法。say扩展相关代码大家请看这篇文章。PHP7扩展开发之hello word文中已经详细介绍了如何创建一个扩展和提供了源码下载。
实现array_concat方法
array_concat方法的PHP扩展源码:
12345678910111213141516171819202122232425262728293031323334353637
PHP_FUNCTION(array_concat){zval *arr, *prefix, *entry, *prefix_entry, value;zend_string *string_key, *result;zend_ulong num_key;if(zend_parse_parameters(ZEND_NUM_ARGS(),"aa", &arr, &prefix) == FAILURE) {return;}array_init_size(return_value, zend_hash_num_elements(Z_ARRVAL_P(arr)));ZEND_HASH_FOREACH_KEY_VAL(Z_ARRVAL_P(arr), num_key, string_key, entry) {if(string_key && zend_hash_exists(Z_ARRVAL_P(prefix), string_key)) {prefix_entry = zend_hash_find(Z_ARRVAL_P(prefix), string_key);if(Z_TYPE_P(entry) == IS_STRING && prefix_entry != NULL && Z_TYPE_P(prefix_entry) == IS_STRING) {result = strpprintf(0,"%s%s", Z_STRVAL_P(prefix_entry), Z_STRVAL_P(entry));ZVAL_STR(&value, result);zend_hash_update(Z_ARRVAL_P(return_value), string_key, &value);}}elseif(string_key == NULL && zend_hash_index_exists(Z_ARRVAL_P(prefix), num_key)){prefix_entry = zend_hash_index_find(Z_ARRVAL_P(prefix), num_key);if(Z_TYPE_P(entry) == IS_STRING && prefix_entry != NULL && Z_TYPE_P(prefix_entry) == IS_STRING) {result = strpprintf(0,"%s%s", Z_STRVAL_P(prefix_entry), Z_STRVAL_P(entry));ZVAL_STR(&value, result);zend_hash_index_update(Z_ARRVAL_P(return_value), num_key, &value);}}elseif(string_key) {zend_hash_update(Z_ARRVAL_P(return_value), string_key, entry);zval_add_ref(entry);}else{zend_hash_index_update(Z_ARRVAL_P(return_value), num_key, entry);zval_add_ref(entry);}}ZEND_HASH_FOREACH_END();}
PHP中的数组本质上就是一个哈希。
对于哈希处理的方法主要集中在Zend/zend_hash.h中。
对于数组的操作方法主要集中在Zend/API.h。数组的方法其实就是对哈希处理方法的一层包装。
数组操作的方法主要是以add_assoc_和add_index_开头的一些列方法。
下面是代码中涉及的一些方法。
zend_hash_num_elements获取数组的元素个数。
array_init_size(return_value, zend_hash_num_elements(Z_ARRVAL_P(arr)));初始化一个数组。在PHP扩展中,我们是通过return_value这个变量设置方法的返回值。因此,我们直接修改这个return_value变量即可。感兴趣的话,可以把宏方法PHP_FUNCTION展开看下。
PHP7提供了一套宏方法用于遍历哈希和对哈希进行操作。这些宏方法主要放在Zend/zend_hash.h文件中。如,代码中的ZEND_HASH_FOREACH_KEY_VAL就是一个变量哈希的宏。是不是和PHP代码中的foreach有点像?
在这里我们把代码中用到的哈希相关的方法做下整理说明:
ZEND_HASH_FOREACH_KEY_VAL 和 ZEND_HASH_FOREACH_END 配合使用,实现foreach的效果。
zend_hash_exists 检测指定的key在哈希中是否存在。key为字符串。
zend_hash_index_exists 检测指定的key在哈希中是否存在。key为数字。
zend_hash_find 根据key查找指定的值。key为字符串。
zend_hash_index_find 根据key查找指定的值。key为数字。
zend_hash_update 更新指定key的值。key为字符串。
zend_hash_index_update 更新指定key的值。key为数字。
基本上有这些方法,你就可以对数组进行一些基本操作了。方法命名也很有规律,key为字符串和数字提供了两套。
zval_add_ref(entry);给数组的值,增加一次引用计数。zend_hash_update方法只自动给string_key自动增加了一次引用计数。数组return_value共用数组arr的值。因此,我们需要手动增加一次引用计数。
PHP7哈希相关的文章大家可以看看
http://jpauli.github.io/2016/04/08/hashtables.html
我们下篇接着聊。
作者:信海龙(21CTO社区会员)
博客:www.bo56.com
作者:信海龙(21CTO社区会员)
博客:www.bo56.com