modify plugin name
This commit is contained in:
+118
-118
@@ -27,140 +27,140 @@ class DatabasesProviders extends Providers
|
|||||||
{
|
{
|
||||||
|
|
||||||
|
|
||||||
public int $timer = 30000;
|
public int $timer = 30000;
|
||||||
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @var EventProvider
|
* @var EventProvider
|
||||||
*/
|
*/
|
||||||
#[Inject(EventProvider::class)]
|
#[Inject(EventProvider::class)]
|
||||||
public EventProvider $provider;
|
public EventProvider $provider;
|
||||||
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @param LocalService $application
|
* @param LocalService $application
|
||||||
* @return void
|
* @return void
|
||||||
* @throws ConfigException
|
* @throws ConfigException
|
||||||
* @throws Exception
|
* @throws Exception
|
||||||
*/
|
*/
|
||||||
public function onImport(LocalService $application): void
|
public function onImport(LocalService $application): void
|
||||||
{
|
{
|
||||||
$databases = Config::get('databases.connections', []);
|
$databases = Config::get('databases.connections', []);
|
||||||
if (empty($databases)) {
|
if (empty($databases)) {
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
$this->provider->on(OnWorkerStart::class, [$this, 'check']);
|
$this->provider->on(OnWorkerStart::class, [$this, 'check']);
|
||||||
$this->provider->on(OnTaskerStart::class, [$this, 'check']);
|
$this->provider->on(OnTaskerStart::class, [$this, 'check']);
|
||||||
$this->provider->on(OnWorkerExit::class, [$this, 'exit'], 9999);
|
$this->provider->on(OnWorkerExit::class, [$this, 'exit'], 9999);
|
||||||
foreach ($databases as $key => $database) {
|
foreach ($databases as $key => $database) {
|
||||||
$application->set($key, $this->_settings($database));
|
$application->set($key, $this->_settings($database));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @param OnWorkerExit $exit
|
* @param OnWorkerExit $exit
|
||||||
* @return void
|
* @return void
|
||||||
* @throws ConfigException
|
* @throws ConfigException
|
||||||
* @throws Exception
|
* @throws Exception
|
||||||
*/
|
*/
|
||||||
public function exit(OnWorkerExit $exit): void
|
public function exit(OnWorkerExit $exit): void
|
||||||
{
|
{
|
||||||
Timer::clearAll();
|
Timer::clearAll();
|
||||||
$databases = Config::get('databases.connections', []);
|
$databases = Config::get('databases.connections', []);
|
||||||
if (!empty($databases)) {
|
if (!empty($databases)) {
|
||||||
$connection = Kiri::getDi()->get(PoolConnection::class);
|
$connection = Kiri::getDi()->get(PoolConnection::class);
|
||||||
foreach ($databases as $database) {
|
foreach ($databases as $database) {
|
||||||
$connection->disconnect($database['cds']);
|
$connection->disconnect($database['cds']);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @param $name
|
* @param $name
|
||||||
* @return Connection
|
* @return Connection
|
||||||
* @throws Exception
|
* @throws Exception
|
||||||
*/
|
*/
|
||||||
public function get($name): Connection
|
public function get($name): Connection
|
||||||
{
|
{
|
||||||
return Kiri::service()->get($name);
|
return Kiri::service()->get($name);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @param OnTaskerStart|OnWorkerStart $start
|
* @param OnTaskerStart|OnWorkerStart $start
|
||||||
* @return void
|
* @return void
|
||||||
*/
|
*/
|
||||||
public function check(OnTaskerStart|OnWorkerStart $start): void
|
public function check(OnTaskerStart|OnWorkerStart $start): void
|
||||||
{
|
{
|
||||||
Timer::tick($this->timer, static function () use ($start) {
|
Timer::tick(60000, static function () use ($start) {
|
||||||
$valid = $count = 0;
|
$valid = $count = 0;
|
||||||
$logger = Kiri::getDi()->get(LoggerInterface::class);
|
$logger = Kiri::getDi()->get(LoggerInterface::class);
|
||||||
$databases = Config::get('databases.connections', []);
|
$databases = Config::get('databases.connections', []);
|
||||||
if (!empty($databases)) {
|
if (!empty($databases)) {
|
||||||
DatabasesProviders::each($databases, $logger, $count, $valid);
|
DatabasesProviders::each($databases, $logger, $count, $valid);
|
||||||
}
|
}
|
||||||
$const = 'Worker %d db client has %d, valid %d';
|
$const = 'Worker %d db client has %d, valid %d';
|
||||||
$logger->alert(sprintf($const, $start->workerId, $count, $valid));
|
$logger->alert(sprintf($const, $start->workerId, $count, $valid));
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @param $databases
|
* @param $databases
|
||||||
* @param LoggerInterface $logger
|
* @param LoggerInterface $logger
|
||||||
* @param $count
|
* @param $count
|
||||||
* @param $valid
|
* @param $valid
|
||||||
* @return void
|
* @return void
|
||||||
*/
|
*/
|
||||||
public static function each($databases, LoggerInterface $logger, &$count, &$valid): void
|
public static function each($databases, LoggerInterface $logger, &$count, &$valid): void
|
||||||
{
|
{
|
||||||
$connection = Kiri::getDi()->get(PoolConnection::class);
|
$connection = Kiri::getDi()->get(PoolConnection::class);
|
||||||
foreach ($databases as $database) {
|
foreach ($databases as $database) {
|
||||||
try {
|
try {
|
||||||
[$total, $success] = $connection->check($database['cds']);
|
[$total, $success] = $connection->check($database['cds']);
|
||||||
|
|
||||||
$count += $total;
|
$count += $total;
|
||||||
$valid += $success;
|
$valid += $success;
|
||||||
|
|
||||||
if (isset($database['slaveConfig']) && isset($database['slaveConfig']['cds'])) {
|
if (isset($database['slaveConfig']) && isset($database['slaveConfig']['cds'])) {
|
||||||
if ($database['slaveConfig']['cds'] != $database['cds']) {
|
if ($database['slaveConfig']['cds'] != $database['cds']) {
|
||||||
[$total, $success] = $connection->check($database['slaveConfig']['cds']);
|
[$total, $success] = $connection->check($database['slaveConfig']['cds']);
|
||||||
|
|
||||||
$count += $total;
|
$count += $total;
|
||||||
$valid += $success;
|
$valid += $success;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
} catch (\Throwable $throwable) {
|
} catch (\Throwable $throwable) {
|
||||||
$logger->error($throwable->getMessage());
|
$logger->error($throwable->getMessage());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @param $database
|
* @param $database
|
||||||
* @return array
|
* @return array
|
||||||
*/
|
*/
|
||||||
private function _settings($database): array
|
private function _settings($database): array
|
||||||
{
|
{
|
||||||
$clientPool = $database['pool'] ?? ['min' => 1, 'max' => 5, 'tick' => 60];
|
$clientPool = $database['pool'] ?? ['min' => 1, 'max' => 5, 'tick' => 60];
|
||||||
return [
|
return [
|
||||||
'id' => $database['id'],
|
'id' => $database['id'],
|
||||||
'cds' => $database['cds'],
|
'cds' => $database['cds'],
|
||||||
'class' => Connection::class,
|
'class' => Connection::class,
|
||||||
'username' => $database['username'],
|
'username' => $database['username'],
|
||||||
'password' => $database['password'],
|
'password' => $database['password'],
|
||||||
'tablePrefix' => $database['tablePrefix'],
|
'tablePrefix' => $database['tablePrefix'],
|
||||||
'database' => $database['database'],
|
'database' => $database['database'],
|
||||||
'connect_timeout' => $database['connect_timeout'] ?? 30,
|
'connect_timeout' => $database['connect_timeout'] ?? 30,
|
||||||
'read_timeout' => $database['read_timeout'] ?? 10,
|
'read_timeout' => $database['read_timeout'] ?? 10,
|
||||||
'pool' => $clientPool,
|
'pool' => $clientPool,
|
||||||
'attributes' => $database['attributes'] ?? [],
|
'attributes' => $database['attributes'] ?? [],
|
||||||
'charset' => $database['charset'] ?? 'utf8mb4',
|
'charset' => $database['charset'] ?? 'utf8mb4',
|
||||||
'slaveConfig' => $database['slaveConfig']
|
'slaveConfig' => $database['slaveConfig']
|
||||||
];
|
];
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -4,12 +4,10 @@ namespace Database\Mysql;
|
|||||||
|
|
||||||
use Exception;
|
use Exception;
|
||||||
use Kiri;
|
use Kiri;
|
||||||
use Kiri\Abstracts\Config;
|
|
||||||
use Kiri\Events\EventProvider;
|
use Kiri\Events\EventProvider;
|
||||||
use Kiri\Pool\StopHeartbeatCheck;
|
use Kiri\Pool\StopHeartbeatCheck;
|
||||||
use Kiri\Server\Events\OnWorkerExit;
|
use Kiri\Server\Events\OnWorkerExit;
|
||||||
use PDOStatement;
|
use PDOStatement;
|
||||||
use Swoole\Timer;
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
*
|
*
|
||||||
|
|||||||
Reference in New Issue
Block a user