Catch exception

aklivity · Dec 17, 2024 · 661a4bf · 661a4bf
1 parent 11f3e09
commit 661a4bf
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 11 deletions.
diff --git a/...nding-pgsql/src/main/java/io/aklivity/zilla/runtime/binding/pgsql/parser/PgsqlParser.java b/...nding-pgsql/src/main/java/io/aklivity/zilla/runtime/binding/pgsql/parser/PgsqlParser.java
@@ -141,15 +141,21 @@ private void parser(
         String sql,
         PostgreSqlParserBaseListener listener)
     {
-        sql = sql.replace("\u0000", "");
-
-        CharStream input = CharStreams.fromString(sql);
-        lexer.reset();
-        lexer.setInputStream(input);
-
-        tokens.setTokenSource(lexer);
-        parser.setTokenStream(tokens);
-
-        walker.walk(listener, parser.root());
+        try
+        {
+            sql = sql.replace("\u0000", "");
+
+            CharStream input = CharStreams.fromString(sql);
+            lexer.reset();
+            lexer.setInputStream(input);
+
+            tokens.setTokenSource(lexer);
+            parser.setTokenStream(tokens);
+
+            walker.walk(listener, parser.root());
+        }
+        catch (Exception ignore)
+        {
+        }
     }
 }
diff --git a/.../io/aklivity/zilla/runtime/binding/risingwave/internal/stream/RisingwaveProxyFactory.java b/.../io/aklivity/zilla/runtime/binding/risingwave/internal/stream/RisingwaveProxyFactory.java
@@ -753,7 +753,9 @@ private void doParseQuery(
                     .ifPresent(statement ->
                     {
                         String command = parser.parseCommand(statement);
-                        final PgsqlTransform transform = clientTransforms.get(RisingwaveCommandType.valueOf(command.getBytes()));
+                        final PgsqlTransform transform = command == null
+                            ? clientTransforms.get(RisingwaveCommandType.UNKNOWN_COMMAND)
+                            : clientTransforms.get(RisingwaveCommandType.valueOf(command.getBytes()));
                         transform.transform(this, traceId, authorizationId, statement);
                     });
             }