graphql-java/src/main/java/graphql/parser/Parser.java at master · HKMOpen/graphql-java

32 lines (24 loc) · 1.05 KB

package graphql.parser;

import graphql.language.Document;

import graphql.parser.antlr.GraphqlLexer;

import graphql.parser.antlr.GraphqlParser;

import org.antlr.v4.runtime.ANTLRInputStream;

import org.antlr.v4.runtime.BailErrorStrategy;

import org.antlr.v4.runtime.CommonTokenStream;

import org.antlr.v4.runtime.atn.PredictionMode;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class Parser {

public Document parseDocument(String input) {

GraphqlLexer lexer = new GraphqlLexer(new ANTLRInputStream(input));

CommonTokenStream tokens = new CommonTokenStream(lexer);

GraphqlParser parser = new GraphqlParser(tokens);

parser.removeErrorListeners();

parser.getInterpreter().setPredictionMode(PredictionMode.SLL);

parser.setErrorHandler(new BailErrorStrategy());

GraphqlParser.DocumentContext document = parser.document();

GraphqlAntlrToLanguage antlrToLanguage = new GraphqlAntlrToLanguage();

antlrToLanguage.visitDocument(document);

return antlrToLanguage.result;

}

Provide feedback