From ae610dcbb713bb08d60346b3b5e03a4e7541b74d Mon Sep 17 00:00:00 2001 From: Stefan Haller Date: Wed, 9 Oct 2024 15:08:01 +0200 Subject: [PATCH] Extract helper function for easier testing --- pkg/utils/io.go | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/pkg/utils/io.go b/pkg/utils/io.go index d31c5fc07..98d026429 100644 --- a/pkg/utils/io.go +++ b/pkg/utils/io.go @@ -2,6 +2,7 @@ package utils import ( "bufio" + "io" "os" ) @@ -12,14 +13,18 @@ func ForEachLineInFile(path string, f func(string, int)) error { } defer file.Close() - reader := bufio.NewReader(file) + forEachLineInStream(file, f) + + return nil +} + +func forEachLineInStream(reader io.Reader, f func(string, int)) { + bufferedReader := bufio.NewReader(reader) for i := 0; true; i++ { - line, err := reader.ReadString('\n') + line, err := bufferedReader.ReadString('\n') if err != nil { break } f(line, i) } - - return nil }