Go語言文件開關及讀寫操作示例

更新時間：2022年08月01日 14:25:34 作者：夢想橡皮擦

這篇文章主要為大家介紹了Go語言文件開關及讀寫操作的示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪

?? 實戰(zhàn)場景

本篇博客為大家再次帶來 Go 語言的基礎知識，這次要學習的內(nèi)容是 Go 中的文件操作。

打開關閉文件

在 Go 中操作文件，首先要做的就是導入 os 模塊，該模塊中具備相關函數(shù)定義。

package main
import (
	"fmt"
	"os"
)
func main() {
	// 打開文件
	file, err := os.Open("./ca.txt")
	if err != nil {
		fmt.Println("文件打開失敗", err)
		return
	}
	print(file) // 輸出地址 0xc000006028
	// 關閉文件
	file.Close()
}

如果文件打開失敗，直接展示錯誤信息，正確的話，會返回 file 對象的地址。

讀取文件

使用 Read 函數(shù)讀取文件，其定義如下所示：

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

從定義中可以發(fā)現(xiàn)，Read 函數(shù)會返回 2 個值，第一個是讀取的字節(jié)數(shù)，第二個是當讀取報錯時，返回錯誤信息，示例代碼如下所示：

package main
import (
	"fmt"
	"io"
	"os"
)
func main() {
	// 打開文件
	file, err := os.Open("./ca.txt")
	if err != nil {
		fmt.Println("打開文件失敗，錯誤信息：", err)
		return
	}
	// 為了防止忘記關閉文件，可使用defer注冊文件關閉語句
	defer file.Close()
	// Read方法讀取數(shù)據(jù)
	var tmp = make([]byte, 128)
	// 讀取文件
	n, err := file.Read(tmp)
	// 碰到 io.EOF 表示文件讀取到末尾
	if err == io.EOF {
		fmt.Println("文件讀取完畢")
		return
	}
	if err != nil {
		fmt.Println("讀取文件失敗，錯誤信息：", err)
		return
	}
	fmt.Printf("讀取了 %d 字節(jié)數(shù)據(jù)\n", n)
	fmt.Println(string(tmp[:n]))
}

輸出如下內(nèi)容：

讀取了 24 字節(jié)數(shù)據(jù)
姓名：夢想橡皮擦

讀取到 24 個字節(jié)，標準的中文占 3 個字節(jié)。

但是上述代碼最大的問題就是，我們給變量 tmp，即數(shù)組 tmp 僅設置了 128 字節(jié)，如果待讀取的文件超過了改值，例如 txt 內(nèi)容如下所示：

姓名：夢想橡皮擦1
姓名：夢想橡皮擦2
姓名：夢想橡皮擦3
姓名：夢想橡皮擦4
姓名：夢想橡皮擦5

此時運行代碼，就會發(fā)現(xiàn) Read 函數(shù)讀取到第 128 個字節(jié)的時候，就會停止讀取，間接導致讀取到的數(shù)據(jù)亂碼。

讀取了 128 字節(jié)數(shù)據(jù)
姓名：夢想橡皮擦1
姓名：夢想橡皮擦2
姓名：夢想橡皮擦3
姓名：夢想橡皮擦4
姓名：夢想橡??

解決該問題，需要用到 for 循環(huán)讀取文件數(shù)據(jù)，代碼如下：

package main
import (
	"fmt"
	"io"
	"os"
)
func main() {
	// 打開文件
	file, err := os.Open("./ca.txt")
	if err != nil {
		fmt.Println("打開文件失敗，錯誤信息：", err)
		return
	}
	defer file.Close()
	// 聲明一個 content 切片
	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("讀取文件失敗，錯誤信息：", err)
			return
		}
		// 追加數(shù)據(jù)
		content = append(content, tmp[:n]...)
	}
	fmt.Println(string(content))
}

上述代碼聲明了一個動態(tài)數(shù)組，即切片，解決了讀取數(shù)據(jù)不足的問題。

bufio 讀取文件

該函數(shù)可以一次讀取一行數(shù)據(jù)，示例代碼如下：

func main() {
	file, err := os.Open("./ca.txt")
	if err != nil {
		fmt.Println("打開文件失敗，錯誤信息：", err)
		return
	}
	defer file.Close()
	reader := bufio.NewReader(file)
	for {
		// 逐行讀取數(shù)據(jù)
		line, err := reader.ReadString('\n')
		if err == io.EOF {
			if len(line) != 0 {
				fmt.Println(line)
			}
			fmt.Println("文件讀取完畢")
			break
		}
		if err != nil {
			fmt.Println("讀取文件失敗，錯誤信息：", err)
			return
		}
		fmt.Print(line)
	}
}

該函數(shù)與 Python 中的 readline 具備同樣的效果，接下來演示一下讀取整個文件，代碼如下：

package main
import (
	"fmt"
	"io/ioutil"
)
func main() {
	content, err := ioutil.ReadFile("./ca.txt")
	if err != nil {
		fmt.Println("打開文件失敗，錯誤信息：", err)
		return
	}
	fmt.Println(string(content))
}

注意導入模塊的時候，直接導入了 io 包中的 ioutil 函數(shù)。

寫文件

使用 os.OpenFile() 函數(shù)可以指定模式打開文件，即可以寫入的方式打開文件。

寫入文件最常用的函數(shù)是 Write和 WriteString，示例代碼如下所示：

package main
import (
	"fmt"
	"os"
)
func main() {
	file, err := os.OpenFile("ca.txt", os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0666)
	if err != nil {
		fmt.Println("打開文件失敗，錯誤信息:", err)
		return
	}
	defer file.Close()
	name := "橡皮擦"
	file.Write([]byte(name))  //寫入字節(jié)切片
	file.WriteString("夢想橡皮擦") //寫入字符串
}

使用該方式寫入文件，會覆蓋之前的文件內(nèi)容，查看結果可直接打開目錄下的 ca.txt 進行查看。

我們將 os.OpenFile("ca.txt", os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0666) 中的參數(shù)進行單獨的說明。

第一個參數(shù)表示文件名，第二個參數(shù)是打開文件的模式，具體模式如下所示：

os.O_WRONLY：只寫；

os.O_CREATE：創(chuàng)建文件；

os.O_RDONLY：只讀；

os.O_RDWR：讀寫；

os.O_TRUNC：清空；

os.O_APPEND：追加。

可以看到與 Python 中的文件模式基本一致。

最后一個參數(shù)值 0666 表示文件權限（八進制），這與 linux 文件權限數(shù)字一致，讀 r 等于 4（感謝博友指出該錯誤），寫 w 表示 2，執(zhí)行 x 等于 1，所以 666 表示的是文件可讀，可寫。

在文件寫入部分，還存在 bufio.NewWriter 整行寫入，ioutil.WriteFile 整體寫入等內(nèi)容，可以學習的時候重點查閱一下。