Skip to content

Commit ae610dc

Browse files
committed
Extract helper function for easier testing
1 parent d11e11d commit ae610dc

File tree

1 file changed

+9
-4
lines changed

1 file changed

+9
-4
lines changed

pkg/utils/io.go

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@ package utils
22

33
import (
44
"bufio"
5+
"io"
56
"os"
67
)
78

@@ -12,14 +13,18 @@ func ForEachLineInFile(path string, f func(string, int)) error {
1213
}
1314
defer file.Close()
1415

15-
reader := bufio.NewReader(file)
16+
forEachLineInStream(file, f)
17+
18+
return nil
19+
}
20+
21+
func forEachLineInStream(reader io.Reader, f func(string, int)) {
22+
bufferedReader := bufio.NewReader(reader)
1623
for i := 0; true; i++ {
17-
line, err := reader.ReadString('\n')
24+
line, err := bufferedReader.ReadString('\n')
1825
if err != nil {
1926
break
2027
}
2128
f(line, i)
2229
}
23-
24-
return nil
2530
}

0 commit comments

Comments
 (0)