From ed0035a7078fa79c19860114e9e9f8bfdd4b0cbf Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=90=91=E6=9E=97?= <as2252258@163.com>
Date: Wed, 19 Apr 2023 10:51:08 +0800
Subject: [PATCH] =?UTF-8?q?=E5=8F=98=E6=9B=B4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Connection.php         |  1 -
 DatabasesProviders.php | 23 +++++------------------
 2 files changed, 5 insertions(+), 19 deletions(-)

diff --git a/Connection.php b/Connection.php
index b34ac3c..9ec51e4 100644
--- a/Connection.php
+++ b/Connection.php
@@ -18,7 +18,6 @@ use Database\Mysql\Schema;
 use Exception;
 use Kiri;
 use Kiri\Abstracts\Component;
-use Psr\Container\ContainerInterface;
 use Kiri\Di\Context;
 use Kiri\Pool\Pool;
 use Kiri\Events\EventProvider;
diff --git a/DatabasesProviders.php b/DatabasesProviders.php
index 9118e95..739e9dd 100644
--- a/DatabasesProviders.php
+++ b/DatabasesProviders.php
@@ -8,13 +8,8 @@ use Exception;
 use Kiri;
 use Kiri\Abstracts\Config;
 use Kiri\Abstracts\Providers;
-use Kiri\Pool\Connection as PoolConnection;
-use Kiri\Events\EventProvider;
-use Psr\Container\ContainerExceptionInterface;
-use Psr\Container\NotFoundExceptionInterface;
 use Swoole\Timer;
 use Kiri\Di\LocalService;
-use Kiri\Di\Inject\Container;
 
 /**
  * Class DatabasesProviders
@@ -44,7 +39,7 @@ class DatabasesProviders extends Providers
 	}
 
 
-	public function start()
+	public function start(): void
 	{
 		if (!Kiri\Di\Context::inCoroutine()) {
 			return;
@@ -55,13 +50,9 @@ class DatabasesProviders extends Providers
 				return;
 			}
 
-			$connection = Kiri::getDi()->get(PoolConnection::class);
+			$connection = Kiri::getDi()->get(Kiri\Pool\Pool::class);
 			foreach ($databases as $database) {
-				$connection->flush($database['cds'] . 'master', $database['pool']['min'] ?? 1);
-
-				$slaveCds = ($database['slaveConfig']['cds'] ?? $database['cds']) . 'slave';
-
-				$connection->flush($slaveCds, $database['pool']['min'] ?? 1);
+				$connection->flush($database['cds'], $database['pool']['min'] ?? 1);
 			}
 		});
 	}
@@ -76,13 +67,9 @@ class DatabasesProviders extends Providers
 		Timer::clearAll();
 		$databases = Config::get('databases.connections', []);
 		if (!empty($databases)) {
-			$connection = Kiri::getDi()->get(PoolConnection::class);
+			$connection = Kiri::getDi()->get(Kiri\Pool\Pool::class);
 			foreach ($databases as $database) {
-				$connection->disconnect($database['cds'] . 'master');
-
-				$slaveCds = ($database['slaveConfig']['cds'] ?? $database['cds']) . 'slave';
-
-				$connection->disconnect($slaveCds);
+				$connection->clean($database['cds']);
 			}
 		}
 	}