wahlhelfer/includes/classes/class.Database.php

333 lines
6.5 KiB
PHP
Raw Normal View History

2014-05-15 16:49:45 +00:00
<?php
/**
* MySQL Wrapper Class in PHP5
*/
class database
{
/**
* The result from the last query executed.
* @var Variant
*/
private $queryResult;
/**
* The first word in a query (ex: SELECT, INSERT, UPDATE, DELETE).
* @var String
*/
private $queryType;
/**
* The last insert id.
* @var Integer
*/
private $lastInsertId;
/**
* Constructor.
* @param String $host - The host to connect to.
* @param String $database - The database to connect to.
* @param String $username - The username of the db to connect to.
* @param String $password - The password of the db to connect to.
* @param Boolean $persistent - Is this a persistent connection or not.
*/
public function __construct()
{
$queryResult = '';
$queryType = '';
$lastInsertId = '';
}
/**
* Executes a command on the database.
* @param String $sql - the query to run.
* @return Mixed If True returns an array of rows. False if no rows.
*/
public function executeQuery($sql = '')
{
// Check to see that the parameters are not empty.
try
{
if (!empty($sql))
{
// Execute the query.
$this->runQuery($sql);
return $this;
}
// Parameters are empty.
else
{
trigger_error('You need to provide a query.', E_USER_ERROR);
}
}
catch(exception $e)
{
return $e;
}
}
/**
* Executes a sql query.
* @param String $query - The sql statement.
* @return Boolean True for success, False if not.
*/
private function runQuery($query = NULL)
{
// Check to see if the sql statement variable is set.
try
{
if (!is_null($query))
{
// Determine the query type. (SELECT, UPDATE, INSERT, DELETE etc.)
$this->queryType = $this->getQueryType($query);
if (!$this->queryResult = mysql_query($query))
{
trigger_error('Error in sql query.', E_USER_ERROR);
}
// Check if the query is an insert. If it is, get the id.
if ($this->queryType == 'INSERT')
{
$this->lastInsertId = mysql_insert_id();
}
}
}
catch(exception $e)
{
return $e;
}
}
/**
* Gets the last insert id.
* @return Variant The ID generated for an AUTO_INCREMENT column or False.
*/
public function getLastInsertId()
{
return $this->lastInsertId;
}
/**
* Free result memory.
* @return Returns True on success or False on failure.
*/
private function freeResult()
{
return mysql_free_result($this->queryResult);
}
/**
* To determine the query type used in the query. ex: SELECT, INSERT. In order to run the getAffectedRows(); function below we need to determine the query type.
* @param String $query - The sql statement.
* @return String The first word in the query.
*/
private function getQueryType($query = '')
{
$query = explode(' ', $query);
return strtoupper($query[0]);
}
/**
* Verifies that the last query executed successfully.
* @return True if the last query executed, False if not.
*/
public function querySucceeded()
{
if (!$this->queryResult)
{
return FALSE;
}
else
{
return TRUE;
}
}
/**
* Gets the number of rows affected by the last query executed.
* @return Integer - The number of rows affected.
*/
public function getAffectedRows()
{
try
{
if ($this->querySucceeded())
{
// Retrieves the number of rows from a result set. This command is only valid for statements like SELECT or SHOW that return an actual result set.
if (($this->queryType == 'SELECT') || ($this->queryType == 'SHOW'))
{
return mysql_num_rows($this->queryResult);
}
else
{
// To retrieve the number of rows affected by a INSERT, UPDATE, REPLACE or DELETE query, use mysql_affected_rows().
return mysql_affected_rows();
}
}
else
{
return 0;
}
}
catch(exception $e)
{
return $e;
}
}
/**
* Verifies that the current INSERT sql call ran successfully.
* @return Mixed The last insert id if successful, false if not.
*/
public function wasInserted()
{
try
{
if ($this->queryType == 'INSERT' && $this->querySucceeded())
{
return $this->getLastInsertId();
}
else {
return FALSE;
}
}
catch(exception $e)
{
return $e;
}
}
/**
* Verifies that the current UPDATE sql call ran successfully.
* @return Boolean True if successful, false if not.
*/
public function wasUpdated()
{
try
{
if ($this->queryType == 'UPDATE' && $this->querySucceeded()) {
return TRUE;
}
else {
return FALSE;
}
}
catch(exception $e)
{
return $e;
}
}
/**
* Verifies that the current DELETE sql call ran successfully.
* @return Boolean True if successful, false if not.
*/
public function wasDeleted()
{
try
{
if ($this->queryType == 'DELETE' && $this->querySucceeded())
{
return TRUE;
}
else {
return FALSE;
}
}
catch(exception $e)
{
return $e;
}
}
/**
* Return the result as an array.
* @return Mixed An array of rows if succcessful, false if not.
*/
public function asArray()
{
// If the last query ran was unsuccessfull, then return false.
try
{
if (!$this->queryResult)
{
return FALSE;
}
else
{
if (!$this->getAffectedRows() == 0)
{
$rows = array();
while ($row = mysql_fetch_assoc($this->queryResult))
{
array_push($rows, $row);
}
$this->freeResult();
return $rows;
}
else
{
return FALSE;
}
}
}
catch(exception $e)
{
return $e;
}
}
/**
* Return the result as an object.
* @return Mixed An array of object rows if succcessful, false if not.
*/
public function asObject()
{
// If the last query ran was unsuccessfull, then return false.
try
{
if (!$this->queryResult)
{
return FALSE;
}
else
{
if (!$this->getAffectedRows() == 0)
{
$rows = array();
while ($row = mysql_fetch_object($this->queryResult))
{
array_push($rows, $row);
}
$this->freeResult();
return $rows;
}
else
{
return FALSE;
}
}
}
catch(exception $e)
{
return $e;
}
}
// Function to protect from SQL Injection
public function clean($str)
{
$str = @trim($str);
if(get_magic_quotes_gpc()) {
$str = stripslashes($str);
}
return mysql_real_escape_string($str);
}
}
?>