Verifica conexão ao banco dados na rota de recuperação de senha

/admin/forgot_password - Tratamento de erro em MyModel->fetch ()
jairpro · Sep 19, 2020 · b099feb · b099feb
1 parent 1a1bfb4
commit b099feb
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/modules/my-model/MyModel.php b/modules/my-model/MyModel.php
@@ -446,6 +446,9 @@ function fetchAll($params=null, $sql=null) {
 
   function fetch($params=null, $sql=null) {
     $this->setup($params, $sql);
+    if (!$this->sth) {
+      return false;
+    }
     $value = $this->sth->fetch();
     return $value;
   }

diff --git a/src/controllers/ForgotPasswordController.php b/src/controllers/ForgotPasswordController.php
@@ -26,10 +26,12 @@ function store($req, $res) {
     if (!isset($data['user'])) {
       $res->status(400)->send(['error' => "Missing the user field."]);
     }
-
     $user = $data['user'];
 
     $model = new $this->modelClass();
+    if (!$model->setup()) {
+      $res->status(500)->send(['error'=>'Database connection failure.']);
+    }
 
     $found = $model->findOne([
       'or'=>[