改进TP3.2.3中Model.class.php中的getField方法

2018-05-31 后端 
  1. 改进TP3.2.3中Model.class.php中的getField方法，使之支持带逗号的函数
  2. 增加的代码只有改进-begin到改进-end之间的代码，其他的就是注释原语句。
  3. 原来的getField：
  4. 支持：getField('id, user_name, role_id',true);
  5. 不支持：getField(array('id',"concat(user_name,'[',id,']') as user_name","concat(user_name,'[',role_id,']') as role_id"),true);返回都是null
  6. 改进后：
  7. 支持：getField('id, user_name, role_id',true);
  8. 支持：getField(array('id',"concat(user_name,'[',id,']') as user_name","concat(user_name,'[',role_id,']') as role_id"),true);
  9. 支持：getField("concat(userName,'[',id,']') as id",true);
  10. 支持：getField(array("concat(userName,'[',id,']') as id"),true);
  11. =============代码详情======================
  12. publicfunction getField($field, $sepa =null)
  13. {
  14. 	$options['field']= $field;
  15. 	$options = $this->_parseOptions($options);
  16. // 判断查询缓存
  17. if(isset($options['cache']))
  18. {
  19. 		$cache = $options['cache'];
  20. 		$key = is_string($cache['key'])? $cache['key']: md5($sepa . serialize($options));
  21. 		$data = S($key,'', $cache);
  22. if(false!== $data)
  23. {
  24. return $data;
  25. }
  26. }
  28. 	$is_multi =false;// 是否是多字段
  29. 	$is_fn =false;// 单字段是否是函数
  31. // 改进-begin
  32. if(is_array($field))
  33. {
  34. if(count($field)>1)
  35. {
  36. 			$is_multi =true;
  37. 			$_field = $field;
  38. }
  39. else
  40. {
  41. 			$field = $field[0];
  42. 			$is_fn = strpos($field,'(');
  43. }
  44. }
  45. else
  46. {
  47. 		$field = trim($field);
  48. 		$is_fn = strpos($field,'(');
  49. if(false=== $is_fn && strpos($field,','))
  50. {
  51. 			$is_multi =true;
  52. 			$_field = explode(',', $field);
  53. }
  54. }
  56. // 改进-end
  57. // 此处为原代码语句，注释之! $field = trim($field);
  58. // 此处为原代码语句，注释之! if (strpos($field, ',') && false !== $sepa) { // 多字段
  60. if($is_multi &&false!== $sepa)
  61. {
  62. // 多字段
  63. if(!isset($options['limit']))
  64. {
  65. 			$options['limit']= is_numeric($sepa)? $sepa :'';
  66. }
  67. 		$resultSet = $this->db->select($options);
  68. if(!empty($resultSet))
  69. {
  70. if(is_string($resultSet))
  71. {
  72. return $resultSet;
  73. }
  75. // 此处为原代码语句，注释之! $_field = explode(',', $field);
  76. 			$field = array_keys($resultSet[0]);
  77. 			$key1 = array_shift($field);
  78. 			$key2 = array_shift($field);
  79. 			$cols = array();
  80. 			$count = count($_field);
  81. foreach($resultSet as $result)
  82. {
  83. 				$name = $result[$key1];
  84. if(2== $count)
  85. {
  86. 					$cols[$name]= $result[$key2];
  87. }
  88. else
  89. {
  90. 					$cols[$name]= is_string($sepa)? implode($sepa, array_slice($result,1)): $result;
  91. }
  92. }
  94. if(isset($cache))
  95. {
  96. 				S($key, $cols, $cache);
  97. }
  99. return $cols;
  100. }
  101. }
  102. else
  103. {
  104. // 查找一条记录
  105. // 返回数据个数
  106. if(true!== $sepa)
  107. {
  108. // 当sepa指定为true的时候 返回所有数据
  109. 			$options['limit']= is_numeric($sepa)? $sepa :1;
  110. }
  112. 		$result = $this->db->select($options);
  113. if(!empty($result))
  114. {
  115. if(is_string($result))
  116. {
  117. return $result;
  118. }
  120. if(true!== $sepa &&1== $options['limit'])
  121. {
  122. 				$data = reset($result[0]);
  123. if(isset($cache))
  124. {
  125. 					S($key, $data, $cache);
  126. }
  127. return $data;
  128. }
  130. foreach($result as $val)
  131. {
  132. // 此处为原代码语句，注释之! $array[] = $val[$field];
  133. 				$array[]= $is_fn ? current($val): $val[$field];
  134. }
  136. if(isset($cache))
  137. {
  138. 				S($key, $array, $cache);
  139. }
  140. return $array;
  141. }
  142. }
  143. returnnull;
  144. }