Add ParseQueryWithLimit (#304)

Signed-off-by: Steve Coffman <steve@khanacademy.org>
vektah · Jun 12, 2024 · 36a3658 · 36a3658
1 parent d457fc0
commit 36a3658
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/parser/query.go b/parser/query.go
@@ -9,7 +9,15 @@ import (
 func ParseQuery(source *Source) (*QueryDocument, error) {
 	p := parser{
 		lexer:         lexer.New(source),
-		maxTokenLimit: 0, // 0 is the default value
+		maxTokenLimit: 0, // 0 means unlimited
+	}
+	return p.parseQueryDocument(), p.err
+}
+
+func ParseQueryWithTokenLimit(source *Source, maxTokenLimit int) (*QueryDocument, error) {
+	p := parser{
+		lexer:         lexer.New(source),
+		maxTokenLimit: maxTokenLimit,
 	}
 	return p.parseQueryDocument(), p.err
 }