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

文件是什么佩谷？

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

打開和關閉文件
os.Open()函數(shù)能夠打開一個文件谐檀，返回一個*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)
它接收一個字節(jié)切片框喳，返回讀取的字節(jié)數(shù)和可能的具體錯誤课幕，讀到文件末尾時會返回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方法讀取數(shù)據(jù)
    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字節(jié)數(shù)據(jù)\n", n)
    fmt.Println(string(tmp[:n]))
}

循環(huán)讀取
使用for循環(huán)讀取文件中的所有數(shù)據(jù)。

func main() {
    // 只讀方式打開當前目錄下的main.go文件
    file, err := os.Open("./main.go")
    if err != nil {
        fmt.Println("open file failed!, err:", err)
        return
    }
    defer file.Close()
    // 循環(huán)讀取文件
    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方法能夠讀取完整的文件，只需要將文件名作為參數(shù)傳入润绎。

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()函數(shù)能夠以指定模式打開文件撬碟，從而實現(xiàn)文件寫入相關功能。

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

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

模式含義
os.O_WRONLY 只寫
os.O_CREATE 創(chuàng)建文件
os.O_RDONLY 只讀
os.O_RDWR 讀寫
os.O_TRUNC 清空
os.O_APPEND 追加
perm：文件權限呢蛤，一個八進制數(shù)。r（讀）04棍郎，w（寫）02其障，x（執(zhí)行）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))       //寫入字節(jié)切片數(shù)據(jù)
    file.WriteString("hello 小王子") //直接寫入字符串數(shù)據(jù)
}

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") //將數(shù)據(jù)先寫入緩存
    }
    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
    }
}

NewReader() 接收一個接口類型

練習
copyFile
借助io.Copy()實現(xiàn)一個拷貝文件函數(shù)涂佃。

// CopyFile 拷貝文件函數(shù)
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()
    // 以寫|創(chuàng)建的方式打開目標文件
    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!")
}

實現(xiàn)一個cat命令
使用文件操作相關知識励翼，模擬實現(xiàn)linux平臺cat命令的功能。

package main

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

// cat命令實現(xiàn)
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() // 解析命令行參數(shù)
    if flag.NArg() == 0 {
        // 如果沒有參數(shù)默認從標準輸入讀取內容
        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))
    }
}

image.png

14.文件

14.文件

讀取文件

文件寫入操作