Merge pull request #184 from wenki96/main

fix foreign key bugs
polardb · Nov 8, 2023 · 547cd18 · 547cd18
2 parents d20495b + 80a7b19
commit 547cd18
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/...utor/src/main/java/com/alibaba/polardbx/executor/handler/ddl/LogicalCommonDdlHandler.java b/...utor/src/main/java/com/alibaba/polardbx/executor/handler/ddl/LogicalCommonDdlHandler.java
@@ -535,12 +535,16 @@ protected void rewriteOriginSqlWithForeignKey(BaseDdlOperation logicalDdlPlan, E
         // rewrite origin sql for different naming behaviours in 5.7 & 8.0
         boolean createTableWithFk = logicalDdlPlan.getDdlType() == DdlType.CREATE_TABLE
             && !((LogicalCreateTable) logicalDdlPlan).getSqlCreateTable().getAddedForeignKeys().isEmpty();
-        boolean alterTableAddFk = logicalDdlPlan.getDdlType() == DdlType.ALTER_TABLE
-            && ((LogicalAlterTable) logicalDdlPlan).getSqlAlterTable().getAlters().get(0).getKind()
-            == SqlKind.ADD_FOREIGN_KEY;
-        boolean alterTableDropFk = logicalDdlPlan.getDdlType() == DdlType.ALTER_TABLE
-            && ((LogicalAlterTable) logicalDdlPlan).getSqlAlterTable().getAlters().get(0).getKind()
-            == SqlKind.DROP_FOREIGN_KEY;
+        boolean alterTableAddFk =
+            logicalDdlPlan.getDdlType() == DdlType.ALTER_TABLE && logicalDdlPlan instanceof LogicalAlterTable
+                && ((LogicalAlterTable) logicalDdlPlan).getSqlAlterTable().getAlters().size() == 1
+                && ((LogicalAlterTable) logicalDdlPlan).getSqlAlterTable().getAlters().get(0).getKind()
+                == SqlKind.ADD_FOREIGN_KEY;
+        boolean alterTableDropFk =
+            logicalDdlPlan.getDdlType() == DdlType.ALTER_TABLE && logicalDdlPlan instanceof LogicalAlterTable
+                && ((LogicalAlterTable) logicalDdlPlan).getSqlAlterTable().getAlters().size() == 1
+                && ((LogicalAlterTable) logicalDdlPlan).getSqlAlterTable().getAlters().get(0).getKind()
+                == SqlKind.DROP_FOREIGN_KEY;
         if (createTableWithFk) {
             ec.getDdlContext().setForeignKeyOriginalSql(
                 ((LogicalCreateTable) logicalDdlPlan).getSqlCreateTable().toString());