mirror-fastadminnet/framework
1
0
Fork 0
mirror of https://gitee.com/fastadminnet/framework.git synced 2026-07-07 07:12:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

修正BELONGS_TO关联查询

Browse Source
This commit is contained in:
thinkphp
2016-04-23 20:26:26 +08:00
parent 8cca5524a7
commit 265a80905d
2 changed files with 16 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
library/think/Model.php
View File

@@ -729,17 +729,17 @@ abstract class Model implements \JsonSerializable, \ArrayAccess
* BELONGS TO 关联定义 * BELONGS TO 关联定义
* @access public * @access public
* @param string $model 模型名 * @param string $model 模型名
* @param string $localKey 关联主键
* @param string $foreignKey 关联外键 * @param string $foreignKey 关联外键
* @param string $otherKey 关联主键
* @return \think\db\Query|string * @return \think\db\Query|string
*/ */
public function belongsTo($model, $localKey = '', $foreignKey = '') public function belongsTo($model, $foreignKey = '', $otherKey = '')
{ {
// 记录当前关联信息 // 记录当前关联信息
$model = $this->parseModel($model); $model = $this->parseModel($model);
$foreignKey = $foreignKey ?: $this->pk; $foreignKey = $foreignKey ?: Loader::parseName(basename(str_replace('\\', '/', $model))) . '_id';
$localKey = $localKey ?: Loader::parseName(basename(str_replace('\\', '/', $model))) . '_id'; $otherKey = $otherKey ?: (new $model)->getPk();
return $this->relation->belongsTo($model, $foreignKey, $localKey); return $this->relation->belongsTo($model, $foreignKey, $otherKey);
} }
/** /**

15
library/think/model/Relation.php
View File

@@ -74,7 +74,6 @@ class Relation
// 判断关联类型执行查询 // 判断关联类型执行查询
switch ($this->type) { switch ($this->type) {
case self::HAS_ONE: case self::HAS_ONE:
case self::BELONGS_TO:
$result = $relation->where($foreignKey, $this->parent->$localKey)->find(); $result = $relation->where($foreignKey, $this->parent->$localKey)->find();
if (false === $result) { if (false === $result) {
$class = $this->model; $class = $this->model;
@@ -82,6 +81,14 @@ class Relation
$result->$foreignKey = $this->parent->$localKey; $result->$foreignKey = $this->parent->$localKey;
} }
break; break;
case self::BELONGS_TO:
$result = $relation->where($localKey, $this->parent->$foreignKey)->find();
if (false === $result) {
$class = $this->model;
$result = new $class;
$result->$localKey = $this->parent->$foreignKey;
}
break;
case self::HAS_MANY: case self::HAS_MANY:
$result = $relation->where($foreignKey, $this->parent->$localKey)->select(); $result = $relation->where($foreignKey, $this->parent->$localKey)->select();
break; break;
@@ -368,17 +375,17 @@ class Relation
* BELONGS TO 关联定义 * BELONGS TO 关联定义
* @access public * @access public
* @param string $model 模型名 * @param string $model 模型名
* @param string $localKey 关联主键
* @param string $foreignKey 关联外键 * @param string $foreignKey 关联外键
* @param string $localKey 关联主键
* @return \think\db\Query|string * @return \think\db\Query|string
*/ */
public function belongsTo($model, $localKey, $foreignKey) public function belongsTo($model, $foreignKey, $otherKey)
{ {
// 记录当前关联信息 // 记录当前关联信息
$this->type = self::BELONGS_TO; $this->type = self::BELONGS_TO;
$this->model = $model; $this->model = $model;
$this->foreignKey = $foreignKey; $this->foreignKey = $foreignKey;
$this->localKey = $localKey; $this->localKey = $otherKey;
// 返回关联的模型对象 // 返回关联的模型对象
return $this; return $this;