mirror-fastadminnet/framework
1
0
Fork 0
mirror of https://gitee.com/fastadminnet/framework.git synced 2026-07-07 15:12:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

修正Driver类的get方法

Browse Source
This commit is contained in:
thinkphp
2016-04-07 14:54:27 +08:00
parent a9980abba8
commit 2a56e81fce
1 changed files with 43 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
library/think/db/Driver.php
View File

@@ -1004,20 +1004,24 @@ abstract class Driver
/** /**
* 得到某个字段的值 或者多个字段列数组 * 得到某个字段的值 或者多个字段列数组
* @access public * @access public
* @return string * @param string|array $field 字段名
* @param bool $resultSet 是否需要返回字段列表
* @return mixed
*/ */
public function get($field, $resultSet = false) public function get($field, $resultSet = false)
{ {
$options['field'] = $field;
// 返回数据个数 // 返回数据个数
if (!$resultSet) { if (!$resultSet) {
$options['limit'] = 1; $pdo = $this->field($field)->fetchPdo(true)->find();
} return $pdo->fetchColumn();
$result = $this->options($options)->select(); } else {
if (1 == $options['limit']) { $pdo = $this->field($field)->fetchPdo(true)->select();
$data = reset($result[0]); if (1 == $pdo->columnCount()) {
return $data; return $pdo->fetchAll(PDO::FETCH_COLUMN);
}
$result = $pdo->fetchAll(PDO::FETCH_ASSOC);
} }
$fields = array_keys($result[0]); $fields = array_keys($result[0]);
$count = count($fields); $count = count($fields);
$key1 = array_shift($fields); $key1 = array_shift($fields);
@@ -1027,33 +1031,61 @@ abstract class Driver
$array[$val[$key1]] = $val; $array[$val[$key1]] = $val;
} elseif (2 == $count) { } elseif (2 == $count) {
$array[$val[$key1]] = $val[$key2]; $array[$val[$key1]] = $val[$key2];
} else {
$array[] = $val[$key1];
} }
} }
return $array; return $array;
} }
/**
* COUNT查询
* @access public
* @param string $field 字段名
* @return integer
*/
public function count($field = '*') public function count($field = '*')
{ {
return $this->get('COUNT(' . $field . ') AS tp_count'); return $this->get('COUNT(' . $field . ') AS tp_count');
} }
/**
* SUM查询
* @access public
* @param string $field 字段名
* @return integer
*/
public function sum($field = '*') public function sum($field = '*')
{ {
return $this->get('SUM(' . $field . ') AS tp_sum'); return $this->get('SUM(' . $field . ') AS tp_sum');
} }
/**
* MIN查询
* @access public
* @param string $field 字段名
* @return integer
*/
public function min($field = '*') public function min($field = '*')
{ {
return $this->get('MIN(' . $field . ') AS tp_min'); return $this->get('MIN(' . $field . ') AS tp_min');
} }
/**
* MAX查询
* @access public
* @param string $field 字段名
* @return integer
*/
public function max($field = '*') public function max($field = '*')
{ {
return $this->get('MAX(' . $field . ') AS tp_max'); return $this->get('MAX(' . $field . ') AS tp_max');
} }
/**
* AVG查询
* @access public
* @param string $field 字段名
* @return integer
*/
public function avg($field = '*') public function avg($field = '*')
{ {
return $this->get('AVG(' . $field . ') AS tp_avg'); return $this->get('AVG(' . $field . ') AS tp_avg');
@@ -1902,7 +1934,7 @@ abstract class Driver
$options = $this->_parseOptions(); $options = $this->_parseOptions();
$sql = $this->buildSelectSql($options); $sql = $this->buildSelectSql($options);
$resultSet = $this->query($sql, $this->getBindParams(), !empty($options['fetch_sql']) ? true : false, !empty($options['master']) ? true : false, isset($options['fetch_pdo']) ? $options['fetch_pdo'] : true); $resultSet = $this->query($sql, $this->getBindParams(), !empty($options['fetch_sql']) ? true : false, !empty($options['master']) ? true : false, isset($options['fetch_pdo']) ? $options['fetch_pdo'] : false);
if (!empty($resultSet)) { if (!empty($resultSet)) {
if (is_string($resultSet)) { if (is_string($resultSet)) {