最近在研究ThinkPHP,我用的版本是2.1,数据库为MSSQL2005,在按示例操作的时候出现了一个问题:
BUG:Model类的对象使用count进行记录统计的时候结果总是为1
下面我们分析下源码,看看具体的执行流程:
1.Action中的代码如下
- $agentInfo = new Model("tblagentinfo");
- $agentInfoCount = $agentInfo->count();
下面我们一步步追踪。
(1)首先我们进入Model类的源码,位于:ThinkPHP/Lib/Think/Core/Model.class.php 。
我们搜索count方法。发现没有这个方法的定义,但我们可以搜索到这么一个方法__call。该方法是PHP5之后新增的魔术方法,作用就是如果调用类中不存在的方法,PHP会自动执行__call方法.具体用法这边我们不做介绍,可以参考这里:
我们看看__call中有如下的代码片段:
- elseif(in_array(strtolower($method),array('count','sum','min','max','avg'),true)){
- // 统计查询的实现
- $field = isset($args[0])?$args[0]:'*';
- return $this->getField(strtoupper($method).'('.$field.') AS tp_'.$method);
我们发现在调用Model类对象的count方法时,调用了Model对象的getField方法。根据代码我们能分析传递给getField方法的参数是个字符串:"count(*) as tp_count"
(2)我们再进入getField方法看看。
我们发现该方法中存在个if else语句,根据我们传入的参数现在执行的代码如下:
- else{ // 查找一条记录
- $options['limit'] = 1;
- $result = $this->db->select($options);
- if(!emptyempty($result)) {
- return reset($result[0]);
- }
- }
我们发现这里对$options变量做了个赋值并且又调用了Db类的select方法。其实这里TP调用的应该是DbMssql类的select方法,因为DbMssql继承自Db类,且没有对Db类的select进行重写,因此我们进入Db类看下。
(3)在Db类的select方法中有如下的代码片段:
- $sql = str_replace(
- array('%TABLE%','%DISTINCT%','%FIELDS%','%JOIN%','%WHERE%','%GROUP%','%HAVING%','%ORDER%','%LIMIT%'),
- array(
- $this->parseTable($options['table']),
- $this->parseDistinct(isset($options['distinct'])?$options['distinct']:false),
- $this->parseField(isset($options['field'])?$options['field']:'*'),
- $this->parseJoin(isset($options['join'])?$options['join']:''),
- $this->parseWhere(isset($options['where'])?$options['where']:''),
- $this->parseGroup(isset($options['group'])?$options['group']:''),
- $this->parseHaving(isset($options['having'])?$options['having']:''),
- $this->parseOrder(isset($options['order'])?$options['order']:''),
- $this->parseLimit(isset($options['limit'])?$options['limit']:'')
- ),$this->selectSql);
- $sql .= $this->parseLock(isset($options['lock'])?$options['lock']:false);
- return $this->query($sql);
这里的代码是对查询SQL进行了拼凑,这里大家我们要留意$selectSql变量,这个变量实际是在DbMssql中的定义的。接着代码又调用了query方法,在该方法中执行了具体的数据库查询操作,并将查询结果传入了该类中的getAll方法,在getAll方法中将查询结果存于$result数组中进行了返回。
(4)到目前为止执行的流程已经结束,下面就是将$result数组一层层的返回。我们回到第2部的代码片段。在调用了Db类的select后对返回的$result进行了判断并将最终的结果reset($result[0])返回给了Action。
(5)好,现在结果已经返回了,我们看下具体的问题出现在哪。我们开启TP的调试功能找出TP执行的SQL的具体代码,如下:
- SELECT T1.* FROM (SELECT ROW_NUMBER() OVER ( ORDER BY rand()) AS ROW_NUMBER, thinkphp.* FROM (SELECT COUNT(*) AS tp_count FROM tblagentinfo) AS thinkphp) AS T1 WHERE (T1.ROW_NUMBER BETWEEN 1 AND 1)
该语句执行的结果如下图:
(6)到这里问题出现的原因已经呼之欲出了,就是因为第4步中返回的结果是第一个字段的值,因此无论你怎么查询,结果都是1.那我们该如何处理呢?
我的处理方法如下:
将DbMsql类中全局变量$selectSql定义进行了修改,修改后的语句如下:
- protected $selectSql = 'SELECT T1.* FROM (SELECT thinkphp.*, ROW_NUMBER() OVER (%ORDER%) AS ROW_NUMBER FROM (SELECT %DISTINCT% %FIELDS% FROM %TABLE%%JOIN%%WHERE%%GROUP%%HAVING%) AS thinkphp) AS T1 WHERE %LIMIT%'
就是将查询结果中tp_count字段放在第一个字段即可。