This commit is contained in:
2023-04-16 01:45:33 +08:00
parent 3cd2e8d889
commit f6d4b0fbca
12 changed files with 24 additions and 417 deletions
+1 -3
View File
@@ -5,8 +5,6 @@ namespace Database\Annotation;
use Attribute; use Attribute;
use Database\Base\Getter;
use Kiri\Annotation\AbstractAttribute;
/** /**
@@ -14,7 +12,7 @@ use Kiri\Annotation\AbstractAttribute;
* @package Annotation\Model * @package Annotation\Model
* @deprecated * @deprecated
*/ */
#[Attribute(Attribute::TARGET_METHOD)] class Get extends AbstractAttribute #[Attribute(Attribute::TARGET_METHOD)] class Get
{ {
+1 -2
View File
@@ -4,7 +4,6 @@
namespace Database\Annotation; namespace Database\Annotation;
use Kiri\Annotation\AbstractAttribute;
use Exception; use Exception;
@@ -12,7 +11,7 @@ use Exception;
* Class Relation * Class Relation
* @package Annotation\Model * @package Annotation\Model
*/ */
#[\Attribute(\Attribute::TARGET_METHOD)] class Relation extends AbstractAttribute #[\Attribute(\Attribute::TARGET_METHOD)] class Relation
{ {
+1 -3
View File
@@ -4,14 +4,12 @@
namespace Database\Annotation; namespace Database\Annotation;
use Database\Base\Setter;
use Kiri\Annotation\AbstractAttribute;
/** /**
* @deprecated * @deprecated
*/ */
#[\Attribute(\Attribute::TARGET_METHOD)] class Set extends AbstractAttribute #[\Attribute(\Attribute::TARGET_METHOD)] class Set
{ {
+1 -2
View File
@@ -1,13 +1,12 @@
<?php <?php
declare(strict_types=1);
namespace Database; namespace Database;
use Co\Channel; use Co\Channel;
use Exception; use Exception;
use Kiri\Di\LocalService; use Kiri\Di\LocalService;
use Kiri\Error\StdoutLoggerInterface;
use Swoole\Coroutine; use Swoole\Coroutine;
use Swoole\Process;
use Symfony\Component\Console\Command\Command; use Symfony\Component\Console\Command\Command;
use Symfony\Component\Console\Input\InputArgument; use Symfony\Component\Console\Input\InputArgument;
use Symfony\Component\Console\Input\InputInterface; use Symfony\Component\Console\Input\InputInterface;
+12 -2
View File
@@ -12,10 +12,9 @@ namespace Database;
use Exception; use Exception;
use Kiri\Abstracts\Component; use Kiri\Abstracts\Component;
use Kiri\Di\Context; use Kiri\Di\Container;
use Kiri\Exception\ConfigException; use Kiri\Exception\ConfigException;
use PDO; use PDO;
use PDOStatement;
/** /**
* Class Command * Class Command
@@ -41,6 +40,17 @@ class Command extends Component
public array $params = []; public array $params = [];
/**
* @param array $params
* @throws Exception
*/
public function __construct(array $params = [])
{
parent::__construct();
Container::configure($this, $params);
}
/** /**
* @return int|bool * @return int|bool
* @throws Exception * @throws Exception
-8
View File
@@ -18,19 +18,12 @@ use Database\Mysql\Schema;
use Exception; use Exception;
use Kiri; use Kiri;
use Kiri\Abstracts\Component; use Kiri\Abstracts\Component;
use Kiri\Abstracts\Config;
use Kiri\Di\ContainerInterface; use Kiri\Di\ContainerInterface;
use Kiri\Di\Context; use Kiri\Di\Context;
use Kiri\Annotation\Inject;
use Kiri\Pool\Pool; use Kiri\Pool\Pool;
use Kiri\Events\EventProvider; use Kiri\Events\EventProvider;
use Kiri\Exception\ConfigException;
use Kiri\Exception\NotFindClassException; use Kiri\Exception\NotFindClassException;
use Kiri\Pool\Connection as PoolConnection;
use Kiri\Server\Events\OnWorkerExit;
use PDO; use PDO;
use Psr\Container\ContainerExceptionInterface;
use Psr\Container\NotFoundExceptionInterface;
use ReflectionException; use ReflectionException;
/** /**
@@ -103,7 +96,6 @@ class Connection extends Component
*/ */
public function init(): void public function init(): void
{ {
$this->eventProvider->on(OnWorkerExit::class, [$this, 'clear_connection'], 9999);
$this->eventProvider->on(BeginTransaction::class, [$this, 'beginTransaction'], 0); $this->eventProvider->on(BeginTransaction::class, [$this, 'beginTransaction'], 0);
$this->eventProvider->on(Rollback::class, [$this, 'rollback'], 0); $this->eventProvider->on(Rollback::class, [$this, 'rollback'], 0);
$this->eventProvider->on(Commit::class, [$this, 'commit'], 0); $this->eventProvider->on(Commit::class, [$this, 'commit'], 0);
+4 -11
View File
@@ -9,15 +9,12 @@ use Kiri;
use Kiri\Abstracts\Config; use Kiri\Abstracts\Config;
use Kiri\Abstracts\Providers; use Kiri\Abstracts\Providers;
use Kiri\Pool\Connection as PoolConnection; use Kiri\Pool\Connection as PoolConnection;
use Kiri\Exception\ConfigException;
use Kiri\Events\EventProvider; use Kiri\Events\EventProvider;
use Kiri\Annotation\Inject;
use Kiri\Server\Events\OnWorkerStart;
use Psr\Container\ContainerExceptionInterface; use Psr\Container\ContainerExceptionInterface;
use Psr\Container\NotFoundExceptionInterface; use Psr\Container\NotFoundExceptionInterface;
use Kiri\Server\Events\OnWorkerExit;
use Swoole\Timer; use Swoole\Timer;
use Kiri\Di\LocalService; use Kiri\Di\LocalService;
use Kiri\Di\Inject\Container;
/** /**
* Class DatabasesProviders * Class DatabasesProviders
@@ -29,7 +26,7 @@ class DatabasesProviders extends Providers
/** /**
* @var EventProvider * @var EventProvider
*/ */
#[Inject(EventProvider::class)] #[Container(EventProvider::class)]
public EventProvider $provider; public EventProvider $provider;
@@ -49,15 +46,13 @@ class DatabasesProviders extends Providers
if (empty($databases)) { if (empty($databases)) {
return; return;
} }
$this->provider->on(OnWorkerExit::class, [$this, 'exit'], 9999);
$this->provider->on(OnWorkerStart::class, [$this, 'start']);
foreach ($databases as $key => $database) { foreach ($databases as $key => $database) {
$application->set($key, $this->_settings($database)); $application->set($key, $this->_settings($database));
} }
} }
public function start(OnWorkerStart $start) public function start()
{ {
if (!Kiri\Di\Context::inCoroutine()) { if (!Kiri\Di\Context::inCoroutine()) {
return; return;
@@ -81,12 +76,10 @@ class DatabasesProviders extends Providers
/** /**
* @param OnWorkerExit $exit
* @return void * @return void
* @throws ConfigException
* @throws Exception * @throws Exception
*/ */
public function exit(OnWorkerExit $exit): void public function exit(): void
{ {
Timer::clearAll(); Timer::clearAll();
$databases = Config::get('databases.connections', []); $databases = Config::get('databases.connections', []);
-386
View File
@@ -1,386 +0,0 @@
<?php
namespace Database\Mysql;
use Co\WaitGroup;
use Database\Db;
use Exception;
use Kiri;
use Kiri\Events\EventProvider;
use Kiri\Pool\StopHeartbeatCheck;
use Kiri\Server\Events\OnWorkerExit;
use PDOStatement;
use Kiri\Server\WorkerStatus;
use Kiri\Server\Abstracts\StatusEnum;
use Swoole\Coroutine;
use Swoole\Timer;
/**
*/
class PDO implements StopHeartbeatCheck
{
const DB_ERROR_MESSAGE = 'The system is busy, please try again later.';
private ?\PDO $pdo = null;
private int $_transaction = 0;
public string $dbname;
public string $cds;
public string $username;
public string $password;
public string $charset;
public int $connect_timeout;
public int $read_timeout;
/** @var WaitGroup */
private WaitGroup $group;
private int $_timerId = -1;
public array $attributes = [];
/**
* @param array $config
*/
public function __construct(array $config)
{
$this->dbname = $config['dbname'];
$this->cds = $config['cds'];
$this->username = $config['username'];
$this->password = $config['password'];
$this->connect_timeout = $config['connect_timeout'] ?? 30;
$this->read_timeout = $config['read_timeout'] ?? 10;
$this->charset = $config['charset'] ?? 'utf8mb4';
$this->attributes = $config['attributes'] ?? [];
$this->group = new WaitGroup();
}
/**
* @return void
* @throws Exception
*/
public function init(): void
{
$eventProvider = Kiri::getDi()->get(EventProvider::class);
$eventProvider->on(OnWorkerExit::class, [$this, 'onWorkerExit']);
$this->_timerId = Timer::tick(60000, [$this, 'check']);
}
/**
* @return bool
*/
public function inTransaction(): bool
{
return $this->_transaction > 0;
}
/**
* @param Kiri\Server\Events\OnWorkerExit $exit
* @return void
*/
public function onWorkerExit(OnWorkerExit $exit): void
{
$this->stopHeartbeatCheck();
}
/**
* @param string $name
* @param array $arguments
* @return mixed
*/
public function __call(string $name, array $arguments)
{
return $this->_pdo()->{$name}(...$arguments);
}
/**
* @param string $sql
* @return PDOStatement|bool
*/
public function prepare(string $sql): PDOStatement|bool
{
return $this->_pdo()->prepare($sql);
}
/**
*
*/
public function stopHeartbeatCheck(): void
{
$this->pdo = null;
if ($this->_timerId > -1) {
Timer::clear($this->_timerId);
$this->_timerId = -1;
}
}
/**
*
*/
public function beginTransaction()
{
if ($this->_transaction == 0) {
$this->_pdo()->beginTransaction();
}
$this->_transaction++;
}
/**
*
*/
public function commit()
{
$this->_transaction--;
if ($this->_transaction == 0) {
$this->_pdo()->commit();
}
}
/**
*
*/
public function rollback()
{
$this->_transaction--;
if ($this->_transaction == 0) {
$this->_pdo()->rollBack();
}
}
/**
* @param string $sql
* @param array $params
* @return bool|array|null
* @throws Exception
*/
public function fetchAll(string $sql, array $params = []): bool|null|array
{
$pdo = $this->queryPrev($sql, $params);
// $pdo->closeCursor();
return $pdo->fetchAll(\PDO::FETCH_ASSOC);
}
/**
* @param string $sql
* @param array $params
* @return bool|array|null
* @throws Exception
*/
public function fetch(string $sql, array $params = []): bool|null|array
{
$pdo = $this->queryPrev($sql, $params);
// $pdo->closeCursor();
return $pdo->fetch(\PDO::FETCH_ASSOC);
}
/**
* @param string $sql
* @param array $params
* @return bool|array|null
* @throws \Exception
*/
public function fetchColumn(string $sql, array $params = []): bool|null|array
{
$pdo = $this->queryPrev($sql, $params);
// $pdo->closeCursor();
return $pdo->fetchColumn(\PDO::FETCH_ASSOC);
}
/**
* @param string $sql
* @param array $params
* @return int
* @throws Exception
*/
public function rowCount(string $sql, array $params = []): int
{
$pdo = $this->queryPrev($sql, $params);
// $pdo->closeCursor();
return $pdo->rowCount();
}
/**
* @param string $sql
* @param array $params
* @return PDOStatement
* @throws Exception
*/
private function queryPrev(string $sql, array $params = []): PDOStatement
{
try {
// if ($this->_timerId === -1) {
// $this->_timerId = Timer::tick(6000, [$this, 'check']);
// }
// $this->_last = time();
if (($statement = $this->_pdo()->query($sql, \PDO::FETCH_ASSOC)) === false) {
throw new Exception($this->_pdo()->errorInfo()[1]);
}
return $this->bindValue($statement, $params);
} catch (\PDOException|\Throwable $throwable) {
if (str_contains($throwable->getMessage(), 'MySQL server has gone away')) {
$this->pdo = null;
return $this->queryPrev($sql, $params);
}
throw new Exception($throwable->getMessage());
}
}
/**
* @return bool
*/
public function check(): bool
{
return true;
// try {
// if ($this->_last == 0) $this->_last = time();
// if (time() - $this->_last >= 600) {
// return $result = false;
// } else if (!($this->pdo instanceof \PDO)) {
// return $result = false;
// }
// $this->_pdo()->getAttribute(\PDO::ATTR_SERVER_INFO);
// $result = true;
// } catch (\Throwable $throwable) {
// if (!str_contains($throwable->getMessage(), 'Idle dis')) {
// Kiri::getLogger()->error($throwable->getMessage());
// }
// $result = false;
// } finally {
// return $this->afterCheck($result);
// }
}
/**
* @param bool $result
* @return bool
*/
private function afterCheck(bool $result): bool
{
$container = Kiri::getDi()->get(WorkerStatus::class);
if (!$result || $container->is(StatusEnum::EXIT)) {
$this->pdo = null;
// $result = Timer::clear($this->_timerId);
// $this->_timerId = -1;
}
return $result;
}
/**
* @param PDOStatement $statement
* @param array $params
* @return PDOStatement
*/
private function bindValue(PDOStatement $statement, array $params = []): PDOStatement
{
// if (empty($params)) return $statement;
// foreach ($params as $key => $param) {
// $statement->bindValue($key, $param);
// }
$statement->execute($params);
return $statement;
}
/**
* @param string $sql
* @param array $params
* @return int
* @throws Exception
*/
public function execute(string $sql, array $params = []): int
{
$pdo = $this->_pdo();
// if ($this->_timerId === -1) {
// $this->_timerId = Timer::tick(6000, [$this, 'check']);
// }
// $this->_last = time();
if (!(($prepare = $pdo->prepare($sql)) instanceof PDOStatement)) {
throw new Exception($prepare->errorInfo()[2] ?? static::DB_ERROR_MESSAGE);
}
if ($prepare->execute($params) === false) {
throw new Exception($prepare->errorInfo()[2] ?? static::DB_ERROR_MESSAGE);
}
$result = (int)$pdo->lastInsertId();
$prepare->closeCursor();
return $result == 0 ? true : $result;
}
/**
* @return array
*/
public function errorInfo(): array
{
return $this->pdo->errorInfo();
}
/**
* @return \PDO
*/
public function _pdo(): \PDO
{
if (!($this->pdo instanceof \PDO)) {
$this->newClient();
}
return $this->pdo;
}
/**
* @return void
*/
private function newClient(): void
{
$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_PERSISTENT => true,
\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);
}
if (Db::inTransactionsActive()) {
$link->beginTransaction();
}
$this->pdo = $link;
}
}
+1
View File
@@ -1,5 +1,6 @@
<?php <?php
declare(strict_types=1);
namespace Database; namespace Database;
+1
View File
@@ -1,4 +1,5 @@
<?php <?php
declare(strict_types=1);
namespace Database\Traits; namespace Database\Traits;
+1
View File
@@ -1,4 +1,5 @@
<?php <?php
declare(strict_types=1);
namespace Database\Traits; namespace Database\Traits;
+1
View File
@@ -1,4 +1,5 @@
<?php <?php
declare(strict_types=1);
namespace Database\Traits; namespace Database\Traits;