80 lines
1.7 KiB
PHP
80 lines
1.7 KiB
PHP
<?php
|
|
namespace catcher\traits\db;
|
|
|
|
use catcher\CatchModelCollection;
|
|
use think\Collection;
|
|
|
|
/**
|
|
* 重写 think\Model 的方法
|
|
*
|
|
* Trait RewriteTrait
|
|
* @package catcher\traits\db
|
|
*/
|
|
trait RewriteTrait
|
|
{
|
|
/**
|
|
* 初始化
|
|
*
|
|
* CatchModel constructor.
|
|
* @param array $data
|
|
*/
|
|
public function __construct(array $data = [])
|
|
{
|
|
parent::__construct($data);
|
|
|
|
$this->hidden = array_merge($this->hidden, $this->defaultHiddenFields());
|
|
}
|
|
|
|
/**
|
|
* hidden model fields
|
|
*
|
|
* @return array
|
|
*/
|
|
protected function defaultHiddenFields(): array
|
|
{
|
|
return [$this->deleteTime];
|
|
}
|
|
|
|
/**
|
|
* 重写 hidden 方法，支持合并 hidden 属性
|
|
*
|
|
* @param array $hidden
|
|
* @return $this
|
|
*/
|
|
public function hidden(array $hidden = [])
|
|
{
|
|
/**
|
|
* 合并属性
|
|
*/
|
|
if (!count($this->hidden)) {
|
|
$this->hidden = array_merge($this->hidden, $hidden);
|
|
|
|
return $this;
|
|
}
|
|
|
|
$this->hidden = $hidden;
|
|
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* rewrite collection
|
|
*
|
|
* @time 2020年10月20日
|
|
* @param array|iterable $collection
|
|
* @param string|null $resultSetType
|
|
* @return CatchModelCollection|mixed
|
|
*/
|
|
public function toCollection(iterable $collection = [], string $resultSetType = null): Collection
|
|
{
|
|
$resultSetType = $resultSetType ?: $this->resultSetType;
|
|
|
|
if ($resultSetType && false !== strpos($resultSetType, '\\')) {
|
|
$collection = new $resultSetType($collection);
|
|
} else {
|
|
$collection = new CatchModelCollection($collection);
|
|
}
|
|
|
|
return $collection;
|
|
}
|
|
} |