在PHP編程中，在遍歷數(shù)組的時(shí)候經(jīng)常需要先計(jì)算數(shù)組的長(zhǎng)度作為循環(huán)結(jié)束的判斷條件，而在PHP里面對(duì)數(shù)組的操作是很頻繁的，因此count也算是一個(gè)常用函數(shù)，下面研究一下count函數(shù)的具體實(shí)現(xiàn)。
我在github上有對(duì)PHP源碼更詳細(xì)的注解。感興趣的可以圍觀一下，給個(gè)star。PHP5.4源碼注解?？梢酝ㄟ^(guò)commit記錄查看已添加的注解。
count
int count ( mixed $array_or_countable [, int $mode = COUNT_NORMAL ] )
count函數(shù)計(jì)算數(shù)組或者對(duì)象里面的所有元素個(gè)數(shù)。
對(duì)于對(duì)象來(lái)說(shuō)，如果你安裝了SPL擴(kuò)展，可以通過(guò)實(shí)現(xiàn)Countable接口來(lái)調(diào)用count函數(shù)。Countable接口有且僅有一個(gè)方法Countable::count()，該方法的返回count()函數(shù)的返回值。
參數(shù)說(shuō)明
mode
如果參數(shù)mode設(shè)為COUNT_RECURSIVE(或1)，count()會(huì)遞歸地計(jì)算該數(shù)組。在計(jì)算多維數(shù)組的時(shí)候特別有用。
如果第一個(gè)參數(shù)不是數(shù)組或者實(shí)現(xiàn)Countable接口的對(duì)象，count函數(shù)將返回1。
注意：count函數(shù)可以檢測(cè)遞歸避免無(wú)限循環(huán)，但會(huì)在遇到無(wú)限遞歸或得到比期望值大的時(shí)候返回E_WARNING提示。
運(yùn)行示例
普通應(yīng)用
$arr1 = array(1, 2, 3, 4, 5);
$val1 = count($arr1); // 5
多維數(shù)組
$arr2 = array('apple', 'banana', array('cat', 'camel'), 'dog');
$val2_1 = count($arr2); // 4
$val2_2 = count($arr2, 1); // 6
數(shù)字和字符串
$str = "hello world";
$int_val = 1;
$val3 = count($str); // 1
$val4 = count($int_val); // 1
普通對(duì)象
class User {
private $name;
private $address;
}
$user = new User();
$val5 = count($user); // 1
$val6 = count((array) $user); // 2
array-like對(duì)象
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
實(shí)現(xiàn)Countable接口對(duì)象
class User implements Countable {
public function count() {
return 3;
}
}
$user3 = new User();
$val8 = count($user3); // 3
運(yùn)行步驟
進(jìn)入switch語(yǔ)句檢測(cè)參數(shù)類(lèi)型
如果是NULL，直接返回0
如果是數(shù)組，調(diào)用php_count_recursive函數(shù)機(jī)選數(shù)組元素個(gè)數(shù)
如果是對(duì)象，先檢查是否為數(shù)組對(duì)象（array-like object），如果是，則計(jì)算數(shù)組對(duì)象的數(shù)量
否則，如果對(duì)象實(shí)現(xiàn)了Countable接口，則調(diào)用Countable的count方法
最后，其他類(lèi)型比如整型數(shù)組或字符串，都返回1。
源碼解讀
如果是普通數(shù)組，count函數(shù)會(huì)調(diào)用php_count_recursive函數(shù)實(shí)現(xiàn)其功能的運(yùn)行步驟如下：
如果當(dāng)前hash Bucket被遞歸訪問(wèn)的次數(shù)大于1，說(shuō)明重復(fù)遞歸，染回E_WARNING錯(cuò)誤
否則計(jì)算當(dāng)前數(shù)組層數(shù)的數(shù)組元素個(gè)數(shù)
如果有遞歸參數(shù)選項(xiàng)，則繼續(xù)遞歸訪問(wèn)
如果參數(shù)是對(duì)象類(lèi)型，實(shí)現(xiàn)時(shí)會(huì)先判斷handler是否被定義。而handler是PHP內(nèi)核中對(duì)象的結(jié)構(gòu)體，其中包含有count_elements字段，實(shí)際上是一個(gè)函數(shù)。如果某個(gè)對(duì)象表現(xiàn)得想數(shù)組一樣，即通常說(shuō)的array-like object，那么就會(huì)執(zhí)行count_elements函數(shù)。具體實(shí)現(xiàn)是類(lèi)繼承PHP的ArrayObject，并在類(lèi)里面實(shí)現(xiàn)count函數(shù)，具體調(diào)用的就是count函數(shù)，如果類(lèi)沒(méi)有實(shí)現(xiàn)count函數(shù)，則count返回0，否則返回對(duì)象的count函數(shù)的返回值。
如果是其他的數(shù)據(jù)類(lèi)型
1、字符串
2、數(shù)字
3、對(duì)象分支中兩個(gè)if判斷都為false的情況，即沒(méi)有繼承ArrayObject且沒(méi)有實(shí)現(xiàn)Countable接口。
這些類(lèi)型通通返回1。
需要注意的是，如果需要計(jì)算的是對(duì)象的屬性數(shù)量，可以先將對(duì)象轉(zhuǎn)換成數(shù)組，然后調(diào)用count函數(shù)。如：
$count_value = count((array) $user);
小結(jié)
閱讀count函數(shù)的源碼過(guò)程中，在其中一步卡住了，就是if (Z_OBJ_HT_P(array)->count_elements)這一步，因?yàn)槭冀K無(wú)法寫(xiě)出進(jìn)入這個(gè)分支的demo，在網(wǎng)上搜索了很多資料也未果，因此請(qǐng)教了TIPI的reeze，最終得到了想要的答案。