fanbinit/rhymix
1
0
Fork 0
mirror of https://github.com/Lastorder-DC/rhymix.git synced 2026-01-07 02:31:40 +09:00
Code Issues Projects Releases Packages Wiki Activity

Replace all wildcard columns with '1' in count-only subquery #1575

Browse source
This commit is contained in:
Kijin Sung 2021-01-29 14:40:28 +09:00
parent 96eae68105
commit 957be16a25
4 changed files with 44 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

32
common/framework/parsers/dbquery/query.php
View file

@ -50,10 +50,10 @@ class Query extends VariableBase
* @param string $prefix
* @param array $args
* @param array $column_list
* @param bool $count_only
* @param int $count_only
* @return string
*/
public function getQueryString(string $prefix = '', array $args, array $column_list = [], bool $count_only = false): string
public function getQueryString(string $prefix = '', array $args, array $column_list = [], int $count_only = 0): string
{
// Save the query information.
$this->_prefix = $prefix;
@ -110,10 +110,10 @@ class Query extends VariableBase
/**
* Generate a SELECT query string.
*
* @param bool $count_only
* @param int $count_only
* @return string
*/
protected function _getSelectQueryString(bool $count_only = false): string
protected function _getSelectQueryString(int $count_only = 0): string
{
// Initialize the query string.
$result = 'SELECT';
@ -133,18 +133,23 @@ class Query extends VariableBase
{
if ($column instanceof self)
{
$subquery = $column->getQueryString($this->_prefix, $this->_args);
$has_subquery_columns = true;
$subquery_count_only = $count_only ? $count_only + 1 : 0;
$subquery = $column->getQueryString($this->_prefix, $this->_args, [], $subquery_count_only);
foreach ($column->getQueryParams() as $param)
{
$this->_params[] = $param;
}
$columns[] = sprintf('(%s) AS %s', $subquery, self::quoteName($column->alias));
$has_subquery_columns = true;
}
elseif ($column->is_expression && !$column->is_wildcard)
{
$columns[] = $column->name . ($column->alias ? (' AS ' . self::quoteName($column->alias)) : '');
}
elseif ($column->is_wildcard && $count_only >= 1 && !$this->select_distinct)
{
$columns[] = '1';
}
else
{
$columns[] = self::quoteName($column->name) . ($column->alias ? (' AS ' . self::quoteName($column->alias)) : '');
@ -154,19 +159,12 @@ class Query extends VariableBase
}
// Replace the column list if this is a count-only query.
if ($count_only)
if ($count_only == 1)
{
$count_wrap = ($this->groupby || $this->select_distinct || $has_subquery_columns || preg_match('/\bDISTINCT\b/i', $column_list));
if ($count_wrap)
{
if ($column_list === '*' || preg_match('/\\.\\*/', $column_list))
{
$result .= ' 1';
}
else
{
$result .= ($this->select_distinct ? ' DISTINCT ' : ' ') . $column_list;
}
$result .= ($this->select_distinct ? ' DISTINCT ' : ' ') . $column_list;
}
else
{
@ -234,13 +232,13 @@ class Query extends VariableBase
}
// Compose the ORDER BY clause.
if ($this->navigation && count($this->navigation->orderby) && !$count_only)
if ($this->navigation && count($this->navigation->orderby) && ($count_only != 1 || $count_wrap))
{
$result .= ' ORDER BY ' . $this->_arrangeOrderBy($this->navigation);
}
// Compose the LIMIT/OFFSET clause.
if ($this->navigation && $this->navigation->list_count && !$count_only)
if ($this->navigation && $this->navigation->list_count && ($count_only != 1 || $count_wrap))
{
$result .= ' LIMIT ' . $this->_arrangeLimitOffset($this->navigation);
}