PHP实例开发源码-StartBBS 起点开源社区系统 php版 v2.0.0.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\Collection; use think\Container; 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 { /** * 数据库连接对象列表 * @var array */ protected static $connections = []; /** * 当前数据库连接对象 * @var Connection */ protected $connection; /** * 当前模型对象 * @var Model */ protected $model; /** * 当前数据表名称（不含前缀） * @var string */ protected $name = ''; /** * 当前数据表主键 * @var string|array */ protected $pk; /** * 当前数据表前缀 * @var string */ protected $prefix = ''; /** * 当前查询参数 * @var array */ protected $options = []; /** * 当前参数绑定 * @var array */ protected $bind = []; /** * 事件回调 * @var array */ private static $event = []; /** * 扩展查询方法 * @var array */ private static $extend = []; /** * 日期查询表达式 * @var array */ protected $timeRule = [ 'today' => ['today', 'tomorrow'], 'yesterday' => ['yesterday', 'today'], 'week' => ['this week 00:00:00', 'next week 00:00:00'], 'last week' => ['last week 00:00:00', 'this week 00:00:00'], 'month' => ['first Day of this month 00:00:00', 'first Day of next month 00:00:00'], 'last month' => ['first Day of last month 00:00:00', 'first Day of this month 00:00:00'], 'year' => ['this year 1/1', 'next year 1/1'], 'last year' => ['last year 1/1', 'this year 1/1'], ]; /** * 日期查询快捷定义 * @var array */ protected $timeExp = ['d' => 'today', 'w' => 'week', 'm' => 'month', 'y' => 'year']; /** * 架构函数 * @access public */ public function __construct(Connection $connection = null) { if (is_null($connection)) { $this->connection = Connection::instance(); } else { $this->connection = $connection; } $this->prefix = $this->connection->getConfig('prefix'); } /** * 创建一个新的查询对象 * @access public * @return Query */ public function newQuery() { return new static($this->connection); } /** * 利用__call方法实现一些特殊的Model方法 * @access public * @param string $method 方法名称 * @param array $args 调用参数 * @return mixed * @throws DbException * @throws Exception */ public function __call($method, $args) { if (isset(self::$extend[strtolower($method)])) { // 调用扩展查询方法 array_unshift($args, $this); return Container::getInstance()->invoke(self::$extend[strtolower($method)], $args); } elseif (strtolower(substr($method, 0, 5)) == 'getby') { // 根据某个字段获取记录 $field = Loader::parseName(substr($method, 5)); return $this->where($field, '=', $args[0])->find(); } elseif (strtolower(substr($method, 0, 10)) == 'getfieldby') { // 根据某个字段获取记录的某个值 $name = Loader::parseName(substr($method, 10)); return $this->where($name, '=', $args[0])->value($args[1]); } elseif (strtolower(substr($method, 0, 7)) == 'whereor') { $name = Loader::parseName(substr($method, 7)); array_unshift($args, $name); return call_user_func_array([$this, 'whereOr'], $args); } elseif (strtolower(substr($method, 0, 5)) == 'where') { $name = Loader::parseName(substr($method, 5)); array_unshift($args, $name); return call_user_func_array([$this, 'where'], $args); } 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:' . static::class . '->' . $method); } } /** * 扩展查询方法 * @access public * @param string|array $method 查询方法名 * @param callable $callback * @return void */ public static function extend($method, $callback = null) { if (is_array($method)) { foreach ($method as $key => $val) { self::$extend[strtolower($key)] = $val; } } else { self::$extend[strtolower($method)] = $callback; } } /** * 设置当前的数据库Connection对象 * @access public * @param Connection $connection * @return $this */ public function setConnection(Connection $connection) { $this->connection = $connection; $this->prefix = $this->connection->getConfig('prefix'); return $this; } /** * 获取当前的数据库Connection对象 * @access public * @return Connection */ public function getConnection() { return $this->connection; } /** * 指定模型 * @access public * @param Model $model 模型对象实例 * @return $this */ public function model(Model $model) { $this->model = $model; return $this; } /** * 获取当前的模型对象 * @access public * @return Model|null */ public function getModel() { return $this->model ? $this->model->setQuery($this) : null; } /** * 指定当前数据表名（不含前缀） * @access public * @param string $name * @return $this */ public function name($name) { $this->name = $name; return $this; } /** * 获取当前的数据表名称 * @access public * @return string */ public function getName() { return $this->name ?: $this->model->getName(); } /** * 得到当前或者指定名称的数据表 * @access public * @param string $name * @return string */ public function getTable($name = '') { if (empty($name) && isset($this->options['table'])) { return $this->options['table']; } $name = $name ?: $this->name; return $this->prefix . Loader::parseName($name); } /** * 切换数据库连接 * @access public * @param mixed $config 连接配置 * @param bool|string $name 连接标识 true 强制重新连接 * @return $this|object * @throws Exception */ public function connect($config = [], $name = false) { $this->connection = Connection::instance($config, $name); $query = $this->connection->getConfig('query