First commit

source/class/db/db_driver_mysqli.php

@@ -0,0 +1,250 @@
+<?php
+
+/**
+ *      [Discuz!] (C)2001-2099 Comsenz Inc.
+ *      This is NOT a freeware, use is subject to license terms
+ *
+ *      $Id: db_driver_mysqli.php 36278 2016-12-09 07:52:35Z nemohou $
+ */
+
+if(!defined('IN_DISCUZ')) {
+	exit('Access Denied');
+}
+
+class db_driver_mysqli
+{
+	var $tablepre;
+	var $version = '';
+	var $drivertype = 'mysqli';
+	var $querynum = 0;
+	var $slaveid = 0;
+	var $curlink;
+	var $link = array();
+	var $config = array();
+	var $sqldebug = array();
+	var $map = array();
+
+	function db_mysql($config = array()) {
+		if(!empty($config)) {
+			$this->set_config($config);
+		}
+	}
+
+	function set_config($config) {
+		$this->config = &$config;
+		$this->tablepre = $config['1']['tablepre'];
+		if(!empty($this->config['map'])) {
+			$this->map = $this->config['map'];
+			for($i = 1; $i <= 100; $i++) {
+				if(isset($this->map['forum_thread'])) {
+					$this->map['forum_thread_'.$i] = $this->map['forum_thread'];
+				}
+				if(isset($this->map['forum_post'])) {
+					$this->map['forum_post_'.$i] = $this->map['forum_post'];
+				}
+				if(isset($this->map['forum_attachment']) && $i <= 10) {
+					$this->map['forum_attachment_'.($i-1)] = $this->map['forum_attachment'];
+				}
+			}
+			if(isset($this->map['common_member'])) {
+				$this->map['common_member_archive'] =
+				$this->map['common_member_count'] = $this->map['common_member_count_archive'] =
+				$this->map['common_member_status'] = $this->map['common_member_status_archive'] =
+				$this->map['common_member_profile'] = $this->map['common_member_profile_archive'] =
+				$this->map['common_member_field_forum'] = $this->map['common_member_field_forum_archive'] =
+				$this->map['common_member_field_home'] = $this->map['common_member_field_home_archive'] =
+				$this->map['common_member_validate'] = $this->map['common_member_verify'] =
+				$this->map['common_member_verify_info'] = $this->map['common_member'];
+			}
+		}
+	}
+
+	function connect($serverid = 1) {
+
+		if(empty($this->config) || empty($this->config[$serverid])) {
+			$this->halt('config_db_not_found');
+		}
+
+		$this->link[$serverid] = $this->_dbconnect(
+			$this->config[$serverid]['dbhost'],
+			$this->config[$serverid]['dbuser'],
+			$this->config[$serverid]['dbpw'],
+			$this->config[$serverid]['dbcharset'],
+			$this->config[$serverid]['dbname'],
+			$this->config[$serverid]['pconnect']
+			);
+		$this->curlink = $this->link[$serverid];
+
+	}
+
+	function _dbconnect($dbhost, $dbuser, $dbpw, $dbcharset, $dbname, $pconnect, $halt = true) {
+		mysqli_report(MYSQLI_REPORT_OFF);
+		if (intval($pconnect) === 1) $dbhost = 'p:' . $dbhost;
+		$link = new mysqli();
+		if(!$link->real_connect($dbhost, $dbuser, $dbpw, $dbname, null, null, MYSQLI_CLIENT_COMPRESS)) {
+			$halt && $this->halt('notconnect', $this->errno());
+		} else {
+			$this->curlink = $link;
+			$link->options(MYSQLI_OPT_LOCAL_INFILE, false);
+			$link->set_charset($dbcharset ? $dbcharset : $this->config[1]['dbcharset']);
+			$serverset = 'sql_mode=\'\',';
+			$serverset .= 'character_set_client=binary';
+			$serverset && $link->query("SET $serverset");
+		}
+		return $link;
+	}
+
+	function table_name($tablename) {
+		if(!empty($this->map) && !empty($this->map[$tablename])) {
+			$id = $this->map[$tablename];
+			if(!$this->link[$id]) {
+				$this->connect($id);
+			}
+			$this->curlink = $this->link[$id];
+		} else {
+			$this->curlink = $this->link[1];
+		}
+		return $this->tablepre.$tablename;
+	}
+
+	function select_db($dbname) {
+		return $this->curlink->select_db($dbname);
+	}
+
+	function fetch_array($query, $result_type = MYSQLI_ASSOC) {
+		return $query ? $query->fetch_array($result_type) : null;
+	}
+
+	function fetch_first($sql) {
+		return $this->fetch_array($this->query($sql));
+	}
+
+	function result_first($sql) {
+		return $this->result($this->query($sql), 0);
+	}
+
+	public function query($sql, $silent = false, $unbuffered = false) {
+		if(defined('DISCUZ_DEBUG') && DISCUZ_DEBUG) {
+			$starttime = microtime(true);
+		}
+
+		if('UNBUFFERED' === $silent) {
+			$silent = false;
+			$unbuffered = true;
+		} elseif('SILENT' === $silent) {
+			$silent = true;
+			$unbuffered = false;
+		}
+
+		$resultmode = $unbuffered ? MYSQLI_USE_RESULT : MYSQLI_STORE_RESULT;
+
+		if(!($query = $this->curlink->query($sql, $resultmode))) {
+			if(in_array($this->errno(), array(2006, 2013)) && substr($silent, 0, 5) != 'RETRY') {
+				$this->connect();
+				return $this->query($sql, 'RETRY'.$silent);
+			}
+			if(!$silent) {
+				$this->halt($this->error(), $this->errno(), $sql);
+			}
+		}
+
+		if(defined('DISCUZ_DEBUG') && DISCUZ_DEBUG) {
+			$this->sqldebug[] = array($sql, number_format((microtime(true) - $starttime), 6), debug_backtrace(), $this->curlink);
+		}
+
+		$this->querynum++;
+		return $query;
+	}
+
+	function affected_rows() {
+		return $this->curlink->affected_rows;
+	}
+
+	function error() {
+		return $this->curlink->error;
+	}
+
+	function errno() {
+		return $this->curlink->errno;
+	}
+
+	function result($query, $row = 0) {
+		if(!$query || $query->num_rows == 0) {
+			return null;
+		}
+		$query->data_seek($row);
+		$assocs = $query->fetch_row();
+		return $assocs[0];
+	}
+
+	function num_rows($query) {
+		$query = $query ? $query->num_rows : 0;
+		return $query;
+	}
+
+	function num_fields($query) {
+		return $query ? $query->field_count : null;
+	}
+
+	function free_result($query) {
+		return $query ? $query->free() : false;
+	}
+
+	function insert_id() {
+		return ($id = $this->curlink->insert_id) >= 0 ? $id : $this->result($this->query("SELECT last_insert_id()"), 0);
+	}
+
+	function fetch_row($query) {
+		$query = $query ? $query->fetch_row() : null;
+		return $query;
+	}
+
+	function fetch_fields($query) {
+		return $query ? $query->fetch_field() : null;
+	}
+
+	function version() {
+		if(empty($this->version)) {
+			$this->version = $this->curlink->server_info;
+		}
+		return $this->version;
+	}
+
+	function escape_string($str) {
+		return $this->curlink->escape_string($str);
+	}
+
+	function close() {
+		return $this->curlink->close();
+	}
+
+	function halt($message = '', $code = 0, $sql = '') {
+		throw new DbException($message, $code, $sql);
+	}
+
+	function begin_transaction() {
+		if (PHP_VERSION < '5.5') {
+			return $this->curlink->autocommit(false);
+		}
+		return $this->curlink->begin_transaction();
+	}
+
+	function commit() {
+		$cr = $this->curlink->commit();
+		if (PHP_VERSION < '5.5') {
+			$this->curlink->autocommit(true);
+		}
+		return $cr;
+	}
+
+	function rollback() {
+		$rr = $this->curlink->rollback();
+		if (PHP_VERSION < '5.5') {
+			$this->curlink->autocommit(true);
+		}
+		return $rr;
+	}
+
+}
+
+?>

source/class/db/db_driver_mysqli_slave.php

@@ -0,0 +1,96 @@
+<?php
+
+/**
+ *      [Discuz!] (C)2001-2099 Comsenz Inc.
+ *      This is NOT a freeware, use is subject to license terms
+ *
+ *      $Id: db_driver_mysqli_slave.php 33959 2013-09-06 04:27:59Z nemohou $
+ */
+
+if(!defined('IN_DISCUZ')) {
+	exit('Access Denied');
+}
+class db_driver_mysqli_slave extends db_driver_mysqli
+{
+
+	public $slaveid = null;
+
+	public $slavequery = 0;
+
+	public $slaveexcept = false;
+
+	public $excepttables = array();
+
+	public $tablename = '';
+
+	protected $_weighttable = array();
+
+	public $serverid = null;
+
+	function set_config($config) {
+		parent::set_config($config);
+
+		if($this->config['common']['slave_except_table']) {
+			$this->excepttables = explode(',', str_replace(' ', '', $this->config['common']['slave_except_table']));
+		}
+	}
+
+	public function table_name($tablename) {
+		$this->tablename = $tablename;
+		if(!$this->slaveexcept && $this->excepttables) {
+			$this->slaveexcept = in_array($tablename, $this->excepttables, true);
+		}
+		$this->serverid = isset($this->map[$this->tablename]) ? $this->map[$this->tablename] : 1;
+		return $this->tablepre.$tablename;
+	}
+
+	protected function _slave_connect() {
+		if(!empty($this->config[$this->serverid]['slave'])) {
+			$this->_choose_slave();
+			if($this->slaveid) {
+				if(!isset($this->link[$this->slaveid])) {
+					$this->connect($this->slaveid);
+				}
+				$this->slavequery ++;
+				$this->curlink = $this->link[$this->slaveid];
+			}
+			return true;
+		} else {
+			return false;
+		}
+	}
+
+	protected function _choose_slave(){
+		if(!isset($this->_weighttable[$this->serverid])) {
+			foreach ($this->config[$this->serverid]['slave'] as $key => $value) {
+				$this->_weighttable[$this->serverid] .= str_repeat($key, 1 + intval($value['weight']));
+			}
+		}
+		$sid = $this->_weighttable[$this->serverid][mt_rand(0, strlen($this->_weighttable[$this->serverid]) -1)];
+		$this->slaveid = $this->serverid.'_'.$sid;
+		if(!isset($this->config[$this->slaveid])) {
+			$this->config[$this->slaveid] = $this->config[$this->serverid]['slave'][$sid];
+		}
+	}
+
+	protected function _master_connect() {
+		if ($this->serverid === null) {
+			$this->serverid = 1;
+		}
+		if(!$this->link[$this->serverid]) {
+			$this->connect($this->serverid);
+		}
+		$this->curlink = $this->link[$this->serverid];
+	}
+
+	public function query($sql, $silent = false, $unbuffered = false) {
+		if(!(!$this->slaveexcept && strtoupper(substr($sql, 0 , 6)) === 'SELECT' && strpos(strtoupper($sql), 'FOR UPDATE') === FALSE && $this->_slave_connect())) {
+			$this->_master_connect();
+		}
+		$this->tablename = '';
+		$this->slaveexcept = false;
+		return parent::query($sql, $silent, $unbuffered);
+	}
+
+}
+?>