改名

Database/ActiveQuery.php

@@ -0,0 +1,406 @@
+<?php
+/**
+ * Created by PhpStorm.
+ * User: whwyy
+ * Date: 2018/4/4 0004
+ * Time: 14:42
+ */
+
+namespace Database;
+
+use Snowflake\Abstracts\Component;
+use Database\Orm\Select;
+use Database\Traits\QueryTrait;
+use Exception;
+use Snowflake\Snowflake;
+
+/**
+ * Class ActiveQuery
+ * @package Database
+ */
+class ActiveQuery extends Component
+{
+
+	use QueryTrait;
+
+	/** @var array */
+	public $with = [];
+
+	/** @var bool */
+	public $asArray = FALSE;
+
+	/** @var bool */
+	public $useCache = FALSE;
+
+	/** @var Connection $db */
+	public $db = NULL;
+
+	/**
+	 * @var array
+	 * 参数绑定
+	 */
+	public $attributes = [];
+
+
+	/** @var ActiveRecord */
+	public $modelClass;
+
+	/**
+	 * Comply constructor.
+	 * @param $model
+	 * @param array $config
+	 */
+	public function __construct($model, $config = [])
+	{
+		$this->modelClass = $model;
+		parent::__construct($config);
+	}
+
+	/**
+	 *
+	 */
+	public function afterInit()
+	{
+		$this->debug(get_called_class() . ' AFTER INIT.');
+		$this->clear();
+	}
+
+	/**
+	 * 清除不完整数据
+	 */
+	public function clear()
+	{
+		$this->db = null;
+		$this->useCache = false;
+		$this->with = [];
+	}
+
+	/**
+	 * @param $key
+	 * @param $value
+	 * @return $this
+	 */
+	public function addParam($key, $value)
+	{
+		$this->attributes[$key] = $value;
+		return $this;
+	}
+
+	/**
+	 * @param array $values
+	 * @return $this
+	 */
+	public function addParams(array $values)
+	{
+		foreach ($values as $key => $val) {
+			$this->addParam($key, $val);
+		}
+		return $this;
+	}
+
+	/**
+	 * @param $name
+	 * @return $this
+	 */
+	public function with($name)
+	{
+		if (empty($name)) {
+			return $this;
+		}
+		if (is_string($name)) {
+			$name = explode(',', $name);
+		}
+		foreach ($name as $key => $val) {
+			array_push($this->with, $val);
+		}
+		return $this;
+	}
+
+	/**
+	 * @param bool $isArray
+	 * @return $this
+	 */
+	public function asArray($isArray = TRUE)
+	{
+		$this->asArray = $isArray;
+		return $this;
+	}
+
+	/**
+	 * @return ActiveRecord
+	 * @throws
+	 */
+	public function first()
+	{
+		$data = $this->command($this->oneLimit()->adaptation())->one();
+		if (empty($data)) {
+			return NULL;
+		}
+		$newModel = Snowflake::createObject($this->modelClass);
+		$newModel = $this->populate($newModel, $data);
+		if ($this->asArray) {
+			return $newModel->toArray();
+		}
+		return $newModel;
+	}
+
+	/**
+	 * @return array|Collection
+	 */
+	public function get()
+	{
+		return $this->all();
+	}
+
+
+	/**
+	 * @throws Exception
+	 */
+	public function flush()
+	{
+		$sql = $this->getChange()->truncate($this->getTable());
+		return $this->command($sql)->flush();
+	}
+
+
+	/**
+	 * @param int $size
+	 * @param callable $callback
+	 * @param mixed $param
+	 * @param int $offset
+	 * @param int $total
+	 * @throws Exception
+	 *
+	 *
+	 * $pagination = new Pagination($this);
+	 * $pagination->setOffset($offset);
+	 * $pagination->setLimit($size);
+	 * $pagination->setMax($total);
+	 * $pagination->setCallback($callback);
+	 * $pagination->search($param);
+	 */
+	public function plunk(int $size, callable $callback, $param = null, $offset = 0, $total = -1)
+	{
+		if (!is_callable($callback, true)) {
+			return;
+		}
+		$data = $this->limit($offset, $size)->get();
+		if ($param !== null) {
+			call_user_func($callback, $data, $param);
+		} else {
+			call_user_func($callback, $data);
+		}
+		if ($data->size() < $size) {
+			return;
+		}
+		unset($data);
+		$this->plunk($size, $callback, $param, $offset + $size, $total);
+	}
+
+	/**
+	 * @param string $field
+	 * @param string $setKey
+	 *
+	 * @return array|null
+	 * @throws Exception
+	 */
+	public function column(string $field, $setKey = '')
+	{
+		return $this->all()->column($field, $setKey);
+	}
+
+	/**
+	 * @return array|Collection
+	 * @throws
+	 */
+	public function all()
+	{
+		$data = $this->command($this->adaptation())->all();
+		$collect = new Collection();
+		$collect->setModel($this->modelClass);
+		if (empty($data) || !is_array($data)) {
+			return $this->asArray ? [] : $collect;
+		}
+
+		$_tmp = [];
+		$model = Snowflake::createObject($this->modelClass);
+		foreach ($data as $key => $val) {
+			$_tmp[] = $this->populate(clone $model, $val);
+		}
+		$collect->setItems($_tmp);
+		if ($this->asArray) {
+			return $collect->toArray();
+		}
+		return $collect;
+	}
+
+	/**
+	 * @param ActiveRecord $model
+	 * @param $data
+	 * @return ActiveRecord
+	 * @throws Exception
+	 */
+	private function populate($model, $data)
+	{
+		$model = $model::populate($data);
+		if (empty($this->with) || !is_array($this->with)) {
+			return $model;
+		}
+		foreach ($this->with as $val) {
+			$method = 'get' . ucfirst($val);
+			if (!method_exists($model, $method)) {
+				continue;
+			}
+			$model->setRelate($val, $method);
+		}
+		return $model;
+	}
+
+	/**
+	 * @return int
+	 * @throws Exception
+	 */
+	public function count()
+	{
+		$data = $this->command($this->getBuild()->count($this))->one();
+		if ($data && is_array($data)) {
+			return (int)array_shift($data);
+		}
+		return 0;
+	}
+
+
+	/**
+	 * @param array $data
+	 * @return array|Command|bool|int|string
+	 * @throws Exception
+	 */
+	public function batchUpdate(array $data)
+	{
+		return $this->getDb()->createCommand()
+			->batchUpdate($this->getTable(), $data, $this->getCondition())
+			->exec();
+	}
+
+	/**
+	 * @param array $data
+	 * @return bool
+	 * @throws Exception
+	 */
+	public function batchInsert(array $data)
+	{
+		return $this->getDb()->createCommand()
+			->batchInsert($this->getTable(), $data)
+			->exec();
+	}
+
+	/**
+	 * @param $filed
+	 *
+	 * @return null
+	 * @throws Exception
+	 */
+	public function value($filed)
+	{
+		$first = $this->first()->toArray();
+		return $first[$filed] ?? null;
+	}
+
+	/**
+	 * @return bool
+	 * @throws Exception
+	 */
+	public function exists()
+	{
+		return (bool)$this->command($this->adaptation())->fetchColumn();
+	}
+
+	/**
+	 * @return bool
+	 * @throws Exception
+	 */
+	public function deleteAll()
+	{
+		return $this->command($this->getChange()->delete($this))->delete();
+	}
+
+	/**
+	 * @return bool
+	 * @throws Exception
+	 */
+	public function delete()
+	{
+		return $this->deleteAll();
+	}
+
+	/**
+	 * @return string
+	 * @throws Exception
+	 */
+	public function getCondition()
+	{
+		return $this->getBuild()->getWhere($this->where);
+	}
+
+	/**
+	 * @return string
+	 * @throws Exception
+	 */
+	public function adaptation()
+	{
+		return $this->getBuild()->getQuery($this);
+	}
+
+	/**
+	 * @param $sql
+	 * @param array $attr
+	 * @return Command
+	 * @throws Exception
+	 */
+	private function command($sql, $attr = [])
+	{
+		if (!empty($attr) && is_array($attr)) {
+			$attr = array_merge($this->attributes, $attr);
+		} else {
+			$attr = $this->attributes;
+		}
+		return $this->getDb()->createCommand($sql, $attr)
+			->setModelName($this->modelClass);
+	}
+
+	/**
+	 * @return Select
+	 * @throws Exception
+	 */
+	public function getBuild()
+	{
+		return $this->getDb()->getSchema()->getQueryBuilder();
+	}
+
+	/**
+	 * @return orm\Change
+	 * @throws Exception
+	 */
+	public function getChange()
+	{
+		return $this->getDb()->getSchema()->getChange();
+	}
+
+	/**
+	 * @return Connection
+	 * @throws Exception
+	 */
+	public function getDb()
+	{
+		return $this->modelClass::getDb();
+	}
+
+	/**
+	 * @return mixed
+	 * @throws Exception
+	 */
+	public function getPrimary()
+	{
+		return $this->modelClass::getPrimary();
+	}
+}