在PHP編程中,在遍歷數組的時候經常需要先計算數組的長度作為循環結束的判斷條件,而在PHP裡面對數組的操作是很頻繁的,因此count也算是一個常用函數,下面研究一下count函數的具體實現。
我在github上有對PHP源碼更詳細的注解。感興趣的可以圍觀一下,給個star。PHP5.4源碼注解。可以通過commit記錄查看已添加的注解。
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,最終得到了想要的答案。不懂就要問,哈哈。