alter/drop add if exist flag

MyCATApache · May 12, 2022 · 17d839f · 17d839f
1 parent cd20926
commit 17d839f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/mycat2/src/main/java/io/mycat/sqlhandler/ddl/AlterTableSQLHandler.java b/mycat2/src/main/java/io/mycat/sqlhandler/ddl/AlterTableSQLHandler.java
@@ -53,6 +53,7 @@ protected Future<Void> onExecute(SQLRequest<SQLAlterTableStatement> request, Myc
             public Future<Void> get() {
                 try {
                     SQLAlterTableStatement sqlAlterTableStatement = request.getAst();
+                    sqlAlterTableStatement.setIfExists(true);
                     SQLExprTableSource tableSource = sqlAlterTableStatement.getTableSource();
                     resolveSQLExprTableSource(tableSource, dataContext);
                     String schema = SQLUtils.normalize(sqlAlterTableStatement.getSchema());

diff --git a/mycat2/src/main/java/io/mycat/sqlhandler/ddl/DropTableSQLHandler.java b/mycat2/src/main/java/io/mycat/sqlhandler/ddl/DropTableSQLHandler.java
@@ -45,6 +45,7 @@ protected Future<Void> onExecute(SQLRequest<SQLDropTableStatement> request, Myca
             public Future<Void> get() {
                 try {
                     SQLDropTableStatement ast = request.getAst();
+                    ast.setIfExists(true);
                     List<SQLExprTableSource> tableSources = ast.getTableSources();
                     if (tableSources.size() != 1) {
                         throw new UnsupportedOperationException("unsupported drop multi table :" + tableSources.get(0));