From 26e172680d953a736efa97e2b553814dea185a3d Mon Sep 17 00:00:00 2001
From: crazywhalecc <jesse2061@outlook.com>
Date: Tue, 3 Jan 2023 00:48:31 +0800
Subject: [PATCH] add sql validation

---
 src/ZM/Store/Database/DBPool.php | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/src/ZM/Store/Database/DBPool.php b/src/ZM/Store/Database/DBPool.php
index 6137d359..2da79b7f 100644
--- a/src/ZM/Store/Database/DBPool.php
+++ b/src/ZM/Store/Database/DBPool.php
@@ -63,6 +63,20 @@ class DBPool
             case SwooleDriver::class:
                 self::$pools[$name] = new SwooleObjectPool($size, \PDO::class, $connect_str, ...$args);
         }
+        switch ($config['type']) {
+            case 'sqlite':
+                /** @var \PDO $pool */
+                $pool = self::$pools[$name]->get();
+                $a = $pool->query('select sqlite_version();')->fetchAll()[0][0] ?? '';
+                if (str_starts_with($a, '3')) {
+                    logger()->debug('sqlite ' . $name . ' connected');
+                }
+                self::$pools[$name]->put($pool);
+                break;
+            case 'mysql':
+                // TODO: 编写验证 MySQL 连接有效性的功能
+                break;
+        }
     }
 
     /**