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

改进HAS ONE BELONGS_TO 关联预载入查询

Browse Source
This commit is contained in:
thinkphp
2016-04-12 15:24:09 +08:00
parent 880562b42c
commit c5cf46c392
2 changed files with 72 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
library/think/db/Driver.php
View File

@@ -2057,19 +2057,25 @@ abstract class Driver
protected function parsePkWhere($data)
{
$pk = $this->getTableInfo('', 'pk');
// 获取当前数据表
if (!empty($this->options['alias'])) {
$alias = $this->options['alias'];
}
if (is_string($pk)) {
$key = isset($alias) ? $alias . '.' . $pk : $pk;
// 根据主键查询
if (is_array($data)) {
$where[$pk] = isset($data[$pk]) ? $data[$pk] : ['in', $data];
$where[$key] = isset($data[$pk]) ? $data[$pk] : ['in', $data];
} else {
$where[$pk] = strpos($data, ',') ? ['IN', $data] : $data;
$where[$key] = strpos($data, ',') ? ['IN', $data] : $data;
}
$this->options['where']['AND'] = $where;
} elseif (is_array($pk) && is_array($data) && !empty($data)) {
// 根据复合主键查询
foreach ($pk as $key) {
if (isset($data[$key])) {
$where[$key] = $data[$key];
$attr = isset($alias) ? $alias . '.' . $key : $key;
$where[$attr] = $data[$key];
} else {
throw new Exception('miss complex primary data');
}