TemporaryTableService.php

Go to the documentation of this file.

<?php
namespace Magento\Framework\DB;

use Magento\Framework\DB\Adapter\AdapterInterface;

class TemporaryTableService
{
    const INDEX_METHOD_HASH = 'HASH';
    const DB_ENGINE_INNODB = 'INNODB';

    private $allowedIndexMethods;

    private $allowedEngines;

    private $random;

    private $createdTableAdapters = [];

    public function __construct(
        \Magento\Framework\Math\Random $random,
        $allowedIndexMethods = [],
        $allowedEngines = []
    ) {
        $this->random = $random;
        $this->allowedIndexMethods = $allowedIndexMethods;
        $this->allowedEngines = $allowedEngines;
    }

    public function createFromSelect(
        Select $select,
        AdapterInterface $adapter,
        array $indexes = [],
        $indexMethod = self::INDEX_METHOD_HASH,
        $dbEngine = self::DB_ENGINE_INNODB
    ) {
        if (!in_array($indexMethod, $this->allowedIndexMethods)) {
            throw new \InvalidArgumentException(
                sprintf('indexMethod must be one of %s', implode(',', $this->allowedIndexMethods))
            );
        }

        if (!in_array($dbEngine, $this->allowedEngines)) {
            throw new \InvalidArgumentException(
                sprintf('dbEngine must be one of %s', implode(',', $this->allowedEngines))
            );
        }

        $name = $this->random->getUniqueHash('tmp_select_');

        $indexStatements = [];
        foreach ($indexes as $indexName => $columns) {
            $renderedColumns = implode(',', array_map([$adapter, 'quoteIdentifier'], $columns));

            $indexType = sprintf(
                'INDEX %s USING %s',
                $adapter->quoteIdentifier($indexName),
                $indexMethod
            );

            if ($indexName === 'PRIMARY') {
                $indexType = 'PRIMARY KEY';
            } elseif (strpos($indexName, 'UNQ_') === 0) {
                $indexType = sprintf('UNIQUE %s', $adapter->quoteIdentifier($indexName));
            }

            $indexStatements[] = sprintf('%s(%s)', $indexType, $renderedColumns);
        }

        $statement = sprintf(
            'CREATE TEMPORARY TABLE %s %s ENGINE=%s IGNORE (%s)',
            $adapter->quoteIdentifier($name),
            $indexStatements ? '(' . implode(',', $indexStatements) . ')' : '',
            $adapter->quoteIdentifier($dbEngine),
            "{$select}"
        );

        $adapter->query(
            $statement,
            $select->getBind()
        );

        $this->createdTableAdapters[$name] = $adapter;

        return $name;
    }

    public function dropTable($name)
    {
        if (!empty($this->createdTableAdapters)) {
            if (isset($this->createdTableAdapters[$name]) && !empty($name)) {
                $adapter = $this->createdTableAdapters[$name];
                $adapter->dropTemporaryTable($name);
                unset($this->createdTableAdapters[$name]);
                return true;
            }
        }
        return false;
    }
}