Go 筆記之常用標準庫

一 文件操作

本文主要介紹了Go語言中文件讀寫的相關操作。

文件是什么？

計算機中的文件是存儲在外部介質（通常是磁盤）上的數據集合，文件分為文本文件和二進制文件。

1.1 文件打開與關閉

os.Open()函數能夠打開一個文件，返回一個*File和一個err。對得到的文件實例調用close()方法能夠關閉文件。

package main import ( "fmt" "os" ) func main() { // 只讀方式打開當前目錄下的main.go文件 file, err := os.Open("./main.go") if err != nil { fmt.Println("open file failed!, err:", err) return } // 關閉文件 file.Close() }

為了防止文件忘記關閉，我們通常使用defer注冊文件關閉語句。

1.2 文件讀取

1.2.1 file.Read()

基本使用

Read方法定義如下：

func (f *File) Read(b []byte) (n int, err error)

它接收一個字節切片，返回讀取的字節數和可能的具體錯誤，讀到文件末尾時會返回0和io.EOF。 舉個例子：

func main() { // 只讀方式打開當前目錄下的main.go文件 file, err := os.Open("./main.go") if err != nil { fmt.Println("open file failed!, err:", err) return } defer file.Close() // 使用Read方法讀取數據 var tmp = make([]byte, 128) n, err := file.Read(tmp) if err == io.EOF { fmt.Println("文件讀完了") return } if err != nil { fmt.Println("read file failed, err:", err) return } fmt.Printf("讀取了%d字節數據\n", n) fmt.Println(string(tmp[:n])) }

循環讀取

使用for循環讀取文件中的所有數據。

func main() { // 只讀方式打開當前目錄下的main.go文件 file, err := os.Open("./main.go") if err != nil { fmt.Println("open file failed!, err:", err) return } defer file.Close() // 循環讀取文件 var content []byte var tmp = make([]byte, 128) for { n, err := file.Read(tmp) if err == io.EOF { fmt.Println("文件讀完了") break } if err != nil { fmt.Println("read file failed, err:", err) return } content = append(content, tmp[:n]...) } fmt.Println(string(content)) }

1.2.2 bufio讀取文件

bufio是在file的基礎上封裝了一層API，支持更多的功能。

package main import ( "bufio" "fmt" "io" "os" ) // bufio按行讀取示例 func main() { file, err := os.Open("./main.go") if err != nil { fmt.Println("open file failed, err:", err) return } defer file.Close() reader := bufio.NewReader(file) for { line, err := reader.ReadString('\n') //注意是字符 if err == io.EOF { fmt.Println("文件讀完了") break } if err != nil { fmt.Println("read file failed, err:", err) return } fmt.Print(line) } }

1.2.3 ioutil 讀取整個文件

io/ioutil包的ReadFile方法能夠讀取完整的文件，只需要將文件名作為參數傳入。

package main import ( "fmt" "io/ioutil" ) // ioutil.ReadFile讀取整個文件 func main() { content, err := ioutil.ReadFile("./main.go") if err != nil { fmt.Println("read file failed, err:", err) return } fmt.Println(string(content)) }

1.3 文件寫入

os.OpenFile()函數能夠以指定模式打開文件，從而實現文件寫入相關功能。

func OpenFile(name string, flag int, perm FileMode) (*File, error) { ... }

其中：

name：要打開的文件名 flag：打開文件的模式。 模式有以下幾種：

perm：文件權限，一個八進制數。r（讀）04，w（寫）02，x（執行）01。

1.3.1 Write和WriteString

func main() { file, err := os.OpenFile("xx.txt", os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0666) if err != nil { fmt.Println("open file failed, err:", err) return } defer file.Close() str := "hello 沙河" file.Write([]byte(str)) //寫入字節切片數據 file.WriteString("hello 小王子") //直接寫入字符串數據 }

1.3.2 bufio.NewWriter

func main() { file, err := os.OpenFile("xx.txt", os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0666) if err != nil { fmt.Println("open file failed, err:", err) return } defer file.Close() writer := bufio.NewWriter(file) for i := 0; i < 10; i++ { writer.WriteString("hello沙河\n") //將數據先寫入緩存 } writer.Flush() //將緩存中的內容寫入文件 }

1.3.3 ioutil.WriteFile

func main() { str := "hello 沙河" err := ioutil.WriteFile("./xx.txt", []byte(str), 0666) if err != nil { fmt.Println("write file failed, err:", err) return } }

1.4 測試

1.4.1 copyFile

借助io.Copy()實現一個拷貝文件函數。

// CopyFile 拷貝文件函數 func CopyFile(dstName, srcName string) (written int64, err error) { // 以讀方式打開源文件 src, err := os.Open(srcName) if err != nil { fmt.Printf("open %s failed, err:%v.\n", srcName, err) return } defer src.Close() // 以寫|創建的方式打開目標文件 dst, err := os.OpenFile(dstName, os.O_WRONLY|os.O_CREATE, 0644) if err != nil { fmt.Printf("open %s failed, err:%v.\n", dstName, err) return } defer dst.Close() return io.Copy(dst, src) //調用io.Copy()拷貝內容 } func main() { _, err := CopyFile("dst.txt", "src.txt") if err != nil { fmt.Println("copy file failed, err:", err) return } fmt.Println("copy done!") }

1.4.2 實現一個cat命令

使用文件操作相關知識，模擬實現linux平臺cat命令的功能。

package main import ( "bufio" "flag" "fmt" "io" "os" ) // cat命令實現 func cat(r *bufio.Reader) { for { buf, err := r.ReadBytes('\n') //注意是字符 if err == io.EOF { break } fmt.Fprintf(os.Stdout, "%s", buf) } } func main() { flag.Parse() // 解析命令行參數 if flag.NArg() == 0 { // 如果沒有參數默認從標準輸入讀取內容 cat(bufio.NewReader(os.Stdin)) } // 依次讀取每個指定文件的內容并打印到終端 for i := 0; i < flag.NArg(); i++ { f, err := os.Open(flag.Arg(i)) if err != nil { fmt.Fprintf(os.Stdout, "reading from %s failed, err:%v\n", flag.Arg(i), err) continue } cat(bufio.NewReader(f)) } }

Go

版權聲明：本文內容由網絡用戶投稿，版權歸原作者所有，本站不擁有其著作權，亦不承擔相應法律責任。如果您發現本站中有涉嫌抄襲或描述失實的內容，請聯系我們jiasou666@gmail.com 處理，核實后本網站將在24小時內刪除侵權內容。

版權聲明：本文內容由網絡用戶投稿，版權歸原作者所有，本站不擁有其著作權，亦不承擔相應法律責任。如果您發現本站中有涉嫌抄襲或描述失實的內容，請聯系我們jiasou666@gmail.com 處理，核實后本網站將在24小時內刪除侵權內容。