修改array_combine函数，增加可选参数use_strict

无野

array_combine是PHP5一个standard的array函数。
它的作用是创建一个数组，用一个数组的值作为其键名，另一个数组的值作为其值。
    array array_combine ( array $keys, array $values )
    返回一个 array，用来自 keys 数组的值作为键名，来自 values 数组的值作为相应的值。
    如果两个数组的单元数不同或者数组为空时返回 FALSE。

注意粗体字的描述，也就是说，如果你使用array_combine，在没有做好兼容性的情况下，便有因为两数组单元素不同而导致程序异常。
首先不讨论这种异常对于程序来说是好处还是坏处，我认为至少函数应该要给程序员一个可选的严格模式。

我决定对array_combine做这个修改，即添加一个boolean参数use_strict，默认值为true。这样可以兼容原生的函数。
vi php-5.3.1/ext/standard/array.c, 将array_combine函数原型源码修改成如下：

1. /* {{{ proto array array_combine(array keys, array values)
   
2.    Creates an array by using the elements of the first parameter as keys and the elements of the second as the corresponding values
   
3. */
   
4. PHP_FUNCTION(array_combine)
   
5. {
   
6.         zval *values, *keys;
   
7.         HashPosition pos_values, pos_keys;
   
8.         zval **entry_keys, **entry_values;
   
9.         int num_keys, num_values;
   
10.         zend_bool use_strict = 1;
    
11. 
    
12.         if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "aa|b", &keys, &values, &use_strict) == FAILURE) {
    
13.                 return;
    
14.         }
    
15. 
    
16.         num_keys = zend_hash_num_elements(Z_ARRVAL_P(keys));
    
17.         num_values = zend_hash_num_elements(Z_ARRVAL_P(values));
    
18.         if (use_strict){
    
19.                 if (num_keys != num_values) {
    
20.                         php_error_docref(NULL TSRMLS_CC, E_WARNING, "Both parameters should have an equal number of elements");
    
21.                         RETURN_FALSE;
    
22.                 }
    
23.         }
    
24.         if (!num_keys) {
    
25.                 php_error_docref(NULL TSRMLS_CC, E_WARNING, "Both parameters should have at least 1 element");
    
26.                 RETURN_FALSE;
    
27.         }
    
28. 
    
29.         array_init_size(return_value, num_keys);
    
30. 
    
31.         zend_hash_internal_pointer_reset_ex(Z_ARRVAL_P(keys), &pos_keys);
    
32.         zend_hash_internal_pointer_reset_ex(Z_ARRVAL_P(values), &pos_values);
    
33.         while (zend_hash_get_current_data_ex(Z_ARRVAL_P(keys), (void **)&entry_keys, &pos_keys) == SUCCESS &&
    
34.                 zend_hash_get_current_data_ex(Z_ARRVAL_P(values), (void **)&entry_values, &pos_values) == SUCCESS
    
35.         ) {
    
36.                 if (Z_TYPE_PP(entry_keys) == IS_LONG) {
    
37.                         zval_add_ref(entry_values);
    
38.                         add_index_zval(return_value, Z_LVAL_PP(entry_keys), *entry_values);
    
39.                 } else {
    
40.                         zval key, *key_ptr = *entry_keys;
    
41. 
    
42.                         if (Z_TYPE_PP(entry_keys) != IS_STRING) {
    
43.                                 key = **entry_keys;
    
44.                                 zval_copy_ctor(&key);
    
45.                                 convert_to_string(&key);
    
46.                                 key_ptr = &key;
    
47.                         }
    
48. 
    
49.                         zval_add_ref(entry_values);
    
50.                         add_assoc_zval_ex(return_value, Z_STRVAL_P(key_ptr), Z_STRLEN_P(key_ptr) + 1, *entry_values);
    
51. 
    
52.                         if (key_ptr != *entry_keys) {
    
53.                                 zval_dtor(&key);
    
54.                         }
    
55.                 }
    
56. 
    
57.                 zend_hash_move_forward_ex(Z_ARRVAL_P(keys), &pos_keys);
    
58.                 zend_hash_move_forward_ex(Z_ARRVAL_P(values), &pos_values);
    
59.         }
    
60. }
    
61. /* }}} */
    

复制代码

这样，array_combine就变成了：
    array array_combine ( array $keys, array $values [, bool $use_strict] )

因为我在一台测试上做这个事，所以我重编译了PHP。

我们来测一下修改的效果。
使用默认的严格模式:

1. $a = array('green', 'red');
   
2. $b = array('avocado', 'apple', 'banana');
   
3. $c = array_combine($a, $b);
   
4. print_r($c);

复制代码

1. [root@tekeq test]# /wwws/php/bin/php test.php
   
2. PHP Warning:  array_combine(): Both parameters should have an equal number of elements in /root/test/test.php on line 5

复制代码

使用非严格模式:

1. $a = array('green', 'red');
   
2. $b = array('avocado', 'apple', 'banana');
   
3. $c = array_combine($a, $b, false); //使用非严格模式
   
4. print_r($c);

复制代码

1. [root@tekeq test]# /wwws/php/bin/php test.php
   
2. Array
   
3. (
   
4.     [green] => avocado
   
5.     [red] => apple
   
6. )

复制代码

看起来一切正常，就这么办。

认为有需要的同学，可以改一下，这修改后的是兼容以前的。
如果觉得没必要就算了，我是有需要才改，不是改着玩。

相关文章:http://tekeq.com/tech/array_combine/

renxiao2003

学习了，不错。有创意。

maochanglu

我有点吃饱了撑着的感觉。

无野

我有点吃饱了撑着的感觉。
maochanglu 发表于 2010-05-30 10:27

你还真没说错，哈哈
我最近晚上老感觉吃饱了会撑着，所以决定以后晚上只吃水果，不吃饭。
想找回以前身轻如燕的感觉。。。