SelectHydrator.php

Go to the documentation of this file.

<?php
namespace Magento\Analytics\ReportXml;

use Magento\Framework\App\ResourceConnection;
use Magento\Framework\DB\Select;
use Magento\Framework\ObjectManagerInterface;

class SelectHydrator
{
    private $predefinedSelectParts =
        [
            Select::DISTINCT,
            Select::COLUMNS,
            Select::UNION,
            Select::FROM,
            Select::WHERE,
            Select::GROUP,
            Select::HAVING,
            Select::ORDER,
            Select::LIMIT_COUNT,
            Select::LIMIT_OFFSET,
            Select::FOR_UPDATE
        ];

    private $selectParts;

    private $resourceConnection;

    private $objectManager;

    public function __construct(
        ResourceConnection $resourceConnection,
        ObjectManagerInterface $objectManager,
        $selectParts = []
    ) {
        $this->resourceConnection = $resourceConnection;
        $this->objectManager = $objectManager;
        $this->selectParts = $selectParts;
    }

    private function getSelectParts()
    {
        return array_merge($this->predefinedSelectParts, $this->selectParts);
    }

    public function extract(Select $select)
    {
        $parts = [];
        foreach ($this->getSelectParts() as $partName) {
            $parts[$partName] = $select->getPart($partName);
        }
        return $parts;
    }

    public function recreate(array $selectParts)
    {
        $select = $this->resourceConnection->getConnection()->select();

        $select = $this->processColumns($select, $selectParts);

        foreach ($selectParts as $partName => $partValue) {
            $select->setPart($partName, $partValue);
        }

        return $select;
    }

    private function processColumns(Select $select, array &$selectParts)
    {
        if (!empty($selectParts[Select::COLUMNS]) && is_array($selectParts[Select::COLUMNS])) {
            $part = [];

            foreach ($selectParts[Select::COLUMNS] as $columnEntry) {
                list($correlationName, $column, $alias) = $columnEntry;
                if (is_array($column) && !empty($column['class'])) {
                    $expression = $this->objectManager->create(
                        $column['class'],
                        isset($column['arguments']) ? $column['arguments'] : []
                    );
                    $part[] = [$correlationName, $expression, $alias];
                } else {
                    $part[] = $columnEntry;
                }
            }

            $select->setPart(Select::COLUMNS, $part);
            unset($selectParts[Select::COLUMNS]);
        }

        return $select;
    }
}