diff options
| -rw-r--r-- | src/Statements/CreateStatement.php | 51 | ||||
| -rw-r--r-- | tests/Builder/CreateStatementTest.php | 8 | ||||
| -rw-r--r-- | tests/Parser/CreateStatementTest.php | 2 | ||||
| -rw-r--r-- | tests/data/parser/parseCreateTableErr3.in | 1 | ||||
| -rw-r--r-- | tests/data/parser/parseCreateTableErr3.out | 1 | ||||
| -rw-r--r-- | tests/data/parser/parseCreateTableLike.in | 1 | ||||
| -rw-r--r-- | tests/data/parser/parseCreateTableLike.out | 1 |
7 files changed, 59 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)) { diff --git a/tests/Builder/CreateStatementTest.php b/tests/Builder/CreateStatementTest.php index cfdad92..f25a36c 100644 --- a/tests/Builder/CreateStatementTest.php +++ b/tests/Builder/CreateStatementTest.php @@ -80,6 +80,7 @@ class CreateStatementTest extends TestCase public function testBuilderTable() { + /* Assertion 1 */ $stmt = new CreateStatement(); $stmt->name = new Expression('', 'test', ''); @@ -105,6 +106,7 @@ class CreateStatementTest extends TestCase $stmt->build() ); + /* Assertion 2 */ $query = "CREATE TABLE `jos_core_acl_aro` (\n" . " `id` int(11) NOT NULL,\n" . @@ -119,6 +121,12 @@ class CreateStatementTest extends TestCase ") ENGINE=InnoDB DEFAULT CHARSET=latin1"; $parser = new Parser($query); $this->assertEquals($query, $parser->statements[0]->build()); + + /* Assertion 3 */ + $query = "CREATE TABLE `table_copy` LIKE `old_table`"; + $parser = new Parser($query); + $this->assertEquals($query, $parser->statements[0]->build()); + } public function testBuilderPartitions() diff --git a/tests/Parser/CreateStatementTest.php b/tests/Parser/CreateStatementTest.php index 2dcc9a9..758e5c4 100644 --- a/tests/Parser/CreateStatementTest.php +++ b/tests/Parser/CreateStatementTest.php @@ -29,8 +29,10 @@ class CreateStatementTest extends TestCase array('parser/parseCreateTable4'), array('parser/parseCreateTableErr1'), array('parser/parseCreateTableErr2'), + array('parser/parseCreateTableErr3'), array('parser/parseCreateTableSelect'), array('parser/parseCreateTableAsSelect'), + array('parser/parseCreateTableLike'), array('parser/parseCreateTrigger'), array('parser/parseCreateUser'), array('parser/parseCreateView'), diff --git a/tests/data/parser/parseCreateTableErr3.in b/tests/data/parser/parseCreateTableErr3.in new file mode 100644 index 0000000..2596969 --- /dev/null +++ b/tests/data/parser/parseCreateTableErr3.in @@ -0,0 +1 @@ +CREATE TABLE `table_copy` LIKE ;
\ No newline at end of file diff --git a/tests/data/parser/parseCreateTableErr3.out b/tests/data/parser/parseCreateTableErr3.out new file mode 100644 index 0000000..125faee --- /dev/null +++ b/tests/data/parser/parseCreateTableErr3.out @@ -0,0 +1 @@ +a:4:{s:5:"query";s:32:"CREATE TABLE `table_copy` LIKE ;";s:5:"lexer";O:15:"SqlParser\Lexer":8:{s:6:"strict";b:0;s:3:"str";s:32:"CREATE TABLE `table_copy` LIKE ;";s:3:"len";i:32;s:4:"last";i:32;s:4:"list";O:20:"SqlParser\TokensList":3:{s:6:"tokens";a:10:{i:0;O:15:"SqlParser\Token":5:{s:5:"token";s:6:"CREATE";s:5:"value";s:6:"CREATE";s:4:"type";i:1;s:5:"flags";i:3;s:8:"position";i:0;}i:1;O:15:"SqlParser\Token":5:{s:5:"token";s:1:" ";s:5:"value";s:1:" ";s:4:"type";i:3;s:5:"flags";i:0;s:8:"position";i:6;}i:2;O:15:"SqlParser\Token":5:{s:5:"token";s:5:"TABLE";s:5:"value";s:5:"TABLE";s:4:"type";i:1;s:5:"flags";i:3;s:8:"position";i:7;}i:3;O:15:"SqlParser\Token":5:{s:5:"token";s:1:" ";s:5:"value";s:1:" ";s:4:"type";i:3;s:5:"flags";i:0;s:8:"position";i:12;}i:4;O:15:"SqlParser\Token":5:{s:5:"token";s:12:"`table_copy`";s:5:"value";s:10:"table_copy";s:4:"type";i:8;s:5:"flags";i:2;s:8:"position";i:13;}i:5;O:15:"SqlParser\Token":5:{s:5:"token";s:1:" ";s:5:"value";s:1:" ";s:4:"type";i:3;s:5:"flags";i:0;s:8:"position";i:25;}i:6;O:15:"SqlParser\Token":5:{s:5:"token";s:4:"LIKE";s:5:"value";s:4:"LIKE";s:4:"type";i:1;s:5:"flags";i:3;s:8:"position";i:26;}i:7;O:15:"SqlParser\Token":5:{s:5:"token";s:1:" ";s:5:"value";s:1:" ";s:4:"type";i:3;s:5:"flags";i:0;s:8:"position";i:30;}i:8;O:15:"SqlParser\Token":5:{s:5:"token";s:1:";";s:5:"value";s:1:";";s:4:"type";i:9;s:5:"flags";i:0;s:8:"position";i:31;}i:9;O:15:"SqlParser\Token":5:{s:5:"token";N;s:5:"value";N;s:4:"type";i:9;s:5:"flags";i:0;s:8:"position";N;}}s:5:"count";i:10;s:3:"idx";i:10;}s:9:"delimiter";s:1:";";s:12:"delimiterLen";i:1;s:6:"errors";a:0:{}}s:6:"parser";O:16:"SqlParser\Parser":5:{s:4:"list";r:8;s:6:"strict";b:0;s:6:"errors";a:0:{}s:10:"statements";a:1:{i:0;O:36:"SqlParser\Statements\CreateStatement":17:{s:4:"name";O:31:"SqlParser\Components\Expression":7:{s:8:"database";N;s:5:"table";s:10:"table_copy";s:6:"column";N;s:4:"expr";s:12:"`table_copy`";s:5:"alias";N;s:8:"function";N;s:8:"subquery";N;}s:13:"entityOptions";N;s:6:"fields";N;s:6:"select";N;s:4:"like";N;s:11:"partitionBy";N;s:13:"partitionsNum";N;s:14:"subpartitionBy";N;s:16:"subpartitionsNum";N;s:10:"partitions";N;s:5:"table";N;s:6:"return";N;s:10:"parameters";N;s:4:"body";a:0:{}s:7:"options";O:33:"SqlParser\Components\OptionsArray":1:{s:7:"options";a:1:{i:6;s:5:"TABLE";}}s:5:"first";i:0;s:4:"last";i:8;}}s:8:"brackets";i:0;}s:6:"errors";a:2:{s:5:"lexer";a:0:{}s:6:"parser";a:1:{i:0;a:3:{i:0;s:26:"A table name was expected.";i:1;r:58;i:2;i:0;}}}}
\ No newline at end of file diff --git a/tests/data/parser/parseCreateTableLike.in b/tests/data/parser/parseCreateTableLike.in new file mode 100644 index 0000000..07528ed --- /dev/null +++ b/tests/data/parser/parseCreateTableLike.in @@ -0,0 +1 @@ +CREATE TABLE `table_copy` LIKE `old_table`;
\ No newline at end of file diff --git a/tests/data/parser/parseCreateTableLike.out b/tests/data/parser/parseCreateTableLike.out new file mode 100644 index 0000000..963c5aa --- /dev/null +++ b/tests/data/parser/parseCreateTableLike.out @@ -0,0 +1 @@ +a:4:{s:5:"query";s:43:"CREATE TABLE `table_copy` LIKE `old_table`;";s:5:"lexer";O:15:"SqlParser\Lexer":8:{s:6:"strict";b:0;s:3:"str";s:43:"CREATE TABLE `table_copy` LIKE `old_table`;";s:3:"len";i:43;s:4:"last";i:43;s:4:"list";O:20:"SqlParser\TokensList":3:{s:6:"tokens";a:11:{i:0;O:15:"SqlParser\Token":5:{s:5:"token";s:6:"CREATE";s:5:"value";s:6:"CREATE";s:4:"type";i:1;s:5:"flags";i:3;s:8:"position";i:0;}i:1;O:15:"SqlParser\Token":5:{s:5:"token";s:1:" ";s:5:"value";s:1:" ";s:4:"type";i:3;s:5:"flags";i:0;s:8:"position";i:6;}i:2;O:15:"SqlParser\Token":5:{s:5:"token";s:5:"TABLE";s:5:"value";s:5:"TABLE";s:4:"type";i:1;s:5:"flags";i:3;s:8:"position";i:7;}i:3;O:15:"SqlParser\Token":5:{s:5:"token";s:1:" ";s:5:"value";s:1:" ";s:4:"type";i:3;s:5:"flags";i:0;s:8:"position";i:12;}i:4;O:15:"SqlParser\Token":5:{s:5:"token";s:12:"`table_copy`";s:5:"value";s:10:"table_copy";s:4:"type";i:8;s:5:"flags";i:2;s:8:"position";i:13;}i:5;O:15:"SqlParser\Token":5:{s:5:"token";s:1:" ";s:5:"value";s:1:" ";s:4:"type";i:3;s:5:"flags";i:0;s:8:"position";i:25;}i:6;O:15:"SqlParser\Token":5:{s:5:"token";s:4:"LIKE";s:5:"value";s:4:"LIKE";s:4:"type";i:1;s:5:"flags";i:3;s:8:"position";i:26;}i:7;O:15:"SqlParser\Token":5:{s:5:"token";s:1:" ";s:5:"value";s:1:" ";s:4:"type";i:3;s:5:"flags";i:0;s:8:"position";i:30;}i:8;O:15:"SqlParser\Token":5:{s:5:"token";s:11:"`old_table`";s:5:"value";s:9:"old_table";s:4:"type";i:8;s:5:"flags";i:2;s:8:"position";i:31;}i:9;O:15:"SqlParser\Token":5:{s:5:"token";s:1:";";s:5:"value";s:1:";";s:4:"type";i:9;s:5:"flags";i:0;s:8:"position";i:42;}i:10;O:15:"SqlParser\Token":5:{s:5:"token";N;s:5:"value";N;s:4:"type";i:9;s:5:"flags";i:0;s:8:"position";N;}}s:5:"count";i:11;s:3:"idx";i:11;}s:9:"delimiter";s:1:";";s:12:"delimiterLen";i:1;s:6:"errors";a:0:{}}s:6:"parser";O:16:"SqlParser\Parser":5:{s:4:"list";r:8;s:6:"strict";b:0;s:6:"errors";a:0:{}s:10:"statements";a:1:{i:0;O:36:"SqlParser\Statements\CreateStatement":17:{s:4:"name";O:31:"SqlParser\Components\Expression":7:{s:8:"database";N;s:5:"table";s:10:"table_copy";s:6:"column";N;s:4:"expr";s:12:"`table_copy`";s:5:"alias";N;s:8:"function";N;s:8:"subquery";N;}s:13:"entityOptions";N;s:6:"fields";N;s:6:"select";N;s:4:"like";O:31:"SqlParser\Components\Expression":7:{s:8:"database";N;s:5:"table";s:9:"old_table";s:6:"column";N;s:4:"expr";s:11:"`old_table`";s:5:"alias";N;s:8:"function";N;s:8:"subquery";N;}s:11:"partitionBy";N;s:13:"partitionsNum";N;s:14:"subpartitionBy";N;s:16:"subpartitionsNum";N;s:10:"partitions";N;s:5:"table";r:98;s:6:"return";N;s:10:"parameters";N;s:4:"body";a:0:{}s:7:"options";O:33:"SqlParser\Components\OptionsArray":1:{s:7:"options";a:1:{i:6;s:5:"TABLE";}}s:5:"first";i:0;s:4:"last";i:8;}}s:8:"brackets";i:0;}s:6:"errors";a:2:{s:5:"lexer";a:0:{}s:6:"parser";a:0:{}}}
\ No newline at end of file |