PageRenderTime 51ms CodeModel.GetById 19ms RepoModel.GetById 0ms app.codeStats 0ms

/ThinkPHP/Lib/Core/Db.class.php

https://github.com/huangshan/yuqing
PHP | 919 lines | 572 code | 42 blank | 305 comment | 81 complexity | 0cb2ced5c5a8b0cdebfb831a04f5fe2f MD5 | raw file
Possible License(s): MPL-2.0-no-copyleft-exception 
    

  1. <?php

    2. 

  2. // +----------------------------------------------------------------------

    3. 

  3. // | ThinkPHP [ WE CAN DO IT JUST THINK IT ]

    4. 

  4. // +----------------------------------------------------------------------

    5. 

  5. // | Copyright (c) 2006-2012 http://thinkphp.cn All rights reserved.

    6. 

  6. // +----------------------------------------------------------------------

    7. 

  7. // | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )

    8. 

  8. // +----------------------------------------------------------------------

    9. 

  9. // | Author: liu21st <liu21st@gmail.com>

    10. 

  10. // +----------------------------------------------------------------------

    11. 

  11. 

    12. 

  12. /**

    13. 

  13.  * ThinkPHP 数据库中间层实现类

    14. 

  14.  * @category   Think

    15. 

  15.  * @package  Think

    16. 

  16.  * @subpackage  Core

    17. 

  17.  * @author    liu21st <liu21st@gmail.com>

    18. 

  18.  */

    19. 

  19. class Db {

    20. 

  20.     // 数据库类型

    21. 

  21.     protected $dbType     = null;

    22. 

  22.     // 是否自动释放查询结果

    23. 

  23.     protected $autoFree   = false;

    24. 

  24.     // 当前操作所属的模型名

    25. 

  25.     protected $model      = '_think_';

    26. 

  26.     // 是否使用永久连接

    27. 

  27.     protected $pconnect   = false;

    28. 

  28.     // 当前SQL指令

    29. 

  29.     protected $queryStr   = '';

    30. 

  30.     protected $modelSql   = array();

    31. 

  31.     // 最后插入ID

    32. 

  32.     protected $lastInsID  = null;

    33. 

  33.     // 返回或者影响记录数

    34. 

  34.     protected $numRows    = 0;

    35. 

  35.     // 返回字段数

    36. 

  36.     protected $numCols    = 0;

    37. 

  37.     // 事务指令数

    38. 

  38.     protected $transTimes = 0;

    39. 

  39.     // 错误信息

    40. 

  40.     protected $error      = '';

    41. 

  41.     // 数据库连接ID 支持多个连接

    42. 

  42.     protected $linkID     = array();

    43. 

  43.     // 当前连接ID

    44. 

  44.     protected $_linkID    = null;

    45. 

  45.     // 当前查询ID

    46. 

  46.     protected $queryID    = null;

    47. 

  47.     // 是否已经连接数据库

    48. 

  48.     protected $connected  = false;

    49. 

  49.     // 数据库连接参数配置

    50. 

  50.     protected $config     = '';

    51. 

  51.     // 数据库表达式

    52. 

  52.     protected $comparison = array('eq'=>'=','neq'=>'<>','gt'=>'>','egt'=>'>=','lt'=>'<','elt'=>'<=','notlike'=>'NOT LIKE','like'=>'LIKE','in'=>'IN','notin'=>'NOT IN');

    53. 

  53.     // 查询表达式

    54. 

  54.     protected $selectSql  = 'SELECT%DISTINCT% %FIELD% FROM %TABLE%%JOIN%%WHERE%%GROUP%%HAVING%%ORDER%%LIMIT% %UNION%%COMMENT%';

    55. 

  55.     // 参数绑定

    56. 

  56.     protected $bind       = array();

    57. 

  57. 

    58. 

  58.     /**

    59. 

  59.      * 取得数据库类实例

    60. 

  60.      * @static

    61. 

  61.      * @access public

    62. 

  62.      * @return mixed 返回数据库驱动类

    63. 

  63.      */

    64. 

  64.     public static function getInstance() {

    65. 

  65.         $args = func_get_args();

    66. 

  66.         return get_instance_of(__CLASS__,'factory',$args);

    67. 

  67.     }

    68. 

  68. 

    69. 

  69.     /**

    70. 

  70.      * 加载数据库 支持配置文件或者 DSN

    71. 

  71.      * @access public

    72. 

  72.      * @param mixed $db_config 数据库配置信息

    73. 

  73.      * @return string

    74. 

  74.      */

    75. 

  75.     public function factory($db_config='') {

    76. 

  76.         // 读取数据库配置

    77. 

  77.         $db_config = $this->parseConfig($db_config);

    78. 

  78.         if(empty($db_config['dbms']))

    79. 

  79.             throw_exception(L('_NO_DB_CONFIG_'));

    80. 

  80.         // 数据库类型

    81. 

  81.         $this->dbType = ucwords(strtolower($db_config['dbms']));

    82. 

  82.         $class = 'Db'. $this->dbType;

    83. 

  83.         // 检查驱动类

    84. 

  84.         if(class_exists($class)) {

    85. 

  85.             $db = new $class($db_config);

    86. 

  86.             // 获取当前的数据库类型

    87. 

  87.             if( 'pdo' != strtolower($db_config['dbms']) )

    88. 

  88.                 $db->dbType = strtoupper($this->dbType);

    89. 

  89.             else

    90. 

  90.                 $db->dbType = $this->_getDsnType($db_config['dsn']);

    91. 

  91.         }else {

    92. 

  92.             // 类没有定义

    93. 

  93.             throw_exception(L('_NO_DB_DRIVER_').': ' . $class);

    94. 

  94.         }

    95. 

  95.         return $db;

    96. 

  96.     }

    97. 

  97. 

    98. 

  98.     /**

    99. 

  99.      * 根据DSN获取数据库类型 返回大写

    100. 

  100.      * @access protected

    101. 

  101.      * @param string $dsn  dsn字符串

    102. 

  102.      * @return string

    103. 

  103.      */

    104. 

  104.     protected function _getDsnType($dsn) {

    105. 

  105.         $match  =  explode(':',$dsn);

    106. 

  106.         $dbType = strtoupper(trim($match[0]));

    107. 

  107.         return $dbType;

    108. 

  108.     }

    109. 

  109. 

    110. 

  110.     /**

    111. 

  111.      * 分析数据库配置信息，支持数组和DSN

    112. 

  112.      * @access private

    113. 

  113.      * @param mixed $db_config 数据库配置信息

    114. 

  114.      * @return string

    115. 

  115.      */

    116. 

  116.     private function parseConfig($db_config='') {

    117. 

  117.         if ( !empty($db_config) && is_string($db_config)) {

    118. 

  118.             // 如果DSN字符串则进行解析

    119. 

  119.             $db_config = $this->parseDSN($db_config);

    120. 

  120.         }elseif(is_array($db_config)) { // 数组配置

    121. 

  121.              $db_config =   array_change_key_case($db_config);

    122. 

  122.              $db_config = array(

    123. 

  123.                   'dbms'      =>  $db_config['db_type'],

    124. 

  124.                   'username'  =>  $db_config['db_user'],

    125. 

  125.                   'password'  =>  $db_config['db_pwd'],

    126. 

  126.                   'hostname'  =>  $db_config['db_host'],

    127. 

  127.                   'hostport'  =>  $db_config['db_port'],

    128. 

  128.                   'database'  =>  $db_config['db_name'],

    129. 

  129.                   'dsn'       =>  $db_config['db_dsn'],

    130. 

  130.                   'params'    =>  $db_config['db_params'],

    131. 

  131.              );

    132. 

  132.         }elseif(empty($db_config)) {

    133. 

  133.             // 如果配置为空，读取配置文件设置

    134. 

  134.             if( C('DB_DSN') && 'pdo' != strtolower(C('DB_TYPE')) ) { // 如果设置了DB_DSN 则优先

    135. 

  135.                 $db_config =  $this->parseDSN(C('DB_DSN'));

    136. 

  136.             }else{

    137. 

  137.                 $db_config = array (

    138. 

  138.                     'dbms'      =>  C('DB_TYPE'),

    139. 

  139.                     'username'  =>  C('DB_USER'),

    140. 

  140.                     'password'  =>  C('DB_PWD'),

    141. 

  141.                     'hostname'  =>  C('DB_HOST'),

    142. 

  142.                     'hostport'  =>  C('DB_PORT'),

    143. 

  143.                     'database'  =>  C('DB_NAME'),

    144. 

  144.                     'dsn'       =>  C('DB_DSN'),

    145. 

  145.                     'params'    =>  C('DB_PARAMS'),

    146. 

  146.                 );

    147. 

  147.             }

    148. 

  148.         }

    149. 

  149.         return $db_config;

    150. 

  150.     }

    151. 

  151. 

    152. 

  152.     /**

    153. 

  153.      * 初始化数据库连接

    154. 

  154.      * @access protected

    155. 

  155.      * @param boolean $master 主服务器

    156. 

  156.      * @return void

    157. 

  157.      */

    158. 

  158.     protected function initConnect($master=true) {

    159. 

  159.         if(1 == C('DB_DEPLOY_TYPE'))

    160. 

  160.             // 采用分布式数据库

    161. 

  161.             $this->_linkID = $this->multiConnect($master);

    162. 

  162.         else

    163. 

  163.             // 默认单数据库

    164. 

  164.             if ( !$this->connected ) $this->_linkID = $this->connect();

    165. 

  165.     }

    166. 

  166. 

    167. 

  167.     /**

    168. 

  168.      * 连接分布式服务器

    169. 

  169.      * @access protected

    170. 

  170.      * @param boolean $master 主服务器

    171. 

  171.      * @return void

    172. 

  172.      */

    173. 

  173.     protected function multiConnect($master=false) {

    174. 

  174.         static $_config = array();

    175. 

  175.         if(empty($_config)) {

    176. 

  176.             // 缓存分布式数据库配置解析

    177. 

  177.             foreach ($this->config as $key=>$val){

    178. 

  178.                 $_config[$key]      =   explode(',',$val);

    179. 

  179.             }

    180. 

  180.         }

    181. 

  181.         // 数据库读写是否分离

    182. 

  182.         if(C('DB_RW_SEPARATE')){

    183. 

  183.             // 主从式采用读写分离

    184. 

  184.             if($master)

    185. 

  185.                 // 主服务器写入

    186. 

  186.                 $r  =   floor(mt_rand(0,C('DB_MASTER_NUM')-1));

    187. 

  187.             else{

    188. 

  188.                 if(is_numeric(C('DB_SLAVE_NO'))) {// 指定服务器读

    189. 

  189.                     $r = C('DB_SLAVE_NO');

    190. 

  190.                 }else{

    191. 

  191.                     // 读操作连接从服务器

    192. 

  192.                     $r = floor(mt_rand(C('DB_MASTER_NUM'),count($_config['hostname'])-1));   // 每次随机连接的数据库

    193. 

  193.                 }

    194. 

  194.             }

    195. 

  195.         }else{

    196. 

  196.             // 读写操作不区分服务器

    197. 

  197.             $r = floor(mt_rand(0,count($_config['hostname'])-1));   // 每次随机连接的数据库

    198. 

  198.         }

    199. 

  199.         $db_config = array(

    200. 

  200.             'username'  =>  isset($_config['username'][$r])?$_config['username'][$r]:$_config['username'][0],

    201. 

  201.             'password'  =>  isset($_config['password'][$r])?$_config['password'][$r]:$_config['password'][0],

    202. 

  202.             'hostname'  =>  isset($_config['hostname'][$r])?$_config['hostname'][$r]:$_config['hostname'][0],

    203. 

  203.             'hostport'  =>  isset($_config['hostport'][$r])?$_config['hostport'][$r]:$_config['hostport'][0],

    204. 

  204.             'database'  =>  isset($_config['database'][$r])?$_config['database'][$r]:$_config['database'][0],

    205. 

  205.             'dsn'       =>  isset($_config['dsn'][$r])?$_config['dsn'][$r]:$_config['dsn'][0],

    206. 

  206.             'params'    =>  isset($_config['params'][$r])?$_config['params'][$r]:$_config['params'][0],

    207. 

  207.         );

    208. 

  208.         return $this->connect($db_config,$r);

    209. 

  209.     }

    210. 

  210. 

    211. 

  211.     /**

    212. 

  212.      * DSN解析

    213. 

  213.      * 格式： mysql://username:passwd@localhost:3306/DbName

    214. 

  214.      * @static

    215. 

  215.      * @access public

    216. 

  216.      * @param string $dsnStr

    217. 

  217.      * @return array

    218. 

  218.      */

    219. 

  219.     public function parseDSN($dsnStr) {

    220. 

  220.         if( empty($dsnStr) ){return false;}

    221. 

  221.         $info = parse_url($dsnStr);

    222. 

  222.         if($info['scheme']){

    223. 

  223.             $dsn = array(

    224. 

  224.             'dbms'      =>  $info['scheme'],

    225. 

  225.             'username'  =>  isset($info['user']) ? $info['user'] : '',

    226. 

  226.             'password'  =>  isset($info['pass']) ? $info['pass'] : '',

    227. 

  227.             'hostname'  =>  isset($info['host']) ? $info['host'] : '',

    228. 

  228.             'hostport'  =>  isset($info['port']) ? $info['port'] : '',

    229. 

  229.             'database'  =>  isset($info['path']) ? substr($info['path'],1) : ''

    230. 

  230.             );

    231. 

  231.         }else {

    232. 

  232.             preg_match('/^(.*?)\:\/\/(.*?)\:(.*?)\@(.*?)\:([0-9]{1, 6})\/(.*?)$/',trim($dsnStr),$matches);

    233. 

  233.             $dsn = array (

    234. 

  234.             'dbms'      =>  $matches[1],

    235. 

  235.             'username'  =>  $matches[2],

    236. 

  236.             'password'  =>  $matches[3],

    237. 

  237.             'hostname'  =>  $matches[4],

    238. 

  238.             'hostport'  =>  $matches[5],

    239. 

  239.             'database'  =>  $matches[6]

    240. 

  240.             );

    241. 

  241.         }

    242. 

  242.         $dsn['dsn'] =  ''; // 兼容配置信息数组

    243. 

  243.         return $dsn;

    244. 

  244.      }

    245. 

  245. 

    246. 

  246.     /**

    247. 

  247.      * 数据库调试 记录当前SQL

    248. 

  248.      * @access protected

    249. 

  249.      */

    250. 

  250.     protected function debug() {

    251. 

  251.         $this->modelSql[$this->model]   =  $this->queryStr;

    252. 

  252.         $this->model  =   '_think_';

    253. 

  253.         // 记录操作结束时间

    254. 

  254.         if (C('DB_SQL_LOG')) {

    255. 

  255.             G('queryEndTime');

    256. 

  256.             trace($this->queryStr.' [ RunTime:'.G('queryStartTime','queryEndTime',6).'s ]','','SQL');

    257. 

  257.         }

    258. 

  258.     }

    259. 

  259. 

    260. 

  260.     /**

    261. 

  261.      * 设置锁机制

    262. 

  262.      * @access protected

    263. 

  263.      * @return string

    264. 

  264.      */

    265. 

  265.     protected function parseLock($lock=false) {

    266. 

  266.         if(!$lock) return '';

    267. 

  267.         if('ORACLE' == $this->dbType) {

    268. 

  268.             return ' FOR UPDATE NOWAIT ';

    269. 

  269.         }

    270. 

  270.         return ' FOR UPDATE ';

    271. 

  271.     }

    272. 

  272. 

    273. 

  273.     /**

    274. 

  274.      * set分析

    275. 

  275.      * @access protected

    276. 

  276.      * @param array $data

    277. 

  277.      * @return string

    278. 

  278.      */

    279. 

  279.     protected function parseSet($data) {

    280. 

  280.         foreach ($data as $key=>$val){

    281. 

  281.             if(is_array($val) && 'exp' == $val[0]){

    282. 

  282.                 $set[]  =   $this->parseKey($key).'='.$this->escapeString($val[1]);

    283. 

  283.             }elseif(is_scalar($val)) { // 过滤非标量数据

    284. 

  284.               if(C('DB_BIND_PARAM') && 0 !== strpos($val,':')){

    285. 

  285.                 $name   =   md5($key);

    286. 

  286.                 $set[]  =   $this->parseKey($key).'=:'.$name;

    287. 

  287.                 $this->bindParam($name,$val);

    288. 

  288.               }else{

    289. 

  289.                 $set[]  =   $this->parseKey($key).'='.$this->parseValue($val);

    290. 

  290.               }

    291. 

  291.             }

    292. 

  292.         }

    293. 

  293.         return ' SET '.implode(',',$set);

    294. 

  294.     }

    295. 

  295. 

    296. 

  296.      /**

    297. 

  297.      * 参数绑定

    298. 

  298.      * @access protected

    299. 

  299.      * @param string $name 绑定参数名

    300. 

  300.      * @param mixed $value 绑定值

    301. 

  301.      * @return void

    302. 

  302.      */

    303. 

  303.     protected function bindParam($name,$value){

    304. 

  304.         $this->bind[':'.$name]  =   $value;

    305. 

  305.     }

    306. 

  306. 

    307. 

  307.      /**

    308. 

  308.      * 参数绑定分析

    309. 

  309.      * @access protected

    310. 

  310.      * @param array $bind

    311. 

  311.      * @return array

    312. 

  312.      */

    313. 

  313.     protected function parseBind($bind){

    314. 

  314.         $bind           =   array_merge($this->bind,$bind);

    315. 

  315.         $this->bind     =   array();

    316. 

  316.         return $bind;

    317. 

  317.     }

    318. 

  318. 

    319. 

  319.     /**

    320. 

  320.      * 字段名分析

    321. 

  321.      * @access protected

    322. 

  322.      * @param string $key

    323. 

  323.      * @return string

    324. 

  324.      */

    325. 

  325.     protected function parseKey(&$key) {

    326. 

  326.         return $key;

    327. 

  327.     }

    328. 

  328.     

    329. 

  329.     /**

    330. 

  330.      * value分析

    331. 

  331.      * @access protected

    332. 

  332.      * @param mixed $value

    333. 

  333.      * @return string

    334. 

  334.      */

    335. 

  335.     protected function parseValue($value) {

    336. 

  336.         if(is_string($value)) {

    337. 

  337.             $value =  '\''.$this->escapeString($value).'\'';

    338. 

  338.         }elseif(isset($value[0]) && is_string($value[0]) && strtolower($value[0]) == 'exp'){

    339. 

  339.             $value =  $this->escapeString($value[1]);

    340. 

  340.         }elseif(is_array($value)) {

    341. 

  341.             $value =  array_map(array($this, 'parseValue'),$value);

    342. 

  342.         }elseif(is_bool($value)){

    343. 

  343.             $value =  $value ? '1' : '0';

    344. 

  344.         }elseif(is_null($value)){

    345. 

  345.             $value =  'null';

    346. 

  346.         }

    347. 

  347.         return $value;

    348. 

  348.     }

    349. 

  349. 

    350. 

  350.     /**

    351. 

  351.      * field分析

    352. 

  352.      * @access protected

    353. 

  353.      * @param mixed $fields

    354. 

  354.      * @return string

    355. 

  355.      */

    356. 

  356.     protected function parseField($fields) {

    357. 

  357.         if(is_string($fields) && strpos($fields,',')) {

    358. 

  358.             $fields    = explode(',',$fields);

    359. 

  359.         }

    360. 

  360.         if(is_array($fields)) {

    361. 

  361.             // 完善数组方式传字段名的支持

    362. 

  362.             // 支持 'field1'=>'field2' 这样的字段别名定义

    363. 

  363.             $array   =  array();

    364. 

  364.             foreach ($fields as $key=>$field){

    365. 

  365.                 if(!is_numeric($key))

    366. 

  366.                     $array[] =  $this->parseKey($key).' AS '.$this->parseKey($field);

    367. 

  367.                 else

    368. 

  368.                     $array[] =  $this->parseKey($field);

    369. 

  369.             }

    370. 

  370.             $fieldsStr = implode(',', $array);

    371. 

  371.         }elseif(is_string($fields) && !empty($fields)) {

    372. 

  372.             $fieldsStr = $this->parseKey($fields);

    373. 

  373.         }else{

    374. 

  374.             $fieldsStr = '*';

    375. 

  375.         }

    376. 

  376.         //TODO 如果是查询全部字段，并且是join的方式，那么就把要查的表加个别名，以免字段被覆盖

    377. 

  377.         return $fieldsStr;

    378. 

  378.     }

    379. 

  379. 

    380. 

  380.     /**

    381. 

  381.      * table分析

    382. 

  382.      * @access protected

    383. 

  383.      * @param mixed $table

    384. 

  384.      * @return string

    385. 

  385.      */

    386. 

  386.     protected function parseTable($tables) {

    387. 

  387.         if(is_array($tables)) {// 支持别名定义

    388. 

  388.             $array   =  array();

    389. 

  389.             foreach ($tables as $table=>$alias){

    390. 

  390.                 if(!is_numeric($table))

    391. 

  391.                     $array[] =  $this->parseKey($table).' '.$this->parseKey($alias);

    392. 

  392.                 else

    393. 

  393.                     $array[] =  $this->parseKey($table);

    394. 

  394.             }

    395. 

  395.             $tables  =  $array;

    396. 

  396.         }elseif(is_string($tables)){

    397. 

  397.             $tables  =  explode(',',$tables);

    398. 

  398.             array_walk($tables, array(&$this, 'parseKey'));

    399. 

  399.         }

    400. 

  400.         return implode(',',$tables);

    401. 

  401.     }

    402. 

  402. 

    403. 

  403.     /**

    404. 

  404.      * where分析

    405. 

  405.      * @access protected

    406. 

  406.      * @param mixed $where

    407. 

  407.      * @return string

    408. 

  408.      */

    409. 

  409.     protected function parseWhere($where) {

    410. 

  410.         $whereStr = '';

    411. 

  411.         if(is_string($where)) {

    412. 

  412.             // 直接使用字符串条件

    413. 

  413.             $whereStr = $where;

    414. 

  414.         }else{ // 使用数组表达式

    415. 

  415.             $operate  = isset($where['_logic'])?strtoupper($where['_logic']):'';

    416. 

  416.             if(in_array($operate,array('AND','OR','XOR'))){

    417. 

  417.                 // 定义逻辑运算规则 例如 OR XOR AND NOT

    418. 

  418.                 $operate    =   ' '.$operate.' ';

    419. 

  419.                 unset($where['_logic']);

    420. 

  420.             }else{

    421. 

  421.                 // 默认进行 AND 运算

    422. 

  422.                 $operate    =   ' AND ';

    423. 

  423.             }

    424. 

  424.             foreach ($where as $key=>$val){

    425. 

  425.                 $whereStr .= '( ';

    426. 

  426.                 if(0===strpos($key,'_')) {

    427. 

  427.                     // 解析特殊条件表达式

    428. 

  428.                     $whereStr   .= $this->parseThinkWhere($key,$val);

    429. 

  429.                 }else{

    430. 

  430.                     // 查询字段的安全过滤

    431. 

  431.                     if(!preg_match('/^[A-Z_\|\&\-.a-z0-9\(\)\,]+$/',trim($key))){

    432. 

  432.                         throw_exception(L('_EXPRESS_ERROR_').':'.$key);

    433. 

  433.                     }

    434. 

  434.                     // 多条件支持

    435. 

  435.                     $multi  = is_array($val) &&  isset($val['_multi']);

    436. 

  436.                     $key    = trim($key);

    437. 

  437.                     if(strpos($key,'|')) { // 支持 name|title|nickname 方式定义查询字段

    438. 

  438.                         $array =  explode('|',$key);

    439. 

  439.                         $str   =  array();

    440. 

  440.                         foreach ($array as $m=>$k){

    441. 

  441.                             $v =  $multi?$val[$m]:$val;

    442. 

  442.                             $str[]   = '('.$this->parseWhereItem($this->parseKey($k),$v).')';

    443. 

  443.                         }

    444. 

  444.                         $whereStr .= implode(' OR ',$str);

    445. 

  445.                     }elseif(strpos($key,'&')){

    446. 

  446.                         $array =  explode('&',$key);

    447. 

  447.                         $str   =  array();

    448. 

  448.                         foreach ($array as $m=>$k){

    449. 

  449.                             $v =  $multi?$val[$m]:$val;

    450. 

  450.                             $str[]   = '('.$this->parseWhereItem($this->parseKey($k),$v).')';

    451. 

  451.                         }

    452. 

  452.                         $whereStr .= implode(' AND ',$str);

    453. 

  453.                     }else{

    454. 

  454.                         $whereStr .= $this->parseWhereItem($this->parseKey($key),$val);

    455. 

  455.                     }

    456. 

  456.                 }

    457. 

  457.                 $whereStr .= ' )'.$operate;

    458. 

  458.             }

    459. 

  459.             $whereStr = substr($whereStr,0,-strlen($operate));

    460. 

  460.         }

    461. 

  461.         return empty($whereStr)?'':' WHERE '.$whereStr;

    462. 

  462.     }

    463. 

  463. 

    464. 

  464.     // where子单元分析

    465. 

  465.     protected function parseWhereItem($key,$val) {

    466. 

  466.         $whereStr = '';

    467. 

  467.         if(is_array($val)) {

    468. 

  468.             if(is_string($val[0])) {

    469. 

  469.                 if(preg_match('/^(EQ|NEQ|GT|EGT|LT|ELT)$/i',$val[0])) { // 比较运算

    470. 

  470.                     $whereStr .= $key.' '.$this->comparison[strtolower($val[0])].' '.$this->parseValue($val[1]);

    471. 

  471.                 }elseif(preg_match('/^(NOTLIKE|LIKE)$/i',$val[0])){// 模糊查找

    472. 

  472.                     if(is_array($val[1])) {

    473. 

  473.                         $likeLogic  =   isset($val[2])?strtoupper($val[2]):'OR';

    474. 

  474.                         if(in_array($likeLogic,array('AND','OR','XOR'))){

    475. 

  475.                             $likeStr    =   $this->comparison[strtolower($val[0])];

    476. 

  476.                             $like       =   array();

    477. 

  477.                             foreach ($val[1] as $item){

    478. 

  478.                                 $like[] = $key.' '.$likeStr.' '.$this->parseValue($item);

    479. 

  479.                             }

    480. 

  480.                             $whereStr .= '('.implode(' '.$likeLogic.' ',$like).')';                          

    481. 

  481.                         }

    482. 

  482.                     }else{

    483. 

  483.                         $whereStr .= $key.' '.$this->comparison[strtolower($val[0])].' '.$this->parseValue($val[1]);

    484. 

  484.                     }

    485. 

  485.                 }elseif('exp'==strtolower($val[0])){ // 使用表达式

    486. 

  486.                     $whereStr .= ' ('.$key.' '.$val[1].') ';

    487. 

  487.                 }elseif(preg_match('/IN/i',$val[0])){ // IN 运算

    488. 

  488.                     if(isset($val[2]) && 'exp'==$val[2]) {

    489. 

  489.                         $whereStr .= $key.' '.strtoupper($val[0]).' '.$val[1];

    490. 

  490.                     }else{

    491. 

  491.                         if(is_string($val[1])) {

    492. 

  492.                              $val[1] =  explode(',',$val[1]);

    493. 

  493.                         }

    494. 

  494.                         $zone      =   implode(',',$this->parseValue($val[1]));

    495. 

  495.                         $whereStr .= $key.' '.strtoupper($val[0]).' ('.$zone.')';

    496. 

  496.                     }

    497. 

  497.                 }elseif(preg_match('/BETWEEN/i',$val[0])){ // BETWEEN运算

    498. 

  498.                     $data = is_string($val[1])? explode(',',$val[1]):$val[1];

    499. 

  499.                     $whereStr .=  ' ('.$key.' '.strtoupper($val[0]).' '.$this->parseValue($data[0]).' AND '.$this->parseValue($data[1]).' )';

    500. 

  500.                 }else{

    501. 

  501.                     throw_exception(L('_EXPRESS_ERROR_').':'.$val[0]);

    502. 

  502.                 }

    503. 

  503.             }else {

    504. 

  504.                 $count = count($val);

    505. 

  505.                 $rule  = isset($val[$count-1])?strtoupper($val[$count-1]):'';

    506. 

  506.                 if(in_array($rule,array('AND','OR','XOR'))) {

    507. 

  507.                     $count  = $count -1;

    508. 

  508.                 }else{

    509. 

  509.                     $rule   = 'AND';

    510. 

  510.                 }

    511. 

  511.                 for($i=0;$i<$count;$i++) {

    512. 

  512.                     $data = is_array($val[$i])?$val[$i][1]:$val[$i];

    513. 

  513.                     if('exp'==strtolower($val[$i][0])) {

    514. 

  514.                         $whereStr .= '('.$key.' '.$data.') '.$rule.' ';

    515. 

  515.                     }else{

    516. 

  516.                         $op = is_array($val[$i])?$this->comparison[strtolower($val[$i][0])]:'=';

    517. 

  517.                         $whereStr .= '('.$key.' '.$op.' '.$this->parseValue($data).') '.$rule.' ';

    518. 

  518.                     }

    519. 

  519.                 }

    520. 

  520.                 $whereStr = substr($whereStr,0,-4);

    521. 

  521.             }

    522. 

  522.         }else {

    523. 

  523.             //对字符串类型字段采用模糊匹配

    524. 

  524.             if(C('DB_LIKE_FIELDS') && preg_match('/('.C('DB_LIKE_FIELDS').')/i',$key)) {

    525. 

  525.                 $val  =  '%'.$val.'%';

    526. 

  526.                 $whereStr .= $key.' LIKE '.$this->parseValue($val);

    527. 

  527.             }else {

    528. 

  528.                 $whereStr .= $key.' = '.$this->parseValue($val);

    529. 

  529.             }

    530. 

  530.         }

    531. 

  531.         return $whereStr;

    532. 

  532.     }

    533. 

  533. 

    534. 

  534.     /**

    535. 

  535.      * 特殊条件分析

    536. 

  536.      * @access protected

    537. 

  537.      * @param string $key

    538. 

  538.      * @param mixed $val

    539. 

  539.      * @return string

    540. 

  540.      */

    541. 

  541.     protected function parseThinkWhere($key,$val) {

    542. 

  542.         $whereStr   = '';

    543. 

  543.         switch($key) {

    544. 

  544.             case '_string':

    545. 

  545.                 // 字符串模式查询条件

    546. 

  546.                 $whereStr = $val;

    547. 

  547.                 break;

    548. 

  548.             case '_complex':

    549. 

  549.                 // 复合查询条件

    550. 

  550.                 $whereStr = substr($this->parseWhere($val),6);

    551. 

  551.                 break;

    552. 

  552.             case '_query':

    553. 

  553.                 // 字符串模式查询条件

    554. 

  554.                 parse_str($val,$where);

    555. 

  555.                 if(isset($where['_logic'])) {

    556. 

  556.                     $op   =  ' '.strtoupper($where['_logic']).' ';

    557. 

  557.                     unset($where['_logic']);

    558. 

  558.                 }else{

    559. 

  559.                     $op   =  ' AND ';

    560. 

  560.                 }

    561. 

  561.                 $array   =  array();

    562. 

  562.                 foreach ($where as $field=>$data)

    563. 

  563.                     $array[] = $this->parseKey($field).' = '.$this->parseValue($data);

    564. 

  564.                 $whereStr   = implode($op,$array);

    565. 

  565.                 break;

    566. 

  566.         }

    567. 

  567.         return $whereStr;

    568. 

  568.     }

    569. 

  569. 

    570. 

  570.     /**

    571. 

  571.      * limit分析

    572. 

  572.      * @access protected

    573. 

  573.      * @param mixed $lmit

    574. 

  574.      * @return string

    575. 

  575.      */

    576. 

  576.     protected function parseLimit($limit) {

    577. 

  577.         return !empty($limit)?   ' LIMIT '.$limit.' ':'';

    578. 

  578.     }

    579. 

  579. 

    580. 

  580.     /**

    581. 

  581.      * join分析

    582. 

  582.      * @access protected

    583. 

  583.      * @param mixed $join

    584. 

  584.      * @return string

    585. 

  585.      */

    586. 

  586.     protected function parseJoin($join) {

    587. 

  587.         $joinStr = '';

    588. 

  588.         if(!empty($join)) {

    589. 

  589.             if(is_array($join)) {

    590. 

  590.                 foreach ($join as $key=>$_join){

    591. 

  591.                     if(false !== stripos($_join,'JOIN'))

    592. 

  592.                         $joinStr .= ' '.$_join;

    593. 

  593.                     else

    594. 

  594.                         $joinStr .= ' LEFT JOIN ' .$_join;

    595. 

  595.                 }

    596. 

  596.             }else{

    597. 

  597.                 $joinStr .= ' LEFT JOIN ' .$join;

    598. 

  598.             }

    599. 

  599.         }

    600. 

  600. 		//将__TABLE_NAME__这样的字符串替换成正规的表名,并且带上前缀和后缀

    601. 

  601. 		$joinStr = preg_replace("/__([A-Z_-]+)__/esU",C("DB_PREFIX").".strtolower('$1')",$joinStr);

    602. 

  602.         return $joinStr;

    603. 

  603.     }

    604. 

  604. 

    605. 

  605.     /**

    606. 

  606.      * order分析

    607. 

  607.      * @access protected

    608. 

  608.      * @param mixed $order

    609. 

  609.      * @return string

    610. 

  610.      */

    611. 

  611.     protected function parseOrder($order) {

    612. 

  612.         if(is_array($order)) {

    613. 

  613.             $array   =  array();

    614. 

  614.             foreach ($order as $key=>$val){

    615. 

  615.                 if(is_numeric($key)) {

    616. 

  616.                     $array[] =  $this->parseKey($val);

    617. 

  617.                 }else{

    618. 

  618.                     $array[] =  $this->parseKey($key).' '.$val;

    619. 

  619.                 }

    620. 

  620.             }

    621. 

  621.             $order   =  implode(',',$array);

    622. 

  622.         }

    623. 

  623.         return !empty($order)?  ' ORDER BY '.$order:'';

    624. 

  624.     }

    625. 

  625. 

    626. 

  626.     /**

    627. 

  627.      * group分析

    628. 

  628.      * @access protected

    629. 

  629.      * @param mixed $group

    630. 

  630.      * @return string

    631. 

  631.      */

    632. 

  632.     protected function parseGroup($group) {

    633. 

  633.         return !empty($group)? ' GROUP BY '.$group:'';

    634. 

  634.     }

    635. 

  635. 

    636. 

  636.     /**

    637. 

  637.      * having分析

    638. 

  638.      * @access protected

    639. 

  639.      * @param string $having

    640. 

  640.      * @return string

    641. 

  641.      */

    642. 

  642.     protected function parseHaving($having) {

    643. 

  643.         return  !empty($having)?   ' HAVING '.$having:'';

    644. 

  644.     }

    645. 

  645. 

    646. 

  646.     /**

    647. 

  647.      * comment分析

    648. 

  648.      * @access protected

    649. 

  649.      * @param string $comment

    650. 

  650.      * @return string

    651. 

  651.      */

    652. 

  652.     protected function parseComment($comment) {

    653. 

  653.         return  !empty($comment)?   ' /* '.$comment.' */':'';

    654. 

  654.     }

    655. 

  655. 

    656. 

  656.     /**

    657. 

  657.      * distinct分析

    658. 

  658.      * @access protected

    659. 

  659.      * @param mixed $distinct

    660. 

  660.      * @return string

    661. 

  661.      */

    662. 

  662.     protected function parseDistinct($distinct) {

    663. 

  663.         return !empty($distinct)?   ' DISTINCT ' :'';

    664. 

  664.     }

    665. 

  665. 

    666. 

  666.     /**

    667. 

  667.      * union分析

    668. 

  668.      * @access protected

    669. 

  669.      * @param mixed $union

    670. 

  670.      * @return string

    671. 

  671.      */

    672. 

  672.     protected function parseUnion($union) {

    673. 

  673.         if(empty($union)) return '';

    674. 

  674.         if(isset($union['_all'])) {

    675. 

  675.             $str  =   'UNION ALL ';

    676. 

  676.             unset($union['_all']);

    677. 

  677.         }else{

    678. 

  678.             $str  =   'UNION ';

    679. 

  679.         }

    680. 

  680.         foreach ($union as $u){

    681. 

  681.             $sql[] = $str.(is_array($u)?$this->buildSelectSql($u):$u);

    682. 

  682.         }

    683. 

  683.         return implode(' ',$sql);

    684. 

  684.     }

    685. 

  685. 

    686. 

  686.     /**

    687. 

  687.      * 插入记录

    688. 

  688.      * @access public

    689. 

  689.      * @param mixed $data 数据

    690. 

  690.      * @param array $options 参数表达式

    691. 

  691.      * @param boolean $replace 是否replace

    692. 

  692.      * @return false | integer

    693. 

  693.      */

    694. 

  694.     public function insert($data,$options=array(),$replace=false) {

    695. 

  695.         $values  =  $fields    = array();

    696. 

  696.         $this->model  =   $options['model'];

    697. 

  697.         foreach ($data as $key=>$val){

    698. 

  698.             if(is_array($val) && 'exp' == $val[0]){

    699. 

  699.                 $fields[]   =  $this->parseKey($key);

    700. 

  700.                 $values[]   =  $this->escapeString($val[1]);

    701. 

  701.             }elseif(is_scalar($val)) { // 过滤非标量数据

    702. 

  702.               $fields[]   =  $this->parseKey($key);

    703. 

  703.               if(C('DB_BIND_PARAM') && 0 !== strpos($val,':')){

    704. 

  704.                 $name       =   md5($key);

    705. 

  705.                 $values[]   =   ':'.$name;

    706. 

  706.                 $this->bindParam($name,$val);

    707. 

  707.               }else{

    708. 

  708.                 $values[]   =  $this->parseValue($val);

    709. 

  709.               }                

    710. 

  710.             }

    711. 

  711.         }

    712. 

  712.         $sql   =  ($replace?'REPLACE':'INSERT').' INTO '.$this->parseTable($options['table']).' ('.implode(',', $fields).') VALUES ('.implode(',', $values).')';

    713. 

  713.         $sql   .= $this->parseLock(isset($options['lock'])?$options['lock']:false);

    714. 

  714.         $sql   .= $this->parseComment(!empty($options['comment'])?$options['comment']:'');

    715. 

  715.         return $this->execute($sql,$this->parseBind(!empty($options['bind'])?$options['bind']:array()));

    716. 

  716.     }

    717. 

  717. 

    718. 

  718.     /**

    719. 

  719.      * 通过Select方式插入记录

    720. 

  720.      * @access public

    721. 

  721.      * @param string $fields 要插入的数据表字段名

    722. 

  722.      * @param string $table 要插入的数据表名

    723. 

  723.      * @param array $option  查询数据参数

    724. 

  724.      * @return false | integer

    725. 

  725.      */

    726. 

  726.     public function selectInsert($fields,$table,$options=array()) {

    727. 

  727.         $this->model  =   $options['model'];

    728. 

  728.         if(is_string($fields))   $fields    = explode(',',$fields);

    729. 

  729.         array_walk($fields, array($this, 'parseKey'));

    730. 

  730.         $sql   =    'INSERT INTO '.$this->parseTable($table).' ('.implode(',', $fields).') ';

    731. 

  731.         $sql   .= $this->buildSelectSql($options);

    732. 

  732.         return $this->execute($sql,$this->parseBind(!empty($options['bind'])?$options['bind']:array()));

    733. 

  733.     }

    734. 

  734. 

    735. 

  735.     /**

    736. 

  736.      * 更新记录

    737. 

  737.      * @access public

    738. 

  738.      * @param mixed $data 数据

    739. 

  739.      * @param array $options 表达式

    740. 

  740.      * @return false | integer

    741. 

  741.      */

    742. 

  742.     public function update($data,$options) {

    743. 

  743.         $this->model  =   $options['model'];

    744. 

  744.         $sql   = 'UPDATE '

    745. 

  745.             .$this->parseTable($options['table'])

    746. 

  746.             .$this->parseSet($data)

    747. 

  747.             .$this->parseWhere(!empty($options['where'])?$options['where']:'')

    748. 

  748.             .$this->parseOrder(!empty($options['order'])?$options['order']:'')

    749. 

  749.             .$this->parseLimit(!empty($options['limit'])?$options['limit']:'')

    750. 

  750.             .$this->parseLock(isset($options['lock'])?$options['lock']:false)

    751. 

  751.             .$this->parseComment(!empty($options['comment'])?$options['comment']:'');

    752. 

  752.         return $this->execute($sql,$this->parseBind(!empty($options['bind'])?$options['bind']:array()));

    753. 

  753.     }

    754. 

  754. 

    755. 

  755.     /**

    756. 

  756.      * 删除记录

    757. 

  757.      * @access public

    758. 

  758.      * @param array $options 表达式

    759. 

  759.      * @return false | integer

    760. 

  760.      */

    761. 

  761.     public function delete($options=array()) {

    762. 

  762.         $this->model  =   $options['model'];

    763. 

  763.         $sql   = 'DELETE FROM '

    764. 

  764.             .$this->parseTable($options['table'])

    765. 

  765.             .$this->parseWhere(!empty($options['where'])?$options['where']:'')

    766. 

  766.             .$this->parseOrder(!empty($options['order'])?$options['order']:'')

    767. 

  767.             .$this->parseLimit(!empty($options['limit'])?$options['limit']:'')

    768. 

  768.             .$this->parseLock(isset($options['lock'])?$options['lock']:false)

    769. 

  769.             .$this->parseComment(!empty($options['comment'])?$options['comment']:'');

    770. 

  770.         return $this->execute($sql,$this->parseBind(!empty($options['bind'])?$options['bind']:array()));

    771. 

  771.     }

    772. 

  772. 

    773. 

  773.     /**

    774. 

  774.      * 查找记录

    775. 

  775.      * @access public

    776. 

  776.      * @param array $options 表达式

    777. 

  777.      * @return mixed

    778. 

  778.      */

    779. 

  779.     public function select($options=array()) {

    780. 

  780.         $this->model  =   $options['model'];

    781. 

  781.         $sql    = $this->buildSelectSql($options);

    782. 

  782.         $cache  =  isset($options['cache'])?$options['cache']:false;

    783. 

  783.         if($cache) { // 查询缓存检测

    784. 

  784.             $key    =  is_string($cache['key'])?$cache['key']:md5($sql);

    785. 

  785.             $value  =  S($key,'',$cache);

    786. 

  786.             if(false !== $value) {

    787. 

  787.                 return $value;

    788. 

  788.             }

    789. 

  789.         }

    790. 

  790.         $result   = $this->query($sql,$this->parseBind(!empty($options['bind'])?$options['bind']:array()));

    791. 

  791.         if($cache && false !== $result ) { // 查询缓存写入

    792. 

  792.             S($key,$result,$cache);

    793. 

  793.         }

    794. 

  794.         return $result;

    795. 

  795.     }

    796. 

  796. 

    797. 

  797.     /**

    798. 

  798.      * 生成查询SQL

    799. 

  799.      * @access public

    800. 

  800.      * @param array $options 表达式

    801. 

  801.      * @return string

    802. 

  802.      */

    803. 

  803.     public function buildSelectSql($options=array()) {

    804. 

  804.         if(isset($options['page'])) {

    805. 

  805.             // 根据页数计算limit

    806. 

  806.             if(strpos($options['page'],',')) {

    807. 

  807.                 list($page,$listRows) =  explode(',',$options['page']);

    808. 

  808.             }else{

    809. 

  809.                 $page = $options['page'];

    810. 

  810.             }

    811. 

  811.             $page    =  $page?$page:1;

    812. 

  812.             $listRows=  isset($listRows)?$listRows:(is_numeric($options['limit'])?$options['limit']:20);

    813. 

  813.             $offset  =  $listRows*((int)$page-1);

    814. 

  814.             $options['limit'] =  $offset.','.$listRows;

    815. 

  815.         }

    816. 

  816.         if(C('DB_SQL_BUILD_CACHE')) { // SQL创建缓存

    817. 

  817.             $key    =  md5(serialize($options));

    818. 

  818.             $value  =  S($key);

    819. 

  819.             if(false !== $value) {

    820. 

  820.                 return $value;

    821. 

  821.             }

    822. 

  822.         }

    823. 

  823.         $sql  =   $this->parseSql($this->selectSql,$options);

    824. 

  824.         $sql .= $this->parseLock(isset($options['lock'])?$options['lock']:false);

    825. 

  825.         if(isset($key)) { // 写入SQL创建缓存

    826. 

  826.             S($key,$sql,array('expire'=>0,'length'=>C('DB_SQL_BUILD_LENGTH'),'queue'=>C('DB_SQL_BUILD_QUEUE')));

    827. 

  827.         }

    828. 

  828.         return $sql;

    829. 

  829.     }

    830. 

  830. 

    831. 

  831.     /**

    832. 

  832.      * 替换SQL语句中表达式

    833. 

  833.      * @access public

    834. 

  834.      * @param array $options 表达式

    835. 

  835.      * @return string

    836. 

  836.      */

    837. 

  837.     public function parseSql($sql,$options=array()){

    838. 

  838.         $sql   = str_replace(

    839. 

  839.             array('%TABLE%','%DISTINCT%','%FIELD%','%JOIN%','%WHERE%','%GROUP%','%HAVING%','%ORDER%','%LIMIT%','%UNION%','%COMMENT%'),

    840. 

  840.             array(

    841. 

  841.                 $this->parseTable($options['table']),

    842. 

  842.                 $this->parseDistinct(isset($options['distinct'])?$options['distinct']:false),

    843. 

  843.                 $this->parseField(!empty($options['field'])?$options['field']:'*'),

    844. 

  844.                 $this->parseJoin(!empty($options['join'])?$options['join']:''),

    845. 

  845.                 $this->parseWhere(!empty($options['where'])?$options['where']:''),

    846. 

  846.                 $this->parseGroup(!empty($options['group'])?$options['group']:''),

    847. 

  847.                 $this->parseHaving(!empty($options['having'])?$options['having']:''),

    848. 

  848.                 $this->parseOrder(!empty($options['order'])?$options['order']:''),

    849. 

  849.                 $this->parseLimit(!empty($options['limit'])?$options['limit']:''),

    850. 

  850.                 $this->parseUnion(!empty($options['union'])?$options['union']:''),

    851. 

  851.                 $this->parseComment(!empty($options['comment'])?$options['comment']:'')

    852. 

  852.             ),$sql);

    853. 

  853.         return $sql;

    854. 

  854.     }

    855. 

  855. 

    856. 

  856.     /**

    857. 

  857.      * 获取最近一次查询的sql语句 

    858. 

  858.      * @param string $model  模型名

    859. 

  859.      * @access public

    860. 

  860.      * @return string

    861. 

  861.      */

    862. 

  862.     public function getLastSql($model='') {

    863. 

  863.         return $model?$this->modelSql[$model]:$this->queryStr;

    864. 

  864.     }

    865. 

  865. 

    866. 

  866.     /**

    867. 

  867.      * 获取最近插入的ID

    868. 

  868.      * @access public

    869. 

  869.      * @return string

    870. 

  870.      */

    871. 

  871.     public function getLastInsID() {

    872. 

  872.         return $this->lastInsID;

    873. 

  873.     }

    874. 

  874. 

    875. 

  875.     /**

    876. 

  876.      * 获取最近的错误信息

    877. 

  877.      * @access public

    878. 

  878.      * @return string

    879. 

  879.      */

    880. 

  880.     public function getError() {

    881. 

  881.         return $this->error;

    882. 

  882.     }

    883. 

  883. 

    884. 

  884.     /**

    885. 

  885.      * SQL指令安全过滤

    886. 

  886.      * @access public

    887. 

  887.      * @param string $str  SQL字符串

    888. 

  888.      * @return string

    889. 

  889.      */

    890. 

  890.     public function escapeString($str) {

    891. 

  891.         return addslashes($str);

    892. 

  892.     }

    893. 

  893. 

    894. 

  894.     /**

    895. 

  895.      * 设置当前操作模型

    896. 

  896.      * @access public

    897. 

  897.      * @param string $model  模型名

    898. 

  898.      * @return void

    899. 

  899.      */

    900. 

  900.     public function setModel($model){

    901. 

  901.         $this->model =  $model;

    902. 

  902.     }

    903. 

  903. 

    904. 

  904.    /**

    905. 

  905.      * 析构方法

    906. 

  906.      * @access public

    907. 

  907.      */

    908. 

  908.     public function __destruct() {

    909. 

  909.         // 释放查询

    910. 

  910.         if ($this->queryID){

    911. 

  911.             $this->free();

    912. 

  912.         }

    913. 

  913.         // 关闭连接

    914. 

  914.         $this->close();

    915. 

  915.     }

    916. 

  916. 

    917. 

  917.     // 关闭数据库 由驱动类定义

    918. 

  918.     public function close(){}

    919. 

  919. }
    920.