diff options
Diffstat (limited to 'src/Statements/CreateStatement.php')
-rw-r--r-- | src/Statements/CreateStatement.php | 51 |
1 files changed, 45 insertions, 6 deletions
diff --git a/src/Statements/CreateStatement.php b/src/Statements/CreateStatement.php index 3ccf78f..38dde32 100644 --- a/src/Statements/CreateStatement.php +++ b/src/Statements/CreateStatement.php @@ -11,6 +11,7 @@ namespace SqlParser\Statements; use SqlParser\Parser; use SqlParser\Statement; use SqlParser\Token; +use SqlParser\Context; use SqlParser\TokensList; use SqlParser\Components\ArrayObj; use SqlParser\Components\DataType; @@ -188,6 +189,15 @@ class CreateStatement extends Statement public $select; /** + * If `CREATE TABLE ... LIKE` + * + * Used by `CREATE TABLE` + * + * @var Expression + */ + public $like; + + /** * Expression used for partitioning. * * @var string @@ -282,6 +292,11 @@ class CreateStatement extends Statement . OptionsArray::build($this->options) . ' ' . Expression::build($this->name) . ' ' . $this->select->build(); + } elseif ($this->options->has('TABLE') && !is_null($this->like)) { + return 'CREATE ' + . OptionsArray::build($this->options) . ' ' + . Expression::build($this->name) . ' LIKE ' + . Expression::build($this->like); } elseif ($this->options->has('TABLE')) { $partition = ''; @@ -389,17 +404,41 @@ class CreateStatement extends Statement $list, static::$DB_OPTIONS ); - } elseif ($this->options->has('TABLE') && ($token->type == Token::TYPE_KEYWORD) && ($token->value == 'SELECT')) { + } elseif ($this->options->has('TABLE') + && ($token->type == Token::TYPE_KEYWORD) + && ($token->value == 'SELECT') + ) { /* CREATE TABLE ... SELECT */ $this->select = new SelectStatement($parser, $list); - } elseif ( - $this->options->has('TABLE') && - ($token->type == Token::TYPE_KEYWORD) && ($token->value == 'AS') && - ($list->tokens[$nextidx]->type == Token::TYPE_KEYWORD) && ($list->tokens[$nextidx]->value == 'SELECT') - ) { + } elseif ($this->options->has('TABLE') + && ($token->type == Token::TYPE_KEYWORD) && ($token->value == 'AS') + && ($list->tokens[$nextidx]->type == Token::TYPE_KEYWORD) + && ($list->tokens[$nextidx]->value == 'SELECT') + ) { /* CREATE TABLE ... AS SELECT */ $list->idx = $nextidx; $this->select = new SelectStatement($parser, $list); + } elseif ($this->options->has('TABLE') + && $token->type == Token::TYPE_KEYWORD + && $token->value == 'LIKE' + ) { + /* CREATE TABLE `new_tbl` LIKE 'orig_tbl' */ + $list->idx = $nextidx; + $this->like = $this->table = Expression::parse( + $parser, + $list, + array( + 'parseField' => 'table', + 'breakOnAlias' => true, + ) + ); + // The 'LIKE' keyword was found, but no table_name was found next to it + if ($this->like == null) { + $parser->error( + __('A table name was expected.'), + $list->tokens[$list->idx] + ); + } } elseif ($this->options->has('TABLE')) { $this->fields = CreateDefinition::parse($parser, $list); if (empty($this->fields)) { |