Update Database.php

jannisnikoy · May 2, 2024 · f3764e7 · f3764e7
1 parent 9e78e21
commit f3764e7
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/Database.php b/src/Database.php
@@ -61,11 +61,13 @@ public function query(string $sql): array {
 
         $this->statement->execute();
 
-        if (strtoupper(substr(ltrim($sql), 0, 6)) === "SELECT") {
+        $substr = strtoupper(substr(ltrim($sql), 0, 6));
+
+        if ($substr == "INSERT" || $substr == "UPDATE" || $substr == "DELETE") {
+            return [$this->statement->rowCount()];
+        }else {
             $this->result = $this->statement->fetchAll(\PDO::FETCH_OBJ);
             return $this->result;
-        } else {
-            return [$this->statement->rowCount()];
         }
     }