container-search/src/main/java/com/yahoo/search/yql/ParserBase.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39

// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.search.yql;

import com.google.common.collect.Sets;

import org.antlr.v4.runtime.Parser;
import org.antlr.v4.runtime.TokenStream;
import org.antlr.v4.runtime.tree.ParseTree;

import java.util.Set;

/**
 * Provides semantic helper functions to Parser.
 */
abstract class ParserBase extends Parser {

    private static String arrayRuleName = "array";

    public ParserBase(TokenStream input) {
        super(input);
    }
    
    private Set<String> arrayParameters = Sets.newHashSet();

    public void registerParameter(String name, String typeName) {
        if (typeName.equals(arrayRuleName)) {
            arrayParameters.add(name);
       }
    }

    public boolean isArrayParameter(ParseTree nameNode) {
        String name = nameNode.getText();
        if (name.startsWith("@")) {
            name = name.substring(1);
        }
        return name != null && arrayParameters.contains(name);
    }
    
}