fix: remove regex from migration tool (#9254)

* remove regex * checkstyle fixes * add SET/UNSET to ksql-parser * ApplyMigrationTest modification * remove prepare() method/use StatementContext * more fundemental changes on CommandParser * change tests (CommandParserTest, ApplyMigrationCommandTest)
confluentinc · Jul 14, 2022 · 5089065 · 5089065
1 parent d679a4a
commit 5089065
Show file tree

Hide file tree

Showing 6 changed files with 625 additions and 1,090 deletions.
diff --git a/ksqldb-parser/src/main/antlr4/io/confluent/ksql/parser/SqlBase.g4 b/ksqldb-parser/src/main/antlr4/io/confluent/ksql/parser/SqlBase.g4
@@ -665,4 +665,4 @@ WS
 // when splitting statements with DelimiterLexer
 UNRECOGNIZED
     : .
-    ;
+    ;
diff --git a/ksqldb-parser/src/main/java/io/confluent/ksql/parser/VariableSubstitutor.java b/ksqldb-parser/src/main/java/io/confluent/ksql/parser/VariableSubstitutor.java
@@ -158,6 +158,19 @@ public Void visitVariableIdentifier(final SqlBaseParser.VariableIdentifierContex
       return null;
     }
 
+    @Override
+    public Void visitSetProperty(final SqlBaseParser.SetPropertyContext context) {
+      lookupVariables(context.STRING(0).getText());
+      lookupVariables(context.STRING(1).getText());
+      return null;
+    }
+
+    @Override
+    public Void visitUnsetProperty(final SqlBaseParser.UnsetPropertyContext context) {
+      lookupVariables(context.STRING().getText());
+      return null;
+    }
+
     private String getIdentifierText(final String value) {
       final char firstChar = value.charAt(0);
       final char lastChar = value.charAt(value.length() - 1);