fix inital buffer size (r3labs#129)

Correctly handle maxBufferSize for scanner buffer
peteut · Mar 8, 2022 · 9a15ccd · 9a15ccd
1 parent c592a97
commit 9a15ccd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/event.go b/event.go
@@ -29,7 +29,8 @@ type EventStreamReader struct {
 // NewEventStreamReader creates an instance of EventStreamReader.
 func NewEventStreamReader(eventStream io.Reader, maxBufferSize int) *EventStreamReader {
 	scanner := bufio.NewScanner(eventStream)
-	scanner.Buffer(make([]byte, 4096), maxBufferSize)
+	initBufferSize := minPosInt(4096, maxBufferSize)
+	scanner.Buffer(make([]byte, initBufferSize), maxBufferSize)
 
 	split := func(data []byte, atEOF bool) (int, []byte, error) {
 		if atEOF && len(data) == 0 {