一篇文章带你搞定Go语言基础之文件操作

Go进阶者

发布于 2021-01-22 10:56:56

3040

发布于 2021-01-22 10:56:56

文章被收录于专栏：Go语言进阶学习Go语言进阶学习

前言

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(目标文件，原文件)
}

执行结果