2025.3.11

vendor/webman/database/src/DatabaseManager.php

@@ -0,0 +1,82 @@
+<?php
+
+namespace Webman\Database;
+
+use Illuminate\Database\DatabaseManager as BaseDatabaseManager;
+use Throwable;
+use Webman\Context;
+use Workerman\Coroutine\Pool;
+
+/**
+ * Class DatabaseManager
+ */
+class DatabaseManager extends BaseDatabaseManager
+{
+
+    /**
+     * @var Pool[]
+     */
+    protected static array $pools = [];
+
+    /**
+     * Get connection
+     *
+     * @param $name
+     * @return mixed
+     * @throws Throwable
+     */
+    public function connection($name = null)
+    {
+        $name = $name ?: $this->getDefaultConnection();
+        [$database, $type] = $this->parseConnectionName($name);
+        
+        $key = "database.connections.$name";
+        $connection = Context::get($key);
+        if (!$connection) {
+            if (!isset(static::$pools[$name])) {
+                $poolConfig = $this->app['config']['database.connections'][$name]['pool'] ?? [];
+                $pool = new Pool($poolConfig['max_connections'] ?? 6, $poolConfig);
+                $pool->setConnectionCreator(function () use ($database, $type) {
+                    return $this->configure($this->makeConnection($database), $type);
+                });
+                $pool->setConnectionCloser(function ($connection) {
+                    $this->closeAndFreeConnection($connection);
+                });
+                $pool->setHeartbeatChecker(function ($connection) {
+                    $connection->select('select 1');
+                });
+                static::$pools[$name] = $pool;
+            }
+            try {
+                $connection = static::$pools[$name]->get();
+                Context::set($key, $connection);
+            } finally {
+                // We cannot use Coroutine::defer() because we may not be in a coroutine environment currently.
+                Context::onDestroy(function () use ($connection, $name) {
+                    try {
+                        $connection && static::$pools[$name]->put($connection);
+                    } catch (Throwable) {
+                        // ignore
+                    }
+                });
+            }
+        }
+        return $connection;
+    }
+
+    /**
+     * Close connection.
+     *
+     * @param $connection
+     * @return void
+     */
+    protected function closeAndFreeConnection($connection): void
+    {
+        $connection->disconnect();
+        $clearProperties = function () {
+            $this->queryGrammar = null;
+        };
+        $clearProperties->call($connection);
+    }
+
+}