基于PHP的Guojiz轻社区源码.zip

<?php // +---------------------------------------------------------------------- // | ThinkPHP [ WE CAN DO IT JUST THINK ] // +---------------------------------------------------------------------- // | Copyright (c) 2006~2018 http://thinkphp.cn All rights reserved. // +---------------------------------------------------------------------- // | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 ) // +---------------------------------------------------------------------- // | Author: liu21st <liu21st@gmail.com> // +---------------------------------------------------------------------- namespace think\db; use PDO; use think\App; use think\Cache; use think\Collection; use think\Config; use think\Db; use think\db\exception\BindParamException; use think\db\exception\DataNotFoundException; use think\db\exception\ModelNotFoundException; use think\Exception; use think\exception\DbException; use think\exception\PDOException; use think\Loader; use think\Model; use think\model\Relation; use think\model\relation\OneToOne; use think\Paginator; class Query { // 数据库Connection对象实例 protected $connection; // 数据库Builder对象实例 protected $builder; // 当前模型类名称 protected $model; // 当前数据表名称（含前缀） protected $table = ''; // 当前数据表名称（不含前缀） protected $name = ''; // 当前数据表主键 protected $pk; // 当前数据表前缀 protected $prefix = ''; // 查询参数 protected $options = []; // 参数绑定 protected $bind = []; // 数据表信息 protected static $info = []; // 回调事件 private static $event = []; // 读取主库 protected static $readMaster = []; /** * 构造函数 * @access public * @param Connection $connection 数据库对象实例 * @param Model $model 模型对象 */ public function __construct(Connection $connection = null, $model = null) { $this->connection = $connection ?: Db::connect([], true); $this->prefix = $this->connection->getConfig('prefix'); $this->model = $model; // 设置当前连接的Builder对象 $this->setBuilder(); } /** * 利用__call方法实现一些特殊的Model方法 * @access public * @param string $method 方法名称 * @param array $args 调用参数 * @return mixed * @throws DbException * @throws Exception */ public function __call($method, $args) { if (strtolower(substr($method, 0, 5)) == 'getby') { // 根据某个字段获取记录 $field = Loader::parseName(substr($method, 5)); $where[$field] = $args[0]; return $this->where($where)->find(); } elseif (strtolower(substr($method, 0, 10)) == 'getfieldby') { // 根据某个字段获取记录的某个值 $name = Loader::parseName(substr($method, 10)); $where[$name] = $args[0]; return $this->where($where)->value($args[1]); } elseif ($this->model && method_exists($this->model, 'scope' . $method)) { // 动态调用命名范围 $method = 'scope' . $method; array_unshift($args, $this); call_user_func_array([$this->model, $method], $args); return $this; } else { throw new Exception('method not exist:' . __CLASS__ . '->' . $method); } } /** * 获取当前的数据库Connection对象 * @access public * @return Connection */ public function getConnection() { return $this->connection; } /** * 切换当前的数据库连接 * @access public * @param mixed $config * @return $this */ public function connect($config) { $this->connection = Db::connect($config); $this->setBuilder(); $this->prefix = $this->connection->getConfig('prefix'); return $this; } /** * 设置当前的数据库Builder对象 * @access protected * @return void */ protected function setBuilder() { $class = $this->connection->getBuilder(); $this->builder = new $class($this->connection, $this); } /** * 获取当前的模型对象实例 * @access public * @return Model|null */ public function getModel() { return $this->model; } /** * 设置后续从主库读取数据 * @access public * @param bool $allTable * @return void */ public function readMaster($allTable = false) { if ($allTable) { $table = '*'; } else { $table = isset($this->options['table']) ? $this->options['table'] : $this->getTable(); } static::$readMaster[$table] = true; return $this; } /** * 获取当前的builder实例对象 * @access public * @return Builder */ public function getBuilder() { return $this->builder; } /** * 指定默认的数据表名（不含前缀） * @access public * @param string $name * @return $this */ public function name($name) { $this->name = $name; return $this; } /** * 指定默认数据表名（含前缀） * @access public * @param string $table 表名 * @return $this */ public function setTable($table) { $this->table = $table; return $this; } /** * 得到当前或者指定名称的数据表 * @access public * @param string $name * @return string */ public function getTable($name = '') { if ($name || empty($this->table)) { $name = $name ?: $this->name; $tableName = $this->prefix; if ($name) { $tableName .= Loader::parseName($name); } } else { $tableName = $this->table; } return $tableName; } /** * 将SQL语句中的__TABLE_NAME__字符串替换成带前缀的表名（小写） * @access public * @param string $sql sql语句 * @return string */ public function parseSqlTable($sql) { if (false !== strpos($sql, '__')) { $prefix = $this->prefix; $sql = preg_replace_callback("/__([A-Z0-9_-]+)__/sU", function ($match) use ($prefix) { return $prefix . strtolower($match[1]); }, $sql); } return $sql; } /** * 执行查询 返回数据集 * @access public * @param string $sql sql指令 * @param array $bind 参数绑定 * @param boolean $master 是否在主服务器读操作 * @param bool|string $class 指定返回的数据集对象 * @return mixed * @throws BindParamException * @throws PDOException */ public function query($sql, $bind = [], $master = false, $class = false) { return $this->connection->query($sql, $bind, $master, $class); } /** * 执行语句 * @access public * @param string $sql sql指令 * @param array $bind 参数绑定 * @return int * @throws BindParamException * @throws PDOException */ public function execute($sql, $bind = []) { return $this->connection->execute($sql, $bind, $this); } /** * 获取最近插入的ID * @access public * @param string $sequence 自增序列名 * @return string */ public function getLastInsID($sequence = null) { return $this->connection->getLastInsID($sequence); } /** * 获取最近一次查询的sql语句 * @access public * @return string */ public function getLastSql() { return $this->connection->getLastSql(); } /**