[PHP] Database class for MySQL which uses the PDO extension

PHP Data Object(PDO)를 이용하여 MySQL DB를 사용하기 위한 Class 이다. Prepared statements 와 named placeholder 를 사용하도록 코드를 작성했다. Class 사용 전 DB_NAME, DB_USER 등의 정보를 설정해야 한다. 아래 코드는 PHP 7.0.0 버전 이상에서 사용할 수 있다.

<?php

// DB 설정
define('DB_HOST', 'localhost');
define('DB_NAME', '');
define('DB_USER', '');
define('DB_PASS', '');

define('DB_ERROR_MODE', ''); // SILENT, WARNING

class DB
{
    public $pdo;
    public $error;

    protected $host;
    protected $user;
    protected $pass;
    protected $dbname;
    protected $stmt;
    protected $options;

    public function __construct(array $options = array())
    {
        require_once NT_CONFIG_PATH.DIRECTORY_SEPARATOR.'db.php';

        $this->options = array();

        switch (DB_ERROR_MODE) {
            case 'SILENT':
                $this->options[\PDO::ATTR_ERRMODE] = \PDO::ERRMODE_SILENT;
                break;
            case 'WARNING':
                $this->options[\PDO::ATTR_ERRMODE] = \PDO::ERRMODE_WARNING;
                break;
            default:
                $this->options[\PDO::ATTR_ERRMODE] = \PDO::ERRMODE_EXCEPTION;
                break;
        }

        if (!empty($options))
            $this->options = array_merge($this->options, $options);

        $this->host   = DB_HOST;
        $this->user   = DB_USER;
        $this->pass   = DB_PASS;
        $this->dbname = DB_NAME;

        try {
            $this->pdo = new \PDO('mysql:host='.$this->host.';dbname='.$this->dbname.';charset=utf8', $this->user, $this->pass, $this->options);
        } catch(\Exception $e) {
            $this->error = 'Connection failed: ' . $e->getMessage();
        }
    }

    public function prepare(string $query)
    {
        $this->stmt = $this->pdo->prepare($query);
    }

    public function bindValue($placeholder, $value, $type = null)
    {
        if (is_null($type)) {
            switch (true) {
                case is_int($value):
                    $type = \PDO::PARAM_INT;
                    break;
                case is_bool($value):
                    $type = \PDO::PARAM_BOOL;
                    break;
                case is_null($value):
                    $type = \PDO::PARAM_NULL;
                    break;
                default:
                    $type = \PDO::PARAM_STR;
                    break;
            }
        }

        $this->stmt->bindValue($placeholder, $value, $type);
    }

    public function bindValueArray(array $params)
    {
        if (!empty($params)) {
            foreach ($params as $key => $val) {
                $this->bindValue($key, $val);
            }
        }
    }

    public function execute(array $param = array())
    {
        if (is_array($param) && !empty($param)) {
            foreach($param as $key => $val) {
                $this->bindValue($key, $val);
            }
        }

        return $this->stmt->execute();
    }

    public function fetchAll($args = \PDO::FETCH_ASSOC)
    {
        return $this->stmt->fetchAll($args);
    }

    public function fetch($args = \PDO::FETCH_ASSOC)
    {
        return $this->stmt->fetch($args);
    }

    public function rowCount()
    {
        return $this->stmt->rowCount();
    }

    public function lastInsertId(){
        return $this->pdo->lastInsertId();
    }

    public function beginTransaction(){
        return $this->pdo->beginTransaction();
    }

    public function commit(){
        return $this->pdo->commit();
    }

    public function rollBack(){
        return $this->pdo->rollBack();
    }

    public function debugDumpParams(){
        return $this->stmt->debugDumpParams();
    }

    public function errorInfo()
    {
        $error = $this->stmt->errorInfo();
        $this->error = $error[0] . ' ' . $error[2];
    }

    public function close()
    {
        $this->pdo = null;
    }

    public function __call($method, $args)
    {
        if (method_exists($this->pdo, $method)) {
            return call_user_func_array(array(&$this->pdo, $method), $args);
        } else if (method_exists($this->stmt, $method)) {
            return call_user_func_array(array(&$this->stmt, $method), $args);
        } else {
            return 'Method \'' . $method . '\' not exists';
        }
    }
}

편리

PHP와 MariaDB, jQuery 등을 사용해 게시판, 쇼핑몰 솔루션을 개발합니다. 그누보드5와 영카트5 개발에 참여 했습니다. Linux와 Nginx는 물론 WordPress, Git 등에도 관심이 많습니다. 자전거 타기 및 사진 촬영을 취미로 하고 있습니다.

Leave a Reply

Your email address will not be published. Required fields are marked *

This site uses Akismet to reduce spam. Learn how your comment data is processed.