as2252258/kiri-databases
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 2 Wiki Activity

变更

Browse Source
This commit is contained in:
向林
2023-04-02 23:23:55 +08:00
parent f1faeb038f
commit cd9327f5c2
2 changed files with 284 additions and 278 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Connection.php
+3 -6
View File
@@ -145,7 +145,7 @@ class Connection extends Component
* @return PDO * @return PDO
* @throws Exception * @throws Exception
*/ */
public function getMasterClient(): PDO public function getMasterClient(): Mysql\PDO
{ {
return $this->connection->get([ return $this->connection->get([
'cds' => $this->cds, 'cds' => $this->cds,
@@ -163,7 +163,7 @@ class Connection extends Component
* @return PDO * @return PDO
* @throws Exception * @throws Exception
*/ */
public function getSlaveClient(): PDO public function getSlaveClient(): Mysql\PDO
{ {
return $this->getMasterClient(); return $this->getMasterClient();
} }
@@ -185,14 +185,11 @@ class Connection extends Component
* @return PDO * @return PDO
* @throws Exception * @throws Exception
*/ */
public function getPdo(bool $restore = false): PDO public function getPdo(bool $restore = false): Mysql\PDO
{ {
if ($restore === true) { if ($restore === true) {
return Context::setContext($this->cds, $this->getMasterClient()); return Context::setContext($this->cds, $this->getMasterClient());
} }
if (!Db::inTransactionsActive()) {
return $this->getMasterClient();
}
if (!Context::hasContext($this->cds)) { if (!Context::hasContext($this->cds)) {
return Context::setContext($this->cds, $this->getMasterClient()); return Context::setContext($this->cds, $this->getMasterClient());
} else { } else {
Mysql/PDO.php
+281 -272
View File
@@ -18,331 +18,340 @@ use Swoole\Timer;
class PDO implements StopHeartbeatCheck class PDO implements StopHeartbeatCheck
{ {
const DB_ERROR_MESSAGE = 'The system is busy, please try again later.'; const DB_ERROR_MESSAGE = 'The system is busy, please try again later.';
private ?\PDO $pdo = null; private ?\PDO $pdo = null;
private int $_transaction = 0; private int $_transaction = 0;
private int $_last = 0; private int $_last = 0;
public string $dbname; public string $dbname;
public string $cds; public string $cds;
public string $username; public string $username;
public string $password; public string $password;
public string $charset; public string $charset;
public int $connect_timeout; public int $connect_timeout;
public int $read_timeout; public int $read_timeout;
private int $_timerId = -1; private int $_timerId = -1;
public array $attributes = []; public array $attributes = [];
/** /**
* @param array $config * @param array $config
*/ */
public function __construct(array $config) public function __construct(array $config)
{ {
$this->dbname = $config['dbname']; $this->dbname = $config['dbname'];
$this->cds = $config['cds']; $this->cds = $config['cds'];
$this->username = $config['username']; $this->username = $config['username'];
$this->password = $config['password']; $this->password = $config['password'];
$this->connect_timeout = $config['connect_timeout'] ?? 30; $this->connect_timeout = $config['connect_timeout'] ?? 30;
$this->read_timeout = $config['read_timeout'] ?? 10; $this->read_timeout = $config['read_timeout'] ?? 10;
$this->charset = $config['charset'] ?? 'utf8mb4'; $this->charset = $config['charset'] ?? 'utf8mb4';
$this->attributes = $config['attributes'] ?? []; $this->attributes = $config['attributes'] ?? [];
} }
/** /**
* @return void * @return void
* @throws Exception * @throws Exception
*/ */
public function init(): void public function init(): void
{ {
$eventProvider = Kiri::getDi()->get(EventProvider::class); $eventProvider = Kiri::getDi()->get(EventProvider::class);
$eventProvider->on(OnWorkerExit::class, [$this, 'onWorkerExit']); $eventProvider->on(OnWorkerExit::class, [$this, 'onWorkerExit']);
$this->_timerId = Timer::tick(60000, [$this, 'check']); $this->_timerId = Timer::tick(60000, [$this, 'check']);
} }
/** /**
* @return bool * @return bool
*/ */
public function inTransaction(): bool public function inTransaction(): bool
{ {
return $this->_transaction > 0; return $this->_transaction > 0;
} }
/** /**
* @param Kiri\Server\Events\OnWorkerExit $exit * @param Kiri\Server\Events\OnWorkerExit $exit
* @return void * @return void
*/ */
public function onWorkerExit(OnWorkerExit $exit): void public function onWorkerExit(OnWorkerExit $exit): void
{ {
$this->stopHeartbeatCheck(); $this->stopHeartbeatCheck();
} }
/** /**
* *
*/ */
public function stopHeartbeatCheck(): void public function stopHeartbeatCheck(): void
{ {
$this->pdo = null; $this->pdo = null;
if ($this->_timerId > -1) { if ($this->_timerId > -1) {
Timer::clear($this->_timerId); Timer::clear($this->_timerId);
$this->_timerId = -1; $this->_timerId = -1;
} }
} }
/** /**
* *
*/ */
public function beginTransaction() public function beginTransaction()
{ {
if ($this->_transaction == 0) { if ($this->_transaction == 0) {
$this->_pdo()->beginTransaction(); $this->_pdo()->beginTransaction();
} }
$this->_transaction++; $this->_transaction++;
} }
/** /**
* *
*/ */
public function commit() public function commit()
{ {
$this->_transaction--; $this->_transaction--;
if ($this->_transaction == 0) { if ($this->_transaction == 0) {
$this->_pdo()->commit(); $this->_pdo()->commit();
} }
} }
/** /**
* *
*/ */
public function rollback() public function rollback()
{ {
$this->_transaction--; $this->_transaction--;
if ($this->_transaction == 0) { if ($this->_transaction == 0) {
$this->_pdo()->rollBack(); $this->_pdo()->rollBack();
} }
} }
/** /**
* @param string $sql * @param string $sql
* @param array $params * @param array $params
* @return bool|array|null * @return bool|array|null
* @throws Exception * @throws Exception
*/ */
public function fetchAll(string $sql, array $params = []): bool|null|array public function fetchAll(string $sql, array $params = []): bool|null|array
{ {
$pdo = $this->queryPrev($sql, $params); $pdo = $this->queryPrev($sql, $params);
$result = $pdo->fetchAll(\PDO::FETCH_ASSOC); $result = $pdo->fetchAll(\PDO::FETCH_ASSOC);
$pdo->closeCursor(); $pdo->closeCursor();
return $result; return $result;
} }
/** /**
* @param string $sql * @param string $sql
* @param array $params * @param array $params
* @return bool|array|null * @return bool|array|null
* @throws Exception * @throws Exception
*/ */
public function fetch(string $sql, array $params = []): bool|null|array public function fetch(string $sql, array $params = []): bool|null|array
{ {
$pdo = $this->queryPrev($sql, $params); $pdo = $this->queryPrev($sql, $params);
$result = $pdo->fetch(\PDO::FETCH_ASSOC); $result = $pdo->fetch(\PDO::FETCH_ASSOC);
$pdo->closeCursor(); $pdo->closeCursor();
return $result; return $result;
} }
/** /**
* @param string $sql * @param string $sql
* @param array $params * @param array $params
* @return bool|array|null * @return bool|array|null
* @throws \Exception * @throws \Exception
*/ */
public function fetchColumn(string $sql, array $params = []): bool|null|array public function fetchColumn(string $sql, array $params = []): bool|null|array
{ {
$pdo = $this->queryPrev($sql, $params); $pdo = $this->queryPrev($sql, $params);
$result = $pdo->fetchColumn(\PDO::FETCH_ASSOC); $result = $pdo->fetchColumn(\PDO::FETCH_ASSOC);
$pdo->closeCursor(); $pdo->closeCursor();
return $result; return $result;
} }
/** /**
* @param string $sql * @param string $sql
* @param array $params * @param array $params
* @return int * @return int
* @throws Exception * @throws Exception
*/ */
public function count(string $sql, array $params = []): int public function rowCount(string $sql, array $params = []): int
{ {
$pdo = $this->queryPrev($sql, $params); $pdo = $this->queryPrev($sql, $params);
$result = $pdo->rowCount(); $result = $pdo->rowCount();
$pdo->closeCursor(); $pdo->closeCursor();
return $result; return $result;
} }
/** /**
* @param string $sql * @param string $sql
* @param array $params * @param array $params
* @return PDOStatement * @return PDOStatement
* @throws Exception * @throws Exception
*/ */
private function queryPrev(string $sql, array $params = []): PDOStatement private function queryPrev(string $sql, array $params = []): PDOStatement
{ {
try { try {
if ($this->_timerId === -1) { if ($this->_timerId === -1) {
$this->_timerId = Timer::tick(6000, [$this, 'check']); $this->_timerId = Timer::tick(6000, [$this, 'check']);
} }
$this->_last = time(); $this->_last = time();
if (($statement = $this->_pdo()->query($sql)) === false) { if (($statement = $this->_pdo()->query($sql)) === false) {
throw new Exception($this->_pdo()->errorInfo()[1]); throw new Exception($this->_pdo()->errorInfo()[1]);
} }
return $this->bindValue($statement, $params); return $this->bindValue($statement, $params);
} catch (\PDOException|\Throwable $throwable) { } catch (\PDOException|\Throwable $throwable) {
if (str_contains($throwable->getMessage(), 'MySQL server has gone away')) { if (str_contains($throwable->getMessage(), 'MySQL server has gone away')) {
$this->pdo = null; $this->pdo = null;
return $this->queryPrev($sql, $params); return $this->queryPrev($sql, $params);
} }
throw new Exception($throwable->getMessage()); throw new Exception($throwable->getMessage());
} }
} }
/** /**
* @return bool * @return bool
*/ */
public function check(): bool public function check(): bool
{ {
try { try {
if ($this->_last == 0) $this->_last = time(); if ($this->_last == 0) $this->_last = time();
if (time() - $this->_last >= 600) { if (time() - $this->_last >= 600) {
return $result = false; return $result = false;
} else if (!($this->pdo instanceof \PDO)) { } else if (!($this->pdo instanceof \PDO)) {
return $result = false; return $result = false;
} }
$this->pdo->getAttribute(\PDO::ATTR_SERVER_INFO); $this->pdo->getAttribute(\PDO::ATTR_SERVER_INFO);
$result = true; $result = true;
} catch (\Throwable $throwable) { } catch (\Throwable $throwable) {
if (!str_contains($throwable->getMessage(), 'Idle dis')) { if (!str_contains($throwable->getMessage(), 'Idle dis')) {
Kiri::getLogger()->error($throwable->getMessage()); Kiri::getLogger()->error($throwable->getMessage());
} }
$result = false; $result = false;
} finally { } finally {
return $this->afterCheck($result); return $this->afterCheck($result);
} }
} }
/** /**
* @param bool $result * @param bool $result
* @return bool * @return bool
*/ */
private function afterCheck(bool $result): bool private function afterCheck(bool $result): bool
{ {
$container = Kiri::getDi()->get(WorkerStatus::class); $container = Kiri::getDi()->get(WorkerStatus::class);
if (!$result || $container->is(StatusEnum::EXIT)) { if (!$result || $container->is(StatusEnum::EXIT)) {
$this->pdo = null; $this->pdo = null;
$result = Timer::clear($this->_timerId); $result = Timer::clear($this->_timerId);
$this->_timerId = -1; $this->_timerId = -1;
} }
return $result; return $result;
} }
/** /**
* @param PDOStatement $statement * @param PDOStatement $statement
* @param array $params * @param array $params
* @return PDOStatement * @return PDOStatement
*/ */
private function bindValue(PDOStatement $statement, array $params = []): PDOStatement private function bindValue(PDOStatement $statement, array $params = []): PDOStatement
{ {
if (empty($params)) return $statement; if (empty($params)) return $statement;
foreach ($params as $key => $param) { foreach ($params as $key => $param) {
$statement->bindValue($key, $param); $statement->bindValue($key, $param);
} }
return $statement; return $statement;
} }
/** /**
* @param string $sql * @param string $sql
* @param array $params * @param array $params
* @return int * @return int
* @throws Exception * @throws Exception
*/ */
public function execute(string $sql, array $params = []): int public function execute(string $sql, array $params = []): int
{ {
$pdo = $this->_pdo(); $pdo = $this->_pdo();
if ($this->_timerId === -1) { if ($this->_timerId === -1) {
$this->_timerId = Timer::tick(6000, [$this, 'check']); $this->_timerId = Timer::tick(6000, [$this, 'check']);
} }
$this->_last = time(); $this->_last = time();
if (!(($prepare = $pdo->prepare($sql)) instanceof PDOStatement)) { if (!(($prepare = $pdo->prepare($sql)) instanceof PDOStatement)) {
throw new Exception($prepare->errorInfo()[2] ?? static::DB_ERROR_MESSAGE); throw new Exception($prepare->errorInfo()[2] ?? static::DB_ERROR_MESSAGE);
} }
if ($prepare->execute($params) === false) { if ($prepare->execute($params) === false) {
throw new Exception($prepare->errorInfo()[2] ?? static::DB_ERROR_MESSAGE); throw new Exception($prepare->errorInfo()[2] ?? static::DB_ERROR_MESSAGE);
} }
$result = (int)$pdo->lastInsertId(); $result = (int)$pdo->lastInsertId();
$prepare->closeCursor(); $prepare->closeCursor();
return $result == 0 ? true : $result; return $result == 0 ? true : $result;
} }
/** /**
* @return \PDO * @return array
*/ */
public function _pdo(): \PDO public function errorInfo(): array
{ {
if (!($this->pdo instanceof \PDO)) { return $this->pdo->errorInfo();
$this->pdo = $this->newClient(); }
}
return $this->pdo;
}
/** /**
* @return \PDO * @return \PDO
*/ */
private function newClient(): \PDO public function _pdo(): \PDO
{ {
$link = new \PDO('mysql:dbname=' . $this->dbname . ';host=' . $this->cds, $this->username, $this->password, [ if (!($this->pdo instanceof \PDO)) {
\PDO::ATTR_EMULATE_PREPARES => false, $this->pdo = $this->newClient();
\PDO::ATTR_CASE => \PDO::CASE_NATURAL, }
\PDO::ATTR_TIMEOUT => $this->connect_timeout, return $this->pdo;
\PDO::MYSQL_ATTR_INIT_COMMAND => 'SET NAMES ' . $this->charset }
]);
$link->setAttribute(\PDO::ATTR_ERRMODE, \PDO::ERRMODE_EXCEPTION);
$link->setAttribute(\PDO::ATTR_STRINGIFY_FETCHES, false); /**
$link->setAttribute(\PDO::ATTR_ORACLE_NULLS, \PDO::NULL_EMPTY_STRING); * @return \PDO
foreach ($this->attributes as $key => $attribute) { */
$link->setAttribute($key, $attribute); private function newClient(): \PDO
} {
return $link; $link = new \PDO('mysql:dbname=' . $this->dbname . ';host=' . $this->cds, $this->username, $this->password, [
} \PDO::ATTR_EMULATE_PREPARES => false,
\PDO::ATTR_CASE => \PDO::CASE_NATURAL,
\PDO::ATTR_TIMEOUT => $this->connect_timeout,
\PDO::MYSQL_ATTR_INIT_COMMAND => 'SET NAMES ' . $this->charset
]);
$link->setAttribute(\PDO::ATTR_ERRMODE, \PDO::ERRMODE_EXCEPTION);
$link->setAttribute(\PDO::ATTR_STRINGIFY_FETCHES, false);
$link->setAttribute(\PDO::ATTR_ORACLE_NULLS, \PDO::NULL_EMPTY_STRING);
foreach ($this->attributes as $key => $attribute) {
$link->setAttribute($key, $attribute);
}
return $link;
}
} }