add comysqli.php

core/framework/db/comysqli.php

@@ -0,0 +1,693 @@
+<?php
+/**
+ * mysqli驱动
+ *
+ *
+ * @package
+ */
+
+
+defined('InShopNC') or exit('Access Invalid!');
+
+require_once(BASE_CORE_PATH . '/framework/function/statistic_tm.php');
+
+class Db
+{
+    const ErrUnConnect = 2006;
+    const ErrLock = 1205;
+
+    private static $ifTransacting = false;
+
+    private function __construct()
+    {
+        if (!extension_loaded('mysqli')) {
+            throw_exception("Db Error: mysqli is not install");
+        } else {
+            self::init_link();
+        }
+    }
+
+    private static function init_link()
+    {
+        $cid = Co::getCid();
+        $link = CoMysqliPool::instance()->find($cid);
+
+        $reset = false;
+        if (!isset($link['master']) || (is_object($link['master']) == false)) {
+            $link = self::connect($link,'master');
+            $reset = true;
+        }
+
+        if (!isset($link['slave']) || (is_object($link['slave']) == false)) {
+            $link = self::connect($link,'slave');
+            $reset = true;
+        }
+
+        if($reset) {
+            CoMysqliPool::instance()->reset($cid,$link);
+        }
+
+        return $link;
+    }
+
+    private static function connect($link,$host = 'slave')
+    {
+        self::closeLink($link,$host);
+
+        $conf = C('db.' . $host);
+        $link[$host] = mysqli_init();
+        $link[$host]->real_connect($conf['dbhost'], $conf['dbuser'], $conf['dbpwd'], $conf['dbname'], $conf['dbport'],NULL,MYSQLI_CLIENT_FOUND_ROWS);
+
+        if (mysqli_connect_errno()) {
+            $err_no = mysqli_connect_errno();
+            $error = mysqli_connect_error();
+            throw_exception("Db Error: database connect failed errno={$err_no} err={$error}");
+        }
+
+        switch (strtoupper($conf['dbcharset'])) {
+            case 'UTF-8':
+                $query_string = "
+		                 SET CHARACTER_SET_CLIENT = utf8mb4,
+		                 CHARACTER_SET_CONNECTION = utf8mb4,
+		                 CHARACTER_SET_DATABASE = utf8mb4,
+		                 CHARACTER_SET_RESULTS = utf8mb4,
+		                 CHARACTER_SET_SERVER = utf8mb4,
+		                 COLLATION_CONNECTION = utf8mb4_unicode_ci,
+		                 COLLATION_DATABASE = utf8mb4_unicode_ci,
+		                 COLLATION_SERVER = utf8mb4_unicode_ci,
+		                 sql_mode=''";
+                break;
+            case 'GBK':
+                $query_string = "
+		   			    SET CHARACTER_SET_CLIENT = gbk,
+		                 CHARACTER_SET_CONNECTION = gbk,
+		                 CHARACTER_SET_DATABASE = gbk,
+		                 CHARACTER_SET_RESULTS = gbk,
+		                 CHARACTER_SET_SERVER = gbk,
+		                 COLLATION_CONNECTION = gbk_chinese_ci,
+		                 COLLATION_DATABASE = gbk_chinese_ci,
+		                 COLLATION_SERVER = gbk_chinese_ci,
+		                 sql_mode=''";
+                break;
+            default:
+                $error = "Db Error: charset is Invalid";
+                throw_exception($error);
+        }
+
+        //进行编码声明
+        if (!$link[$host]->query($query_string)) {
+            throw_exception("Db Error: " . mysqli_error($link[$host]));
+        }
+
+        return $link;
+    }
+
+    private static function closeLink($link,$host)
+    {
+        if (isset($link[$host]) && is_object($link[$host])) {
+            $link[$host]->close();
+        }
+        $link[$host] = null;
+    }
+
+    public static function ping($host = 'master')
+    {
+        $cid = Co::getCid();
+        $link = CoMysqliPool::instance()->find($cid);
+
+        if (is_object($link[$host])) {
+            return $link[$host]->ping();
+        }
+    }
+
+    /**
+     * 执行查询
+     *
+     * @param string $sql
+     * @return mixed
+     */
+    public static function query($sql, $host = 'master')
+    {
+        $link = self::init_link();
+        $count = 0;
+        do {
+            $staer = new statistic_tm();
+            $query = $link[$host]->query($sql);
+            $staer->end();
+
+            if ($query === false)
+            {
+                $eno = mysqli_errno($link[$host]);
+                $emsg = mysqli_error($link[$host]);
+
+                $error = "DbError eno={$eno} msg={$emsg}";
+                Log::record("DbError eno={$eno} msg={$emsg} sql={$sql}", Log::ERR);
+
+                if ($eno == self::ErrUnConnect) {
+                    if ($count > 0) return false;
+                    $link = self::connect($link,$host);
+                    CoMysqliPool::instance()->reset($cid,$link);
+                    $count++;
+                } elseif (C('debug')) {
+                    throw_exception($error . '<br/>' . $sql);
+                    return false;
+                }
+                elseif(self::ping($host) == false)
+                {
+                    if ($count > 0) return false;
+                    $link = self::connect($link,$host);
+                    CoMysqliPool::instance()->reset($cid,$link);
+                    $count++;
+                }
+                else {
+                    return false;
+                }
+            }
+            else {
+                Log::record($sql . " [ RunTime:" . $staer->elapsed(6) . "s ]", Log::SQL);
+                return $query;
+            }
+        } while (true);
+    }
+
+    /**
+     * 取得数组
+     *
+     * @param string $sql
+     * @return array
+     */
+    public static function getAll($sql, $host = 'slave')
+    {
+        $result = self::query($sql, $host);
+        if ($result === false) return [];
+
+        $array = [];
+        while ($tmp = mysqli_fetch_array($result, MYSQLI_ASSOC)) {
+            $array[] = $tmp;
+        }
+        return !empty($array) ? $array : null;
+    }
+
+    /**
+     * SELECT查询
+     *
+     * @param array $param 参数
+     * @param object $obj_page 分类对象
+     * @return array
+     */
+    public static function select($param, $obj_page = '', $host = 'slave')
+    {
+        static $_cache = [];
+        if (empty($param)) throw_exception('Db Error: select param is empty!');
+
+        if (empty($param['field'])) {
+            $param['field'] = '*';
+        }
+        if (empty($param['count'])) {
+            $param['count'] = 'count(*)';
+        }
+
+        if (isset($param['index'])) {
+            $param['index'] = 'USE INDEX (' . $param['index'] . ')';
+        }
+
+        if (trim($param['where']) != '') {
+            if (strtoupper(substr(trim($param['where']), 0, 5)) != 'WHERE') {
+                if (strtoupper(substr(trim($param['where']), 0, 3)) == 'AND') {
+                    $param['where'] = substr(trim($param['where']), 3);
+                }
+                $param['where'] = 'WHERE ' . $param['where'];
+            }
+        } else {
+            $param['where'] = '';
+        }
+
+        $param['where_group'] = '';
+        if (!empty($param['group'])) {
+            $param['where_group'] .= ' group by ' . $param['group'];
+        }
+        $param['where_order'] = '';
+        if (!empty($param['order'])) {
+            $param['where_order'] .= ' order by ' . $param['order'];
+        }
+
+        //判断是否是联表
+        $tmp_table = explode(',', $param['table']);
+        if (!empty($tmp_table) && count($tmp_table) > 1) {
+            //判断join表数量和join条件是否一致
+            if ((count($tmp_table) - 1) != count($param['join_on'])) {
+                throw_exception('Db Error: join number is wrong!');
+            }
+
+            //trim 掉空白字符
+            foreach ($tmp_table as $key => $val) {
+                $tmp_table[$key] = trim($val);
+            }
+
+            //拼join on 语句
+            $tmp_sql = '';
+            for ($i = 1; $i < count($tmp_table); $i++) {
+                $tmp_sql .= $param['join_type'] . ' `' . DBPRE . $tmp_table[$i] . '` as `' . $tmp_table[$i] . '` ON ' . $param['join_on'][$i - 1] . ' ';
+            }
+            $sql = 'SELECT ' . $param['field'] . ' FROM `' . DBPRE . $tmp_table[0] . '` as `' . $tmp_table[0] . '` ' . $tmp_sql . ' ' . $param['where'] . $param['where_group'] . $param['where_order'];
+
+            //如果有分页，那么计算信息总数
+            $count_sql = 'SELECT ' . $param['count'] . ' as count FROM `' . DBPRE . $tmp_table[0] . '` as `' . $tmp_table[0] . '` ' . $tmp_sql . ' ' . $param['where'] . $param['where_group'];
+        } else {
+            $sql = 'SELECT ' . $param['field'] . ' FROM `' . DBPRE . $param['table'] . '` as `' . $param['table'] . '` ' . $param['index'] . ' ' . $param['where'] . $param['where_group'] . $param['where_order'];
+            $count_sql = 'SELECT ' . $param['count'] . ' as count FROM `' . DBPRE . $param['table'] . '` as `' . $param['table'] . '` ' . $param['index'] . ' ' . $param['where'];
+        }
+
+        //limit ，如果有分页对象的话，那么优先分页对象
+        if ($obj_page instanceof Page) {
+            $count_query = self::query($count_sql, $host);
+            $count_fetch = mysqli_fetch_array($count_query, MYSQLI_ASSOC);
+            $obj_page->setTotalNum($count_fetch['count']);
+            $param['limit'] = $obj_page->getLimitStart() . "," . $obj_page->getEachNum();
+        }
+
+        if ($param['limit'] != '') {
+            $sql .= ' limit ' . $param['limit'];
+        }
+        if ($param['cache'] !== false) {
+            $key = is_string($param['cache_key']) ? $param['cache_key'] : md5($sql);
+            if (isset($_cache[$key])) return $_cache[$key];
+        }
+
+        $result = self::query($sql, $host);
+        if ($result === false) {
+            $result = [];
+        }
+        $array = [];
+        while ($tmp = mysqli_fetch_array($result, MYSQLI_ASSOC)) {
+            $array[] = $tmp;
+        }
+
+        if ($param['cache'] !== false && !isset($_cache[$key])) {
+            $_cache[$key] = $array;
+        }
+        return $array;
+    }
+
+    /**
+     * 插入操作
+     *
+     * @param string $table_name 表名
+     * @param array $insert_array 待插入数据
+     * @return mixed
+     */
+    public static function insert($table_name, $insert_array = [], $host = 'master')
+    {
+        if (!is_array($insert_array)) return false;
+        $fields = [];
+        $value = [];
+        foreach ($insert_array as $key => $val) {
+            $fields[] = self::parseKey($key);
+            $value[] = self::parseValue($val);
+        }
+        $sql = 'INSERT INTO `' . DBPRE . $table_name . '` (' . implode(',', $fields) . ') VALUES(' . implode(',', $value) . ')';
+
+        //当数据库没有自增ID的情况下，返回 是否成功
+        $result = self::query($sql, $host);
+        $insert_id = self::getLastId($host);
+        return $insert_id ? $insert_id : $result;
+    }
+
+    /**
+     * 批量插入
+     *
+     * @param string $table_name 表名
+     * @param array $insert_array 待插入数据
+     * @return mixed
+     */
+    public static function insertAll($table_name, $insert_array = [], $host = 'master')
+    {
+        if (!is_array($insert_array[0])) return false;
+        $fields = array_keys($insert_array[0]);
+        array_walk($fields, [__CLASS__, 'parseKey']);
+        $values = [];
+        foreach ($insert_array as $data) {
+            $value = array();
+            foreach ($data as $key => $val) {
+                $val = self::parseValue($val);
+                if (is_scalar($val)) {
+                    $value[] = $val;
+                }
+            }
+            $values[] = '(' . implode(',', $value) . ')';
+        }
+
+        $sql = 'INSERT INTO `' . DBPRE . $table_name . '` (' . implode(',', $fields) . ') VALUES ' . implode(',', $values);
+        $result = self::query($sql, $host);
+        $insert_id = self::getLastId($host);
+        return $insert_id ? $insert_id : $result;
+    }
+
+    /**
+     * 更新操作
+     *
+     * @param string $table_name 表名
+     * @param array $update_array 待更新数据
+     * @param string $where 执行条件
+     * @return bool
+     */
+    public static function update($table_name, $update_array = [], $where = '', $host = 'master')
+    {
+        if (!is_array($update_array)) return false;
+        $string_value = '';
+        foreach ($update_array as $k => $v) {
+            if (is_array($v)) {
+                switch ($v['sign']) {
+                    case 'increase':
+                        $string_value .= " $k = $k + " . $v['value'] . ",";
+                        break;
+                    case 'decrease':
+                        $string_value .= " $k = $k - " . $v['value'] . ",";
+                        break;
+                    case 'calc':
+                        $string_value .= " $k = " . $v['value'] . ",";
+                        break;
+                    default:
+                        $string_value .= " $k = " . self::parseValue($v['value']) . ",";
+                }
+            } else {
+                $string_value .= " $k = " . self::parseValue($v) . ",";
+            }
+        }
+
+        $string_value = trim(trim($string_value), ',');
+        if (trim($where) != '') {
+            if (strtoupper(substr(trim($where), 0, 5)) != 'WHERE') {
+                if (strtoupper(substr(trim($where), 0, 3)) == 'AND') {
+                    $where = substr(trim($where), 3);
+                }
+                $where = ' WHERE ' . $where;
+            }
+        }
+        $sql = 'UPDATE `' . DBPRE . $table_name . '` AS `' . $table_name . '` SET ' . $string_value . ' ' . $where;
+        $result = self::query($sql, $host);
+        return $result;
+    }
+
+    /**
+     * 删除操作
+     *
+     * @param string $table_name 表名
+     * @param string $where 执行条件
+     * @return bool
+     */
+    public static function delete($table_name, $where = '', $host = 'master')
+    {
+        if (trim($where) != '') {
+            if (strtoupper(substr(trim($where), 0, 5)) != 'WHERE') {
+                if (strtoupper(substr(trim($where), 0, 3)) == 'AND') {
+                    $where = substr(trim($where), 3);
+                }
+                $where = ' WHERE ' . $where;
+            }
+            $sql = 'DELETE FROM `' . DBPRE . $table_name . '` ' . $where;
+            return self::query($sql, $host);
+        } else {
+            throw_exception('Db Error: the condition of delete is empty!');
+        }
+    }
+
+    /**
+     * 取得上一步插入产生的ID
+     *
+     * @return int
+     */
+    public static function getLastId($host = 'master')
+    {
+        $link = self::init_link();
+        $id = mysqli_insert_id($link[$host]);
+        if (!$id) {
+            $result = self::query('SELECT last_insert_id() as id', $host);
+            if ($result === false) return false;
+            $id = mysqli_fetch_array($result, MYSQLI_ASSOC);
+            $id = $id['id'];
+        }
+        return $id;
+    }
+
+    /**
+     * 取得一行信息
+     *
+     * @param array $param
+     * @param string $fields
+     * @return array
+     */
+    public static function getRow($param, $fields = '*', $host = 'slave')
+    {
+        $table = $param['table'];
+        $wfield = $param['field'];
+        $value = $param['value'];
+
+        if (is_array($wfield)) {
+            $where = [];
+            foreach ($wfield as $k => $v) {
+                $where[] = $v . "='" . $value[$k] . "'";
+            }
+            $where = implode(' and ', $where);
+        } else {
+            $where = $wfield . "='" . $value . "'";
+        }
+
+        $sql = "SELECT " . $fields . " FROM `" . DBPRE . $table . "` WHERE " . $where;
+        $result = self::query($sql, $host);
+        if ($result === false) return [];
+        return mysqli_fetch_array($result, MYSQLI_ASSOC);
+    }
+
+    /**
+     * 执行REPLACE操作
+     *
+     * @param string $table_name 表名
+     * @param array $replace_array 待更新的数据
+     * @return bool
+     */
+    public static function replace($table_name, $replace_array = [], $host = 'master')
+    {
+        if (!empty($replace_array)) {
+            $string_field = '';
+            $string_value = '';
+            foreach ($replace_array as $k => $v) {
+                $string_field .= " $k ,";
+                $string_value .= " '" . $v . "',";
+            }
+            $sql = 'REPLACE INTO `' . DBPRE . $table_name . '` (' . trim($string_field, ', ') . ') VALUES(' . trim($string_value, ', ') . ')';
+            return self::query($sql, $host);
+        } else {
+            return false;
+        }
+    }
+
+    /**
+     * 返回单表查询记录数量
+     *
+     * @param string $table 表名
+     * @param $condition mixed 查询条件，可以为空，也可以为数组或字符串
+     * @return int
+     */
+    public static function getCount($table, $condition = null, $host = 'slave')
+    {
+        if (!empty($condition) && is_array($condition)) {
+            $where = '';
+            foreach ($condition as $key => $val) {
+                self::parseKey($key);
+                $val = self::parseValue($val);
+                $where .= ' AND ' . $key . '=' . $val;
+            }
+            $where = ' WHERE ' . substr($where, 4);
+        } elseif (is_string($condition)) {
+            if (strtoupper(substr(trim($condition), 0, 3)) == 'AND') {
+                $where = ' WHERE ' . substr(trim($condition), 4);
+            } else {
+                $where = ' WHERE ' . $condition;
+            }
+        }
+
+        $sql = 'SELECT COUNT(*) as `count` FROM `' . DBPRE . $table . '` as `' . $table . '` ' . (isset($where) ? $where : '');
+        $result = self::query($sql, $host);
+        if ($result === false) return 0;
+        $result = mysqli_fetch_array($result, MYSQLI_ASSOC);
+        return $result['count'];
+    }
+
+    /**
+     * 执行SQL语句
+     *
+     * @param string $sql 待执行的SQL
+     * @return
+     */
+    public static function execute($sql, $host = 'master')
+    {
+        $result = self::query($sql, $host);
+        return $result;
+    }
+
+    /**
+     * 列出所有表
+     *
+     * @return array
+     */
+    public static function showTables($host = 'slave')
+    {
+        $sql = 'SHOW TABLES';
+        $result = self::query($sql, $host);
+        if ($result === false) return [];
+
+        $array = [];
+        while ($tmp = mysqli_fetch_array($result, MYSQLI_ASSOC)) {
+            $array[] = $tmp;
+        }
+        return $array;
+    }
+
+    /**
+     * 显示建表语句
+     *
+     * @param string $table
+     * @return string
+     */
+    public static function showCreateTable($table, $host = 'slave')
+    {
+        $sql = 'SHOW CREATE TABLE `' . DBPRE . $table . '`';
+        $result = self::query($sql, $host);
+        if ($result === false) return '';
+        $result = mysqli_fetch_array($result, MYSQLI_ASSOC);
+        return $result['Create Table'];
+    }
+
+    /**
+     * 显示表结构信息
+     *
+     * @param string $table
+     * @return array
+     */
+    public static function showColumns($table, $host = 'slave')
+    {
+        $sql = 'SHOW COLUMNS FROM `' . DBPRE . $table . '`';
+        $result = self::query($sql, $host);
+        if ($result === false) return [];
+        $array = [];
+
+        while ($tmp = mysqli_fetch_array($result, MYSQLI_ASSOC)) {
+            $array[$tmp['Field']] = array(
+                'name' => $tmp['Field'],
+                'type' => $tmp['Type'],
+                'null' => $tmp['Null'],
+                'default' => $tmp['Default'],
+                'primary' => (strtolower($tmp['Key']) == 'pri'),
+                'autoinc' => (strtolower($tmp['Extra']) == 'auto_increment'),
+            );
+        }
+        return $array;
+    }
+
+    /**
+     * 取得服务器信息
+     *
+     * @return string
+     */
+    public static function getServerInfo($host = 'slave')
+    {
+        $link = self::init_link();
+        $result = mysqli_get_server_info($link[$host]);
+        return $result;
+    }
+
+    /**
+     * 格式化字段
+     *
+     * @param string $key 字段名
+     * @return string
+     */
+    public static function parseKey(&$key)
+    {
+        $key = trim($key);
+        if (!preg_match('/[,\'\"\*\(\)`.\s]/', $key)) {
+            $key = '`' . $key . '`';
+        }
+        return $key;
+    }
+
+    /**
+     * 格式化值
+     *
+     * @param mixed $value
+     * @return mixed
+     */
+    public static function parseValue($value)
+    {
+        $value = addslashes(stripslashes($value));//重新加斜线，防止从数据库直接读取出错
+        return "'" . $value . "'";
+    }
+
+    public static function beginTransaction($host = 'master')
+    {
+        $link = self::init_link();
+        if (!self::$ifTransacting)
+        {
+            $result = $link[$host]->autocommit(false);
+            if ($result == false) {
+                $link = self::connect($host);
+                CoMysqliPool::instance()->reset($cid,$link);
+                self::$ifTransacting = false;
+
+                Log::record("autocommit start 2", Log::DEBUG);
+                $link[$host]->autocommit(false);
+            } else {
+                Log::record("autocommit start 1", Log::DEBUG);
+            }
+        }
+        self::$ifTransacting = true;
+    }
+
+    public static function commit($host = 'master')
+    {
+        $link = self::init_link();
+        $ifTransacting = CoMysqliPool::instance()->transaction(Co::getCid());
+        if ($ifTransacting)
+        {
+            $result = $link[$host]->commit();
+            Log::record("autocommit end commit", Log::DEBUG);
+            $link[$host]->autocommit(true);
+            CoMysqliPool::instance()->commit(Co::getCid());
+
+            if (!$result) {
+                $err = mysqli_error($link[$host]);
+                throw_exception("Db Error: {$err}");
+            }
+        }
+    }
+
+    public static function rollback($host = 'master')
+    {
+        $link = self::init_link();
+        $ifTransacting = CoMysqliPool::instance()->transaction(Co::getCid());
+
+        if ($ifTransacting)
+        {
+            $result = $link[$host]->rollback();
+            CoMysqliPool::instance()->rollback(Co::getCid());
+
+            Log::record("autocommit end rollback", Log::DEBUG);
+            $fsuccess = $link[$host]->autocommit(true);
+
+            if (!$result || !$fsuccess) {
+                $err = mysqli_error($link[$host]);
+                $link = self::connect($host);
+                CoMysqliPool::instance()->reset(Co::getCid(),$link);
+
+                throw_exception("Db Error: {$err}");
+            }
+        }
+    }
+
+    public static function affected_rows($host = 'master')
+    {
+        $link = self::init_link();
+        return $link[$host]->affected_rows;
+    }
+}