Merge pull request #275 from Netflix/update-token-limit

Update the schema token limit to INT.MAX_VALUE.
Netflix · Nov 11, 2021 · ea0e317 · ea0e317
2 parents 6c51f2b + 5782c91
commit ea0e317
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/graphql-dgs-codegen-core/src/main/kotlin/com/netflix/graphql/dgs/codegen/CodeGen.kt b/graphql-dgs-codegen-core/src/main/kotlin/com/netflix/graphql/dgs/codegen/CodeGen.kt
@@ -30,6 +30,7 @@ import com.squareup.javapoet.JavaFile
 import com.squareup.kotlinpoet.FileSpec
 import graphql.language.*
 import graphql.parser.Parser
+import graphql.parser.ParserOptions
 import java.io.File
 import java.nio.file.Path
 import java.nio.file.Paths
@@ -80,6 +81,12 @@ class CodeGen(private val config: CodeGenConfig) {
     }
 
     private fun generateForSchema(schema: String): CodeGenResult {
+        val SDL_MAX_ALLOWED_SCHEMA_TOKENS: Int = Int.MAX_VALUE
+        val parserOptions = ParserOptions.getDefaultParserOptions()
+            .transform {
+                it.maxTokens(SDL_MAX_ALLOWED_SCHEMA_TOKENS)
+            }
+        ParserOptions.setDefaultParserOptions(parserOptions)
         document = Parser.parse(schema)
         requiredTypeCollector = RequiredTypeCollector(
             document,