diff --git a/sys/db/db_pdo.php b/sys/db/db_pdo.php index 279c2e1..e2ef609 100644 --- a/sys/db/db_pdo.php +++ b/sys/db/db_pdo.php @@ -54,11 +54,11 @@ abstract class DB_PDO extends PDO { $this->statement =& $query; - if( ! (is_array($data) || is_object($data))) + /*if( ! (is_array($data) || is_object($data))) { trigger_error("Invalid data argument"); return FALSE; - } + }*/ // Bind the parameters foreach($data as $k => $value) @@ -108,7 +108,7 @@ abstract class DB_PDO extends PDO { */ public function get_query_data($statement) { - $this->statement = $statement; + $this->statement =& $statement; // Execute the query $this->statement->execute(); diff --git a/sys/db/query_builder.php b/sys/db/query_builder.php index 377f4fa..5942d56 100644 --- a/sys/db/query_builder.php +++ b/sys/db/query_builder.php @@ -18,7 +18,7 @@ */ class Query_Builder { - private $table, $where_array, $sql, $select_string; + private $table, $where_array, $sql, $select_string, $where_string; /** * Constructor @@ -86,6 +86,12 @@ class Query_Builder { // Replace the star with the selected fields $sql = str_replace('*', $this->select_string, $sql); } + + // Set the where string + if ( ! empty($this->where_string)) + { + $sql .= $this->where_string; + } // Set the limit, if it exists if ($limit !== FALSE) @@ -93,8 +99,15 @@ class Query_Builder { $sql = $this->sql->limit($sql, $limit, $offset); } - //echo $sql."
"; + echo $sql."
"; + // Do prepared statements for anything involving a "where" clause + if ( ! empty($this->where_string)) + { + return $this->db->prepare_execute($sql, array_values($this->where_array)); + } + + // Otherwise, a simple query will do. return $this->db->query($sql); } @@ -154,6 +167,47 @@ class Query_Builder { */ public function where($key, $val=array()) { + // Key and value passed? Add them to the where array + if (is_scalar($key) && is_scalar($val)) + { + $this->where_array[$key] = $val; + } + // Array or object, loop through and add to the where array + elseif ( ! is_scalar($key)) + { + foreach($key as $k => $v) + { + $this->where_array[$k] = $v; + } + } + + // The values are irrelevant until the query is actually run + $fields = array_keys($this->where_array); + + // Array of conditions + $kv_array = array(); + + // Create key/value placeholders + foreach($fields as $f) + { + // Split each key by spaces, incase there + // is an operator such as >, <, !=, etc. + $f_array = explode(' ', trim($f)); + + // Simple key = val + if (count($f_array) === 1) + { + $kv_array[] = $this->db->quote_ident($f_array[0]) . '= ?'; + } + else // Other operators + { + $kv_array[] = $this->db->quote_ident($f_array[0]) . " {$f_array[1]} ?"; + } + } + + // Create the where portion of the string + $this->where_string = ' WHERE '.implode(', ', $kv_array); + // @todo Implement where method return $this; } diff --git a/tests/databases/sqlite.php b/tests/databases/sqlite.php index 01fc692..5274e29 100644 --- a/tests/databases/sqlite.php +++ b/tests/databases/sqlite.php @@ -152,6 +152,13 @@ SQL; $this->assertIsA($query, 'PDOStatement'); } + function TestQBSelectWhereGet() + { + $query = $this->qb->select('id, key as k, val')->where('id >', 1)->get('create_test, 2, 1'); + + $this->assertIsA($query, 'PDOStatement'); + } + function TestQBSelectGet() { $query = $this->qb->select('id, key as k, val')->get('create_test', 2, 1);