Go語言 碎片整理之 文件操作

文件是什麼?
計算機中的文件是存儲在外部介質(通常是磁盤)上的數據集合,文件分爲文本文件和二進制文件。

打開和關閉文件

os.Open()函數能夠打開一個文件,返回一個*File和一個error。對得到的文件實例調用close()方法能夠關閉文件。

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

爲了防止文件忘記關閉,我們通常使用defer註冊文件關閉語句。

讀取文件

file.Read()

基本使用

Read的方法定義如下:
func(f *File) Read(b []byte)(n int, err error)
它接收一個字節切片,返回讀取的字節數和可能的具體錯誤,讀到文件末尾時會返回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的字節數據\n",n)
    fmt.Println(string(tmp[:n]))
}

循環讀取

使用for循環讀取文件中的所有數據。

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的基礎上封裝了一層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讀取整個文件

io/ioutil包的ReadFile方法能夠讀取完整的文件,只需要將文件名作爲參數傳入。

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

文件寫入操作

os.OpenFile()函數能夠以指定的模式打開文件,從而實現文件寫入相關功能。

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

其中:
name 是指要打開的文件名,flag是指打開文件的模式,模式有以下幾種:

Go語言 碎片整理之 文件操作
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

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

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!")
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章