Update PDO.class.php

Added in Stored procedure 'Call' to statement OR condition.
lincanbin · May 14, 2019 · 20670f9 · 20670f9
1 parent bad17b1
commit 20670f9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/PDO.class.php b/src/PDO.class.php
@@ -223,7 +223,7 @@ public function query($query, $params = null, $fetchMode = PDO::FETCH_ASSOC)
 		$rawStatement = explode(" ", $query);
 		$this->Init($query, $params);
 		$statement = strtolower($rawStatement[0]);
-		if ($statement === 'select' || $statement === 'show') {
+		if ($statement === 'select' || $statement === 'show' || $statement === "call") {
 			return $this->sQuery->fetchAll($fetchMode);
 		} elseif ($statement === 'insert' || $statement === 'update' || $statement === 'delete') {
 			return $this->sQuery->rowCount();
@@ -245,7 +245,7 @@ public function iterator($query, $params = null, $fetchMode = PDO::FETCH_ASSOC)
         $rawStatement = explode(" ", $query);
         $this->Init($query, $params, array(PDO::ATTR_CURSOR => PDO::CURSOR_SCROLL));
         $statement = strtolower($rawStatement[0]);
-        if ($statement === 'select' || $statement === 'show') {
+        if ($statement === 'select' || $statement === 'show' || $statement === "call") {
             return new PDOIterator($this->sQuery, $fetchMode);
         } elseif ($statement === 'insert' || $statement === 'update' || $statement === 'delete') {
             return $this->sQuery->rowCount();