Add option to set MaxBuf in html.Parse

Jarcis-cy · Jarcis-cy · commit cbd34d58b5d2 · 2024-06-20T16:53:06.000+08:00
diff --git a/html/parse.go b/html/parse.go
@@ -2366,6 +2366,13 @@ func ParseOptionEnableScripting(enable bool) ParseOption {
 	}
 }
 
+// ParseOptionSetTokenizerMaxBuf sets the maximum buffer size for the tokenizer.
+func ParseOptionSetTokenizerMaxBuf(maxBuf int) ParseOption {
+	return func(p *parser) {
+		p.tokenizer.SetMaxBuf(maxBuf)
+	}
+}
+
 // ParseWithOptions is like Parse, with options.
 func ParseWithOptions(r io.Reader, opts ...ParseOption) (*Node, error) {
 	p := &parser{

Original file line number	Diff line number	Diff line change
`@@ -2366,6 +2366,13 @@ func ParseOptionEnableScripting(enable bool) ParseOption {`
`2366`	`2366`	`}`
`2367`	`2367`	`}`
`2368`	`2368`
	`2369`	`+// ParseOptionSetTokenizerMaxBuf sets the maximum buffer size for the tokenizer.`
	`2370`	`+func ParseOptionSetTokenizerMaxBuf(maxBuf int) ParseOption {`
	`2371`	`+ return func(p *parser) {`
	`2372`	`+ p.tokenizer.SetMaxBuf(maxBuf)`
	`2373`	`+ }`
	`2374`	`+}`
	`2375`	`+`
`2369`	`2376`	`// ParseWithOptions is like Parse, with options.`
`2370`	`2377`	`func ParseWithOptions(r io.Reader, opts ...ParseOption) (*Node, error) {`
`2371`	`2378`	`p := &parser{`