constraint.php

<?php
namespace Bitrix\Perfmon\Sql;
 
class Constraint extends BaseObject
{
    public $columns = [];
 
    public static function create(Tokenizer $tokenizer, $constraintName = '')
    {
        if ($constraintName === false)
        {
            $constraintName = '';
        }
        elseif (!$constraintName)
        {
            $constraintName = $tokenizer->getCurrentToken()->text;
            $tokenizer->nextToken();
            $tokenizer->skipWhiteSpace();
        }
 
        $constraint = new self($constraintName);
 
        $token = $tokenizer->getCurrentToken();
        $level = $token->level;
        $column = '';
        $constraintDefinition = '';
        do
        {
            if ($token->text === '(' && $token->level == $level)
            {
                $column = '';
            }
            elseif (
                $token->text === ','
                || ($token->text === ')' && $token->level == $level)
            )
            {
                $column = trim($column);
                if ($column)
                {
                    $constraint->columns[] = trim($column);
                    $column = '';
                }
            }
            else
            {
                $column .= $token->text;
            }
 
            if ($token->level == $level && $token->text === ',')
            {
                break;
            }
            if ($token->level < $level && $token->text === ')')
            {
                break;
            }
 
            $constraintDefinition .= $token->text;
 
            $token = $tokenizer->nextToken();
        }
        while (!$tokenizer->endOfInput());
 
        $constraint->setBody($constraintDefinition);
 
        return $constraint;
    }
 
    public function getCreateDdl($dbType = '')
    {
        switch ($dbType)
        {
        case 'MYSQL':
        case 'MSSQL':
        case 'ORACLE':
            return 'ALTER TABLE ' . $this->parent->name . ' ADD CONSTRAINT ' . $this->name . ' ' . $this->body;
        default:
            return '// ' . get_class($this) . ':getCreateDdl for database type [' . $dbType . '] not implemented';
        }
    }
 
    public function getDropDdl($dbType = '')
    {
        switch ($dbType)
        {
        case 'MSSQL':
        case 'ORACLE':
            return 'ALTER TABLE ' . $this->parent->name . ' DROP CONSTRAINT ' . $this->name;
        case 'MYSQL':
        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),
        ];
    }
}