[PHP源碼閱讀]count函數(shù)，php源碼count函數(shù)

在PHP編程中，在遍歷數(shù)組的時候經(jīng)常需要先計算數(shù)組的長度作為循環(huán)結束的判斷條件，而在PHP里面對數(shù)組的操作是很頻繁的，因此count也算是一個常用函數(shù)，下面研究一下count函數(shù)的具體實現(xiàn)。

我在github上有對PHP源碼更詳細的注解。感興趣的可以圍觀一下，給個star。PHP5.4源碼注解。可以通過commit記錄查看已添加的注解。

count

int <span>count</span> ( <span>mixed</span> <span>$array_or_countable</span> [, int <span>$mode</span> = COUNT_NORMAL ] )

count函數(shù)計算數(shù)組或者對象里面的所有元素個數(shù)。

對于對象來說，如果你安裝了SPL擴展，可以通過實現(xiàn)Countable接口來調(diào)用count函數(shù)。Countable接口有且僅有一個方法Countable::count()，該方法的返回count()函數(shù)的返回值。

參數(shù)說明

mode

如果參數(shù)mode設為COUNT_RECURSIVE(或1)，count()會遞歸地計算該數(shù)組。在計算多維數(shù)組的時候特別有用。

如果第一個參數(shù)不是數(shù)組或者實現(xiàn)Countable接口的對象，count函數(shù)將返回1。

注意：count函數(shù)可以檢測遞歸避免無限循環(huán)，但會在遇到無限遞歸或得到比期望值大的時候返回E_WARNING提示。

運行示例

普通應用

<span>$arr1</span> = <span>array</span>(1, 2, 3, 4, 5<span>);
</span><span>$val1</span> = <span>count</span>(<span>$arr1</span>); <span>//</span><span> 5</span>

多維數(shù)組

<span>$arr2</span> = <span>array</span>('apple', 'banana', <span>array</span>('cat', 'camel'), 'dog'<span>);
</span><span>$val2_1</span> = <span>count</span>(<span>$arr2</span>); <span>//</span><span> 4</span>
<span>$val2_2</span> = <span>count</span>(<span>$arr2</span>, 1); <span>//</span><span> 6</span>

數(shù)字和字符串

<span>$str</span> = "hello world"<span>;
</span><span>$int_val</span> = 1<span>;
</span><span>$val3</span> = <span>count</span>(<span>$str</span>); <span>//</span><span> 1</span>
<span>$val4</span> = <span>count</span>(<span>$int_val</span>); <span>//</span><span> 1</span>

普通對象

<span>class</span><span> User {
    </span><span>private</span> <span>$name</span><span>;
    </span><span>private</span> <span>$address</span><span>;
}

</span><span>$user</span> = <span>new</span><span> User();
</span><span>$val5</span> = <span>count</span>(<span>$user</span>); <span>//</span><span> 1</span>
<span>$val6</span> = <span>count</span>((<span>array</span>) <span>$user</span>); <span>//</span><span> 2</span>

array-like對象

    <span>class</span> User <span>extends</span><span> ArrayObject {
        </span><span>private</span> <span>$name</span><span>;

        </span><span>public</span> <span>function</span><span> __construct() {
            </span><span>$this</span>->name = 'hhq'<span>;
        }

        </span><span>public</span> <span>function</span><span> getName() {
            </span><span>return</span> <span>$this</span>-><span>name;
        }

        </span><span>public</span> <span>function</span> <span>count</span><span>() {
            </span><span>return</span> 2<span>;
        }

    }

    </span><span>$user2</span> = <span>new</span><span> User();
    </span><span>$val7</span> = <span>count</span>(<span>$user2</span>); <span>//</span><span> 2</span>

實現(xiàn)Countable接口對象

    <span>class</span> User <span>implements</span><span> Countable {
        </span><span>public</span> <span>function</span> <span>count</span><span>() {
            </span><span>return</span> 3<span>;
        }
    }

    </span><span>$user3</span> = <span>new</span><span> User();
    </span><span>$val8</span> = <span>count</span>(<span>$user3</span>); <span>//</span><span> 3</span> 

運行步驟

<p>進入switch語句檢測參數(shù)類型</p>
<p>　　如果是NULL，直接返回0</p>
<p>　　如果是數(shù)組，調(diào)用php_count_recursive函數(shù)機選數(shù)組元素個數(shù)</p>
<p>　　如果是對象，先檢查是否為數(shù)組對象（array-like object），如果是，則計算數(shù)組對象的數(shù)量</p>
<p>　　否則，如果對象實現(xiàn)了Countable接口，則調(diào)用Countable的count方法</p>
<p>　　最后，其他類型比如整型數(shù)組或字符串，都返回1。</p>

源碼解讀

如果是普通數(shù)組，count函數(shù)會調(diào)用php_count_recursive函數(shù)實現(xiàn)其功能的運行步驟如下：

<p>如果當前hash Bucket被遞歸訪問的次數(shù)大于1，說明重復遞歸，染回E_WARNING錯誤</p>
<p>否則計算當前數(shù)組層數(shù)的數(shù)組元素個數(shù)</p>
<p>如果有遞歸參數(shù)選項，則繼續(xù)遞歸訪問</p>

如果參數(shù)是對象類型，實現(xiàn)時會先判斷handler是否被定義。而handler是PHP內(nèi)核中對象的結構體，其中包含有count_elements字段，實際上是一個函數(shù)。如果某個對象表現(xiàn)得想數(shù)組一樣，即通常說的array-like object，那么就會執(zhí)行count_elements函數(shù)。具體實現(xiàn)是類繼承PHP的ArrayObject，并在類里面實現(xiàn)count函數(shù)，具體調(diào)用的就是count函數(shù)，如果類沒有實現(xiàn)count函數(shù)，則count返回0，否則返回對象的count函數(shù)的返回值。

如果是其他的數(shù)據(jù)類型
1、字符串

2、數(shù)字

3、對象分支中兩個if判斷都為false的情況，即沒有繼承ArrayObject且沒有實現(xiàn)Countable接口。

這些類型通通返回1。

需要注意的是，如果需要計算的是對象的屬性數(shù)量，可以先將對象轉換成數(shù)組，然后調(diào)用count函數(shù)。如：
$count_value = count((array) $user);

小結

閱讀count函數(shù)的源碼過程中，在其中一步卡住了，就是if (Z_OBJ_HT_P(array)->count_elements)這一步，因為始終無法寫出進入這個分支的demo，在網(wǎng)上搜索了很多資料也未果，因此請教了TIPI的reeze，最終得到了想要的答案。不懂就要問，哈哈。

?

原創(chuàng)文章，文筆有限，才疏學淺，文中若有不正之處，萬望告知。

如果本文對你有幫助，請點下推薦吧，謝謝^_^

?

最后再安利一下，我在github有對PHP源碼更詳細的注解。感興趣的可以圍觀一下，給個star。PHP5.4源碼注解?？梢酝ㄟ^commit記錄查看已添加的注解。

更多源碼文章，歡迎訪問個人主頁繼續(xù)查看：hoohack