Commit

Author: walkor

README.md

@@ -1 +1,2 @@
-# database
+# Webman database
+This is Webman's database library, which includes database client and connection pool.

composer.json

@@ -0,0 +1,16 @@
+{
+  "name": "webman/database",
+  "type": "library",
+  "license": "MIT",
+  "description": "Webman database",
+  "require": {
+    "workerman/webman-framework": "^2.0.0"
+  },
+  "autoload": {
+    "psr-4": {
+      "Webman\\Database\\": "src"
+    }
+  },
+  "minimum-stability": "dev",
+  "prefer-stable": true
+}

src/DatabaseManager.php

@@ -0,0 +1,63 @@
+<?php
+
+namespace Webman\Database;
+
+use Illuminate\Database\DatabaseManager as BaseDatabaseManager;
+use Throwable;
+use Webman\Context;
+use Webman\Coroutine\Pool;
+
+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) {
+                    $connection->disconnect();
+                });
+                $pool->setHeartbeatChecker(function ($connection) {
+                    return $connection->select('select 1');
+                });
+                static::$pools[$name] = $pool;
+            }
+            try {
+                $connection = static::$pools[$name]->get();
+                Context::set($key, $connection);
+            } finally {
+                Context::onDestroy(function () use ($connection, $name) {
+                    try {
+                        static::$pools[$name]->put($connection);
+                    } catch (Throwable) {
+                        // ignore
+                    }
+                });
+            }
+        }
+        return $connection;
+    }
+
+}

src/Initializer.php

@@ -0,0 +1,98 @@
+<?php
+/**
+ * This file is part of webman.
+ *
+ * Licensed under The MIT License
+ * For full copyright and license information, please see the MIT-LICENSE.txt
+ * Redistributions of files must retain the above copyright notice.
+ *
+ * @author    walkor<[email protected]>
+ * @copyright walkor<[email protected]>
+ * @link      http://www.workerman.net/
+ * @license   http://www.opensource.org/licenses/mit-license.php MIT License
+ */
+
+namespace Webman\Database;
+
+use Illuminate\Container\Container as IlluminateContainer;
+use Webman\Database\Manager as Capsule;
+use Illuminate\Events\Dispatcher;
+use Illuminate\Pagination\Paginator;
+use Illuminate\Pagination\CursorPaginator;
+use Illuminate\Pagination\Cursor;
+use Jenssegers\Mongodb\Connection as JenssegersMongodbConnection;
+use MongoDB\Laravel\Connection as LaravelMongodbConnection;
+use support\Container;
+
+class Initializer
+{
+
+    /**
+     * @param $config
+     * @return void
+     */
+    public static function init($config): void
+    {;
+        $connections = $config['connections'] ?? [];
+        if (!$connections) {
+            return;
+        }
+
+        $capsule = new Capsule(IlluminateContainer::getInstance());
+
+        $capsule->getDatabaseManager()->extend('mongodb', function ($config, $name) {
+            $config['name'] = $name;
+            return class_exists(LaravelMongodbConnection::class) ? new LaravelMongodbConnection($config) : new JenssegersMongodbConnection($config);
+        });
+
+        $default = $config['default'] ?? false;
+        if ($default) {
+            $defaultConfig = $connections[$default] ?? false;
+            if ($defaultConfig) {
+                $capsule->addConnection($defaultConfig, $default);
+                $capsule->getDatabaseManager()->setDefaultConnection($default);
+                unset($connections[$default]);
+            }
+        }
+
+        foreach ($connections as $name => $config) {
+            $capsule->addConnection($config, $name);
+        }
+
+        if (class_exists(Dispatcher::class) && !$capsule->getEventDispatcher()) {
+            $capsule->setEventDispatcher(Container::make(Dispatcher::class, [IlluminateContainer::getInstance()]));
+        }
+
+        $capsule->setAsGlobal();
+
+        $capsule->bootEloquent();
+        
+        // Paginator
+        if (class_exists(Paginator::class)) {
+            if (method_exists(Paginator::class, 'queryStringResolver')) {
+                Paginator::queryStringResolver(function () {
+                    $request = request();
+                    return $request?->queryString();
+                });
+            }
+            Paginator::currentPathResolver(function () {
+                $request = request();
+                return $request ? $request->path(): '/';
+            });
+            Paginator::currentPageResolver(function ($pageName = 'page') {
+                $request = request();
+                if (!$request) {
+                    return 1;
+                }
+                $page = (int)($request->input($pageName, 1));
+                return $page > 0 ? $page : 1;
+            });
+            if (class_exists(CursorPaginator::class)) {
+                CursorPaginator::currentCursorResolver(function ($cursorName = 'cursor') {
+                    return Cursor::fromEncoded(request()->input($cursorName));
+                });
+            }
+        }
+    }
+
+}

src/Manager.php

@@ -0,0 +1,23 @@
+<?php
+
+namespace Webman\Database;
+
+use Illuminate\Database\Connection;
+use Illuminate\Database\Connectors\ConnectionFactory;
+
+class Manager extends \Illuminate\Database\Capsule\Manager
+{
+    /**
+     * Build the database manager instance.
+     *
+     * @return void
+     */
+    protected function setupManager()
+    {
+        $factory = new ConnectionFactory($this->container);
+        $this->manager = new DatabaseManager($this->container, $factory);
+        Connection::resolverFor('mysql', function ($connection, $database, $prefix, $config) {
+            return new MySqlConnection($connection, $database, $prefix, $config);
+        });
+    }
+}

src/MysqlConnection.php

@@ -0,0 +1,39 @@
+<?php
+
+namespace Webman\Database;
+
+use Illuminate\Database\MySqlConnection as BaseMySqlConnection;
+
+class MysqlConnection extends BaseMySqlConnection
+{
+    /**
+     * Disconnect。
+     *
+     * @return void
+     */
+    public function disconnect()
+    {
+        parent::disconnect();
+        // Set $this->queryGrammar to null to avoid memory leaks.
+        $this->queryGrammar = null;
+    }
+
+    /**
+     * Reconnect.
+     *
+     * @return void
+     */
+    public function reconnect()
+    {
+        $result = parent::reconnect();
+        if (!$this->queryGrammar) {
+            $this->useDefaultQueryGrammar();
+        }
+        return $result;
+    }
+
+    public function __destruct()
+    {
+        echo "__destruct\n";
+    }
+}

src/support/Db.php

@@ -0,0 +1,52 @@
+<?php
+/**
+ * This file is part of webman.
+ *
+ * Licensed under The MIT License
+ * For full copyright and license information, please see the MIT-LICENSE.txt
+ * Redistributions of files must retain the above copyright notice.
+ *
+ * @author    walkor<[email protected]>
+ * @copyright walkor<[email protected]>
+ * @link      http://www.workerman.net/
+ * @license   http://www.opensource.org/licenses/mit-license.php MIT License
+ */
+
+namespace support;
+
+use Closure;
+use Illuminate\Database\Capsule\Manager;
+use Illuminate\Database\Connection;
+
+/**
+ * Class Db
+ * @package support
+ * @method static array select(string $query, $bindings = [], $useReadPdo = true)
+ * @method static int insert(string $query, $bindings = [])
+ * @method static int update(string $query, $bindings = [])
+ * @method static int delete(string $query, $bindings = [])
+ * @method static bool statement(string $query, $bindings = [])
+ * @method static mixed transaction(Closure $callback, $attempts = 1)
+ * @method static void beginTransaction()
+ * @method static void rollBack($toLevel = null)
+ * @method static void commit()
+ */
+class Db extends Manager
+{
+    /**
+     * @return Manager
+     */
+    public static function getInstance(): Manager
+    {
+        return static::$instance;
+    }
+
+    /**
+     * @return Connection[]
+     */
+    public static function getConnections(): array
+    {
+        return static::$instance->getDatabaseManager()->getConnections();
+    }
+
+}