Use classmap autoloader for stapibas classes

[stapibas.git] / src / stapibas / PDO.php

diff --git a/src/stapibas/PDO.php b/src/stapibas/PDO.php
index a85008bbb5278dbbaee4b825931a180fcad46daa..65a6b7336848f18dffbbae3cfa8d75490ba72521 100644 (file)
--- a/src/stapibas/PDO.php
+++ b/src/stapibas/PDO.php
@@ -6,8 +6,9 @@ namespace stapibas;
  */
 class PDO extends \PDO
 {
-    public function query()
-    {
+    public function query(
+        string $query, ?int $fetchMode = null, mixed ...$fetchModeArgs
+    ): \PDOStatement|false {
         $args = func_get_args();
         $res = call_user_func_array(array('parent', 'query'), $args);
         if ($res !== false) {
@@ -17,12 +18,23 @@ class PDO extends \PDO
         $this->handleError();
     }
 
+    public function exec(string $statement): int|false
+    {
+        $res = parent::exec($statement);
+        if ($res !== false) {
+            return $res;
+        }
+
+        $this->handleError();
+    }
+
     protected function handleError()
     {
         echo "SQL error\n";
         echo " " . $this->errorCode() . "\n";
         echo " " . implode(' - ', $this->errorInfo()) . "\n";
+        exit(2);
     }
 }
 
-?>
\ No newline at end of file
+?>