Update PdoDatabase.php

getpinga · Feb 3, 2023 · 400ebf6 · 400ebf6
1 parent 7ba20a8
commit 400ebf6
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/PdoDatabase.php b/src/PdoDatabase.php
@@ -89,8 +89,14 @@ private function __construct(PDO $pdoInstance = null, PdoDsn $pdoDsn = null, $pr
 	 * @param bool|null (optional) $preserveOldState whether the old state of the PDO instance should be preserved
 	 * @return static the new instance
 	 */
-	public static function fromPdo(PDO $pdoInstance, $preserveOldState = null) {
-		return new static($pdoInstance, null, $preserveOldState);
+	public static function fromPdo($pdoInstance, $preserveOldState = null) {
+		if ($pdoInstance instanceof PDO) {
+                   return new static($pdoInstance, null, $preserveOldState);
+		} else if ($pdoInstance instanceof Swoole\Coroutine\MySQL) {
+                   return new static($pdoInstance, null, $preserveOldState);
+                } else {
+                   throw new Exception("Invalid argument type. Expected PDO or Swoole\Coroutine\MySQL.");
+                }
 	}
 
 	/**