Go文件开关读写操作
梦想橡皮擦 人气:0⛳️ 实战场景
本篇博客为大家再次带来 Go 语言的基础知识，这次要学习的内容是 Go 中的文件操作。
打开关闭文件
在 Go 中操作文件，首先要做的就是导入 os 模块，该模块中具备相关函数定义。
package main
import (
"fmt"
"os"
)
func main() {
// 打开文件
file, err := os.Open("./ca.txt")
if err != nil {
fmt.Println("文件打开失败", err)
return
}
print(file) // 输出地址 0xc000006028
// 关闭文件
file.Close()
}
如果文件打开失败，直接展示错误信息，正确的话，会返回 file 对象的地址。
读取文件
使用 Read 函数读取文件，其定义如下所示：
func (f *File) Read(b [] byte)(n int,err error)
从定义中可以发现，Read 函数会返回 2 个值，第一个是读取的字节数，第二个是当读取报错时，返回错误信息，示例代码如下所示：
package main
import (
"fmt"
"io"
"os"
)
func main() {
// 打开文件
file, err := os.Open("./ca.txt")
if err != nil {
fmt.Println("打开文件失败，错误信息：", err)
return
}
// 为了防止忘记关闭文件，可使用defer注册文件关闭语句
defer file.Close()
// Read方法读取数据
var tmp = make([]byte, 128)
// 读取文件
n, err := file.Read(tmp)
// 碰到 io.EOF 表示文件读取到末尾
if err == io.EOF {
fmt.Println("文件读取完毕")
return
}
if err != nil {
fmt.Println("读取文件失败，错误信息：", err)
return
}
fmt.Printf("读取了 %d 字节数据\n", n)
fmt.Println(string(tmp[:n]))
}
输出如下内容：
读取了 24 字节数据
姓名：梦想橡皮擦
读取到 24 个字节，标准的中文占 3 个字节。
但是上述代码最大的问题就是，我们给变量 tmp，即数组 tmp 仅设置了 128 字节，如果待读取的文件超过了改值，例如 txt 内容如下所示：
姓名：梦想橡皮擦1
姓名：梦想橡皮擦2
姓名：梦想橡皮擦3
姓名：梦想橡皮擦4
姓名：梦想橡皮擦5
此时运行代码，就会发现 Read 函数读取到第 128 个字节的时候，就会停止读取，间接导致读取到的数据乱码。
读取了 128 字节数据
姓名：梦想橡皮擦1
姓名：梦想橡皮擦2
姓名：梦想橡皮擦3
姓名：梦想橡皮擦4
姓名：梦想橡��
解决该问题，需要用到 for 循环读取文件数据，代码如下：
package main
import (
"fmt"
"io"
"os"
)
func main() {
// 打开文件
file, err := os.Open("./ca.txt")
if err != nil {
fmt.Println("打开文件失败，错误信息：", err)
return
}
defer file.Close()
// 声明一个 content 切片
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("读取文件失败，错误信息：", err)
return
}
// 追加数据
content = append(content, tmp[:n]...)
}
fmt.Println(string(content))
}
上述代码声明了一个动态数组，即切片，解决了读取数据不足的问题。
bufio 读取文件
该函数可以一次读取一行数据，示例代码如下：
func main() {
file, err := os.Open("./ca.txt")
if err != nil {
fmt.Println("打开文件失败，错误信息：", 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("读取文件失败，错误信息：", err)
return
}
fmt.Print(line)
}
}
该函数与 Python 中的 readline 具备同样的效果，接下来演示一下读取整个文件，代码如下：
package main
import (
"fmt"
"io/ioutil"
)
func main() {
content, err := ioutil.ReadFile("./ca.txt")
if err != nil {
fmt.Println("打开文件失败，错误信息：", err)
return
}
fmt.Println(string(content))
}
注意导入模块的时候，直接导入了 io 包中的 ioutil 函数。
写文件
使用 os.OpenFile() 函数可以指定模式打开文件，即可以写入的方式打开文件。
写入文件最常用的函数是 Write和 WriteString，示例代码如下所示：
package main
import (
"fmt"
"os"
)
func main() {
file, err := os.OpenFile("ca.txt", os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0666)
if err != nil {
fmt.Println("打开文件失败，错误信息:", err)
return
}
defer file.Close()
name := "橡皮擦"
file.Write([]byte(name)) //写入字节切片
file.WriteString("梦想橡皮擦") //写入字符串
}
使用该方式写入文件，会覆盖之前的文件内容，查看结果可直接打开目录下的 ca.txt 进行查看。
我们将 os.OpenFile("ca.txt", os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0666) 中的参数进行单独的说明。
第一个参数表示文件名，第二个参数是打开文件的模式，具体模式如下所示：
os.O_WRONLY：只写；
os.O_CREATE：创建文件；
os.O_RDONLY：只读；
os.O_RDWR：读写；
os.O_TRUNC：清空；
os.O_APPEND：追加。
可以看到与 Python 中的文件模式基本一致。
最后一个参数值 0666 表示文件权限（八进制），这与 linux 文件权限数字一致，读 r 等于 4（感谢博友指出该错误），写 w 表示 2，执行 x 等于 1，所以 666 表示的是文件可读，可写。
在文件写入部分，还存在 bufio.NewWriter 整行写入，ioutil.WriteFile 整体写入等内容，可以学习的时候重点查阅一下。
加载全部内容