323 lines
11 KiB
PHP
323 lines
11 KiB
PHP
<?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\model;
|
|
|
|
use think\Db;
|
|
use think\db\Query;
|
|
use think\Model;
|
|
|
|
class Merge extends Model
|
|
{
|
|
|
|
protected $relationModel = []; // HAS ONE 关联的模型列表
|
|
protected $fk = ''; // 外键名 默认为主表名_id
|
|
protected $mapFields = []; // 需要处理的模型映射字段，避免混淆 array( id => 'user.id' )
|
|
|
|
/**
|
|
* 构造函数
|
|
* @access public
|
|
* @param array|object $data 数据
|
|
*/
|
|
public function __construct($data = [])
|
|
{
|
|
parent::__construct($data);
|
|
|
|
// 设置默认外键名 仅支持单一外键
|
|
if (empty($this->fk)) {
|
|
$this->fk = strtolower($this->name) . '_id';
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 查找单条记录
|
|
* @access public
|
|
* @param mixed $data 主键值或者查询条件（闭包）
|
|
* @param string|array $with 关联预查询
|
|
* @param bool $cache 是否缓存
|
|
* @return \think\Model
|
|
*/
|
|
public static function get($data = null, $with = [], $cache = false)
|
|
{
|
|
$query = self::parseQuery($data, $with, $cache);
|
|
$query = self::attachQuery($query);
|
|
return $query->find($data);
|
|
}
|
|
|
|
/**
|
|
* 附加查询表达式
|
|
* @access protected
|
|
* @param \think\db\Query $query 查询对象
|
|
* @return \think\db\Query
|
|
*/
|
|
protected static function attachQuery($query)
|
|
{
|
|
$class = new static();
|
|
$master = $class->name;
|
|
$fields = self::getModelField($query, $master, '', $class->mapFields, $class->field);
|
|
$query->alias($master)->field($fields);
|
|
|
|
foreach ($class->relationModel as $key => $model) {
|
|
$name = is_int($key) ? $model : $key;
|
|
$table = is_int($key) ? $query->getTable($name) : $model;
|
|
$query->join($table . ' ' . $name, $name . '.' . $class->fk . '=' . $master . '.' . $class->getPk());
|
|
$fields = self::getModelField($query, $name, $table, $class->mapFields, $class->field);
|
|
$query->field($fields);
|
|
}
|
|
return $query;
|
|
}
|
|
|
|
/**
|
|
* 获取关联模型的字段 并解决混淆
|
|
* @access protected
|
|
* @param \think\db\Query $query 查询对象
|
|
* @param string $name 模型名称
|
|
* @param string $table 关联表名称
|
|
* @param array $map 字段映射
|
|
* @param array $fields 查询字段
|
|
* @return array
|
|
*/
|
|
protected static function getModelField($query, $name, $table = '', $map = [], $fields = [])
|
|
{
|
|
// 获取模型的字段信息
|
|
$fields = $fields ?: $query->getTableInfo($table, 'fields');
|
|
$array = [];
|
|
foreach ($fields as $field) {
|
|
if ($key = array_search($name . '.' . $field, $map)) {
|
|
// 需要处理映射字段
|
|
$array[] = $name . '.' . $field . ' AS ' . $key;
|
|
} else {
|
|
$array[] = $field;
|
|
}
|
|
}
|
|
return $array;
|
|
}
|
|
|
|
/**
|
|
* 查找所有记录
|
|
* @access public
|
|
* @param mixed $data 主键列表或者查询条件（闭包）
|
|
* @param array|string $with 关联预查询
|
|
* @param bool $cache
|
|
* @return array|false|string
|
|
*/
|
|
public static function all($data = null, $with = [], $cache = false)
|
|
{
|
|
$query = self::parseQuery($data, $with, $cache);
|
|
$query = self::attachQuery($query);
|
|
return $query->select($data);
|
|
}
|
|
|
|
/**
|
|
* 处理写入的模型数据
|
|
* @access public
|
|
* @param string $model 模型名称
|
|
* @param array $data 数据
|
|
* @return array
|
|
*/
|
|
protected function parseData($model, $data)
|
|
{
|
|
$item = [];
|
|
foreach ($data as $key => $val) {
|
|
if ($this->fk != $key && array_key_exists($key, $this->mapFields)) {
|
|
list($name, $key) = explode('.', $this->mapFields[$key]);
|
|
if ($model == $name) {
|
|
$item[$key] = $val;
|
|
}
|
|
} else {
|
|
$item[$key] = $val;
|
|
}
|
|
}
|
|
return $item;
|
|
}
|
|
|
|
/**
|
|
* 保存模型数据 以及关联数据
|
|
* @access public
|
|
* @param mixed $data 数据
|
|
* @param array $where 更新条件
|
|
* @param string $sequence 自增序列名
|
|
* @return false|int
|
|
* @throws \Exception
|
|
*/
|
|
public function save($data = [], $where = [], $sequence = null)
|
|
{
|
|
if (!empty($data)) {
|
|
// 数据自动验证
|
|
if (!$this->validateData($data)) {
|
|
return false;
|
|
}
|
|
// 数据对象赋值
|
|
foreach ($data as $key => $value) {
|
|
$this->setAttr($key, $value, $data);
|
|
}
|
|
if (!empty($where)) {
|
|
$this->isUpdate = true;
|
|
}
|
|
}
|
|
|
|
// 数据自动完成
|
|
$this->autoCompleteData($this->auto);
|
|
|
|
// 自动写入更新时间
|
|
if ($this->autoWriteTimestamp && $this->updateTime && !isset($this->data[$this->updateTime])) {
|
|
$this->setAttr($this->updateTime, null);
|
|
}
|
|
|
|
// 事件回调
|
|
if (false === $this->trigger('before_write', $this)) {
|
|
return false;
|
|
}
|
|
|
|
$db = $this->db();
|
|
$db->startTrans();
|
|
$pk = $this->getPk();
|
|
try {
|
|
if ($this->isUpdate) {
|
|
// 自动写入
|
|
$this->autoCompleteData($this->update);
|
|
|
|
if (false === $this->trigger('before_update', $this)) {
|
|
return false;
|
|
}
|
|
|
|
if (empty($where) && !empty($this->updateWhere)) {
|
|
$where = $this->updateWhere;
|
|
}
|
|
|
|
// 获取有更新的数据
|
|
$data = $this->getChangedData();
|
|
// 保留主键数据
|
|
foreach ($this->data as $key => $val) {
|
|
if ($this->isPk($key)) {
|
|
$data[$key] = $val;
|
|
}
|
|
}
|
|
// 处理模型数据
|
|
$data = $this->parseData($this->name, $data);
|
|
if (is_string($pk) && isset($data[$pk])) {
|
|
if (!isset($where[$pk])) {
|
|
unset($where);
|
|
$where[$pk] = $data[$pk];
|
|
}
|
|
unset($data[$pk]);
|
|
}
|
|
// 写入主表数据
|
|
$result = $db->strict(false)->where($where)->update($data);
|
|
|
|
// 写入附表数据
|
|
foreach ($this->relationModel as $key => $model) {
|
|
$name = is_int($key) ? $model : $key;
|
|
$table = is_int($key) ? $db->getTable($model) : $model;
|
|
// 处理关联模型数据
|
|
$data = $this->parseData($name, $data);
|
|
if (Db::table($table)->strict(false)->where($this->fk, $this->data[$this->getPk()])->update($data)) {
|
|
$result = 1;
|
|
}
|
|
}
|
|
|
|
// 新增回调
|
|
$this->trigger('after_update', $this);
|
|
} else {
|
|
// 自动写入
|
|
$this->autoCompleteData($this->insert);
|
|
|
|
// 自动写入创建时间
|
|
if ($this->autoWriteTimestamp && $this->createTime && !isset($this->data[$this->createTime])) {
|
|
$this->setAttr($this->createTime, null);
|
|
}
|
|
|
|
if (false === $this->trigger('before_insert', $this)) {
|
|
return false;
|
|
}
|
|
|
|
// 处理模型数据
|
|
$data = $this->parseData($this->name, $this->data);
|
|
// 写入主表数据
|
|
$result = $db->name($this->name)->strict(false)->insert($data);
|
|
if ($result) {
|
|
$insertId = $db->getLastInsID($sequence);
|
|
// 写入外键数据
|
|
if ($insertId) {
|
|
if (is_string($pk)) {
|
|
$this->data[$pk] = $insertId;
|
|
}
|
|
$this->data[$this->fk] = $insertId;
|
|
}
|
|
|
|
// 写入附表数据
|
|
$source = $this->data;
|
|
if ($insertId && is_string($pk) && isset($source[$pk]) && $this->fk != $pk) {
|
|
unset($source[$pk]);
|
|
}
|
|
foreach ($this->relationModel as $key => $model) {
|
|
$name = is_int($key) ? $model : $key;
|
|
$table = is_int($key) ? $db->getTable($model) : $model;
|
|
// 处理关联模型数据
|
|
$data = $this->parseData($name, $source);
|
|
Db::table($table)->strict(false)->insert($data);
|
|
}
|
|
}
|
|
// 标记为更新
|
|
$this->isUpdate = true;
|
|
// 新增回调
|
|
$this->trigger('after_insert', $this);
|
|
}
|
|
$db->commit();
|
|
// 写入回调
|
|
$this->trigger('after_write', $this);
|
|
|
|
$this->origin = $this->data;
|
|
return $result;
|
|
} catch (\Exception $e) {
|
|
$db->rollback();
|
|
throw $e;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 删除当前的记录 并删除关联数据
|
|
* @access public
|
|
* @return int
|
|
* @throws \Exception
|
|
*/
|
|
public function delete()
|
|
{
|
|
if (false === $this->trigger('before_delete', $this)) {
|
|
return false;
|
|
}
|
|
|
|
$db = $this->db();
|
|
$db->startTrans();
|
|
try {
|
|
$result = $db->delete($this->data);
|
|
if ($result) {
|
|
// 获取主键数据
|
|
$pk = $this->data[$this->getPk()];
|
|
|
|
// 删除关联数据
|
|
foreach ($this->relationModel as $key => $model) {
|
|
$table = is_int($key) ? $db->getTable($model) : $model;
|
|
$query = new Query;
|
|
$query->table($table)->where($this->fk, $pk)->delete();
|
|
}
|
|
}
|
|
$this->trigger('after_delete', $this);
|
|
$db->commit();
|
|
return $result;
|
|
} catch (\Exception $e) {
|
|
$db->rollback();
|
|
throw $e;
|
|
}
|
|
}
|
|
|
|
}
|