服务端相关 / 40 Go 语言的文件操作

40 Go 语言的文件操作

本文主要介绍 Go 语言中对文件系统的操作,Go 语言内置库中提供了丰富的文件操作函数,所以本文主要介绍使用Go 语言中的内置包——“os” 来实现文件操作。

1. 文件创建/打开

若你想创建/打开一个文件,则可以使用

func OpenFile(name string, flag int, perm uint32) (file *File, err Error)  

参数解释:

  • name:就是文件名,若文件不在执行程序目录下,则可以使用文件的相对路径或者绝对路径。

  • flag:是文件打开模式,其支持模式共有(O_RDONLY|O_WRONLY|O_RDWR三个中必须要选一个):

    • O_RDONLY:以只读方式打开文件;
    • O_WRONLY:以只写方式打开文件;
    • O_RDWR:以读写方式打开文件;
    • O_APPEND:写入时将数据追加到文件;
    • O_CREATE:如果不存在新文件,则创建一个新文件;
    • O_EXCL:与O_CREATE一起使用,文件不能存在;
    • O_SYNC:为同步I/O打开;
    • O_TRUNC:打开时截断常规可写文件;
  • perm:文件权限,一般最高为0777。

代码示例

package main

import (
    "fmt"
    "os"
)

func main() {
    f, err := os.OpenFile("a.txt", os.O_RDWR|os.O_CREATE, 0777)
    if err != nil {
        panic(err)
    }
    fmt.Println(f.Name())
}
  • 第 9 行:以读写的方式打开一个名为a.txt的文件,若文件不存在则创建这个文件,文件权限为0777;
  • 第 13 行:打印这个文件名;

执行结果:

图片描述

2. 文件读取

文件打开之后,就可以从得到的文件对象中读取文件内容。我们在 a.txt 文件中输入 file test 字符串,来验证读取结果。

代码示例:

package main

import (
    "fmt"
    "os"
)

func main() {
    f, err := os.OpenFile("a.txt", os.O_RDWR|os.O_CREATE, 0777)
    if err != nil {
        panic(err)
    }
    b := make([]byte, 10) //最大读取字符数量
    i, err := f.Read(b)
    if err != nil {
        panic(err)
    }
    fmt.Println(i)         //输出读取的字符串的个数
    fmt.Println(string(b)) //输出读取到的字符串
}

执行结果:

图片描述

3. 文件写入

文件写入有两种模式,追加模式和覆写模式。如果在打开文件的时候加上了 O_APPEND 模式,就是追加,如果不加就是覆写。

代码示例:

package main

import (
    "os"
)

func main() {
    f1, err := os.OpenFile("a.txt", os.O_RDWR|os.O_CREATE, 0777)
    if err != nil {
        panic(err)
    }
    f1.WriteString("覆写字符")
    f1.Close()

    f2, err := os.OpenFile("b.txt", os.O_RDWR|os.O_CREATE|os.O_APPEND, 0777)//追加模式打开文件
    if err != nil {
        panic(err)
    }
    f2.WriteString("追加字符")
}

反复执行几遍上述代码,然后打开两个文件,文件内容如下:

覆写的文件中只有"覆写字符"这四个字,因为其写入被不断覆写了。

图片描述

追加的文件中有很多"追加字符",因为其写入是不断的被追加到文件末尾的。

图片描述

4. 小结

本文主要介绍了 Go 语言中自带的 os 包的文件操作,需要注意的是文件的打开中若不确定文件是否存在,就可以加上 O_CREATE。改写文件若是需要追加到文件末尾,记得在打开文件的时候要加上 O_APPEND。