Translator supports NOT LIKE / NOT SIMILAR TO

69f0be2a · Donald H. (Donnie) Pinkston, III · 87630d9d · 69f0be2a
Commit 69f0be2a authored 6 years ago by Donald H. (Donnie) Pinkston, III
Hide whitespace changes
Inline Side-by-side

Showing

with 12 additions and 2 deletions
+12 -2
--- a/src/main/java/edu/caltech/nanodb/sqlparse/NanoSQLTranslator.java
+++ b/src/main/java/edu/caltech/nanodb/sqlparse/NanoSQLTranslator.java
@@ -1002,16 +1002,26 @@ public class NanoSQLTranslator extends NanoSQLBaseVisitor<Object> {

    @Override
    public Object visitExprLike(NanoSQLParser.ExprLikeContext ctx) {
+        boolean invert = (ctx.NOT() != null);
        Expression lhs = (Expression) visit(ctx.expression(0));
        Expression rhs = (Expression) visit(ctx.expression(1));
-        return new StringMatchOperator(StringMatchOperator.Type.LIKE, lhs, rhs);
+
+        StringMatchOperator op = new StringMatchOperator(
+            StringMatchOperator.Type.LIKE, lhs, rhs);
+        op.setInvert(invert);
+        return op;
    }

    @Override
    public Object visitExprSimilarTo(NanoSQLParser.ExprSimilarToContext ctx) {
+        boolean invert = (ctx.NOT() != null);
        Expression lhs = (Expression) visit(ctx.expression(0));
        Expression rhs = (Expression) visit(ctx.expression(1));
-        return new StringMatchOperator(StringMatchOperator.Type.REGEX, lhs, rhs);
+
+        StringMatchOperator op = new StringMatchOperator(
+            StringMatchOperator.Type.REGEX, lhs, rhs);
+        op.setInvert(invert);
+        return op;
    }

    @Override