辅助函数

utils/file/file.go

@@ -1,6 +1,10 @@
 package file
 
-import "os"
+import (
+	"bufio"
+	"io"
+	"os"
+)
 
 // PathExist 路径是否存在
 func PathExist(path string) (bool, error) {
@@ -22,3 +26,54 @@ func IsDir(path string) (bool, error) {
 	}
 	return false, err
 }
+
+// WriterFile 向特定文件写入内容
+func WriterFile(filePath string, content []byte) error {
+	f, err := os.OpenFile(filePath, os.O_WRONLY|os.O_TRUNC|os.O_CREATE, 0644)
+	if err != nil {
+		return err
+	} else {
+		// offset
+		//os.Truncate(filename, 0) //clear
+		n, _ := f.Seek(0, io.SeekEnd)
+		_, err = f.WriteAt(content, n)
+		_ = f.Close()
+	}
+	return nil
+}
+
+// ReadOnce 单次读取文件
+//   - 一次性对整个文件进行读取，小文件读取可以很方便的一次性将文件内容读取出来，而大文件读取会造成性能影响。
+func ReadOnce(filePath string) ([]byte, error) {
+	if file, err := os.Open(filePath); err != nil {
+		return nil, err
+	} else {
+		_ = file.Close()
+		return io.ReadAll(file)
+	}
+}
+
+// ReadBlockHook 分块读取文件
+//   - 将filePath路径对应的文件数据并将读到的每一部分传入hook函数中，当过程中如果产生错误则会返回error。
+//   - 分块读取可以在读取速度和内存消耗之间有一个很好的平衡。
+func ReadBlockHook(filePath string, bufferSize int, hook func(data []byte)) error {
+	file, err := os.Open(filePath)
+	if err != nil {
+		return err
+	}
+	defer func() {
+		_ = file.Close()
+	}()
+	buffer := make([]byte, bufferSize)
+	bufferReader := bufio.NewReader(file)
+	for {
+		successReadSize, err := bufferReader.Read(buffer)
+		hook(buffer[:successReadSize])
+		if err != nil {
+			if err == io.EOF {
+				return nil
+			}
+			return err
+		}
+	}
+}