本文主要介紹了Go語言中檔案讀寫的相關操作，

檔案是什么？

計算機中的檔案是存盤在外部介質（通常是磁盤）上的資料集合，檔案分為文本檔案和二進制檔案，

打開和關閉檔案

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注冊檔案關閉陳述句，

讀取檔案

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))
}

bufio讀取檔案

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

package main

import (
	"bufio"
	"fmt"
	"io"
	"os"
)

// bufio按行讀取示例
func main() {
	file, err := os.Open("./xx.txt")
	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 {
			if len(line) != 0 {
				fmt.Println(line)
			}
			fmt.Println("檔案讀完了")
			break
		}
		if err != nil {
			fmt.Println("read file failed, err:", err)
			return
		}
		fmt.Print(line)
	}
}

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))
}

檔案寫入操作

os.OpenFile()函式能夠以指定模式打開檔案，從而實作檔案寫入相關功能，

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

其中：

name：要打開的檔案名 flag：打開檔案的模式， 模式有以下幾種：

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

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 小王子") //直接寫入字串資料
}

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() //將快取中的內容寫入檔案
}

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
	}
}

練習

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!")
}

實作一個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

上一篇：Go語言標準庫之strconv

下一篇：Go語言標準庫之log