index.php

<?php
namespace Bitrix\Perfmon\Sql;
 
use Bitrix\Main\NotSupportedException;
 
class Index extends BaseObject
{
    public $unique = false;
    public $fulltext = false;
    public $using = '';
    public $columns = [];
 
    public function __construct($name = '', $unique, $fulltext=false)
    {
        parent::__construct($name);
        $this->unique = (bool)$unique;
        $this->fulltext = (bool)$fulltext;
    }
 
    public function addColumn($name)
    {
        $this->columns[] = trim($name);
        $this->setBody(implode(', ', $this->columns));
        return $this;
    }
 
    public static function create(Tokenizer $tokenizer, $unique = false, $fulltext = false, $indexName = '')
    {
        if (!$indexName)
        {
            if ($tokenizer->getCurrentToken()->text !== '(')
            {
                $indexName = $tokenizer->getCurrentToken()->text;
                $tokenizer->nextToken();
                $tokenizer->skipWhiteSpace();
            }
        }
 
        if ($tokenizer->testUpperText('ON'))
        {
            $tokenizer->skipWhiteSpace();
            $tokenizer->nextToken();
            $tokenizer->skipWhiteSpace();
        }
 
        if ($tokenizer->testUpperText('USING'))
        {
            $tokenizer->skipWhiteSpace();
            $indexType = $tokenizer->getCurrentToken()->text;
            if (strtoupper($indexType) !== 'GIN')
            {
                throw new NotSupportedException("'GIN' expected. line:" . $tokenizer->getCurrentToken()->line);
            }
            $fulltext = true;
            $tokenizer->nextToken();
            $tokenizer->skipWhiteSpace();
        }
 
        $index = new self($indexName, $unique, $fulltext);
 
        if ($tokenizer->testText('('))
        {
            $tokenizer->skipWhiteSpace();
            $token = $tokenizer->getCurrentToken();
            $level = $token->level;
            $column = '';
            do
            {
                if ($token->text === ',')
                {
                    $index->addColumn($column);
                    $column = '';
                }
                else
                {
                    $column .= $token->text;
                }
                $token = $tokenizer->nextToken();
            }
            while (!$tokenizer->endOfInput() && $token->level >= $level);
 
            if ($column)
            {
                $index->addColumn($column);
            }
 
            if (!$tokenizer->testText(')'))
            {
                throw new NotSupportedException("')' expected. line:" . $tokenizer->getCurrentToken()->line);
            }
 
            //USING BTREE
            $tokenizer->skipWhiteSpace();
            if ($tokenizer->testText('USING'))
            {
                $tokenizer->skipWhiteSpace();
                $token = $tokenizer->nextToken();
                $index->using = $token->text;
            }
        }
        else
        {
            throw new NotSupportedException("'(' expected. line:" . $tokenizer->getCurrentToken()->line);
        }
 
        return $index;
    }
 
    public static function searchTableName(Tokenizer $tokenizer)
    {
        $lineToken = $tokenizer->getCurrentToken();
        while (!$tokenizer->endOfInput())
        {
            if ($tokenizer->getCurrentToken()->upper === 'ON')
            {
                $tokenizer->nextToken();
                $tokenizer->skipWhiteSpace();
                return;
            }
            $tokenizer->nextToken();
        }
        throw new NotSupportedException('Index: table name not found. line: ' . $lineToken->line);
    }
 
    public function getCreateDdl($dbType = '')
    {
        switch ($dbType)
        {
        case 'MYSQL':
            return 'CREATE ' . ($this->fulltext ? 'FULLTEXT ' : '') . ($this->unique ? 'UNIQUE ' : '') . 'INDEX ' . $this->name . ' ON ' . $this->parent->name . '(' . $this->body . ')';
        case 'PGSQL':
            return 'CREATE ' . ($this->unique ? 'UNIQUE ' : '') . 'INDEX ' . $this->name . ' ON ' . $this->parent->name . ($this->fulltext ? ' USING GIN (' . $this->body . ')' : '(' . $this->body . ')');
        default:
            return '// ' . get_class($this) . ':getDropDdl for database type [' . $dbType . '] not implemented';
        }
    }
 
    public function getDropDdl($dbType = '')
    {
        switch ($dbType)
        {
        case 'MYSQL':
        case 'MSSQL':
            return 'DROP INDEX ' . $this->name . ' ON ' . $this->parent->name;
        case 'ORACLE':
        case 'PGSQL':
            return 'DROP INDEX ' . $this->name;
        default:
            return '// ' . get_class($this) . ':getDropDdl for database type [' . $dbType . '] not implemented';
        }
    }
 
    public function getModifyDdl(BaseObject $target, $dbType = '')
    {
        return [
            $this->getDropDdl($dbType),
            $target->getCreateDdl($dbType),
        ];
    }
}