【discuzX2】/source/class/class_core.php DB及db_mysql分析

admin

1. <?php
   
2. /**
   
3. * Discuz MySQL 类的支持 程序中一般不直接使用此类,而是使用DB类，DB类对db_mysql类中的方法又进行了二次封装
   
4. *
   
5. */
   
6. class db_mysql
   
7. {
   
8.         var $tablepre;
   
9.         var $version = '';
   
10.         var $querynum = 0;
    
11.         var $slaveid = 0;
    
12.         var $curlink;
    
13.         var $link = array();
    
14.         var $config = array();
    
15.         var $sqldebug = array();
    
16.         var $map = array();
    
17. 
    
18.         function db_mysql($config = array()) {
    
19.                 if(!empty($config)) {
    
20.                         $this->set_config($config);
    
21.                 }
    
22.         }
    
23. 
    
24.         function set_config($config) {
    
25.                 $this->config = &$config;
    
26.                 $this->tablepre = $config['1']['tablepre'];
    
27.                 if(!empty($this->config['map'])) {
    
28.                         $this->map = $this->config['map'];
    
29.                 }
    
30.         }
    
31. 
    
32.         function connect($serverid = 1) {
    
33. 
    
34.                 if(empty($this->config) || empty($this->config[$serverid])) {
    
35.                         $this->halt('config_db_not_found');
    
36.                 }
    
37. 
    
38.                 $this->link[$serverid] = $this->_dbconnect(
    
39.                         $this->config[$serverid]['dbhost'],
    
40.                         $this->config[$serverid]['dbuser'],
    
41.                         $this->config[$serverid]['dbpw'],
    
42.                         $this->config[$serverid]['dbcharset'],
    
43.                         $this->config[$serverid]['dbname'],
    
44.                         $this->config[$serverid]['pconnect']
    
45.                         );
    
46.                 $this->curlink = $this->link[$serverid];
    
47. 
    
48.         }
    
49. 
    
50.         function _dbconnect($dbhost, $dbuser, $dbpw, $dbcharset, $dbname, $pconnect) {
    
51.                 $link = null;
    
52.                 $func = empty($pconnect) ? 'mysql_connect' : 'mysql_pconnect';
    
53.                 if(!$link = @$func($dbhost, $dbuser, $dbpw, 1)) {
    
54.                         $this->halt('notconnect');
    
55.                 } else {
    
56.                         $this->curlink = $link;
    
57.                         if($this->version() > '4.1') {
    
58.                                 $dbcharset = $dbcharset ? $dbcharset : $this->config[1]['dbcharset'];
    
59.                                 $serverset = $dbcharset ? 'character_set_connection='.$dbcharset.', character_set_results='.$dbcharset.', character_set_client=binary' : '';
    
60.                                 $serverset .= $this->version() > '5.0.1' ? ((empty($serverset) ? '' : ',').'sql_mode=\'\'') : '';
    
61.                                 $serverset && mysql_query("SET $serverset", $link);
    
62.                         }
    
63.                         $dbname && @mysql_select_db($dbname, $link);
    
64.                 }
    
65.                 return $link;
    
66.         }
    
67. 
    
68.         function table_name($tablename) {
    
69.                 if(!empty($this->map) && !empty($this->map[$tablename])) {
    
70.                         $id = $this->map[$tablename];
    
71.                         if(!$this->link[$id]) {
    
72.                                 $this->connect($id);
    
73.                         }
    
74.                         $this->curlink = $this->link[$id];
    
75.                         return $this->config[$id]['tablepre'].$tablename;
    
76.                 } else {
    
77.                         $this->curlink = $this->link[1];
    
78.                 }
    
79.                 return $this->tablepre.$tablename;
    
80.         }
    
81. 
    
82.         function select_db($dbname) {
    
83.                 return mysql_select_db($dbname, $this->curlink);
    
84.         }
    
85. 
    
86.         function fetch_array($query, $result_type = MYSQL_ASSOC) {
    
87.                 return mysql_fetch_array($query, $result_type);
    
88.         }
    
89. 
    
90.         function fetch_first($sql) {
    
91.                 return $this->fetch_array($this->query($sql));
    
92.         }
    
93. 
    
94.         function result_first($sql) {
    
95.                 return $this->result($this->query($sql), 0);
    
96.         }
    
97. 
    
98.         function query($sql, $type = '') {
    
99. 
    
100.                 if(defined('DISCUZ_DEBUG') && DISCUZ_DEBUG) {
     
101.                         $starttime = dmicrotime();
     
102.                 }
     
103.                 $func = $type == 'UNBUFFERED' && @function_exists('mysql_unbuffered_query') ?
     
104.                 'mysql_unbuffered_query' : 'mysql_query';
     
105.                 if(!($query = $func($sql, $this->curlink))) {
     
106.                         if(in_array($this->errno(), array(2006, 2013)) && substr($type, 0, 5) != 'RETRY') {
     
107.                                 $this->connect();
     
108.                                 return $this->query($sql, 'RETRY'.$type);
     
109.                         }
     
110.                         if($type != 'SILENT' && substr($type, 5) != 'SILENT') {
     
111.                                 $this->halt('query_error', $sql);
     
112.                         }
     
113.                 }
     
114. 
     
115.                 if(defined('DISCUZ_DEBUG') && DISCUZ_DEBUG) {
     
116.                         $this->sqldebug[] = array($sql, number_format((dmicrotime() - $starttime), 6), debug_backtrace());
     
117.                 }
     
118. 
     
119.                 $this->querynum++;
     
120.                 return $query;
     
121.         }
     
122. 
     
123.         function affected_rows() {
     
124.                 return mysql_affected_rows($this->curlink);
     
125.         }
     
126. 
     
127.         function error() {
     
128.                 return (($this->curlink) ? mysql_error($this->curlink) : mysql_error());
     
129.         }
     
130. 
     
131.         function errno() {
     
132.                 return intval(($this->curlink) ? mysql_errno($this->curlink) : mysql_errno());
     
133.         }
     
134. 
     
135.         function result($query, $row = 0) {
     
136.                 $query = @mysql_result($query, $row);
     
137.                 return $query;
     
138.         }
     
139. 
     
140.         function num_rows($query) {
     
141.                 $query = mysql_num_rows($query);
     
142.                 return $query;
     
143.         }
     
144. 
     
145.         function num_fields($query) {
     
146.                 return mysql_num_fields($query);
     
147.         }
     
148. 
     
149.         function free_result($query) {
     
150.                 return mysql_free_result($query);
     
151.         }
     
152. 
     
153.         function insert_id() {
     
154.                 return ($id = mysql_insert_id($this->curlink)) >= 0 ? $id : $this->result($this->query("SELECT last_insert_id()"), 0);
     
155.         }
     
156. 
     
157.         function fetch_row($query) {
     
158.                 $query = mysql_fetch_row($query);
     
159.                 return $query;
     
160.         }
     
161. 
     
162.         function fetch_fields($query) {
     
163.                 return mysql_fetch_field($query);
     
164.         }
     
165. 
     
166.         function version() {
     
167.                 if(empty($this->version)) {
     
168.                         $this->version = mysql_get_server_info($this->curlink);
     
169.                 }
     
170.                 return $this->version;
     
171.         }
     
172. 
     
173.         function close() {
     
174.                 return mysql_close($this->curlink);
     
175.         }
     
176. 
     
177.         function halt($message = '', $sql = '') {
     
178.                 require_once libfile('class/error');
     
179.                 discuz_error::db_error($message, $sql);
     
180.         }
     
181. 
     
182. }
     
183. 
     
184. /**
     
185. * 对Discuz CORE 中 DB Object中的主要方法进行二次封装，方便程序调用
     
186. *
     
187. */
     
188. class DB
     
189. {
     
190. 
     
191.         /**
     
192.          * 返回表名(pre_$table)
     
193.          *
     
194.          * @param 原始表名 $table
     
195.          * @return 增加pre之后的名字
     
196.          */
     
197.         function table($table) {
     
198.                 return DB::_execute('table_name', $table);
     
199.         }
     
200. 
     
201.         /**
     
202.          * 删除一条或者多条记录
     
203.          *
     
204.          * @param string $table 原始表名
     
205.          * @param string $condition 条件语句，不需要写WHERE
     
206.          * @param int $limit 删除条目数
     
207.          * @param boolean $unbuffered 立即返回？
     
208.          */
     
209.         function delete($table, $condition, $limit = 0, $unbuffered = true) {
     
210.                 if(empty($condition)) {
     
211.                         $where = '1';
     
212.                 } elseif(is_array($condition)) {
     
213.                         $where = DB::implode_field_value($condition, ' AND ');
     
214.                 } else {
     
215.                         $where = $condition;
     
216.                 }
     
217.                 $sql = "DELETE FROM ".DB::table($table)." WHERE $where ".($limit ? "LIMIT $limit" : '');
     
218.                 return DB::query($sql, ($unbuffered ? 'UNBUFFERED' : ''));
     
219.         }
     
220. 
     
221.         /**
     
222.          * 插入一条记录
     
223.          *
     
224.          * @param string $table 原始表名
     
225.          * @param array $data 数组field->vlaue 对
     
226.          * @param boolen $return_insert_id 返回 InsertID?
     
227.          * @param boolen $replace 是否是REPLACE模式
     
228.          * @param boolen $silent 屏蔽错误？
     
229.          * @return InsertID or Result
     
230.          */
     
231.         function insert($table, $data, $return_insert_id = false, $replace = false, $silent = false) {
     
232. 
     
233.                 $sql = DB::implode_field_value($data);
     
234. 
     
235.                 $cmd = $replace ? 'REPLACE INTO' : 'INSERT INTO';
     
236. 
     
237.                 $table = DB::table($table);
     
238.                 $silent = $silent ? 'SILENT' : '';
     
239. 
     
240.                 $return = DB::query("$cmd $table SET $sql", $silent);
     
241. 
     
242.                 return $return_insert_id ? DB::insert_id() : $return;
     
243. 
     
244.         }
     
245. 
     
246.         /**
     
247.          * 更新一条或者多条数据记录
     
248.          *
     
249.          * @param string $table 原始表名
     
250.          * @param array $data 数据field-value
     
251.          * @param string $condition 条件语句，不需要写WHERE
     
252.          * @param boolean $unbuffered 迅速返回？
     
253.          * @param boolan $low_priority 延迟更新？
     
254.          * @return result
     
255.          */
     
256.         function update($table, $data, $condition, $unbuffered = false, $low_priority = false) {
     
257.                 $sql = DB::implode_field_value($data);
     
258.                 $cmd = "UPDATE ".($low_priority ? 'LOW_PRIORITY' : '');
     
259.                 $table = DB::table($table);
     
260.                 $where = '';
     
261.                 if(empty($condition)) {
     
262.                         $where = '1';
     
263.                 } elseif(is_array($condition)) {
     
264.                         $where = DB::implode_field_value($condition, ' AND ');
     
265.                 } else {
     
266.                         $where = $condition;
     
267.                 }
     
268.                 $res = DB::query("$cmd $table SET $sql WHERE $where", $unbuffered ? 'UNBUFFERED' : '');
     
269.                 return $res;
     
270.         }
     
271. 
     
272.         /**
     
273.          * 格式化field字段和value，并组成一个字符串
     
274.          *
     
275.          * @param array $array 格式为 key=>value 数组
     
276.          * @param 分割符 $glue
     
277.          * @return string
     
278.          */
     
279.         function implode_field_value($array, $glue = ',') {
     
280.                 $sql = $comma = '';
     
281.                 foreach ($array as $k => $v) {
     
282.                         $sql .= $comma."`$k`='$v'";
     
283.                         $comma = $glue;
     
284.                 }
     
285.                 return $sql;
     
286.         }
     
287. 
     
288.         /**
     
289.          * 返回插入的ID
     
290.          *
     
291.          * @return int
     
292.          */
     
293.         function insert_id() {
     
294.                 return DB::_execute('insert_id');
     
295.         }
     
296. 
     
297.         /**
     
298.          * 依据查询结果，返回一行数据
     
299.          *
     
300.          * @param resourceID $resourceid
     
301.          * @return array
     
302.          */
     
303.         function fetch($resourceid, $type = MYSQL_ASSOC) {
     
304.                 return DB::_execute('fetch_array', $resourceid, $type);
     
305.         }
     
306. 
     
307.         /**
     
308.          * 依据SQL语句，返回第一条查询结果
     
309.          *
     
310.          * @param string $query 查询语句
     
311.          * @return array
     
312.          */
     
313.         function fetch_first($sql) {
     
314.                 DB::checkquery($sql);
     
315.                 return DB::_execute('fetch_first', $sql);
     
316.         }
     
317. 
     
318.         /**
     
319.          * 依据查询结果，返回结果数值
     
320.          *
     
321.          * @param resourceid $resourceid
     
322.          * @return string or int
     
323.          */
     
324.         function result($resourceid, $row = 0) {
     
325.                 return DB::_execute('result', $resourceid, $row);
     
326.         }
     
327. 
     
328.         /**
     
329.          * 依据查询语句，返回结果数值
     
330.          *
     
331.          * @param string $query SQL查询语句
     
332.          * @return unknown
     
333.          */
     
334.         function result_first($sql) {
     
335.                 DB::checkquery($sql);
     
336.                 return DB::_execute('result_first', $sql);
     
337.         }
     
338. 
     
339.         /**
     
340.          * 执行查询
     
341.          *
     
342.          * @param string $sql
     
343.          * @param 类型定义 $type UNBUFFERED OR SILENT
     
344.          * @return Resource OR Result
     
345.          */
     
346.         function query($sql, $type = '') {
     
347.                 DB::checkquery($sql);
     
348.                 return DB::_execute('query', $sql, $type);
     
349.         }
     
350. 
     
351.         /**
     
352.          * 返回select的结果行数
     
353.          *
     
354.          * @param resource $resourceid
     
355.          * @return int
     
356.          */
     
357.         function num_rows($resourceid) {
     
358.                 return DB::_execute('num_rows', $resourceid);
     
359.         }
     
360. 
     
361.         /**
     
362.          * 返回sql语句所影响的记录行数
     
363.          *
     
364.          * @return int
     
365.          */
     
366.         function affected_rows() {
     
367.                 return DB::_execute('affected_rows');
     
368.         }
     
369. 
     
370.         function free_result($query) {
     
371.                 return DB::_execute('free_result', $query);
     
372.         }
     
373. 
     
374.         function error() {
     
375.                 return DB::_execute('error');
     
376.         }
     
377. 
     
378.         function errno() {
     
379.                 return DB::_execute('errno');
     
380.         }
     
381. 
     
382.         function _execute($cmd , $arg1 = '', $arg2 = '') {//DB类中的很多方法都调用了此方法，此方法又调用了 &object()方法，详情请查看&object()方法，其实&object()方法返回一个db_mysql类的实例化对象，而且是statics类型的实例化对象
     
383.                 static $db;
     
384.                 if(empty($db)) $db = & DB::object();//返回db_mysql操作类的实例化对象
     
385.                 $res = $db->$cmd($arg1, $arg2);
     
386.                 return $res;
     
387.         }
     
388. 
     
389.         /**
     
390.          * 返回 DB object 指针
     
391.          *
     
392.          * @return pointer of db object from discuz core
     
393.          */
     
394.         function &object($dbclass = 'db_mysql') {
     
395.                 static $db;
     
396.                 if(empty($db)) $db = new $dbclass();//返回db_mysql数据库操作类的一个statics类型的实例化对象
     
397.                 return $db;
     
398.         }
     
399. 
     
400.         function checkquery($sql) {
     
401.                 static $status = null, $checkcmd = array('SELECT', 'UPDATE', 'INSERT', 'REPLACE', 'DELETE');
     
402.                 if($status === null) $status = getglobal('config/security/querysafe/status');
     
403.                 if($status) {
     
404.                         $cmd = trim(strtoupper(substr($sql, 0, strpos($sql, ' '))));
     
405.                         if(in_array($cmd, $checkcmd)) {
     
406.                                 $test = DB::_do_query_safe($sql);
     
407.                                 if($test < 1) DB::_execute('halt', 'security_error', $sql);
     
408.                         }
     
409.                 }
     
410.                 return true;
     
411.         }
     
412. 
     
413.         function _do_query_safe($sql) {
     
414.                 static $_CONFIG = null;
     
415.                 if($_CONFIG === null) {
     
416.                         $_CONFIG = getglobal('config/security/querysafe');
     
417.                 }
     
418. 
     
419.                 $sql = str_replace(array('\\\\', '\\\'', '\\"', '\'\''), '', $sql);
     
420.                 $mark = $clean = '';
     
421.                 if(strpos($sql, '/') === false && strpos($sql, '#') === false && strpos($sql, '-- ') === false) {
     
422.                         $clean = preg_replace("/'(.+?)'/s", '', $sql);
     
423.                 } else {
     
424.                         $len = strlen($sql);
     
425.                         $mark = $clean = '';
     
426.                         for ($i = 0; $i <$len; $i++) {
     
427.                                 $str = $sql[$i];
     
428.                                 switch ($str) {
     
429.                                         case '\'':
     
430.                                                 if(!$mark) {
     
431.                                                         $mark = '\'';
     
432.                                                         $clean .= $str;
     
433.                                                 } elseif ($mark == '\'') {
     
434.                                                         $mark = '';
     
435.                                                 }
     
436.                                                 break;
     
437.                                         case '/':
     
438.                                                 if(empty($mark) && $sql[$i+1] == '*') {
     
439.                                                         $mark = '/*';
     
440.                                                         $clean .= $mark;
     
441.                                                         $i++;
     
442.                                                 } elseif($mark == '/*' && $sql[$i -1] == '*') {
     
443.                                                         $mark = '';
     
444.                                                         $clean .= '*';
     
445.                                                 }
     
446.                                                 break;
     
447.                                         case '#':
     
448.                                                 if(empty($mark)) {
     
449.                                                         $mark = $str;
     
450.                                                         $clean .= $str;
     
451.                                                 }
     
452.                                                 break;
     
453.                                         case "\n":
     
454.                                                 if($mark == '#' || $mark == '--') {
     
455.                                                         $mark = '';
     
456.                                                 }
     
457.                                                 break;
     
458.                                         case '-':
     
459.                                                 if(empty($mark)&& substr($sql, $i, 3) == '-- ') {
     
460.                                                         $mark = '-- ';
     
461.                                                         $clean .= $mark;
     
462.                                                 }
     
463.                                                 break;
     
464. 
     
465.                                         default:
     
466. 
     
467.                                                 break;
     
468.                                 }
     
469.                                 $clean .= $mark ? '' : $str;
     
470.                         }
     
471.                 }
     
472. 
     
473.                 $clean = preg_replace("/[^a-z0-9_\-\(\)#\*\/\"]+/is", "", strtolower($clean));
     
474. 
     
475.                 if($_CONFIG['afullnote']) {
     
476.                         $clean = str_replace('/**/','',$clean);
     
477.                 }
     
478. 
     
479.                 if(is_array($_CONFIG['dfunction'])) {
     
480.                         foreach($_CONFIG['dfunction'] as $fun) {
     
481.                                 if(strpos($clean, $fun.'(') !== false) return '-1';
     
482.                         }
     
483.                 }
     
484. 
     
485.                 if(is_array($_CONFIG['daction'])) {
     
486.                         foreach($_CONFIG['daction'] as $action) {
     
487.                                 if(strpos($clean,$action) !== false) return '-3';
     
488.                         }
     
489.                 }
     
490. 
     
491.                 if($_CONFIG['dlikehex'] && strpos($clean, 'like0x')) {
     
492.                         return '-2';
     
493.                 }
     
494. 
     
495.                 if(is_array($_CONFIG['dnote'])) {
     
496.                         foreach($_CONFIG['dnote'] as $note) {
     
497.                                 if(strpos($clean,$note) !== false) return '-4';
     
498.                         }
     
499.                 }
     
500. 
     
501.                 return 1;
     
502. 
     
503.         }
     
504. 
     
505. }
     
506. ?>

複製代碼