githubmirror/lazygit

pkg/utils/io.go

30 строк473 B

Stefan Haller

Fix ForEachLineInFile to not lose the last line if it doesn't end with a LF

09 окт 2024, 16:37

09 окт 2024, 16:37696e78f

package utils

import (
	"bufio"
	"io"
	"os"
)

func ForEachLineInFile(path string, f func(string, int)) error {
	file, err := os.Open(path)
	if err != nil {
		return err
	}
	defer file.Close()

	forEachLineInStream(file, f)

	return nil
}

func forEachLineInStream(reader io.Reader, f func(string, int)) {
	bufferedReader := bufio.NewReader(reader)
	for i := 0; true; i++ {
		line, _ := bufferedReader.ReadString('\n')
		if len(line) == 0 {
			break
		}
		f(line, i)
	}
}