Go語言文件開關(guān)及讀寫操作的方法

本文小編為大家詳細介紹“Go語言文件開關(guān)及讀寫操作的方法”，內(nèi)容詳細，步驟清晰，細節(jié)處理妥當，希望這篇“Go語言文件開關(guān)及讀寫操作的方法”文章能幫助大家解決疑惑，下面跟著小編的思路慢慢深入，一起來學(xué)習(xí)新知識吧。

打開關(guān)閉文件

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

package main
import (
	"fmt"
	"os"
)
func main() {
	// 打開文件
	file, err := os.Open("./ca.txt")
	if err != nil {
		fmt.Println("文件打開失敗", err)
		return
	}
	print(file) // 輸出地址 0xc000006028
	// 關(guān)閉文件
	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
	}
	// 為了防止忘記關(guān)閉文件，可使用defer注冊文件關(guān)閉語句
	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 僅設(shè)置了 128 字節(jié)，如果待讀取的文件超過了改值，例如 txt 內(nèi)容如下所示：

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

此時運行代碼，就會發(fā)現(xiàn) Read 函數(shù)讀取到第 128 個字節(jié)的時候，就會停止讀取，間接導(dǎo)致讀取到的數(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))
}

注意導(dǎo)入模塊的時候，直接導(dǎo)入了 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)容，查看結(jié)果可直接打開目錄下的 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 表示文件權(quán)限（八進制），這與 linux 文件權(quán)限數(shù)字一致，讀 r 等于 4（感謝博友指出該錯誤），寫 w 表示 2，執(zhí)行 x 等于 1，所以 666 表示的是文件可讀，可寫。

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

讀到這里，這篇“Go語言文件開關(guān)及讀寫操作的方法”文章已經(jīng)介紹完畢，想要掌握這篇文章的知識點還需要大家自己動手實踐使用過才能領(lǐng)會，如果想了解更多相關(guān)內(nèi)容的文章，歡迎關(guān)注億速云行業(yè)資訊頻道。