归还连接操作封装

itxiao6 · Oct 9, 2020 · 2278975 · 2278975
1 parent 4798336
commit 2278975
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 8 deletions.
diff --git a/README.MD b/README.MD
@@ -44,10 +44,10 @@ $s = microtime(true);
              * 设置 回收数据库连接
              */
             Swoole\Coroutine::defer(function () {
-                $lists = \itxiao6\SwooleDatabase\Utils\Context::get(\itxiao6\SwooleDatabase\PoolManager::class . '_connection') === null ? [] : \itxiao6\SwooleDatabase\Utils\Context::get(\itxiao6\SwooleDatabase\PoolManager::class . '_connection');
-                foreach ($lists as $item) {
-                    \itxiao6\SwooleDatabase\PoolManager::getPool($item['name'])->put($item['pdo']);
-                }
+                /**
+                 * 归还当前协程内的连接
+                 */
+                \itxiao6\SwooleDatabase\PoolManager::recoveryConnection();
             });
             /**
              * 创建表

diff --git a/src/PoolManager.php b/src/PoolManager.php
@@ -2,7 +2,9 @@
 
 
 namespace itxiao6\SwooleDatabase;
+
 use Swoole\Database\PDOPool;
+
 /**
  * 连接池管理(全局)
  * Class PoolManager
@@ -18,9 +20,9 @@ class PoolManager
      * @param string $name
      * @throws \Exception
      */
-    public static function addPool($size = 64,$name='default')
+    public static function addPool($size = 64, $name = 'default')
     {
-        if(isset(self::$pool[$name]) && self::$pool[$name] instanceof PDOPool){
+        if (isset(self::$pool[$name]) && self::$pool[$name] instanceof PDOPool) {
             throw new \Exception('Pool Exist');
         }
         $config = PDOConfig::getConfig($name);
@@ -33,11 +35,23 @@ public static function addPool($size = 64,$name='default')
      * @return PDOPool
      * @throws \Exception
      */
-    public static function getPool($name='default')
+    public static function getPool($name = 'default')
     {
-        if(!(isset(self::$pool[$name]) && self::$pool[$name] instanceof PDOPool)){
+        if (!(isset(self::$pool[$name]) && self::$pool[$name] instanceof PDOPool)) {
             self::addPool();
         }
         return self::$pool[$name];
     }
+
+    /**
+     * 归还当前协程内的所有连接
+     * @throws \Exception
+     */
+    public static function recoveryConnection()
+    {
+        $lists = \itxiao6\SwooleDatabase\Utils\Context::get(\itxiao6\SwooleDatabase\PoolManager::class . '_connection') === null ? [] : \itxiao6\SwooleDatabase\Utils\Context::get(\itxiao6\SwooleDatabase\PoolManager::class . '_connection');
+        foreach ($lists as $item) {
+            \itxiao6\SwooleDatabase\PoolManager::getPool($item['name'])->put($item['pdo']);
+        }
+    }
 }