Go語(yǔ)言 碎片整理之 文件操作

文件是什么？
計(jì)算機(jī)中的文件是存儲(chǔ)在外部介質(zhì)（通常是磁盤(pán)）上的數(shù)據(jù)集合，文件分為文本文件和二進(jìn)制文件。

打開(kāi)和關(guān)閉文件

os.Open()函數(shù)能夠打開(kāi)一個(gè)文件，返回一個(gè)*File和一個(gè)error。對(duì)得到的文件實(shí)例調(diào)用close()方法能夠關(guān)閉文件。

func main(){
    file,err := os.Open("./main.go")
    if err != nil{
        fmt.Println("open file failed!,err",err)
        return 
    }
    fmt.Println("cool!")
    file.Close()
}

為了防止文件忘記關(guān)閉，我們通常使用defer注冊(cè)文件關(guān)閉語(yǔ)句。

讀取文件

file.Read()

基本使用

Read的方法定義如下：
func(f *File) Read(b []byte)(n int, err error)
它接收一個(gè)字節(jié)切片，返回讀取的字節(jié)數(shù)和可能的具體錯(cuò)誤，讀到文件末尾時(shí)會(huì)返回0和io.EOF。

package main
import (
    "fmt"
    "os"
    "io"
)
func main(){
    file,err := os.Open("./main.go")
    if err != nil{
        fmt.Println("open file failed!,err", err)
        return 
    }
    defer file.Close()
    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ù)。

package main
import (
    "fmt"
    "os"
    "io"
)
func main(){
    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的基礎(chǔ)上封裝了一層API,支持更多的功能。

package main
import (
    "fmt"
    "os"
    "io"
    "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.Println(line)
    }
}

使用ioutil讀取整個(gè)文件

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

package main
import (
    "fmt"
    "io/ioutil"
)
func main(){
    content, err := ioutil.ReadFile("./main.go")
    if err != nil{
        fmt.Println("read file failed,err:",err)
        return 

    }
    fmt.Println(string(content))
}

文件寫(xiě)入操作

os.OpenFile()函數(shù)能夠以指定的模式打開(kāi)文件，從而實(shí)現(xiàn)文件寫(xiě)入相關(guān)功能。

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

其中：
name 是指要打開(kāi)的文件名，flag是指打開(kāi)文件的模式，模式有以下幾種：

perm 文件權(quán)限，一個(gè)八進(jìn)制數(shù)。r(讀)04 ，w（寫(xiě)）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))       //寫(xiě)入字節(jié)切片數(shù)據(jù)
    file.WriteString("hello 小王子") //直接寫(xiě)入字符串?dāng)?shù)據(jù)
}

bufio.NewWriter

import (
    "io"
        "fmt"
)
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ù)先寫(xiě)入緩存
    }
    writer.Flush() //將緩存中的內(nèi)容寫(xiě)入文件
}

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

練習(xí)

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_CREATE|os.O_WRONLY,0644)
    if err != nil{
        fmt.Printf("open %s failed,err:%v.\n",dstName,err)
        return 
    }
    defer dst.Close()
    return io.Copy(dst,src)
}
func main(){
    _,err := CopyFile("dst.txt","src.txt")
    if err != nil{
        fmt.Println("copy file failed,err",err)
        return 
    }
    fmt.Println("Copy done!")
}