php/yii2
1
0
Fork 0
mirror of https://github.com/yiisoft/yii2.git synced 2025-08-15 23:04:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update QueryBuilder.php

Browse Source 
solution used **heukirne** for the names of columns.
This commit is contained in:
Fourteenmeister
2014-05-15 12:03:50 +09:00
parent f46c706033
commit f98c414cd1
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
framework/db/mssql/QueryBuilder.php
View File

@ -197,7 +197,7 @@ class QueryBuilder extends \yii\db\QueryBuilder
$select = $query->select = '*';
}
if ($select === '*') {
$columns = $this->getAllColumnNames($query->from[0]);
$columns = $this->getAllColumnNames($query->modelClass);
if ($columns && is_array($columns))
$select = implode(', ', $columns);
else
@ -209,13 +209,14 @@ class QueryBuilder extends \yii\db\QueryBuilder
return $sql;
}
protected function getAllColumnNames($table = null)
protected function getAllColumnNames($modelClass = null)
{
if (!$table) {
if (!$modelClass) {
return null;
}
$columns = $this->db->createCommand("SELECT COLUMN_NAME FROM INFORMATION_SCHEMA.COLUMNS WHERE TABLE_NAME='{$table}'")->queryColumn();
array_walk($columns, create_function('&$str', '$str = "[$str]";'));
$model = new $modelClass;
$schema = $model->getTableSchema();
$columns = array_keys($schema->columns);
return $columns;
}