恒富网

<?php
$dbConfig = require_once(dirname(__FILE__).'/config.php');

class Db{
    public $conn;
    private $host = null;
    private $user     = null;
    private $password = null;
    private $database = null;
    private $tablename = null;
    private $dbConfig = null;
    private $sql = [
        'where'   => null,
        'orderBy' => null,
        'limit'   => null,
    ];


    public function __construct($tablename = '') {
        global $dbConfig;
        $this->dbConfig = $dbConfig;
        $this->tablename = $dbConfig['DB_PREFIX'].$tablename;
        $this->user = $dbConfig['DB_USER'];
        $this->host = $dbConfig['DB_HOST'];
        $this->password = $dbConfig['DB_PWD'];
        $this->database = $dbConfig['DB_NAME'];
        $dsn = 'mysql:dbname='.$this->database.';host='.$this->host.';port=3306';
        try {
            $this->conn = new PDO($dsn, $this->user, $this->password); // also allows an extra parameter of configuration
        } catch(PDOException $e) {
            die('Could not connect to the database:<br/>' . $e);
        }
    }

    public function table($tablename) {
        $this->tablename = $this->dbConfig['DB_PREFIX'].$tablename;
        return $this;
    }

    public function getAll($fields = '*') {
        $querySql = sprintf("SELECT %s FROM %s", $fields, $this->tablename);
        if(!empty($this->sql['where'])) {
            $querySql .= ' WHERE ' . $this->sql['where'];
        }
        if(!empty($this->sql['orderBy'])) {
            $querySql .= ' ORDER BY ' . $this->sql['orderBy'];
        }
        if(!empty($this->sql['limit'])) {
            $querySql .= ' LIMIT ' . $this->sql['limit'];
        }
        return $this->query($querySql);
    }

    public function getOne($fields = '*') {
        $result = $this->getAll($fields);
        return isset($result[0]) ? $result[0] : null;
    }

    public function insert($data) {
        foreach ($data as $key => &$value) {
            $value = addslashes($value);
        }
        $keys = "`".implode('`,`', array_keys($data))."`";
        $values = "'".implode("','", array_values($data))."'";
        $querySql = sprintf("INSERT INTO %s ( %s ) VALUES ( %s )", $this->tablename, $keys, $values);
        return $this->query($querySql);
    }

    public function delete() {
        $querySql = sprintf("DELETE FROM %s WHERE ( %s )", $this->tablename, $this->sql['where']);
        return $this->query($querySql);
    }

    public function update($data) {
        $updateFields = [];
        foreach ($data as $key => $value) {
            $up_value = addslashes($value);
            $updateFields[] = "`$key`='$up_value'";
        }
        $updateFields = implode(',', $updateFields);
        $querySql = sprintf("UPDATE %s SET %s", $this->tablename, $updateFields);

        if(!empty($this->sql['where'])) {
            $querySql .= ' WHERE ' . $this->sql['where'];
        }

        return $this->query($querySql);
    }

    public function query($querySql) {
        $querystr = strtolower(trim(substr($querySql,0,6)));
        $stmt = $this->conn->prepare($querySql);
        $ret = $stmt->execute();

        if(!$ret) print_r($stmt->errorInfo());

        if($querystr == 'select') {
            $retData = $stmt->fetchAll(PDO::FETCH_ASSOC);
            return $retData;
        }elseif($ret && $querystr == 'insert') {
            return $this->conn->lastInsertId();
        }else{
            return $ret;
        }
    }


    public function limit($limit, $limitCount = null) {
        if(!$limitCount) {
            $this->sql['limit'] = $limit;
        }else{
            $this->sql['limit'] = $limit .','. $limitCount;
        }
        return $this;
    }

    public function orderBy($orderBy) {
        $this->sql['orderBy'] = $orderBy;
        return $this;
    }

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

    public function where($where) {
        if(!is_array($where)) {
            return null;
        }
        $crondsArr = [];
        foreach ($where as $key => $value) {
            $fieldValue = $value;
            if(is_array($fieldValue)) {
                $crondsArr[] = "$key ".$fieldValue[0]. ' ' . addslashes($fieldValue[1]);
            }else{
                $fieldValue = addslashes($fieldValue);
                $crondsArr[] = "$key='$fieldValue'";
            }
        }
        $this->sql['where'] = implode(' AND ', $crondsArr);

        return $this;
    }

}