mirror of
https://gitee.com/zoujingli/ThinkAdmin.git
synced 2025-08-24 22:09:45 +08:00
333 lines
9.0 KiB
PHP
333 lines
9.0 KiB
PHP
<?php
|
|
// +----------------------------------------------------------------------
|
|
// | ThinkPHP [ WE CAN DO IT JUST THINK ]
|
|
// +----------------------------------------------------------------------
|
|
// | Copyright (c) 2006~2019 http://thinkphp.cn All rights reserved.
|
|
// +----------------------------------------------------------------------
|
|
// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
|
|
// +----------------------------------------------------------------------
|
|
// | Author: liu21st <liu21st@gmail.com>
|
|
// +----------------------------------------------------------------------
|
|
declare (strict_types = 1);
|
|
|
|
namespace think\model\concern;
|
|
|
|
use think\Collection;
|
|
use think\db\exception\DbException as Exception;
|
|
use think\helper\Str;
|
|
use think\Model;
|
|
use think\model\Collection as ModelCollection;
|
|
use think\model\relation\OneToOne;
|
|
|
|
/**
|
|
* 模型数据转换处理
|
|
*/
|
|
trait Conversion
|
|
{
|
|
/**
|
|
* 数据输出显示的属性
|
|
* @var array
|
|
*/
|
|
protected $visible = [];
|
|
|
|
/**
|
|
* 数据输出隐藏的属性
|
|
* @var array
|
|
*/
|
|
protected $hidden = [];
|
|
|
|
/**
|
|
* 数据输出需要追加的属性
|
|
* @var array
|
|
*/
|
|
protected $append = [];
|
|
|
|
/**
|
|
* 数据输出字段映射
|
|
* @var array
|
|
*/
|
|
protected $mapping = [];
|
|
|
|
/**
|
|
* 数据集对象名
|
|
* @var string
|
|
*/
|
|
protected $resultSetType;
|
|
|
|
/**
|
|
* 数据命名是否自动转为驼峰
|
|
* @var bool
|
|
*/
|
|
protected $convertNameToCamel;
|
|
|
|
/**
|
|
* 转换数据为驼峰命名（用于输出）
|
|
* @access public
|
|
* @param bool $toCamel 是否自动驼峰命名
|
|
* @return $this
|
|
*/
|
|
public function convertNameToCamel(bool $toCamel = true)
|
|
{
|
|
$this->convertNameToCamel = $toCamel;
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* 设置需要附加的输出属性
|
|
* @access public
|
|
* @param array $append 属性列表
|
|
* @return $this
|
|
*/
|
|
public function append(array $append = [])
|
|
{
|
|
$this->append = $append;
|
|
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* 设置附加关联对象的属性
|
|
* @access public
|
|
* @param string $attr 关联属性
|
|
* @param string|array $append 追加属性名
|
|
* @return $this
|
|
* @throws Exception
|
|
*/
|
|
public function appendRelationAttr(string $attr, array $append)
|
|
{
|
|
$relation = Str::camel($attr);
|
|
|
|
if (isset($this->relation[$relation])) {
|
|
$model = $this->relation[$relation];
|
|
} else {
|
|
$model = $this->getRelationData($this->$relation());
|
|
}
|
|
|
|
if ($model instanceof Model) {
|
|
foreach ($append as $key => $attr) {
|
|
$key = is_numeric($key) ? $attr : $key;
|
|
if (isset($this->data[$key])) {
|
|
throw new Exception('bind attr has exists:' . $key);
|
|
}
|
|
|
|
$this->data[$key] = $model->$attr;
|
|
}
|
|
}
|
|
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* 设置需要隐藏的输出属性
|
|
* @access public
|
|
* @param array $hidden 属性列表
|
|
* @return $this
|
|
*/
|
|
public function hidden(array $hidden = [])
|
|
{
|
|
$this->hidden = $hidden;
|
|
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* 设置需要输出的属性
|
|
* @access public
|
|
* @param array $visible
|
|
* @return $this
|
|
*/
|
|
public function visible(array $visible = [])
|
|
{
|
|
$this->visible = $visible;
|
|
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* 设置属性的映射输出
|
|
* @access public
|
|
* @param array $map
|
|
* @return $this
|
|
*/
|
|
public function mapping(array $map)
|
|
{
|
|
$this->mapping = $map;
|
|
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* 转换当前模型对象为数组
|
|
* @access public
|
|
* @return array
|
|
*/
|
|
public function toArray(): array
|
|
{
|
|
$item = [];
|
|
$hasVisible = false;
|
|
|
|
foreach ($this->visible as $key => $val) {
|
|
if (is_string($val)) {
|
|
if (strpos($val, '.')) {
|
|
[$relation, $name] = explode('.', $val);
|
|
$this->visible[$relation][] = $name;
|
|
} else {
|
|
$this->visible[$val] = true;
|
|
$hasVisible = true;
|
|
}
|
|
unset($this->visible[$key]);
|
|
}
|
|
}
|
|
|
|
foreach ($this->hidden as $key => $val) {
|
|
if (is_string($val)) {
|
|
if (strpos($val, '.')) {
|
|
[$relation, $name] = explode('.', $val);
|
|
$this->hidden[$relation][] = $name;
|
|
} else {
|
|
$this->hidden[$val] = true;
|
|
}
|
|
unset($this->hidden[$key]);
|
|
}
|
|
}
|
|
|
|
// 合并关联数据
|
|
$data = array_merge($this->data, $this->relation);
|
|
|
|
foreach ($data as $key => $val) {
|
|
if ($val instanceof Model || $val instanceof ModelCollection) {
|
|
// 关联模型对象
|
|
if (isset($this->visible[$key]) && is_array($this->visible[$key])) {
|
|
$val->visible($this->visible[$key]);
|
|
} elseif (isset($this->hidden[$key]) && is_array($this->hidden[$key])) {
|
|
$val->hidden($this->hidden[$key]);
|
|
}
|
|
// 关联模型对象
|
|
if (!isset($this->hidden[$key]) || true !== $this->hidden[$key]) {
|
|
$item[$key] = $val->toArray();
|
|
}
|
|
} elseif (isset($this->visible[$key])) {
|
|
$item[$key] = $this->getAttr($key);
|
|
} elseif (!isset($this->hidden[$key]) && !$hasVisible) {
|
|
$item[$key] = $this->getAttr($key);
|
|
}
|
|
|
|
if (isset($this->mapping[$key])) {
|
|
// 检查字段映射
|
|
$mapName = $this->mapping[$key];
|
|
$item[$mapName] = $item[$key];
|
|
unset($item[$key]);
|
|
}
|
|
}
|
|
|
|
// 追加属性（必须定义获取器）
|
|
foreach ($this->append as $key => $name) {
|
|
$this->appendAttrToArray($item, $key, $name);
|
|
}
|
|
|
|
if ($this->convertNameToCamel) {
|
|
foreach ($item as $key => $val) {
|
|
$name = Str::camel($key);
|
|
if ($name !== $key) {
|
|
$item[$name] = $val;
|
|
unset($item[$key]);
|
|
}
|
|
}
|
|
}
|
|
|
|
return $item;
|
|
}
|
|
|
|
protected function appendAttrToArray(array &$item, $key, $name)
|
|
{
|
|
if (is_array($name)) {
|
|
// 追加关联对象属性
|
|
$relation = $this->getRelation($key, true);
|
|
$item[$key] = $relation ? $relation->append($name)
|
|
->toArray() : [];
|
|
} elseif (strpos($name, '.')) {
|
|
[$key, $attr] = explode('.', $name);
|
|
// 追加关联对象属性
|
|
$relation = $this->getRelation($key, true);
|
|
$item[$key] = $relation ? $relation->append([$attr])
|
|
->toArray() : [];
|
|
} else {
|
|
$value = $this->getAttr($name);
|
|
$item[$name] = $value;
|
|
|
|
$this->getBindAttr($name, $value, $item);
|
|
}
|
|
}
|
|
|
|
protected function getBindAttr(string $name, $value, array &$item = [])
|
|
{
|
|
$relation = $this->isRelationAttr($name);
|
|
if (!$relation) {
|
|
return false;
|
|
}
|
|
|
|
$modelRelation = $this->$relation();
|
|
|
|
if ($modelRelation instanceof OneToOne) {
|
|
$bindAttr = $modelRelation->getBindAttr();
|
|
|
|
if (!empty($bindAttr)) {
|
|
unset($item[$name]);
|
|
}
|
|
|
|
foreach ($bindAttr as $key => $attr) {
|
|
$key = is_numeric($key) ? $attr : $key;
|
|
|
|
if (isset($item[$key])) {
|
|
throw new Exception('bind attr has exists:' . $key);
|
|
}
|
|
|
|
$item[$key] = $value ? $value->getAttr($attr) : null;
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 转换当前模型对象为JSON字符串
|
|
* @access public
|
|
* @param integer $options json参数
|
|
* @return string
|
|
*/
|
|
public function toJson(int $options = JSON_UNESCAPED_UNICODE): string
|
|
{
|
|
return json_encode($this->toArray(), $options);
|
|
}
|
|
|
|
public function __toString()
|
|
{
|
|
return $this->toJson();
|
|
}
|
|
|
|
// JsonSerializable
|
|
public function jsonSerialize()
|
|
{
|
|
return $this->toArray();
|
|
}
|
|
|
|
/**
|
|
* 转换数据集为数据集对象
|
|
* @access public
|
|
* @param array|Collection $collection 数据集
|
|
* @param string $resultSetType 数据集类
|
|
* @return Collection
|
|
*/
|
|
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 ModelCollection($collection);
|
|
}
|
|
|
|
return $collection;
|
|
}
|
|
|
|
}
|