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

改进模型类对主键的自动读取

Browse Source
This commit is contained in:
thinkphp
2016-04-27 12:42:57 +08:00
parent 8cb52f60dd
commit 6915f0249e
2 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
library/think/Model.php
View File

@@ -30,7 +30,7 @@ abstract class Model implements \JsonSerializable, \ArrayAccess
protected static $event = [];
// 数据表主键 复合主键使用数组定义
protected $pk = 'id';
protected $pk;
// 错误信息
protected $error;
// 当前模型名称
@@ -177,7 +177,7 @@ abstract class Model implements \JsonSerializable, \ArrayAccess
*/
public function getPk($table = '')
{
if (!$this->pk) {
if (empty($this->pk)) {
$this->pk = self::db()->getTableInfo($table, 'pk');
}
return $this->pk;
@@ -272,8 +272,9 @@ abstract class Model implements \JsonSerializable, \ArrayAccess
// 获取自动增长主键
if ($result && $getInsertId) {
$insertId = self::db()->getLastInsID();
if (is_string($this->pk) && $insertId) {
$this->data[$this->pk] = $insertId;
$pk = $this->getPk();
if (is_string($pk) && $insertId) {
$this->data[$pk] = $insertId;
}
}
// 新增回调
@@ -731,7 +732,7 @@ abstract class Model implements \JsonSerializable, \ArrayAccess
{
// 记录当前关联信息
$model = $this->parseModel($model);
$localKey = $localKey ?: $this->pk;
$localKey = $localKey ?: $this->getPk();
$foreignKey = $foreignKey ?: Loader::parseName($this->name) . '_id';
return $this->relation->hasOne($model, $foreignKey, $localKey);
}
@@ -765,7 +766,7 @@ abstract class Model implements \JsonSerializable, \ArrayAccess
{
// 记录当前关联信息
$model = $this->parseModel($model);
$localKey = $localKey ?: $this->pk;
$localKey = $localKey ?: $this->getPk();
$foreignKey = $foreignKey ?: Loader::parseName($this->name) . '_id';
return $this->relation->hasMany($model, $foreignKey, $localKey);
}