golang的文件讀寫操作

1.打開和關(guān)閉文件

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

package main

import (
    "fmt"
    "os"
)

func main() {
    file,err := os.Open("main.go")
    if err != nil{
        fmt.Println("open file failed!err:",err)
        return
    }
    //為了防止文件忘記關(guān)閉，我們通常使用defer注冊(cè)文件關(guān)閉語句
    file.Close()
}

結(jié)果：
open file failed!err: open main.go: no such file or directory

Process finished with exit code 0

2.讀取文件

2.1file.Read()

main.go文件中的內(nèi)容
main wenjain
少時(shí)誦詩(shī)書所所所所所所所所所
sssssss
少時(shí)誦詩(shī)書所所所所所所所所所
少時(shí)誦詩(shī)書所所所所所所所所所

main wenjain
少時(shí)誦詩(shī)書所所所所所所所所所
sssssss
少時(shí)誦詩(shī)書所所所所所所所所所
少時(shí)誦詩(shī)書所所所所所所所所所
main wenjain
少時(shí)誦詩(shī)書所所所所所所所所所
sssssss
少時(shí)誦詩(shī)書所所所所所所所所所
少時(shí)誦詩(shī)書所所所所所所所所所
main wenjain
少時(shí)誦詩(shī)書所所所所所所所所所
sssssss

基本使用
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"
    "io"
    "os"
)

func main() {
    file,err := os.Open("main.go")
    if err != nil{
        fmt.Println("open file failed!err:",err)
        return
    }
    //為了防止文件忘記關(guān)閉，我們通常使用defer注冊(cè)文件關(guān)閉語句
    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)
    }
    fmt.Printf("讀取了%s字節(jié)數(shù)據(jù)\n",n)
    fmt.Println(string(tmp[:n]))
}

結(jié)果：
讀取了%!s(int=128)字節(jié)數(shù)據(jù)
main wenjain
少時(shí)誦詩(shī)書所所所所所所所所所
sssssss
少時(shí)誦詩(shī)書所所所所所所所所所
少時(shí)誦詩(shī)書所所

Process finished with exit code 0

從上面可以看出，只能讀取128字節(jié)的內(nèi)容，并沒有全部讀取到。

2.2循環(huán)讀取

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

package main

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

func main() {
    file,err := os.Open("main.go")
    if err != nil{
        fmt.Println("open file failed!err:",err)
        return
    }
    //為了防止文件忘記關(guān)閉，我們通常使用defer注冊(cè)文件關(guān)閉語句
    defer file.Close()
    //存儲(chǔ)所有的數(shù)據(jù)
    var content []byte
    //存儲(chǔ)每次讀取的128字節(jié)數(shù)據(jù)
    var tmp = make([]byte,128)
    //循環(huán)讀取文件
    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))

}

結(jié)果：
文件讀完了
少時(shí)誦詩(shī)書所所所所所所所所所
sssssss
少時(shí)誦詩(shī)書所所所所所所所所所
少時(shí)誦詩(shī)書所所所所所所所所所

main wenjain
少時(shí)誦詩(shī)書所所所所所所所所所
sssssss
少時(shí)誦詩(shī)書所所所所所所所所所
少時(shí)誦詩(shī)書所所所所所所所所所
main wenjain
少時(shí)誦詩(shī)書所所所所所所所所所
sssssss
少時(shí)誦詩(shī)書所所所所所所所所所
少時(shí)誦詩(shī)書所所所所所所所所所
main wenjain
少時(shí)誦詩(shī)書所所所所所所所所所
sssssss

Process finished with exit code 0

2.3bufio讀取文件

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

package main

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

func main() {
    // 只讀方式打開當(dāng)前目錄下的main.go文件
    file, err := os.Open("./main.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)
    }

}

結(jié)果：
少時(shí)誦詩(shī)書所所所所所所所所所
sssssss
少時(shí)誦詩(shī)書所所所所所所所所所
少時(shí)誦詩(shī)書所所所所所所所所所

main wenjain
少時(shí)誦詩(shī)書所所所所所所所所所
sssssss
少時(shí)誦詩(shī)書所所所所所所所所所
少時(shí)誦詩(shī)書所所所所所所所所所
main wenjain
少時(shí)誦詩(shī)書所所所所所所所所所
sssssss
少時(shí)誦詩(shī)書所所所所所所所所所
少時(shí)誦詩(shī)書所所所所所所所所所
main wenjain
少時(shí)誦詩(shī)書所所所所所所所所所
sssssss
文件讀完了

Process finished with exit code 0

2.4ioutil讀取整個(gè)文件

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

package main

import (
    "fmt"
    "io/ioutil"
)

func main() {
    content,err := ioutil.ReadFile("main.txt")
    if err != nil{
        fmt.Println("read file failed,err:",err)
        return
    }
    fmt.Println(string(content))
}

結(jié)果：
少時(shí)誦詩(shī)書所所所所所所所所所
sssssss
少時(shí)誦詩(shī)書所所所所所所所所所
少時(shí)誦詩(shī)書所所所所所所所所所

main wenjain
少時(shí)誦詩(shī)書所所所所所所所所所
sssssss
少時(shí)誦詩(shī)書所所所所所所所所所
少時(shí)誦詩(shī)書所所所所所所所所所
main wenjain
少時(shí)誦詩(shī)書所所所所所所所所所
sssssss
少時(shí)誦詩(shī)書所所所所所所所所所
少時(shí)誦詩(shī)書所所所所所所所所所
main wenjain
少時(shí)誦詩(shī)書所所所所所所所所所
sssssss

Process finished with exit code 0

3.文件寫入操作

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

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

其中，
name:要打開的文件名。
flag：打開文件的模式。模式有以下幾種。
perm：文件權(quán)限，一個(gè)八進(jìn)制數(shù)。r（讀）04，w（寫）02，x（執(zhí)行）01。

3.1Write和WriteString

package main

import (
    "fmt"
    "os"
)

func main() {
    file,err := os.OpenFile("write.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 vita"
    file.Write([]byte(str)) //寫入字節(jié)切片數(shù)據(jù)
    file.WriteString("hello lili") //直接寫入字符串?dāng)?shù)據(jù)

}

write.txt中內(nèi)容
hello vitahello lili

3.2bufio.NewWriter

package main

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

func main() {
    file,err := os.OpenFile("write.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 麗麗")
    }
    writer.Flush()
}

write.txt中內(nèi)容
hello 麗麗hello 麗麗hello 麗麗hello 麗麗hello 麗麗hello 麗麗hello 麗麗hello 麗麗hello 麗麗hello 麗麗

2.3ioutil.WriteFile

package main

import (
    "fmt"
    "io/ioutil"
)

func main() {
    str := "hello lili"
    err := ioutil.WriteFile("./write.txt", []byte(str), 0666)
    if err != nil {
        fmt.Println("write file failed, err:", err)
        return
    }
}

write.txt中內(nèi)容
hello lili

4.練習(xí)

4.1copyFile

借助io.Copy()實(shí)現(xiàn)一個(gè)拷貝文件函數(shù)。
src.txt中內(nèi)容
src src

package main

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

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)建的方式打開目標(biāo)文件
    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)
}
func main() {
    _,err := CopyFile("dst.txt","src.txt")
    if err != nil{
        fmt.Println("copy file failed,err:",err)
        return
    }
    fmt.Println("copy done!")
}

結(jié)果：
copy done!

Process finished with exit code 0

dst中內(nèi)容：
src src

4.2實(shí)現(xiàn)一個(gè)cat命令

dst.txt中內(nèi)容
dst

src.txt中內(nèi)容
src

注意：由于只有一行內(nèi)容，所以必須要有換行，否則不出現(xiàn)內(nèi)容。可以嘗試下哦~

package main

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

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ù)默認(rèn)從標(biāo)準(zhǔn)輸入讀取內(nèi)容
        cat(bufio.NewReader(os.Stdin))
    }
    //依次讀取每個(gè)指定文件的內(nèi)容并打印到終端
    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))
    }
}

結(jié)果：
驅(qū)蚊器無群無群無  //輸入
驅(qū)蚊器無群無群無  //輸出
dst.txt  //輸入
dst.txt  //輸出
^C
Process finished with exit code 2

? test go build -o "args_demo"    
?  test ./args_demo dst.txt src.txt
dst
src