| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354 | <?phpnamespace KIF\Db;use KIF\Core\Config;use MySQLi as MySQLiResource;use KIF\Debug\Debug;use KIF\Core\Request;use KIF\Cookie;use Exception;class MySQLi {	/**     * 当前的数据库连接对象     *     * @var MySQLiResource     */    private $link;    /**     * 当前连接的数据库名     * @var string     */    private $dbname;        /**     * #private     * 统计查询次数     *     * @var Int     */    static public $queries = 0;    /**     * 通过$this->query 所执行的总时间     * 只有 debug::sql打开才会统计     *     * @var Int     */    private static $intQueriesTotalTime = 0;    /**     * 保存所有已建立的到服务器的连接。     *     * @var Array 结构如下:     * array(     *     '连接唯一标志' => MySQLiResource   //连接唯一标志由 $this->getUniqueFlagOfLink() 获取;MySQLiResource 是到主服务器1的连接。     *     , ...     * )     */    static private $links;    /**     * 调试等级     * 0 不处理(交予外部程序处理)     * 1 显示错误并中断程序     * 2 直接中断程序     *     * @var Int     */    public $debug_level = 1;    /**     * 最近一次执行的语句     *     * @var string     */    private $last_sql;    /**     * 最后一次执行 query 后获取的 mysqli_result 对象     *     * @var mysqli_result     */    private $result;	/**     * from Data Source Name (dsn)     * for example: mysqli://root:851031@localhost:3306/testDb?charset=utf8     *     * @var Array (      * 		'prefix' => '',      * 		'host' => '',      * 		'port' => '',      * 		'user' => '',      * 		'pass' => '',     * 		'params'	=> array(),// 通过dsn传递进来的查询字串     * )     */    private $arrDsn;    /**     * 保存当前主服务器连接的信息     *是对 self::$masterDBInfos 数据结构某一项的引用     * @var array     * 结构如下     * array(// 当前集群下的主服务器     *     'host' => //主机名     *     , 'port'  => //端口     *     , 'user' => //用户名     *     , 'pass' => //密码     *     , 'transactionIds' => //用于存储事务id。Array ('taId1' => true, 'taId2' => false, ...);     *                                                        //键代表事务标志,同时也是用于保存点的标志。值表示是否为顶层事务(只能有一个顶层事务)     *     , 'isRunOnMaster' => Boolean   //$this->query 方法里可检验该值,判断当前查询正连接于主服务器上     *     , 'isUseMaster' => Boolean //指示后续查询是否要使用到主库的连接     *     , 'arrStatusOfUseMaster' => Array(); //当嵌套修改 isUseMaster 的值时,需要有一个结构保存之前的状态,以便恢复     * )     */    private $masterDBInfo;    /**     * 保存所有用于主服务器连接的信息     *     * @var Array  结构如下:     * array(     *     '集群1标志' => array( //     *         '主服务器标志' => array(//该标志通过 $this->getUniqueFlagOfLink() 获取:实质是通过 host, port, user, pass 来区别,所以同一集群如果使用不同的用户连接,会有不同的标志     *             'host' => //主机名     *             , 'port'  => //端口     *             , 'user' => //用户名     *             , 'pass' => //密码     *             , 'transactionIds' => //用于存储事务id。Array ('taId1' => true, 'taId2' => false, ...);     *                                                        //键代表事务标志,同时也是用于保存点的标志。值表示是否为顶层事务(只能有一个顶层事务)     *             , 'isRunOnMaster' => Boolean   //$this->query 方法里可检验该值,判断$this->link是否正链接到主服务器上     *             , 'isUseMaster' => Boolean //指示后续查询是否要使用到主库的连接     *             , 'arrStatusOfUseMaster' => Array();     *         )     *         , ...     *     )     *     , ...     * )     */    static private $masterDBInfos = array();    /**     * 集群信息,即用于候选的从服务器集     * array(     *     'clusterLevel' => (2 表示正常集群、3 表示低效集群)     *     , 'Ignore_Check_Master_Patterns' => array()     *     *     , '集群1标志' => array( // 该标志通过 $this->getMasterHost() 获取:集群只需要通过主服务器host即可区分     *                        array(     *                                  'host' =>     *                                  , 'port'  =>     *                                  , 'user' =>     *                                  , 'pass' =>     *                        )     *                        , ...     *                  )     *     , ...     * )     * @var Array     */    static private $slaveDBInfos;    /**     * 保存已从集群信息中获取的用于连接从服务器的信息     * @author Gxg <gaoxiaogang@gmail.com>     *     * @var Array  结构如下:     * array(// 当前集群下的从库信息     *         'host' => //主机名     *         , 'port'  => //端口     *         , 'user' => //用户名     *         , 'pass' => //密码     * )     */    private $slaveDBInfo;    /**     * 返回当前集群中从服务器的信息     *     * @return false | Array 格式:     * array(     *       'host' => //主机名     *       , 'user' => //用户名     *       , 'pass' => //密码     * )     */    private function getCurrentSlaveInfo() {    	$arrInvalidSlaveInfos = $this->getCurrentInvalidSlaveInfos();        if (isset($this->slaveDBInfo)) {            # 检验是否有效            if (!$arrInvalidSlaveInfos || !in_array($this->slaveDBInfo['host'], $arrInvalidSlaveInfos)) {                return $this->slaveDBInfo;            }        }        $strUniqueFlagOfCluster = $this->getUniqueFlagOfCurrentCluster();        if (!isset(self::$slaveDBInfos[$strUniqueFlagOfCluster])) {        	# 不能使用 $this->slaveDBInfo = null,否则引用的 self::$slaveDBInfos 相应的值也会变成 null            return false;        }        # 把 self::$slaveDBInfos 中已经无效的从库踢掉        if ($arrInvalidSlaveInfos) {            foreach (self::$slaveDBInfos[$strUniqueFlagOfCluster] as $key => $arrSlaveInfo) {                $strSlaveHost = $arrSlaveInfo['host'];                if (in_array($strSlaveHost, $arrInvalidSlaveInfos)) {                    unset(self::$slaveDBInfos[$strUniqueFlagOfCluster][$key]);                }            }        }        if (count(self::$slaveDBInfos[$strUniqueFlagOfCluster]) == 0) {        	return false;        }        $intRandPos = array_rand(self::$slaveDBInfos[$strUniqueFlagOfCluster]);	    $this->slaveDBInfo = & self::$slaveDBInfos[$strUniqueFlagOfCluster][$intRandPos];	    if (!isset($this->slaveDBInfo['user']) || !isset($this->slaveDBInfo['pass'])) {//从服务器没有提供连接帐户,则使用主服务器的帐户	        $this->slaveDBInfo['user'] = $this->masterDBInfo['user'];	        $this->slaveDBInfo['pass'] = $this->masterDBInfo['pass'];	    }	    if (!isset($this->slaveDBInfo['port'])) {	        $this->slaveDBInfo['port'] = $this->masterDBInfo['port'];	    }	    return $this->slaveDBInfo;    }    /**     * 集群中无效的从服务器信息     * 寻找从服务器时,会忽略该结构中列出的主机     * array(     *     '集群1标志' => array('从服务器host', '从服务器host', ...)     *     , ...     * )     *     * @var Array     */    static private $invalidSlaveInfos;    /**     * 增加无效的从服务器信息     *     * @param String $strHost     * @return true     */    private function addInvalidSlaveInfo($strHost) {        $strUniqueFlagOfCluster = $this->getUniqueFlagOfCurrentCluster();        if (!isset(self::$invalidSlaveInfos[$strUniqueFlagOfCluster])            || !in_array($strHost, self::$invalidSlaveInfos[$strUniqueFlagOfCluster])        ) {            self::$invalidSlaveInfos[$strUniqueFlagOfCluster][] = $strHost;        }        return true;    }    /**     * 获取当前集群的唯一标志(即主服务器host)     *     * @return String     */    private function getUniqueFlagOfCurrentCluster() {        //return "{$this->masterDBInfo['host']}_{$this->masterDBInfo['port']}_{$this->masterDBInfo['user']}";        return "{$this->masterDBInfo['host']}:{$this->masterDBInfo['port']}";    }    /**     * 获取当前集群下无效的从服务器信息     *     * @return false | Array     */    private function getCurrentInvalidSlaveInfos() {        $strUniqueFlagOfCluster = $this->getUniqueFlagOfCurrentCluster();        if (!isset(self::$invalidSlaveInfos[$strUniqueFlagOfCluster])) {            return false;        }        return self::$invalidSlaveInfos[$strUniqueFlagOfCluster];    }    /**     * 设置客户端连接字符集     *     * @param String $charset 连接编码。该参数决定了从各种编码的表里取出数据后,以$strCharset指定的编码返回     * 		比如:gbk编码存储的表数据,可以通过指定$strCharset为utf8来返回utf8编码的数据     * @return Boolean true:成功;false:失败.     */    public function setCharset($charset = null) {    	if (is_null($charset)) {    		if (isset($this->arrDsn['params']['charset']) && $this->arrDsn['params']['charset']) {    			$charset = $this->arrDsn['params']['charset'];    		} else {    			$charset = "utf8";    		}    	}        return $boolResult = mysqli_set_charset($this->link, $charset);    }    /**     * 取得客户端连接字符集     *     * @return String     */    public function getCharset() {        return $strResult = mysqli_character_set_name($this->link);    }    /**     * 获取当前连接的唯一标志     *     * @return String     */    private function getUniqueFlagOfLink() {        return "{$this->arrDsn['host']}_{$this->arrDsn['port']}_{$this->arrDsn['user']}_{$this->arrDsn['pass']}";    }    /**     * 构造函数     *     * @param String $dsn For Example: mysqli://root:851031@localhost/testDb     */    public function __construct($dsn) {        $arrUrlInfo = parse_url($dsn);        if (!is_array($arrUrlInfo) || !isset($arrUrlInfo['host']) || !isset($arrUrlInfo['user'])           || !isset($arrUrlInfo['pass']) || !isset($arrUrlInfo['path']))        {        	$this->_halt("构造参数不正确:{$dsn}");        	return false;            //解析出错时的处理        }        $this->arrDsn['host'] = $arrUrlInfo['host'];        $this->arrDsn['user'] = $arrUrlInfo['user'];        $this->arrDsn['pass'] = $arrUrlInfo['pass'];        $dbname = substr($arrUrlInfo['path'], 1);        if (empty($dbname)) {        	return $this->_halt('请先设置数据库名', '21');        }        $this->dbname = $dbname;        isset($arrUrlInfo['scheme']) && $this->arrDsn['prefix'] = $arrUrlInfo['scheme'];        if (!isset($arrUrlInfo['port'])) {        	$this->arrDsn['port'] = 3306;        } else {        	$this->arrDsn['port'] = $arrUrlInfo['port'];        }                # 处理参数        if (isset($arrUrlInfo['query'])) {        	parse_str($arrUrlInfo['query'], $this->arrDsn['params']);        } else {        	$this->arrDsn['params'] = array();        }        $strUniqueFlagOfCluster = $this->arrDsn['host'];        $strUniqueFlagOfMaster = $this->getUniqueFlagOfLink();        if (!isset(self::$masterDBInfos[$strUniqueFlagOfCluster][$strUniqueFlagOfMaster])) {            self::$masterDBInfos[$strUniqueFlagOfCluster][$strUniqueFlagOfMaster] = array(                'host' => $this->arrDsn['host']                , 'port' => $this->arrDsn['port']                , 'user' => $this->arrDsn['user']                , 'pass' => $this->arrDsn['pass']                , 'transactionIds' => null                , 'isRunOnMaster' => false                , 'isUseMaster' => false                , 'arrStatusOfUseMaster' => null            );        }        $this->masterDBInfo = & self::$masterDBInfos[$strUniqueFlagOfCluster][$strUniqueFlagOfMaster];        if (!isset(self::$slaveDBInfos)) {            $this->initSlaveDBInfos();        }    }	/**     * 初始化 self::$slaveDBInfos 数据结构     *     * dbslaves 的结构:     * array(     *     array(     *         'master' => (string),//对应主库的host值     *         'mixed'  => array(//正常从库     *             'mysqld-6.yoka.com',     *             '192.168.0.150',     *             ... ,     *         ),     *         'delay_mixed'    => array(//慢速从库,比如用于提供给爬虫、或提供给翻页的比较后面的页面,无需保证特别好的服务     *             'mysqld-12.yoka.com',     *             array(     *                 'host'   => (string),//必须     *                 'user'   => (string),//必须     *                 'pass'   => (string),//必须     *                 'port'   => (int),//非必须     *             ),     *             ... ,     *         ),     *     )     * );     *     */    private function initSlaveDBInfos() {    	try {    		$dbslavesConfig = Config::getInstance()->get('dbslaves');    	} catch (Exception $e) {    		return false;    	}    	if (!is_array($dbslavesConfig)) {    		return false;    	}    	    	$db_cluster_maps = $dbslavesConfig;    	if (isset($_SERVER['Cluster_User_Level']) && $_SERVER['Cluster_User_Level'] == 3) {    		$cluster_level = 3;    		$strType = 'delay_mixed';    	} else {    		$cluster_level = 2;    		$strType = 'mixed';    	}    	foreach ($db_cluster_maps as $arrClusterInfo) {    		if (!isset($arrClusterInfo['master']) || !isset($arrClusterInfo[$strType]) || !is_array($arrClusterInfo[$strType]) || (count($arrClusterInfo[$strType]) == 0)) {    			# TODO 记录日志    			continue;    		}    		foreach($arrClusterInfo[$strType] as $mixConnectInfo) {                if (is_string($mixConnectInfo)) {//类似 mysqld-6.verycd.com 的值                    self::$slaveDBInfos[$arrClusterInfo['master']][] = array(                        'host'      => $mixConnectInfo                        , 'port'    => null                        , 'user'    => null                        , 'pass'    => null                    );                } elseif (is_array($mixConnectInfo)) {//如果$_SERVER['DataBase_Cluster_Map'] 提供客启端的密码,请使用以下格式 !                    if (isset($mixConnectInfo['host']) && isset($mixConnectInfo['user']) && isset($mixConnectInfo['pass'])) {                    	$tmpPort = isset($mixConnectInfo['port']) ? $mixConnectInfo['port'] : 3306;                        self::$slaveDBInfos[$arrClusterInfo['master']][] = array(                            'host'      => $mixConnectInfo['host']                            , 'port'    => $tmpPort                            , 'user'    => $mixConnectInfo['user']                            , 'pass'    => $mixConnectInfo['pass']                        );                    } else {//                                $this->_halt('该集群:' . $arrClusterInfo['master'] . '在$_SERVER[\'DataBase_Cluster_Map\']里提供的连接信息格式错误');                        # TODO 记下错误日志                        continue;                    }                } else {                    # TODO 记下错误日志//                            $this->_halt('该集群:' . $arrClusterInfo['master'] . '在$_SERVER[\'DataBase_Cluster_Map\']里提供的连接信息格式错误');                    continue;                }            }//end foreach    	}//end foreach    }    /**     * 获取当前主连接唯一标志     *     * @return String     */    private function getUniqueFlagOfCurrentMaster() {        return "{$this->masterDBInfo['host']}_{$this->masterDBInfo['port']}_{$this->masterDBInfo['user']}_{$this->masterDBInfo['pass']}";    }    /**     * 获取当前集群状态下到主服务器的连接     *     * @return mysqli | false   已有连接,返回该连接(即mysqli对象);false:没有连接     */    private function getCurrentMasterLink() {        $strUniqueFlagOfCurrentMaster = $this->getUniqueFlagOfCurrentMaster();        if (isset(self::$links[$strUniqueFlagOfCurrentMaster])) {            if ($this->isLink(self::$links[$strUniqueFlagOfCurrentMaster])) {                return self::$links[$strUniqueFlagOfCurrentMaster];            } else {            	# 曾建立过连接,但中途该连接失效了,应该对此做出处理的。            	if (!empty($this->masterDBInfo['transactionIds'])) {            		return $this->_halt("到主库的连接已失效,且该主库上存在事务,必须中断!");            	}            }        }        return false;    }    /**     * 当前连接是否连到主库     */    private function isRunOnMaster() {        return (boolean) $this->masterDBInfo['isRunOnMaster'];    }    /**     * 获取所有事务信息     * 只有当前集群中主服务器的连接是可用的,才有事务可言     *     * @return false | Array()   false:没有到主服务器的连接或者还没开启事务;     */    private function getTransactions() {        if ($this->getCurrentMasterLink()) {            if (!empty($this->masterDBInfo['transactionIds'])) {                return $this->masterDBInfo['transactionIds'];            }        }        return false;    }    /**     * 当前查询是否正运行在主服务器上并且开启了事务     * 该方法在 $this->query 方法里调用才是最有价值的。     *     * @return Boolean     true:是;false:否     */    private function isRunOnTransaction() {        if ($this->isRunOnMaster() && $this->getTransactions()) {            return true;        }        return false;    }    private function isLink($link) {        if (!($link instanceof MySQLiResource)) return false;        $sinfo = @mysqli_get_host_info($link);        return !empty($sinfo);    }    private function isReadSql($sql) {        static $r_ops = array('select','show','desc');        $sql = strtolower(trim($sql));        foreach ($r_ops as $op) {           if (strpos($sql,$op)===0) return true;        }        return false;    }    /**     * 连接数据库。这里创建的是真实链接,不会重用已有的链接     *     * @return false | MySQLiResource false:连接失败     */     private function connect() {        // 连接数据库服务器        $objMysqli = mysqli_init();        $connect_rs = mysqli_real_connect($objMysqli, $this->arrDsn['host'], $this->arrDsn['user'], $this->arrDsn['pass']                            , null, $this->arrDsn['port'], null                            , MYSQLI_CLIENT_COMPRESS);        if (!$connect_rs) {            return false;        }        # 设置字符集的代码        $this->link = $objMysqli;        $this->setCharset();        return $this->link;    }    /**     * 标志当前连接$this->link连接到主库     *     * @return Boolean     */    private function beginRunOnMaster() {        $this->masterDBInfo['isRunOnMaster'] = true;        return true;    }    /**     * 标志当前连接$this->link离开主库     *     * @return Boolean     */    private function endRunOnMaster() {        $this->masterDBInfo['isRunOnMaster'] = false;        return true;    }    /**     * 该方法只应该由 $this->xconnect()调用     *     * @return Boolean  false:连接失败     */    private function connectMaster() {        # These codes increase by Gxg <gaoxiaogang@gmail.com>        # 保证到当前集群的主服务器的连接唯一        $this->arrDsn['host'] = $this->masterDBInfo['host'];        $this->arrDsn['port'] = $this->masterDBInfo['port'];        $this->arrDsn['user'] = $this->masterDBInfo['user'];        $this->arrDsn['pass'] = $this->masterDBInfo['pass'];        $objCurrentMasterLink = $this->getCurrentMasterLink();        if ($objCurrentMasterLink) {            $this->link = $objCurrentMasterLink;        } else {            if (false === $this->connect()) {                return false;            }            $strUniqueFlagOfLink = $this->getUniqueFlagOfLink();            self::$links[$strUniqueFlagOfLink] = $this->link;        }        $this->beginRunOnMaster();        return true;    }    /**     * 该方法只应该由 $this->xconnect()调用     *     * @return Boolean  false:连接失败     *     */    private function connectSlave() {    	$arrCurrentSlaveInfo = $this->getCurrentSlaveInfo();        if ($arrCurrentSlaveInfo) {            $this->arrDsn['host'] = $arrCurrentSlaveInfo['host'];            $this->arrDsn['port'] = $arrCurrentSlaveInfo['port'];            $this->arrDsn['user'] = $arrCurrentSlaveInfo['user'];            $this->arrDsn['pass'] = $arrCurrentSlaveInfo['pass'];            $strUniqueFlagOfLink = $this->getUniqueFlagOfLink();            if (isset(self::$links[$strUniqueFlagOfLink]) && $this->isLink(self::$links[$strUniqueFlagOfLink])) {                $this->link = self::$links[$strUniqueFlagOfLink];            } else {                if(false === $this->connect()) {                    return false;                }                self::$links[$strUniqueFlagOfLink] = $this->link;            }            return true;        } else {//还是连主服务器            return $this->connectMaster();        }    }        /**     *      * 重新链接。     * 目前在 mysql 2006错误,并且没有运行事务时才开启该方法     * @return Boolean     */    private function reConnect() {    	if (false === $this->connect()) {        	return false;        }        $strUniqueFlagOfLink = $this->getUniqueFlagOfLink();        self::$links[$strUniqueFlagOfLink] = $this->link;        return true;    }    /**     * 判断查询是否要使用主服务器     *     * @return Boolean  true:使用主;false:使用从     */    private function isUseMaster() {    	return (boolean) $this->masterDBInfo['isUseMaster'];    }    /**     * 保存当前状态,并置为$status     *     * @param Boolean $status     * @return String     */    private function changeStatusOfUseMaster($status) {    	($status === true) || $status = false;        # 保存当前状态        $strMasterStatusId = $this->getUniqueMasterStatusId();        $this->masterDBInfo['arrStatusOfUseMaster'][$strMasterStatusId] = $this->masterDBInfo['isUseMaster'];        $this->masterDBInfo['isUseMaster'] = $status;        return $strMasterStatusId;    }    /**     * 开始使用主库     *     * @return String 返回一串标志,供$this->restore 方法使用,用于恢复上一个状态     */    public function beginUseMaster() {        return $this->changeStatusOfUseMaster(true);    }    /**     * 恢复采用 $strMasterStatusId 为句柄保存的上次的状态     *     * @param String $strMasterStatusId     * @return Boolean     *     */    public function restore($strMasterStatusId) {        # 恢复指定状态        if (isset($this->masterDBInfo['arrStatusOfUseMaster'][$strMasterStatusId])) {            $this->masterDBInfo['isUseMaster'] = $this->masterDBInfo['arrStatusOfUseMaster'][$strMasterStatusId];            unset($this->masterDBInfo['arrStatusOfUseMaster'][$strMasterStatusId]);            return true;        }        return false;    }    /**     * 开始使用从库     * 尽量不要使用该接口,除非你明白自己真的需要     *     */    public function beginUseSlave() {        return $this->changeStatusOfUseMaster(false);    }    /**     * 处理连接     *     * @param String $sql     */    protected function xconnect($sql) {    	$isUseSlave = $this->isReadSql($sql) && !$this->isUseMaster() ? true : false;        $intConnectErrorNum = 0;//连接出错次数        while(true) {            if ($isUseSlave) {                $isConnect = $this->connectSlave();            } else {                $isConnect = $this->connectMaster();            }            if (!$isConnect) {//连接失败                ++$intConnectErrorNum;                $strMasterHost = $this->masterDBInfo['host'];                if(4 >= $intConnectErrorNum   //允许四次重试                   && $this->arrDsn['host'] != $strMasterHost //错误不是发生在主服务器上                ) {                    $this->addInvalidSlaveInfo($this->arrDsn['host']);//                    $this->addErrorLog(self::PARAM_NO_IMPORTANCE_ERROR_DIR, $intConnectErrorNum);                    continue;                }                return $this->_halt('服务器连接失败', '01');            }            # 成功就退出循环            break;        }     }    /**     * 执行一条SQL     *     * @param String $sql     * @return resource result     */    public function query($sql) {    	# 每条语句都添加注释,方便debug。比如慢查询日志里知道问题出在哪个文件。	    $sql .= '/* ' . $_SERVER['HTTP_HOST'] . ' in '.$_SERVER['PHP_SELF'] . ' */';	            if (Debug::$open && preg_match('#^\s*select\s#i', $sql)) {            $explain_query = true;        } else {            $explain_query = false;        }        if ($explain_query) {        	# 便于查看不使用缓存时的情况            $sql = preg_replace('#select #i', 'select sql_no_cache ', $sql);        }        $this->last_sql = $sql; // 临时加上        $intQueryErrorNum = 0;//查询出错次数        $intSelectErrorNum = 0;//选择数据库出错次数        while (true) {            $this->xconnect($sql);            $isSelect = mysqli_select_db($this->link, $this->dbname);            # 处理选择数据库错误            if (!$isSelect) {            	if ($this->isRunOnTransaction()) {            		return $this->_halt('进入数据库失败:存在事务,直接停机', '02');            	}            	            	++$intSelectErrorNum;            	if ($intSelectErrorNum > 4) {            		return $this->_halt('进入数据库失败后,重试多次后仍然失败', '02');            	}            	            	// 服务器链接丢失的错误,并且没有运行事务,允许重连            	if ($this->errno() == 2006) {            		if ($this->reConnect()) {            			continue;            		}            	}            	            	if ($this->isRunOnMaster()) {// 主库重连错误,停机            		return $this->_halt('进入数据库失败', '02');            	}            		            	// 否则将这台重库置为无效,尝试重连            	$this->addInvalidSlaveInfo($this->arrDsn['host']);                continue;            }            if (!$this->isReadSql($sql)) {//                # 如果是写入语句,记录开始时间//                $objProcessTimeOfWriteSqlTime = new ProcessTime();//                $objProcessTimeOfWriteSqlTime->start();                # 如果运行于事务中,记录该语句                if ($this->isRunOnTransaction()) {                    $this->masterDBInfo['arrTransactionSqls'][] = $sql;                }            }            $query = mysqli_query($this->link, $sql);//            # 记录慢写入语句//            if (!$this->isReadSql($sql)) {//                if (($runTimeOfWriteSql = $objProcessTimeOfWriteSqlTime->getFinalTime()) > 1) {//                    $this->addSlowWriteSqlLog($runTimeOfWriteSql);//                }//            }            # 处理查询错误            if (!$query) {            	if ($this->isRunOnTransaction()) {            		return $this->_halt('查询数据库失败:存在事务,直接停机', '21');            	}            	++$intQueryErrorNum;            	if ($intQueryErrorNum > 4) {            		return $this->_halt('查询数据库失败后,重试多次后仍然失败', '21');            	}            	            	// 服务器链接丢失的错误,并且没有运行事务,允许重连            	if (in_array($this->errno(), array(2006, 2013))) {            		if ($this->reConnect()) {            			continue;            		}            	}                static $arrConnectErrnos = array(                    1053      //在操作过程中服务器关闭。                    , 1030    //从存储引擎中获得错误                    , 126     //表损坏                );                                if ($this->isRunOnMaster()) {// 主库重连错误,停机            		return $this->_halt('查询主数据库失败', '21');            	}            		                // 否则将这台重库置为无效,尝试重连                if(in_array($this->errno(), $arrConnectErrnos)) {//指定的错误号                    $this->addInvalidSlaveInfo($this->arrDsn['host']);//由于连接失效导致的查询出错,将这台服务器标记为无效                    continue;                }                return $this->_halt('查询数据库失败,不能处理的错误类型', '21');            }            break;        }        # 走到这里,说明成功的执行了写入sql        if ($this->canSetCookieForMasterDBHasWrite($sql)) {// 如果写入语句成功,就写一个保存时间为$tmp_expiration秒的cookie        	$tmp_expiration = 2*60;			Cookie::set(KIF_MASTER_DB_HAS_WRITE_COOKIE_KEY, '1', $tmp_expiration);// 设置$tmp_expiration秒的cookie        }        self::$queries++;        if ($explain_query) {            $begin_microtime = Debug::getTime();            self::$intQueriesTotalTime += $begin_microtime;            $explainSql = 'explain ' . $sql;            $equery = mysqli_query($this->link, $explainSql);            $explain = $this->fetch($equery);            $this->freeResult($equery);            Debug::db($this->getLinkDesc(), $this->dbname, $explainSql, Debug::getTime() - $begin_microtime, $explain);        }        if (!$this->isReadSql($sql)) {        	$begin_microtime = Debug::getTime();        	Debug::db($this->getLinkDesc(), $this->dbname, $sql, Debug::getTime() - $begin_microtime, $query);        }        $this->isRunOnMaster() && $this->endRunOnMaster();        return $query;    }    /**     *     * 判断指定的sql执行后,能否设置后续查询转到主库的cookie     * @param string $sql     * @return boolean     */    private function canSetCookieForMasterDBHasWrite($sql) {    	if (Request::isCLI()) {    		return false;    	}		if ($this->isReadSql($sql)) {			return false;		}		# xhprof_logs表是用来记录慢查询的,没必要因为这个表写入了一次就把后续的请求都转到主库。		if (preg_match('#(xhprof_logs)#', strtolower($sql))) {			return false;		}		return true;    }    public function fetchOne($sql) {    	$begin_microtime = Debug::getTime();        $res = $this->query($sql);        if (!$res) {        	return false;        }        $result = $this->fetch($res);        $this->freeResult($res);        Debug::db($this->getLinkDesc(), $this->dbname, $this->last_sql, Debug::getTime() - $begin_microtime, $result);        return $result;    }    /**     * 获取连接描述,用于Debug输出     */    private function getLinkDesc() {    	$thread_id = mysqli_thread_id($this->link);    	return "mysqli://{$this->arrDsn['user']}:{$this->arrDsn['port']}@{$this->arrDsn['host']} (thread_id: {$thread_id})";    }    /**     * 执行一条SQL并返回此查询包含的所有数据(2维数组)     *     * @param string $sql     * @param string $associateKey 如果指定了$associateKey,返回结果以 每条记录的$associateKey字段做数组下标     * @return false | array     */    public function fetchAll($sql, $associateKey = null) {    	$begin_microtime = Debug::getTime();        $res = $this->query($sql);        if (!$res) {            return false;        }        $result = array();        if ($associateKey) {            while (true) {            	$row = $this->fetch($res);            	if (!$row) {            		break;            	}            	if (isset($row[$associateKey])) {            		$result[$row[$associateKey]] = $row;            	} else {            		$result[] = $row;            	}            }        } else {            while (true) {            	$row = $this->fetch($res);            	if (!$row) {            		break;            	}                $result[] = $row;            }        }        $this->freeResult($res);        Debug::db($this->getLinkDesc(), $this->dbname, $this->last_sql, Debug::getTime() - $begin_microtime, $result);        return $result;    }    /**     * 执行SQL语句并返回第一行第一列     *     * @param string $sql     * @return false | scala     */    public function fetchSclare($sql) {    	$begin_microtime = Debug::getTime();        $result = $this->fetchOne($sql);        if (!$result) {        	return false;        }        $result = array_shift($result);        Debug::db($this->getLinkDesc(), $this->dbname, $this->last_sql, Debug::getTime() - $begin_microtime, $result);        return $result;    }    /**     * 返回上一步 INSERT 查询中产生的 AUTO_INCREMENT 的 ID 号;     * 或者 返回 update 语句中 last_insert_id()函数中表达式的值。     *     * !!请记住,一定紧接在insert 或 update 语句后执行该方法,否则$this->link可能已经指向别的服务器了     *     * @return int | NULL >0:成功取到;0:没取到;NULL:$this->link无效     */    public function insertId() {        return mysqli_insert_id($this->link);    }    /**     * $this->insertId() 的别名     * !!请记住,一定紧接在insert 或 update 语句后执行该方法,否则$this->link可能已经指向别的服务器了     * @return int | NULL >0:成功取到;0:没取到;NULL:$this->link无效     */    public function getLastInsertId() {    	return $this->insertId();    }    /**     * 返回最近一次 INSERT,UPDATE 或 DELETE 查询所影响的记录行数。     * @return int | null 返回值 >= 0:成功;等于 -1:最后一条查询错误;null:$this->link无效     **/    public function affectedRows()    {        return mysqli_affected_rows($this->link);    }    public function fetch($query, $resulttype = MYSQLI_ASSOC) {        return mysqli_fetch_array($query, $resulttype);    }    protected function freeResult($query) {        return mysqli_free_result($query);    }    /**     * 生成唯一的字符串作为事务的唯一id     *     * @return String     */    static private function getUniqueTransactionId() {        return self::getUniqueId('TAId');    }    /**     * 生成唯一的字符串作为保存当前主服务器状态的唯一id     *     * @return String     */    static private function getUniqueMasterStatusId() {        return self::getUniqueId('MSId');    }    /**     * 生成唯一id     *     * @param String $prefix     * @return String     */    static private function getUniqueId($prefix = '') {        if(!is_string($prefix)) {            $prefix = '';        }        return uniqid($prefix . '_'.rand());    }    /**     * 返回上一个错误文本,如果没有出错则返回 ''(空字符串)。     * 如果没有指定连接资源号,则使用上一个成功打开的连接从数据库服务器提取错误信息。     *     * @return String     */    public function error() {        return @mysqli_error($this->link);    }    /**     * 返回上一个错误号     *     * @return int | NULL     */    public function errno() {        return @mysqli_errno($this->link);    }    /**     * 返回上一个连接错误     *     * @return String     */    public function connect_error() {        return @mysqli_connect_error();    }    /**     * 返回上一个连接的错误号     *     * @return int     */    public function connect_errno() {        return @mysqli_connect_errno();    }    /**     * 根据 $this->debug_level 处理一些异常情况     * 1 直接输出错误信息并中断程序     * 2 直接中断程序     * 其他情况不处理错误,返回flase,修改错误代号和本函数所提供的错误信息,最后的是MySQL服务器提供的信息     *     * @param String $msg     * @param String $errorcode     * @return Array     */    function _halt($msg, $errorcode = '00') {        switch ($this->debug_level) {            case 1:                ob_clean();                header("HTTP/1.0 500 Server Error");                header("Expires: ".gmdate("D, d M Y H:i:s", time())." GMT");                header("Last-Modified: ".gmdate("D, d M Y H:i:s", time())." GMT");                header("Cache-Control: private");//                $out = file_get_contents(ROOT_PATH . '/mysql.html');				$out = '$the_error';                $out = str_replace('$the_error', $msg.'<hr />'.$this->error().' No.'.$this->errno()."<!-- {$this->last_sql} -->", $out);//                $this->addErrorLog(null, null, $msg);                echo $out;                exit;                break;            case 2:                ob_clean();                header("HTTP/1.0 500 Server Error");                header("Expires: ".gmdate("D, d M Y H:i:s", time())." GMT");                header("Last-Modified: ".gmdate("D, d M Y H:i:s", time())." GMT");                header("Cache-Control: private");                echo $this->connect_error(), "<br />";                echo $this->connect_errno(), "<br />";                echo "{$msg}<br />";                exit('MySQL.');                break;            default:                $this->errorcode = array($errorcode, $msg.':'.$this->last_sql, $this->errno().": ".$this->error());                return false;                break;        }    }    ########### TRANSACTION ##########    /**     * 开启事务     *     * @return false | string false:失败;string:成功返回事务标志     */    public function startTransaction() {        $strTransactionId = self::getUniqueTransactionId();        if ($this->getTransactions()) {//已存在事务            if ($this->setSavePoint($strTransactionId)) {            	$this->masterDBInfo['transactionIds'][$strTransactionId] = false;                return $strTransactionId;            }        } else {	        # 初次开启事务	        if (true === $this->query('START TRANSACTION;')) {//如果没有建立到当前主服务器的连接,该操作会隐式的建立	            $this->masterDBInfo['transactionIds'][$strTransactionId] = true;	            return $strTransactionId;	        }        }        # 开启事务失败。返回一个独特的字符串,该字符串是不可能出现在 事务id数组中的        return false;    }    /**     * 回滚父事务     *     * @param String $strTransactionId     * @return Boolean     */    private function _rollbackRootTransaction($strTransactionId) {        if ($this->isRootTransaction($strTransactionId)) {//父事务            $this->masterDBInfo['transactionIds'] = null;            $this->masterDBInfo['arrTransactionSqls'] = array();            return $this->query('ROLLBACK;');        }        return false;    }    /**     * 回滚子事务     *     * @param String $strTransactionId     * @return Boolean     */    private function _rollbackSubTransaction($strTransactionId) {        if($this->isSubTransaction($strTransactionId)) {//子事务            $boolStatusTmp = $this->rollbackToSavePoint($strTransactionId);            $this->releaseSavePoint($strTransactionId);            unset($this->masterDBInfo['transactionIds'][$strTransactionId]);            return $boolStatusTmp;        }        return false;    }    /**     * 撤消指定事务     *     * @param String $strTransactionId     * @return Bollean true:成功;false:失败     */    public function rollback($strTransactionId) {        if ($this->isRootTransaction($strTransactionId)) {//父事务            return $this->_rollbackRootTransaction($strTransactionId);        } elseif ($this->isSubTransaction($strTransactionId)) {//子事务            return $this->_rollbackSubTransaction($strTransactionId);        } else {            return false;        }    }    /**     * 提交父事务     *     * @param String $strTransactionId     * @return Boolean     */    private function _commitRootTransaction($strTransactionId) {        if ($this->isRootTransaction($strTransactionId)) {//父事务            $this->masterDBInfo['transactionIds'] = null;            $this->masterDBInfo['arrTransactionSqls'] = array();            return $this->query('COMMIT;');        }        return false;    }    /**     * 提交子事务     *     * @param String $strTransactionId     * @return Boolean     */    private function _commitSubTransaction($strTransactionId) {        if ($this->isSubTransaction($strTransactionId)) {//子事务            $this->releaseSavePoint($strTransactionId);            unset($this->masterDBInfo['transactionIds'][$strTransactionId]);            return true;        }        return false;    }    /**     * 提交指定事务     *     * @param String $strTransactionId     * @return Boolean true:成功;false:失败     */    public function commit($strTransactionId) {        if ($this->isRootTransaction($strTransactionId)) {//父事务            return $this->_commitRootTransaction($strTransactionId);        } elseif ($this->isSubTransaction($strTransactionId)) {//子事务            return $this->_commitSubTransaction($strTransactionId);        } else {            return false;        }    }    /**     * 设置子事务的保存点,用于支持子事务的回滚     *     * @param String $SPId 应该被传递的值是事务的唯一id,即调用self::getUniqueTransactionId()生成的     * @return Boolean  true:成功;false:失败     */    private function setSavePoint($SPId) {        if (true === $this->query("SAVEPOINT {$SPId}")) {            return true;        }        return false;    }    /**     * 获取指定事务的类型(父事务 还是 子事务)     * 只有当前集群中主服务器的连接是可用的,才有事务可言     *     * @param String $strTransactionId     * @return Boolean | null true:父事务;false:子事务;null:无效     */    private function getTransactionTypeById($strTransactionId) {        if ($this->getCurrentMasterLink()) {            if (isset($this->masterDBInfo['transactionIds'][$strTransactionId])) {                return $this->masterDBInfo['transactionIds'][$strTransactionId];            }        }        return null;    }    /**     * 是否父事务     *     * @param String $strTransactionId     * @return Boolean      true:是;false:否     */    private function isRootTransaction($strTransactionId) {        if (true === $this->getTransactionTypeById($strTransactionId)) {            return true;        }        return false;    }    /**     * 是否子事务     *     * @param String $strTransactionId     * @return Boolean      true:是;false:否     */    private function isSubTransaction($strTransactionId) {        if (false === $this->getTransactionTypeById($strTransactionId)) {            return true;        }        return false;    }    /**     * 回滚到指定事务点     *     * @param String $SPId     * @return Boolean  true:成功;false:失败     */    private function rollbackToSavePoint($SPId) {        # 只对子事务的回滚点操作        if ($this->isSubTransaction($SPId)) {            if (true === $this->query("ROLLBACK TO SAVEPOINT {$SPId}")) {                return true;            }        }        return false;    }    /**     * 释放事务保存点     *     * @param String $SPId     * @return Boolean  true:成功;false:失败     */    private function releaseSavePoint($SPId) {        # 只对子事务的回滚点操作        if ($this->isSubTransaction($SPId)) {            if (true === $this->query("RELEASE SAVEPOINT {$SPId}")) {                return true;            }        }        return false;    }}
 |