Support Create table LIKE (#1066)

* fixes #413

* add coverage

Co-authored-by: chyun <[email protected]>

Author: chyun

src/main/java/net/sf/jsqlparser/statement/create/table/CreateTable.java

@@ -29,6 +29,7 @@ public class CreateTable implements Statement {
     private List<ColumnDefinition> columnDefinitions;
     private List<Index> indexes;
     private Select select;
+    private Table likeTable;
     private boolean selectParenthesis;
     private boolean ifNotExists = false;
     private RowMovement rowMovement;
@@ -106,6 +107,15 @@ public void setSelect(Select select, boolean parenthesis) {
         this.selectParenthesis = parenthesis;
     }
 
+    public Table getLikeTable() {
+        return likeTable;
+    }
+
+    public void setLikeTable(Table likeTable, boolean parenthesis) {
+        this.likeTable = likeTable;
+        this.selectParenthesis = parenthesis;
+    }
+
     public boolean isIfNotExists() {
         return ifNotExists;
     }
@@ -160,6 +170,9 @@ public String toString() {
         if (select != null) {
             sql += " AS " + (selectParenthesis ? "(" : "") + select.toString() + (selectParenthesis ? ")" : "");
         }
+        if (likeTable != null) {
+            sql += " LIKE " + (selectParenthesis ? "(" : "") + likeTable.toString() + (selectParenthesis ? ")" : "");
+        }
         return sql;
     }
 

src/main/java/net/sf/jsqlparser/util/deparser/CreateTableDeParser.java

@@ -11,6 +11,7 @@
 
 import java.util.Iterator;
 
+import net.sf.jsqlparser.schema.Table;
 import net.sf.jsqlparser.statement.create.table.ColumnDefinition;
 import net.sf.jsqlparser.statement.create.table.CreateTable;
 import net.sf.jsqlparser.statement.create.table.Index;
@@ -95,6 +96,17 @@ public void deParse(CreateTable createTable) {
                 buffer.append(")");
             }
         }
+        if (createTable.getLikeTable() != null) {
+            buffer.append(" LIKE ");
+            if (createTable.isSelectParenthesis()) {
+                buffer.append("(");
+            }
+            Table table = createTable.getLikeTable();
+            buffer.append(table.getFullyQualifiedName());
+            if (createTable.isSelectParenthesis()) {
+                buffer.append(")");
+            }
+        }
     }
 
 }

src/main/jjtree/net/sf/jsqlparser/parser/JSqlParserCC.jjt

@@ -4048,6 +4048,7 @@ CreateTable CreateTable():
     List<String> idxSpec = new ArrayList<String>();
     Table fkTable = null;
     Select select = null;
+    Table likeTable = null;
     CheckConstraint checkCs = null;
     ExcludeConstraint excludeC = null;
     RowMovement rowMovement = null;
@@ -4201,6 +4202,10 @@ CreateTable CreateTable():
     <K_AS> ( LOOKAHEAD("(" Select() ")") "(" select = Select() { createTable.setSelect(select, true); } ")"
              | select = Select() { createTable.setSelect(select, false); } )
     ]
+    [
+    <K_LIKE> ( LOOKAHEAD("(" Table() ")") "(" likeTable=Table() { createTable.setLikeTable(likeTable, true); } ")"
+                 | likeTable=Table() { createTable.setLikeTable(likeTable, false); } )
+    ]
     {
         createTable.setTable(table);
         if (indexes.size() > 0)

src/test/java/net/sf/jsqlparser/statement/create/CreateTableTest.java

@@ -664,4 +664,16 @@ public void tableMovementWithAS() throws JSQLParserException {
         String sql = "CREATE TABLE test (startdate DATE) DISABLE ROW MOVEMENT AS SELECT 1 FROM dual";
         assertSqlCanBeParsedAndDeparsed(sql);
     }
+
+    @Test
+    public void testCreateTableWithCommentIssue413() throws JSQLParserException {
+        String statement = "CREATE TABLE a LIKE b";
+        assertSqlCanBeParsedAndDeparsed(statement);
+    }
+
+    @Test
+    public void testCreateTableWithCommentIssue413_2() throws JSQLParserException {
+        String statement = "CREATE TABLE a LIKE (b)";
+        assertSqlCanBeParsedAndDeparsed(statement);
+    }
 }