Skip to content

Commit 942cfbb

Browse files
authored
Merge pull request #610 from graphql-java-kickstart/601-schema-max-tokens-limit
601 set schema parser max tokens to max int
2 parents 40035bd + 0758574 commit 942cfbb

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

src/main/kotlin/graphql/kickstart/tools/SchemaParserBuilder.kt

+6-1
Original file line numberDiff line numberDiff line change
@@ -3,11 +3,13 @@ package graphql.kickstart.tools
33
import graphql.language.Definition
44
import graphql.language.Document
55
import graphql.parser.Parser
6+
import graphql.parser.ParserOptions
67
import graphql.schema.GraphQLScalarType
78
import graphql.schema.idl.RuntimeWiring
89
import graphql.schema.idl.SchemaDirectiveWiring
910
import org.antlr.v4.runtime.RecognitionException
1011
import org.antlr.v4.runtime.misc.ParseCancellationException
12+
import kotlin.Int.Companion.MAX_VALUE
1113
import kotlin.reflect.KClass
1214

1315
/**
@@ -169,7 +171,10 @@ class SchemaParserBuilder {
169171
files.forEach { documents.add(parser.parseDocument(readFile(it), it)) }
170172

171173
if (schemaString.isNotEmpty()) {
172-
documents.add(parser.parseDocument(schemaString.toString()))
174+
val options = ParserOptions
175+
.getDefaultParserOptions()
176+
.transform { o -> o.maxTokens(MAX_VALUE) }
177+
documents.add(parser.parseDocument(schemaString.toString(), options))
173178
}
174179
} catch (pce: ParseCancellationException) {
175180
val cause = pce.cause

0 commit comments

Comments
 (0)