111

Author: lvan100

lib/vidl/parser.go

@@ -209,24 +209,28 @@ func parseLogicalAndExpr(ctx ILogicalAndExprContext) Expr {
 
 // parseEqualityExpr converts an EqualityExprContext into an Expr.
 func parseEqualityExpr(ctx IEqualityExprContext) Expr {
+	left := parseRelationalExpr(ctx.RelationalExpr(0))
+
 	var op antlr.TerminalNode
 	if ctx.EQUAL() != nil {
 		op = ctx.EQUAL()
 	} else if ctx.NOT_EQUAL() != nil {
 		op = ctx.NOT_EQUAL()
+	} else {
+		return left
 	}
-	if op != nil {
-		return BinaryExpr{
-			Left:  parseRelationalExpr(ctx.RelationalExpr(0)),
-			Op:    op.GetText(),
-			Right: parseRelationalExpr(ctx.RelationalExpr(1)),
-		}
+
+	return BinaryExpr{
+		Left:  left,
+		Op:    op.GetText(),
+		Right: parseRelationalExpr(ctx.RelationalExpr(1)),
 	}
-	return parseRelationalExpr(ctx.RelationalExpr(0))
 }
 
 // parseRelationalExpr converts a RelationalExprContext into an Expr.
 func parseRelationalExpr(ctx IRelationalExprContext) Expr {
+	left := parseUnaryExpr(ctx.UnaryExpr(0))
+
 	var op antlr.TerminalNode
 	if ctx.LESS_THAN() != nil {
 		op = ctx.LESS_THAN()
@@ -236,15 +240,15 @@ func parseRelationalExpr(ctx IRelationalExprContext) Expr {
 		op = ctx.GREATER_THAN()
 	} else if ctx.GREATER_OR_EQUAL() != nil {
 		op = ctx.GREATER_OR_EQUAL()
+	} else {
+		return left
 	}
-	if op != nil {
-		return BinaryExpr{
-			Left:  parseUnaryExpr(ctx.UnaryExpr(0)),
-			Op:    op.GetText(),
-			Right: parseUnaryExpr(ctx.UnaryExpr(1)),
-		}
+
+	return BinaryExpr{
+		Left:  left,
+		Op:    op.GetText(),
+		Right: parseUnaryExpr(ctx.UnaryExpr(1)),
 	}
-	return parseUnaryExpr(ctx.UnaryExpr(0))
 }
 
 // parseUnaryExpr converts a UnaryExprContext into an Expr.