DatabasePDO.php

Go to the documentation of this file.

<?php

namespace Combu;

use PDO;
use PDOException;

class DatabasePDO extends Database {
    
    private $_errorCode = NULL;
    private $_errorMessage = NULL;
    
    protected function _setConnection() {
        try {
            $dbCharset = "utf8";
            if (defined("GAME_DB_CHARSET") && GAME_DB_CHARSET) {
                $dbCharset = GAME_DB_CHARSET;
            }
            $dbOptions = NULL;
            if (defined("GAME_DB_OPTIONS") && GAME_DB_OPTIONS) {
                $dbOptions = GAME_DB_OPTIONS;
            }
            $dsn = sprintf("%s:host=%s;port=%d;dbname=%s;charset=%s", $this->_dbType, $this->_dbServer, $this->_dbPort, $this->_dbName, $dbCharset);
            $this->_connection = new PDO($dsn, $this->_dbUser, $this->_dbPass, $dbOptions);
        } catch (PDOException $e) {
            $this->_connection = NULL;
            $this->_errorCode = $e->getCode();
            $this->_errorMessage = $e->getMessage();
        }
    }
    
    protected function _selectDatabase() {
    }
    
    public function CloseConnection() {
        $this->_connection = NULL;
    }
    
    public function TestConnection() {
        if ($this->_connection) {
            return TRUE;
        }
        return FALSE;
    }
    
    public function GetError() {
        return $this->_errorMessage;
    }
    
    public function GetErrorNo() {
        return $this->_errorCode;
    }
    
    public function InsertedId($name = NULL) {
        if ($this->_connection) {
            return $this->_connection->lastInsertId($name);
        }
    }
    
    public function AffectedRows() {
        if ($this->_connection) {
            return $this->_connection->rowCount();
        }
        return 0;
    }
    
    private function _Fetch($result, $result_type) {
        if ($result) {
            return $result->fetch($result_type);
        }
        return NULL;
    }

    public function FetchAssoc($result) {
        if ($this->_connection) {
            return $this->_Fetch($result, PDO::FETCH_ASSOC);
        }
        return NULL;
    }

    public function FetchNum($result) {
        if ($this->_connection) {
            return $this->_Fetch($result, PDO::FETCH_NUM);
        }
        return NULL;
    }
    
    public function Query($query) {
        if ($this->_connection && $query) {
            try {
                if (Utils::StartsWith($query, "SELECT", TRUE)) {
                    $stmt = $this->_connection->prepare($query);
                    if ($stmt->execute()) {
                        return $stmt;
                    }
                    return FALSE;
                }
                if ($this->_connection->query($query)) {
                    return TRUE;
                }
            } catch (PDOException $e) {
                $this->_errorNumber = $e->getCode();
                $this->_errorMessage = $e->getMessage();
            }
        }
        return FALSE;
    }
    
    public function Escape($text) {
        if ($this->_connection) {
            $quoted = $this->_connection->quote($text);
            return substr($quoted, 1, strlen($quoted) - 2);
        }
        return FALSE;
    }
    
    public function TransactionStart() {
        if ($this->_connection) {
            return $this->_connection->beginTransaction();
        }
        return FALSE;
    }
    
    public function TransactionCommit() {
        if ($this->_connection) {
            return $this->_connection->commit();
        }
        return FALSE;
    }
    
    public function TransactionRollback() {
        if ($this->_connection) {
            return $this->_connection->rollBack();
        }
        return FALSE;
    }
}