diff --git a/resources/classes/database.php b/resources/classes/database.php index 31fb1ad89c..33cd84ba8c 100644 --- a/resources/classes/database.php +++ b/resources/classes/database.php @@ -450,7 +450,7 @@ include "root.php"; } // Use this function to execute complex queries - public function execute($sql, $parameters = null) { + public function execute($sql, $parameters = null, $return_type = 'all') { //connect to the database if needed if (!$this->db) { @@ -476,9 +476,18 @@ include "root.php"; $message["parameters"] = $parameters; } $this->message = $message; - //$result = $prep_statement->fetchAll(PDO::FETCH_ASSOC); - //unset($prep_statement); - return $prep_statement->fetchAll(PDO::FETCH_ASSOC); + + //return the results + switch($return_type) { + case 'all': + return $prep_statement->fetchAll(PDO::FETCH_ASSOC); + case 'row': + return $prep_statement->fetch(PDO::FETCH_ASSOC); + case 'column'; + return $prep_statement->fetchColumn(); + default: + return $prep_statement->fetchAll(PDO::FETCH_ASSOC); + } } catch(PDOException $e) { $message["message"] = "Bad Request"; @@ -906,7 +915,7 @@ include "root.php"; } //count - public function select($sql, $parameters = null) { + public function select($sql, $parameters = '', $return_type = 'all') { //connect to the database if needed if (!$this->db) { @@ -932,9 +941,18 @@ include "root.php"; $message["parameters"] = $parameters; } $this->message = $message; - //$result = $prep_statement->fetchAll(PDO::FETCH_ASSOC); - //unset($prep_statement); - return $prep_statement->fetchAll(PDO::FETCH_ASSOC); + + //return the results + switch($return_type) { + case 'all': + return $prep_statement->fetchAll(PDO::FETCH_ASSOC); + case 'row': + return $prep_statement->fetch(PDO::FETCH_ASSOC); + case 'column'; + return $prep_statement->fetchColumn(); + default: + return $prep_statement->fetchAll(PDO::FETCH_ASSOC); + } } catch(PDOException $e) { $message["message"] = "Bad Request";