org.graylog.events.conditions.BooleanNumberConditionsVisitor Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of graylog2-server Show documentation
Graylog
There is a newer version: 6.0.6
/**
 * This file is part of Graylog.
 *
 * Graylog is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Graylog is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Graylog.  If not, see .
 */
package org.graylog.events.conditions;

import java.util.Collections;
import java.util.Map;

public class BooleanNumberConditionsVisitor implements ExpressionVisitor {
    private final Map numberReferences;

    public BooleanNumberConditionsVisitor() {
        this.numberReferences = Collections.emptyMap();
    }

    public BooleanNumberConditionsVisitor(Map numberReferences) {
        this.numberReferences = numberReferences;
    }

    private  T accept(Expression expr) {
        return expr.accept(this);
    }

    @Override
    public boolean visit(Expr.True expr) {
        return true;
    }

    @Override
    public boolean visit(Expr.And and) {
        return accept(and.left()) && accept(and.right());
    }

    @Override
    public boolean visit(Expr.Or or) {
        return accept(or.left()) || accept(or.right());
    }

    @Override
    public boolean visit(Expr.Not not) {
        return !accept(not.left());
    }

    @Override
    public boolean visit(Expr.Equal equal) {
        return accept(equal.left()).equals(accept(equal.right()));
    }

    @Override
    public boolean visit(Expr.Greater greater) {
        return accept(greater.left()) > accept(greater.right());
    }

    @Override
    public boolean visit(Expr.GreaterEqual greater) {
        return accept(greater.left()) >= accept(greater.right());
    }

    @Override
    public boolean visit(Expr.Lesser lesser) {
        return accept(lesser.left()) < accept(lesser.right());
    }

    @Override
    public boolean visit(Expr.LesserEqual lesser) {
        return accept(lesser.left()) <= accept(lesser.right());
    }

    @Override
    public double visit(Expr.NumberValue numberValue) {
        return numberValue.value();
    }

    @Override
    public double visit(Expr.NumberReference numberRef) {
        final Double value = numberReferences.get(numberRef.ref());

        if (value == null) {
            throw new IllegalArgumentException("Couldn't find value for reference " + numberRef.toString());
        }

        return value;
    }

    @Override
    public boolean visit(Expr.Group group) {
        return accept(group.child());
    }
}