欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Thinkphp使用mongodb数据库实现多条件查询方法

程序员文章站 2023-12-04 08:29:34
有个项目用了mongodb数据库,查询条件有and也有or,按thinkphp官方手册,使用复合查询(_complex),getlastsql输出查询语句,发现查询条件是空...

有个项目用了mongodb数据库,查询条件有and也有or,按thinkphp官方手册,使用复合查询(_complex),getlastsql输出查询语句,发现查询条件是空的.用字符串模式查询(_string),请求字符串查询(_query)无法满足需求.估计用mongodb的用户不多,thinkphp官方对这方面支持也不够.打开thinkphp的mongodb驱动,thinkphp/extend/driver/db/dbmongo.class.php,找到protected function parsethinkwhere($key,$val)方法,可以发现,switch里没有_complex,也就是说,thinkphp使用mongodb时,根本不支持复合查询.加上:

复制代码 代码如下:

case '_complex'://复合查询
             $arr   = array();
             foreach ($val as $nkey=>$nval){
              if( strpos($nkey,'_')!=0)
              {
               $parsearr=$this->parsewhereitem($nkey,$nval);
               //转换成对象
               $obj=new stdclass();
               foreach ($parsearr as $pkey=>$pval)
               {
                $obj->$pkey=$pval;
               }
               array_push($arr, $obj);
              }
             }
             if(isset($val['_logic']) && strtolower($val['_logic']) == 'or' ) {
              unset($val['_logic']);
              $query['$or']   =  $arr;
             }
             break;

这里之所以要转换成对象,是因为使用thinkphp使用json_encode函数生成查询语句,但是如果数组元素带key,json_encode函数会把数组转换成对象的形式,mongodb不能识别.因为目前只用到or,所以,代码只对or作了处理.
另外,发现个bug(不知道算不算),在parsewhere方法中:

复制代码 代码如下:

foreach ($where as $key=>$val){
            if('_id' != $key && 0===strpos($key,'_')) {
                // 解析特殊条件表达式
                //原 $query=$this->parsethinkwhere($key,$val);
                $query   = array_merge($query,$this->parsethinkwhere($key,$val));
            }else{
                // 查询字段的安全过滤
                if(!preg_match('/^[a-z_\|\&\-.a-z0-9]+$/',trim($key))){
                    throw_exception(l('_error_query_').':'.$key);
                }
                $key = trim($key);
                if(strpos($key,'|')) {
                    $array   =  explode('|',$key);
                    $str   = array();
                    foreach ($array as $k){
                        $str[]   = $this->parsewhereitem($k,$val);
                    }
                    $query['$or'] =    $str;
                }elseif(strpos($key,'&')){
                    $array   =  explode('&',$key);
                    $str   = array();
                    foreach ($array as $k){
                        $str[]   = $this->parsewhereitem($k,$val);
                    }
                    $query   = array_merge($query,$str);
                }else{
                    $str   = $this->parsewhereitem($key,$val);
                    $query   = array_merge($query,$str);
                }
            }
        }

解析特殊条件表达式时,源代码里是$query=$this->parsethinkwhere($key,$val);当特殊表达式在where数组里不是第一个元素时,就出错了,else里的代码得到的$query数组,都没了.