Added better support for aliases column to Query class.
This commit is contained in:
parent
4ace7c8f80
commit
c480e19b6c
|
@ -120,8 +120,41 @@ public function distinct()
|
||||||
public function select()
|
public function select()
|
||||||
{
|
{
|
||||||
$this->select = ($this->distinct) ? 'SELECT DISTINCT ' : 'SELECT ';
|
$this->select = ($this->distinct) ? 'SELECT DISTINCT ' : 'SELECT ';
|
||||||
$this->select .= implode(', ', array_map(array($this, 'wrap'), func_get_args()));
|
|
||||||
|
|
||||||
|
$columns = array();
|
||||||
|
|
||||||
|
foreach (func_get_args() as $column)
|
||||||
|
{
|
||||||
|
// ---------------------------------------------------------
|
||||||
|
// If the column name is being aliases, we will need to
|
||||||
|
// wrap the column name and its alias.
|
||||||
|
// ---------------------------------------------------------
|
||||||
|
if (strpos(strtolower($column), ' as ') !== false)
|
||||||
|
{
|
||||||
|
$segments = explode(' ', $column);
|
||||||
|
|
||||||
|
$columns[] = $this->wrap($segments[0]).' AS '.$this->wrap($segments[2]);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
$columns[] = $this->wrap($column);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
$this->select .= implode(', ', $columns);
|
||||||
|
|
||||||
|
return $this;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Set the FROM clause.
|
||||||
|
*
|
||||||
|
* @param string $from
|
||||||
|
* @return Query
|
||||||
|
*/
|
||||||
|
public function from($from)
|
||||||
|
{
|
||||||
|
$this->from = $from;
|
||||||
return $this;
|
return $this;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue