taoqu/Frame/Library/MySQL/Query.php

<?php

namespace Heanup\Frame\Library\MySQL;

/**
 * 查询生成器
 */
class Query
{

    const PREPARE_PREFIX = ':prepared_';

    private $tableName;
    private $operation = 'SELECT';
    private $field = array();
    private $primaryKey = '';
    private $data = array();
    private $where;
    private $orderBy;
    private $groupBy;
    private $limit = 'LIMIT 2000';
    //替换后的映射表
    private $prepared = array();
    private $sql = '';

    /**
     * 实例
     * @return Query
     */
    public static function newInstance()
    {
        return new Query();
    }

    /**
     * 获取select
     * @return Query
     */
    public function select()
    {
        $this->operation = 'SELECT';
        return $this;
    }

    /**
     * 获取count
     * @return Query
     */
    public function count()
    {
        $this->operation = 'COUNT';
        return $this;
    }

    /**
     * 插入条记录
     * @return Query
     */
    public function insert()
    {
        $this->operation = 'INSERT';
        return $this;
    }

    /**
     * replace into
     * @return Query
     */
    public function replace()
    {
        $this->operation = 'REPLACE';
        return $this;
    }

    /**
     * 更新
     * @return Query
     */
    public function update()
    {
        $this->operation = 'UPDATE';
        return $this;
    }

    /**
     * 删除
     * @return Query
     */
    public function delete()
    {
        $this->operation = 'DELETE';
        return $this;
    }

    /**
     * 设置表名
     * @param string $tableName
     * @return Query
     */
    public function setTableName($tableName)
    {
        if ($tableName) {
            $this->tableName = $this->_prepareFieldName($tableName);
        }
        return $this;
    }

    /**
     * 设置字段
     * @param $field
     * @return Query
     */
    public function setField($field)
    {
        if ($field) {
            if (!is_array($field)) {
                $field = explode(',', $field);
            }
            $this->field = $field;
        }
        return $this;
    }

    /**
     * 设置主键名
     * @param $primaryKey
     * @return $this
     */
    public function setPrimaryKey($primaryKey)
    {
        if ($primaryKey) {
            $this->primaryKey = trim($primaryKey);
        }
        return $this;
    }

    /**
     * 设置where条件
     * @param array $where
     * @param string $whereString
     * @return Query
     */
    public function setWhere(Array $where, $whereString = '')
    {
        if (empty($where) && empty($whereString)) {
            return $this;
        }
        if ($whereString) {
            foreach ($where as $key => $val) {
                $preparedKeys = array();
                if (is_array($val)) {
                    foreach ($val as $row) {
                        $preparedKeys[] = $this->_prepareData($row);
                    }
                } else {
                    $preparedKeys[] = $this->_prepareData($val);
                }
                $key = ':' . ltrim($key, ':');
                $whereString = str_replace($key, implode(',', $preparedKeys), $whereString);
            }
            $this->where = 'WHERE ' . $whereString;
        } elseif ($where) {
            $data = array();
            foreach ($where as $key => $val) {
                if (is_array($val)) {
                    $preparedKeys = array();
                    foreach ($val as $row) {
                        $preparedKeys[] = $this->_prepareData($row);
                    }
                    $data[] = $this->_prepareFieldName($key) . ' IN (' . implode(',', $preparedKeys) . ')';
                } else {
                    $data[] = $this->_prepareFieldName($key) . " = " . $this->_prepareData($val);
                }
            }
            $this->where = 'WHERE ' . implode(' AND ', $data);
        }
        return $this;
    }

    /**
     * 设置排序
     * @param array  $orderBy  array('id' => 1); value 是 true 表示倒序，反之是正序
     * @return Query
     */
    public function setOrderBy(Array $orderBy)
    {
        if ($orderBy && is_array($orderBy)) {
            $tmp = array();
            foreach ($orderBy as $key => $val) {
                if (is_numeric($key)) {
                    continue;
                }
                $order = $val ? 'DESC' : 'ASC';
                $tmp[] = $this->_prepareFieldName($key) . ' ' . $order;
            }
            $tmp = implode(',', $tmp);
            $this->orderBy = "ORDER BY " . $tmp;
        }
        return $this;
    }

    /**
     * 设置范围
     * @param int $limit
     * @param int $offset
     * @return Query
     */
    public function setLimit($limit = 2000, $offset = 0)
    {
        if ($limit) {
            $this->limit = 'LIMIT ' . intval($offset) . ',' . intval($limit);
        } else {
            $this->limit = '';
        }
        return $this;
    }

    /**
     * 设置分组
     * @param $groupBy
     * @return Query
     */
    public function setGroupBy($groupBy)
    {
        if (empty($groupBy)) {
            return $this;
        }
        if (is_array($groupBy)) {
            $tmp = array();
            foreach ($groupBy as $v) {
                $tmp[] = $this->_prepareFieldName($v);
            }
            $this->groupBy = 'GROUP BY ' . implode(",", $tmp);
        } else {
            $this->groupBy = 'GROUP BY ' . $this->_prepareFieldName($groupBy);
        }
        return $this;
    }

    /**
     * 设置数据
     * @param $data
     * @return Query
     */
    public function setData(Array $data)
    {
        if ($data) {
            $this->data = $data;
        }
        return $this;
    }

    /**
     * 获取相关SQL
     * @param boolean $raw
     * @return String $sql
     */
    public function getSQL($raw = false)
    {
        $sql = $this->_getSQL();
        if ($raw) {
            foreach ($this->prepared as $key => $value) {
                $key = ':' . trim($key, ':');
                if (is_array($value)) {
                    foreach ($value as $k => $v) {
                        $value[$k] = addslashes($v);
                    }
                    $value = implode(',', $value);
                } else {
                    $value = "'" . addslashes($value) . "'";
                }
                $sql = str_replace($key, $value, $sql);
            }
        }
        return $sql;
    }

    /**
     * 返回Prepare过的数据
     * @return array
     */
    public function getData()
    {
        $sql = $this->_getSQL();
        $data = array();
        foreach ($this->prepared as $key => $value) {
            if (strpos($sql, $key) !== false) {
                $key = ltrim($key, ':');
                $data[$key] = $value;
            }
        }
        return $data;
    }

    /**
     * 拼SQL
     * @return string
     */
    private function _getSQL()
    {
        if ($this->sql) {
            return $this->sql;
        }
        switch ($this->operation) {
            case 'INSERT':
                $sql = 'INSERT INTO  ' . $this->tableName . ' SET ' . $this->_getData();
                break;
            case 'UPDATE':
                $sql = 'UPDATE ' . $this->tableName . ' SET ' . $this->_getData() . ' ' . $this->where;
                break;
            case 'REPLACE':
                $sql = 'REPLACE INTO  ' . $this->tableName . ' SET ' . $this->_getData();
                break;
            case 'DELETE':
                $sql = 'DELETE FROM ' . $this->tableName . ' ' . $this->where;
                break;
            case 'COUNT':
                $sql = 'SELECT COUNT(*) AS `count`  FROM ' . $this->tableName . ' ' . $this->where . ' ' . $this->groupBy;
                break;
            case 'SELECT':
            default:
                $sql = 'SELECT ' . $this->_getField() . ' FROM ' . $this->tableName . ' ' . $this->where . ' ' . $this->orderBy . ' ' . $this->groupBy . ' ' . $this->limit;
                break;
        }
        return $this->sql = $sql;
    }

    /**
     * 获取字段名称
     * @return String
     */
    private function _getField()
    {
        if (empty($this->field)) {
            $field = '*';
        } else {
            $tmp = array();
            foreach ($this->field as $v) {
                if(strstr($v,'sum')){
                    $tmp[] = $v;
                }else{
                    $tmp[] = $this->_prepareFieldName($v);
                }
            }
            $field = implode(",", $tmp);
        }
        return $field;
    }

    /**
     * 获取prepare后的数据
     * @return String
     */
    private function _getData()
    {
        $data = array();
        foreach ($this->data as $key => $value) {
            //简单字段检查
            if (($this->field AND !in_array($key, $this->field)) || is_null($value)) {
                continue;
            }
            $data[$key] = $this->_prepareFieldName($key) . ' = ' . $this->_prepareData($value);
        }
        $data = implode(",", $data);
        return $data;
    }

    /**
     * 过滤字段名，表名
     * @param String $value
     * @return string
     */
    private function _prepareFieldName($value)
    {
        $value = strtr($value, array(' ' => '', '`' => ''));
        $value = "`" . $value . "`";
        return $value;
    }

    /**
     * prepare替换
     * @param String $value
     * @return string
     */
    private function _prepareData($value)
    {
        $count = count($this->prepared);
        $key = self::PREPARE_PREFIX . $count . '_';
        $this->prepared[$key] = $value;
        return $key;
    }

    /**
     * 魔术输出方法
     * @return String
     */
    public function __toString()
    {
        return $this->getSQL();
    }

}