Go -- 讀取文件內容
阿新 • • 發佈:2017-10-17
簡單的 std func 速查 all pen port 平衡 write
代碼如下:
Golang 的文件讀取方法很多,剛上手時不知道怎麽選擇,所以貼在此處便後速查。
一次性讀取
小文件推薦一次性讀取,這樣程序更簡單,而且速度最快。
代碼如下:func ReadAll(filePth string) ([]byte, error) { f, err := os.Open(filePth) if err != nil { return nil, err } return ioutil.ReadAll(f) }
還有更加簡單的方法,我經常使用ioutil.ReadFile(filePth)
分塊讀取
可在速度和內存占用之間取得很好的平衡。
package main import ( "bufio" "io" "os" ) func processBlock(line []byte) { os.Stdout.Write(line) } func ReadBlock(filePth string, bufSize int, hookfn func([]byte)) error { f, err := os.Open(filePth) if err != nil { return err } defer f.Close() buf := make([]byte, bufSize) //一次讀取多少個字節 bfRd := bufio.NewReader(f) for { n, err := bfRd.Read(buf) hookfn(buf[:n]) // n 是成功讀取字節數 if err != nil { //遇到任何錯誤立即返回,並忽略 EOF 錯誤信息 if err == io.EOF { return nil } return err } } return nil } func main() { ReadBlock("test.txt", 10000, processBlock) }
逐行讀取
逐行讀取有的時候真的很方便,性能可能慢一些,但是僅占用極少的內存空間。
代碼如下:package main import ( "bufio" "io" "os" ) func processLine(line []byte) { os.Stdout.Write(line) } func ReadLine(filePth string, hookfn func([]byte)) error { f, err := os.Open(filePth) if err != nil { return err } defer f.Close() bfRd := bufio.NewReader(f) for { line, err := bfRd.ReadBytes(‘\n‘) hookfn(line) //放在錯誤處理前面,即使發生錯誤,也會處理已經讀取到的數據。 if err != nil { //遇到任何錯誤立即返回,並忽略 EOF 錯誤信息 if err == io.EOF { return nil } return err } } return nil } func main() { ReadLine("test.txt", processLine) }
Go -- 讀取文件內容