tyler.durden/icingaweb2
1
0
Fork 0
mirror of https://github.com/Icinga/icingaweb2.git synced 2025-07-24 06:14:25 +02:00
Code Issues Packages Projects Releases Wiki Activity

RepositoryQuery: Properly handle custom aliases when applying value conversion rules

Browse Source
This commit is contained in:
Johannes Meyer 2015-11-03 14:35:06 +01:00
parent e170d5b79a
commit 477af43a2f
1 changed files with 62 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

73
library/Icinga/Repository/RepositoryQuery.php
View File

@ -47,6 +47,13 @@ class RepositoryQuery implements QueryInterface, SortRules, FilterColumns, Itera
*/ */
protected $iterator; protected $iterator;
/**
* This query's custom aliases
*
* @var array
*/
protected $customAliases;
/** /**
* Create a new repository query * Create a new repository query
* *
@ -124,6 +131,7 @@ class RepositoryQuery implements QueryInterface, SortRules, FilterColumns, Itera
*/ */
protected function prepareQueryColumns($target, array $desiredColumns = null) protected function prepareQueryColumns($target, array $desiredColumns = null)
{ {
$this->customAliases = array();
if (empty($desiredColumns)) { if (empty($desiredColumns)) {
$columns = $this->repository->requireAllQueryColumns($target); $columns = $this->repository->requireAllQueryColumns($target);
} else { } else {
@ -131,9 +139,15 @@ class RepositoryQuery implements QueryInterface, SortRules, FilterColumns, Itera
foreach ($desiredColumns as $customAlias => $columnAlias) { foreach ($desiredColumns as $customAlias => $columnAlias) {
$resolvedColumn = $this->repository->requireQueryColumn($target, $columnAlias, $this); $resolvedColumn = $this->repository->requireQueryColumn($target, $columnAlias, $this);
if ($resolvedColumn !== $columnAlias) { if ($resolvedColumn !== $columnAlias) {
$columns[is_string($customAlias) ? $customAlias : $columnAlias] = $resolvedColumn; if (is_string($customAlias)) {
$columns[$customAlias] = $resolvedColumn;
$this->customAliases[$customAlias] = $columnAlias;
} else {
$columns[$columnAlias] = $resolvedColumn;
}
} elseif (is_string($customAlias)) { } elseif (is_string($customAlias)) {
$columns[$customAlias] = $columnAlias; $columns[$customAlias] = $columnAlias;
$this->customAliases[$customAlias] = $columnAlias;
} else { } else {
$columns[] = $columnAlias; $columns[] = $columnAlias;
} }
@ -143,6 +157,24 @@ class RepositoryQuery implements QueryInterface, SortRules, FilterColumns, Itera
return $columns; return $columns;
} }
/**
* Return the native column alias for the given custom alias
*
* If no custom alias is found with the given name, it is returned unchanged.
*
* @param string $customAlias
*
* @return string
*/
protected function getNativeAlias($customAlias)
{
if (isset($this->customAliases[$customAlias])) {
return $this->customAliases[$customAlias];
}
return $customAlias;
}
/** /**
* Return this query's available filter columns with their optional label as key * Return this query's available filter columns with their optional label as key
* *
@ -467,7 +499,7 @@ class RepositoryQuery implements QueryInterface, SortRules, FilterColumns, Itera
$result = $this->query->fetchOne(); $result = $this->query->fetchOne();
if ($result !== false && $this->repository->providesValueConversion($this->target)) { if ($result !== false && $this->repository->providesValueConversion($this->target)) {
$columns = $this->getColumns(); $columns = $this->getColumns();
$column = isset($columns[0]) ? $columns[0] : key($columns); $column = isset($columns[0]) ? $columns[0] : $this->getNativeAlias(key($columns));
return $this->repository->retrieveColumn($this->target, $column, $result, $this); return $this->repository->retrieveColumn($this->target, $column, $result, $this);
} }
@ -492,7 +524,12 @@ class RepositoryQuery implements QueryInterface, SortRules, FilterColumns, Itera
$alias = $column; $alias = $column;
} }
$result->$alias = $this->repository->retrieveColumn($this->target, $alias, $result->$alias, $this); $result->$alias = $this->repository->retrieveColumn(
$this->target,
$this->getNativeAlias($alias),
$result->$alias,
$this
);
} }
} }
@ -514,7 +551,7 @@ class RepositoryQuery implements QueryInterface, SortRules, FilterColumns, Itera
if (! empty($results) && $this->repository->providesValueConversion($this->target)) { if (! empty($results) && $this->repository->providesValueConversion($this->target)) {
$columns = $this->getColumns(); $columns = $this->getColumns();
$aliases = array_keys($columns); $aliases = array_keys($columns);
$column = is_int($aliases[0]) ? $columns[0] : $aliases[0]; $column = is_int($aliases[0]) ? $columns[0] : $this->getNativeAlias($aliases[0]);
if ($this->repository->providesValueConversion($this->target, $column)) { if ($this->repository->providesValueConversion($this->target, $column)) {
foreach ($results as & $value) { foreach ($results as & $value) {
$value = $this->repository->retrieveColumn($this->target, $column, $value, $this); $value = $this->repository->retrieveColumn($this->target, $column, $value, $this);
@ -542,8 +579,11 @@ class RepositoryQuery implements QueryInterface, SortRules, FilterColumns, Itera
if (! empty($results) && $this->repository->providesValueConversion($this->target)) { if (! empty($results) && $this->repository->providesValueConversion($this->target)) {
$columns = $this->getColumns(); $columns = $this->getColumns();
$aliases = array_keys($columns); $aliases = array_keys($columns);
$colOne = $aliases[0] !== 0 ? $aliases[0] : $columns[0]; $colOne = $aliases[0] !== 0 ? $this->getNativeAlias($aliases[0]) : $columns[0];
$colTwo = count($aliases) < 2 ? $colOne : ($aliases[1] !== 1 ? $aliases[1] : $columns[1]); $colTwo = count($aliases) < 2 ? $colOne : (
$aliases[1] !== 1 ? $this->getNativeAlias($aliases[1]) : $columns[1]
);
if ( if (
$this->repository->providesValueConversion($this->target, $colOne) $this->repository->providesValueConversion($this->target, $colOne)
|| $this->repository->providesValueConversion($this->target, $colTwo) || $this->repository->providesValueConversion($this->target, $colTwo)
@ -588,21 +628,27 @@ class RepositoryQuery implements QueryInterface, SortRules, FilterColumns, Itera
$alias = $column; $alias = $column;
} }
$row->$alias = $this->repository->retrieveColumn($this->target, $alias, $row->$alias, $this); $row->$alias = $this->repository->retrieveColumn(
$this->target,
$this->getNativeAlias($alias),
$row->$alias,
$this
);
} }
foreach (($this->getOrder() ?: array()) as $rule) { foreach (($this->getOrder() ?: array()) as $rule) {
$nativeAlias = $this->getNativeAlias($rule[0]);
if (! array_key_exists($rule[0], $flippedColumns) && property_exists($row, $rule[0])) { if (! array_key_exists($rule[0], $flippedColumns) && property_exists($row, $rule[0])) {
if ($this->repository->providesValueConversion($this->target, $rule[0])) { if ($this->repository->providesValueConversion($this->target, $nativeAlias)) {
$updateOrder = true; $updateOrder = true;
$row->{$rule[0]} = $this->repository->retrieveColumn( $row->{$rule[0]} = $this->repository->retrieveColumn(
$this->target, $this->target,
$rule[0], $nativeAlias,
$row->{$rule[0]} $row->{$rule[0]}
); );
} }
} elseif (array_key_exists($rule[0], $flippedColumns)) { } elseif (array_key_exists($rule[0], $flippedColumns)) {
if ($this->repository->providesValueConversion($this->target, $rule[0])) { if ($this->repository->providesValueConversion($this->target, $nativeAlias)) {
$updateOrder = true; $updateOrder = true;
} }
} }
@ -678,7 +724,12 @@ class RepositoryQuery implements QueryInterface, SortRules, FilterColumns, Itera
$alias = $column; $alias = $column;
} }
$row->$alias = $this->repository->retrieveColumn($this->target, $alias, $row->$alias, $this); $row->$alias = $this->repository->retrieveColumn(
$this->target,
$this->getNativeAlias($alias),
$row->$alias,
$this
);
} }
} }