PHP源码阅读:count 函数
count
int count ( mixed $array_or_countable [, int $mode = COUNT_NORMAL ] )
count函数计算数组或者对象里面的所有元素个数。
对于对象来说,如果你安装了SPL扩展,可以通过实现Countable接口来调用count函数。Countable接口有且仅有一个方法Countable::count(),该方法的返回count()函数的返回值。
参数说明
mode
如果参数mode设为COUNT_RECURSIVE(或1),count()会递归地计算该数组。在计算多维数组的时候特别有用。
如果第一个参数不是数组或者实现Countable接口的对象,count函数将返回1。
注意:count函数可以检测递归避免无限循环,但会在遇到无限递归或得到比期望值大的时候返回E_WARNING提示。
运行示例
普通应用
$arr1 = array(1, 2, 3, 4, 5);$val1 = count($arr1); // 5
多维数组
$arr2 = array('apple', 'banana', array('cat', 'camel'), 'dog');$val2_1 = count($arr2); // 4$val2_2 = count($arr2, 1); // 6
数字和字符串
$str = "hello world";$int_val = 1;$val3 = count($str); // 1$val4 = count($int_val); // 1
普通对象
class User { private $name; private $address;}$user = new User();$val5 = count($user); // 1$val6 = count((array) $user); // 2
array-like对象
class User extends ArrayObject { private $name; public function __construct() { $this->name = 'hhq'; } public function getName() { return $this->name; } public function count() { return 2; }}$user2 = new User();$val7 = count($user2); // 2
实现Countable接口对象
class User implements Countable { public function count() { return 3; }}$user3 = new User();$val8 = count($user3); // 3
运行步骤
进入switch语句检测参数类型
如果是NULL,直接返回0
如果是数组,调用php_count_recursive函数机选数组元素个数
如果是对象,先检查是否为数组对象(array-like object),如果是,则计算数组对象的数量
否则,如果对象实现了Countable接口,则调用Countable的count方法
最后,其他类型比如整型数组或字符串,都返回1。
源码解读
如果是普通数组,count函数会调用php_count_recursive函数实现其功能的运行步骤如下:
如果当前hash Bucket被递归访问的次数大于1,说明重复递归,染回E_WARNING错误
否则计算当前数组层数的数组元素个数
如果有递归参数选项,则继续递归访问
如果参数是对象类型,实现时会先判断handler是否被定义。而handler是PHP内核中对象的结构体,其中包含有count_elements字段,实际上是一个函数。如果某个对象表现得想数组一样,即通常说的array-like object,那么就会执行count_elements函数。具体实现是类继承PHP的ArrayObject,并在类里面实现count函数,具体调用的就是count函数,如果类没有实现count函数,则count返回0,否则返回对象的count函数的返回值。
如果是其他的数据类型 1、字符串
2、数字
3、对象分支中两个if判断都为false的情况,即没有继承ArrayObject且没有实现Countable接口。
这些类型通通返回1。
需要注意的是,如果需要计算的是对象的属性数量,可以先将对象转换成数组,然后调用count函数。如: $count_value = count((array) $user);
小结
阅读count函数的源码过程中,在其中一步卡住了,就是if (Z_OBJ_HT_P(array)->count_elements)这一步,因为始终无法写出进入这个分支的demo,在网上搜索了很多资料也未果,因此请教了TIPI的reeze,最终得到了想要的答案。不懂就要问,哈哈。
原创文章,文笔有限,才疏学浅,文中若有不正之处,万望告知。
打赏支持我写出更多好文章,谢谢!
打赏作者
打赏支持我写出更多好文章,谢谢!
任选一种支付方式
关于作者:hoohack
一个正在努力的菜鸟 个人主页 · 我的文章 · 15 ·
推荐阅读