From 4162d0dcba07fa561f572d386208d24c678b24c1 Mon Sep 17 00:00:00 2001
From: Steve Coffman <steve@khanacademy.org>
Date: Wed, 12 Jun 2024 16:40:28 -0400
Subject: [PATCH] Add ParseQueryWithLimit

Signed-off-by: Steve Coffman <steve@khanacademy.org>
---
 parser/query.go | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/parser/query.go b/parser/query.go
index 7f31b0f4..f653bed5 100644
--- 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
 }