当前位置：网站首页>Go language Go language, understand Go language file operation in one article

Go language Go language, understand Go language file operation in one article

2022-08-04 18:00:00 【dream eraser】

文章目录

- ️ 实战场景

️ 实战场景

本篇博客为大家再次带来 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("梦想橡皮擦") //写入字符串

}