NumericInTreeNode.java

package com.netflix.suro.routing.filter.lang;

import com.google.common.base.Function;
import com.google.common.collect.Lists;
import com.netflix.suro.routing.filter.MessageFilter;
import com.netflix.suro.routing.filter.MessageFilters;
import com.netflix.suro.routing.filter.NumericValuePredicate;
import com.netflix.suro.routing.filter.PathValueMessageFilter;
import org.antlr.runtime.Token;
import org.antlr.runtime.tree.Tree;

import java.util.List;

import static com.netflix.suro.routing.filter.lang.TreeNodeUtil.getXPath;

public class NumericInTreeNode extends MessageFilterBaseTreeNode implements MessageFilterTranslatable {

	@SuppressWarnings("unchecked")
    @Override
	public MessageFilter translate() {
		final String xpath = getXPath(getChild(0));
		
		List children = getChildren();
		return MessageFilters.or(
                Lists.transform(children.subList(1, children.size()), new Function<Object, MessageFilter>() {

                    @Override
                    public MessageFilter apply(Object node) {
                        Number value = ((NumberTreeNode) node).getValue();
                        return new PathValueMessageFilter(xpath, new NumericValuePredicate(value, "="));
                    }

                })
        );
	}

	public NumericInTreeNode(Token t) {
		super(t);
	} 

	public NumericInTreeNode(NumericInTreeNode node) {
		super(node);
	} 

	public Tree dupNode() {
		return new NumericInTreeNode(this);
	} 
}