一篇文章帶你搞定Go語言基礎之文件操作

作者：Go進階者 2021-01-13 08:40:04

開發前端

本次章節我們主要學習文件操作的相關知識，主要分為讀取文件和寫入文件

[[375517]]

前言

Hey，大家好呀，我是碼農，星期八，這次咱們來說一下文件操作。

文件操作就簡單了，像打開word了，excel了，都是文件操作，當然，我們肯定是不能直接操作word的

我們就從最簡單的普通文件開始叭!

打開和關閉文件

package main 
 
import "os" 
 
func main() { 
    //os.Open是只讀模式 
    fileObj， err := os.Open("永不退縮.txt") 
    if err != nil { 
        panic(err) 
} 
    //關閉文件 
    fileObj.Close() 
    //一般情況下使用這種方式關閉文件 
    defer fileObj.Close() 
}

注:如果你使用的是Goland，在學習文件操作這一塊時，盡量不要右擊運行，因為查找文件路徑不一樣

盡可能的編譯成.exe文件，這樣找的就是當前目錄

讀取文件

讀取指定大小

代碼

func main() { 
    //os.Open是只讀模式 
    fileObj， err := os.Open("永不退縮.txt") 
    if err != nil { 
        panic(err) 
} 
    var fileBytes = make([]byte，128) 
    n， err := fileObj.Read(fileBytes) 
    if err != nil { 
        panic(err) 
} 
    fmt.Println(string(fileBytes[:n])) 
    //一般情況下使用這種方式關閉文件 
    defer fileObj.Close() 
}

執行結果

可以發現，根本就沒有讀取完，并且還亂碼了。

讀取整個文件

上述我們只讀取了128個字節，一個中文3個字節，128/3，除下標點符號，應該是40個左右，明顯不夠

那該怎么辦才能讀取所有呢???

上述我們只讀取了128個字節，我們可以在讀取128個字節啊

然后找個罐子將每次讀取的都裝進去，讀取完，裝完，完美

代碼

func main() { 
    //os.Open是只讀模式 
    fileObj， err := os.Open("永不退縮.txt") 
    if err != nil { 
        panic(err) 
} 
    //一般情況下使用這種方式關閉文件 
    defer fileObj.Close() 
 
    var 罐子 []string 
    var 每次讀取字節 = make([]byte，128) 
    for{ 
        n， err := fileObj.Read(每次讀取字節) 
        //err == io.EOF表示讀完了，一定要放在err != nil前面 
        if err == io.EOF { 
            break 
        } 
        if err != nil { 
            panic(err) 
        } 
        var 每次讀取字符串 = string(每次讀取字節[:n]) 
        罐子 = append(罐子，每次讀取字符串) 
} 
    fmt.Println(罐子) 
}

執行結果

會發現還是有亂碼，這是為啥???

這是因為我們每次都是按照字節來讀取一部分，一部分的，但是中文是3個字節，所有有時候可能切錯了，就出現了亂碼

那咋辦呢?沒辦法，默認的讀取文件就這么點功能，更多功能，盡在bufio

bufio

bufio可以理解為原生file操作的一個加強版，更牛叉，功能更多

還是上述這個文件，看看如何完美讀取

代碼

func main() { 
    //os.Open是只讀模式 
    fileObj， err := os.Open("永不退縮.txt") 
    if err != nil { 
        panic(err) 
} 
    //一般情況下使用這種方式關閉文件 
    defer fileObj.Close() 
 
    //需要將文件對象傳進去 
    reader := bufio.NewReader(fileObj) 
    for{ 
        //按行讀取 
        row， err := reader.ReadString('\n')//參數是字符，不是字符串 
        if err == io.EOF { 
            break 
        } 
        if err != nil { 
            panic(err) 
        } 
        fmt.Printf("%v"，row) 
} 
}

執行結果

讀取整個文件

可能又有人說了，啊，星期八，我感覺還是麻煩，有沒有更簡單的辦法讀取文件，肯定有呀，一次性讀取所有呢?答案是肯定的，此時需要用到ioutil包。

代碼

func main() { 
    //os.Open是只讀模式 
    bytes， err := ioutil.ReadFile("永不退縮.txt") 
    if err != nil { 
        panic(err) 
} 
    fmt.Println(string(bytes)) 
}

執行結果

但是這種有個缺點，只能讀取小文件，要是來個10G文件也這樣玩，保證電腦死翹翹!!!

寫入文件

寫入文件的話，我們就需要使用openFile這個方法打開文件

在開始之前呢，先記一下下面幾種模式

os.O_WRONLY 只寫
os.O_RDONLY 只讀
os.O_CREATE 如果文件不存在，則創建文件
os.O_RDWR 可讀可寫
os.O_TRUNC 清空
os.O_APPEND 追加文件

寫入字節和寫入行

代碼

func main() { 
    fileObj， err := os.OpenFile("臨時.txt"， os.O_CREATE|os.O_TRUNC|os.O_WRONLY， 0666) 
    if err != nil { 
        panic(err) 
} 
    defer fileObj.Close() 
    w_content := "我是臨時表內容\n" 
    //Write方法需要將字符串轉成字節 
    fileObj.Write([]byte(w_content)) 
    //WriteString直接寫入字符串 
    fileObj.WriteString(w_content) 
}

執行結果

bufio寫文件

代碼

func main() { 
    fileObj， err := os.OpenFile("臨時.txt"， os.O_CREATE|os.O_TRUNC|os.O_WRONLY， 0666) 
    if err != nil { 
        panic(err) 
} 
    defer fileObj.Close() 
    w_content := "我是臨時表內容\n" 
    writer := bufio.NewWriter(fileObj) 
    writer.Write([]byte(w_content)) 
    writer.WriteString(w_content) 
    //寫完必須刷入 
    writer.Flush() 
}

執行結果

ioutil寫文件

代碼

func main() { 
    w_content := "我是臨時表內容\n" 
    err := ioutil.WriteFile("臨時.txt"， []byte(w_content)， 0666) 
    if err != nil { 
        panic(err) 
} 
}

執行結果

拷貝文件

拷貝文件，就是拷貝文件唄，A文件拷貝到B文件中

其實他的底層還是打開倆文件，把A文件內容寫入到B文件中

代碼

func main() { 
    原文件， err := os.Open("永不退縮.txt") 
    if err != nil { 
        panic(err) 
} 
    defer 原文件.Close() 
    目標文件， err := os.OpenFile("臨時.txt"， os.O_WRONLY|os.O_CREATE， 0666) 
    if err != nil { 
        panic(err) 
} 
    defer 目標文件.Close() 
    //拷貝文件 
    io.Copy(目標文件，原文件) 
}