在PHP編程中，在遍歷數組的時候經常需要先計算數組的長度作為循環(huán)結束的判斷條件，而在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函數可以檢測遞歸避免無限循環(huán)，但會在遇到無限遞歸或得到比期望值大的時候返回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錯誤

否則計算當前數組層數的數組元素個數

如果有遞歸參數選項，則繼續(xù)遞歸訪問

如果參數是對象類型，實現時會先判斷handler是否被定義。而handler是PHP內核中對象的結構體，其中包含有count_elements字段，實際上是一個函數。如果某個對象表現得想數組一樣，即通常說的array-like object，那么就會執(zhí)行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，最終得到了想要的答案。不懂就要問，哈哈。

最新評論