zhamao-framework/src/ZM/DB/DB.php

<?php


namespace ZM\DB;


use Exception;
use framework\Console;
use framework\ZMBuf;
use PDOStatement;
use Swoole\Coroutine;
use Swoole\Database\PDOStatementProxy;
use ZM\Exception\DbException;

class DB
{
    private static $table_list = [];

    /**
     * @throws DbException
     * @throws Exception
     */
    public static function initTableList() {
        if (!extension_loaded("mysqlnd")) throw new Exception("Can not find mysqlnd PHP extension.");
        $result = self::rawQuery("select TABLE_NAME from INFORMATION_SCHEMA.TABLES where TABLE_SCHEMA='" . ZMBuf::globals("sql_config")["sql_database"] . "';", []);
        foreach ($result as $v) {
            self::$table_list[] = $v['TABLE_NAME'];
        }
    }

    /**
     * @param $table_name
     * @param bool $enable_cache
     * @return Table
     * @throws DbException
     */
    public static function table($table_name, $enable_cache = null) {
        if (Table::getTableInstance($table_name) === null) {
            if (in_array($table_name, self::$table_list))
                return new Table($table_name, $enable_cache ?? ZMBuf::globals("sql_config")["sql_enable_cache"]);
            elseif(ZMBuf::$sql_pool !== null){
                throw new DbException("Table " . $table_name . " not exist in database.");
            } else {
                throw new DbException("Database connection not exist or connect failed. Please check sql configuration");
            }
        }
        return Table::getTableInstance($table_name);
    }

    /**
     * @param $line
     * @throws DbException
     */
    public static function statement($line) {
        self::rawQuery($line, []);
    }

    /**
     * @param $line
     * @return bool
     * @throws DbException
     */
    public static function unprepared($line) {
        if (ZMBuf::get("sql_log") === true) {
            $starttime = microtime(true);
        }
        try {
            $conn = ZMBuf::$sql_pool->get();
            if ($conn === false) {
                ZMBuf::$sql_pool->put(null);
                throw new DbException("无法连接SQL！" . $line);
            }
            $result = $conn->query($line) === false ? false : true;
            ZMBuf::$sql_pool->put($conn);
            return $result;
        } catch (DBException $e) {
            if (ZMBuf::get("sql_log") === true) {
                $log =
                    "[" . date("Y-m-d H:i:s") .
                    " " . round(microtime(true) - $starttime, 5) .
                    "] " . $line . " (Error:" . $e->getMessage() . ")\n";
                Coroutine::writeFile(CRASH_DIR . "sql.log", $log, FILE_APPEND);
            }
            Console::warning($e->getMessage());
            throw $e;
        }
    }

    /**
     * @param string $line
     * @param array $params
     * @return mixed
     * @throws DbException
     */
    public static function rawQuery(string $line, $params = []) {
        if (ZMBuf::get("sql_log") === true) {
            $starttime = microtime(true);
        }
        Console::debug("MySQL: ".$line);
        try {
            $conn = ZMBuf::$sql_pool->get();
            if ($conn === false) {
                ZMBuf::$sql_pool->put(null);
                throw new DbException("无法连接SQL！" . $line);
            }
            $ps = $conn->prepare($line);
            if ($ps === false) {
                ZMBuf::$sql_pool->put(null);
                throw new DbException("SQL语句查询错误，" . $line . "，错误信息：" . $conn->error);
            } else {
                if (!($ps instanceof PDOStatement) && !($ps instanceof PDOStatementProxy)) {
                    var_dump($ps);
                    ZMBuf::$sql_pool->put(null);
                    throw new DbException("语句查询错误！返回的不是 PDOStatement" . $line);
                }
                if ($params == []) $result = $ps->execute();
                elseif (!is_array($params)) {
                    $result = $ps->execute([$params]);
                } else $result = $ps->execute($params);
                if ($result !== true) {
                    ZMBuf::$sql_pool->put(null);
                    throw new DBException("语句[$line]错误！" . $ps->errorInfo()[2]);
                    //echo json_encode(debug_backtrace(), 128 | 256);
                }
                ZMBuf::$sql_pool->put($conn);
                if (ZMBuf::get("sql_log") === true) {
                    $log =
                        "[" . date("Y-m-d H:i:s") .
                        " " . round(microtime(true) - $starttime, 4) .
                        "] " . $line . " " . json_encode($params, JSON_UNESCAPED_UNICODE) . "\n";
                    Coroutine::writeFile(CRASH_DIR . "sql.log", $log, FILE_APPEND);
                }
                return $ps->fetchAll();
            }
        } catch (DBException $e) {
            if (ZMBuf::get("sql_log") === true) {
                $log =
                    "[" . date("Y-m-d H:i:s") .
                    " " . round(microtime(true) - $starttime, 4) .
                    "] " . $line . " " . json_encode($params, JSON_UNESCAPED_UNICODE) . " (Error:" . $e->getMessage() . ")\n";
                Coroutine::writeFile(CRASH_DIR . "sql.log", $log, FILE_APPEND);
            }
            if(mb_strpos($e->getMessage(), "has gone away") !== false) {
                zm_sleep(0.2);
                Console::warning("Gone away of MySQL! retrying!");
                return self::rawQuery($line, $params);
            }
            Console::warning($e->getMessage());
            throw $e;
        }
    }

    public static function isTableExists($table) {
        return in_array($table, self::$table_list);
    }
}