关于Go语言中的IO操作详解
作者：景天科技苑
在现代软件开发中,高效的输入输出（I/O）操作是提高程序性能的关键之一,Go语言提供了丰富的I/O操作接口,使得文件读写、网络通信等任务变得简单而高效,本文介绍了关于Go语言中的IO操作,需要的朋友可以参考下
在现代软件开发中，高效的输入输出（I/O）操作是提高程序性能的关键之一。Go语言提供了丰富的I/O操作接口，使得文件读写、网络通信等任务变得简单而高效。
Go语言的I/O操作主要通过标准库中的io包和os包实现。o包提供了一系列用于输入和输出操作的基本接口和原语。这些接口和原语构成了Go语言中处理I/O操作的基础，支持多种I/O设备的读写操作，包括文件、网络连接、内存缓冲区等。而os包则提供了对操作系统功能的封装，包括文件操作、进程管理等。
一、os包简介
os包是Go语言标准库中的一个重要包，提供了操作系统相关的功能接口。通过os包，我们可以执行各种底层操作，如读取和写入文件、获取系统信息、执行外部命令等。
二、常用函数与示例
1. 文件操作
1.1 获取文件信息
Stat：获取文件信息对象，符号链接会跳转。
Lstat：获取文件信息对象，符号链接不跳转。
package main
import (
"fmt"
"os"
)
//看下fileinfo接口
// file
// fileInfo
/*
type FileInfo interface {
Name() string // base name of the file
Size() int64 // length in bytes for regular files; system-dependent for others
Mode() FileMode // file mode bits ： 权限
ModTime() time.Time // modification time
IsDir() bool // abbreviation for Mode().IsDir()
// 获取更加详细的文件信息， *syscall.Stat_t 反射来获取
Sys() any // underlying data source (can return nil)
*/
func main() {
// 获取某个文件的状态
//func Stat(name string) (FileInfo, error)
fileinfo, err := os.Stat("F:\\goworks\\src\\jingtian\\yufa\\type别名\\type别名.go")
if err != nil {
fmt.Println(err)
return
}
fmt.Println(fileinfo.Name()) // 文件名 demo01.go
fmt.Println(fileinfo.IsDir()) // 是否目录 false
fmt.Println(fileinfo.ModTime()) // 文件的修改时间 2024-09-26 10:03:01.0946679 +0800 CST
fmt.Println(fileinfo.Size()) // 文件大小 1542 字节数
fmt.Println(fileinfo.Mode()) // 权限 -rw-rw-rw-
}
1.2 创建文件
Create：创建文件，如果文件存在，则清空原文件。
package main
import (
"fmt"
"os"
)
func main() {
//创建文件，文件不存在就创建文件，如果文件存在，则清空原文件。
// func Create(name string) (*File, error)
file, err := os.Create("example.txt")
if err != nil {
fmt.Println(err)
return
}
//关闭文件，延迟执行
defer file.Close()
//向文件中写内容
// func (f *File) WriteString(s string) (n int, err error)
_, err = file.WriteString("Hello, World!")
if err != nil {
fmt.Println(err)
}
fmt.Println("File created successfully")
}
文件创建，并且写入成功
创建的文件默认权限是0666
1.3 删除文件
Remove：删除文件或目录（单一文件）只能删除空文件夹，文件不存在会报错。如果文件夹中有文件也删不掉
RemoveAll：递归删除文件或目录，文件或目录不存在不报错，返回nil。
package main
import (
"fmt"
"os"
)
func main() {
// 删除文件
// func Remove(name string) error
//文件不存在会报错
err := os.Remove("example.txt")
if err != nil {
fmt.Println(err)
} else {
fmt.Println("File removed successfully")
}
// 递归删除目录
// func RemoveAll(path string) error
//目录不存在返回nil 不会报错
//可以删除指定目录下的目录或文件
err = os.RemoveAll("testdir\\mydir.go")
if err != nil {
fmt.Println(err)
} else {
fmt.Println("Directory removed successfully")
}
}
1.4 重命名或移动文件
Rename：重命名文件或移动文件。
package main
import (
"fmt"
"os"
)
func main() {
// 重命名文件
// func Rename(oldpath, newpath string) error
err := os.Rename("张三.txt", "新张三.txt")
if err != nil {
fmt.Println(err)
} else {
fmt.Println("File renamed successfully")
}
// 移动文件
err = os.Rename("新张三.txt", "testdir\\新张三.txt")
if err != nil {
fmt.Println(err)
} else {
fmt.Println("File moved successfully")
}
}
1.5 打开文件
使用os.Open函数可以打开一个文件，返回一个*os.File类型的文件对象。如果文件打开失败，会返回一个错误。
package main
import (
"fmt"
"os"
)
func main() {
//func Open(name string) (*File, error)
file, err := os.Open("example.txt")
if err != nil {
fmt.Println("Error opening file:", err)
return
}
defer file.Close()
fmt.Println("File opened successfully")
//打印打开的文件，是个内存地址
fmt.Println("Reading file:", file)
}
打开文件常用方式二：os.OpenFile
在Go语言中，文件的读写操作是使用非常频繁的功能之一。os.OpenFile函数是Go标准库中用于打开文件的强大工具，它允许以多种模式打开文件，并可以设置文件的权限。
os.OpenFile函数简介
os.OpenFile函数用法如下：
func OpenFile(name string, flag int, perm FileMode) (File, error)
name：要打开的文件名或路径。
flag：打开文件的模式，可以是多个标志的按位或组合。常见的标志包括：
os.O_RDONLY：只读模式。
os.O_WRONLY：只写模式。
os.O_RDWR：读写模式。
os.O_APPEND：追加模式，在文件末尾写入数据而不覆盖原有数据。只要有这个参数，就会采用追加模式
os.O_CREATE：如果文件不存在则创建文件。
os.O_EXCL：与O_CREATE一起使用，可执行权限。
os.O_SYNC：同步模式，将文件更改刷新到磁盘。
os.O_TRUNC：截断模式，清空文件内容。
perm：文件权限，表示文件的读写权限，默认为0666（即所有用户都可读写）。
返回值是一个指向打开文件的File对象和一个可能出现的错误。如果文件打开成功，则可以通过该文件对象进行读取和写入操作。
使用os.OpenFile写文件
以下是一个使用os.OpenFile函数写文件的示例：
package main
import (
"fmt"
"os"
)
func main() {
// 以追加、创建、读写模式打开文件
file, err := os.OpenFile("./example.txt", os.O_APPEND|os.O_CREATE|os.O_RDWR, 0666)
if err != nil {
fmt.Println("打开文件失败:", err)
return
}
defer file.Close() // 确保在函数执行完毕后关闭文件
// 写入文件
_, err = file.Write([]byte("Hello, Go!\n"))
if err != nil {
fmt.Println("写入文件失败:", err)
return
}
fmt.Println("文件写入成功")
}
将内容追加到文件中
1.6 读取文件
使用os.File对象的Read方法可以从文件中读取数据。Read方法会将数据读取到一个字节切片中，并返回读取的字节数和可能发生的错误。
需要借助 file.Read([]byte)读取 ，将file中的数据读取到 []byte 中， n，err n读取到的行数，err 错误，EOF错误，就代表文件读取完毕了
一直调用read，就代表光标往后移动…
以下是一个使用os.Open函数读文件的示例：
package main
import (
"fmt"
"os"
)
// 读取文件数据
func main() {
// 我们习惯于在建立连接时候通过defer来关闭连接，保证程序不会出任何问题，或者忘记关闭
// 建立连接
file, _ := os.Open("example.txt")
// 关闭连接
defer file.Close()
// 读代码 ,Go 的错误机制，让我们专心可以写业务代码。
// 1、创建一个容器 （二进制文本文件--0100101010 => 读取流到一个容器 => 读取容器的数据）
//一次性读取的内容长度为切片的长度，切片长度为2，容量为1024
bs := make([]byte, 2, 1024) // 缓冲区，可以接受我们读取的数据 这里设置的是一次性读取两个字节
fmt.Printf("字节切片数据类型%T\n", bs)
// 2、读取到缓冲区中。 // 汉字一个汉字占 3个字节
// func (f *File) Read(b []byte) (n int, err error) 返回读到的字节数和错误
// At end of file, Read returns 0, io.EOF. 读到文件末尾返回0和io.EOF
n, err := file.Read(bs)
fmt.Println(n)
fmt.Println(err)
//每次读取光标移动切片的长度，将读取的内容存到切片中。如果读取的文件是文本数据，可以通过string方法将切片中的内容转换为字符串
//如果是MP4呀，音乐，图片等文件，就不要用string转换了
fmt.Println(string(bs)) // 读取到的字符串 ab
//继续往下读
// 光标不停的向下去指向，读取出来的内容就存到我们的容器中。
file.Read(bs)
fmt.Println(string(bs)) // 读取到的字符串 cd
file.Read(bs)
fmt.Println(string(bs)) // 读取到的字符串 e
//读到文件末尾，得到的n是0，err是EOF
n, err = file.Read(bs)
fmt.Println(n)
fmt.Println(err) // EOF ，读取到了文件末尾。err就会返回EOF。
//最后读到了e，将原来切片中的cd的c覆盖了，得到的最终结果是ed
fmt.Println(string(bs)) // 读取到的字符串
}
1.7 IO写
写的话，要考虑文件的权限问题，大致流程如下
建立连接 （设置权限：可读可写，扩充这个文件的append） os.OpenFile
关闭连接
写入文件常用的方法
- file.Write
- file.WriteString
先看下example.txt文件
package main
import (
"fmt"
"os"
)
func main() {
fileName := "example.txt"
// 权限：如果我们要向一个文件中追加内容， O_APPEND, 如果没有，就是从头开始写
file, _ := os.OpenFile(fileName, os.O_WRONLY|os.O_RDONLY|os.O_APPEND, os.ModePerm)
defer file.Close()
// 操作 字节切片写入
bs := []byte{65, 66, 67, 68, 69} // A B C D E ASCII编码
// func (f *File) Write(b []byte) (n int, err error)
//write方法 写入的是二进制数据字节流，即字节切片
n, err := file.Write(bs)
if err != nil {
fmt.Println(err)
}
fmt.Println(n)
// string类型的写入，可以直接将字符串写入到文件中
n, err = file.WriteString("hhahahahah哈哈哈哈哈哈哈")
if err != nil {
fmt.Println(err)
}
fmt.Println(n)
}
再看下example.txt文件，可以看到我们写入的内容全部追加到文件末尾
如果打开文件的方式不包含os.O_APPEND 则是将源文件覆盖写。os.O_APPEND的意思是，在打开文件时，将光标移动到文件末尾
2. 目录操作
2.1 创建目录
Mkdir：使用指定的权限和名称创建一个目录，目录存在会报错。
MkdirAll：递归创建目录，包括任何必要的上级目录，目录存在不报错。
package main
import (
"fmt"
"os"
)
func main() {
// 创建单个目录
// func Mkdir(name string, perm FileMode) error
// os.ModePerm 是0777的权限
//mkidr创建目录，目录存在的情况下，会报错 Cannot create a file when that file already exists.
err := os.Mkdir("singledir", os.ModePerm)
if err != nil {
fmt.Println(err)
} else {
fmt.Println("Single directory created successfully")
}
// 递归创建目录，目录存在不报错
// func MkdirAll(path string, perm FileMode) error
err = os.MkdirAll("nested/dirs/test", os.ModePerm)
if err != nil {
fmt.Println(err)
} else {
fmt.Println("Nested directories created successfully")
}
}
2.2 更改当前工作目录
Getwd：获取当前工作目录。
Chdir：修改当前工作目录。
package main
import (
"fmt"
"os"
)
func main() {
// 获取当前工作目录
path, err := os.Getwd()
if err != nil {
fmt.Println(err)
return
}
fmt.Println("Current working directory:", path)
// 修改当前工作目录
err = os.Chdir("../")
if err != nil {
fmt.Println(err)
return
}
path, err = os.Getwd()
if err != nil {
fmt.Println(err)
return
}
fmt.Println("New working directory:", path)
}
3. 环境变量
3.1 获取环境变量
Environ：返回所有的环境变量，格式为“key=value”的字符串切片。
Getenv：检索并返回名为key的环境变量的值，如果不存在则返回空字符串。
Setenv:
函数定义：func Setenv(key, value string) error
功能：设置名为key的环境变量，如果出错会返回该错误。
package main
import (
"fmt"
"os"
"strings"
)
func main() {
// 设置环境变量
err := os.Setenv("MY_ENV_VAR", "my_value")
if err != nil {
fmt.Println(err)
} else {
fmt.Println("Environment variable set successfully")
}
// 获取并打印刚设置的环境变量
value := os.Getenv("MY_ENV_VAR")
fmt.Println("MY_ENV_VAR:", value)
//查看所有环境变量
envs := os.Environ()
for _, env := range envs {
cache := strings.Split(env, "=")
fmt.Printf("key=%v value=%v\n", cache[0], cache[1])
}
// 清除所有环境变量（谨慎使用）
// err = os.Clearenv()
// if err != nil {
// fmt.Println(err)
// } else {
// fmt.Println("All environment variables cleared")
// }
}
4. 程序退出
Exit：让当前程序以给出的状态码code退出。一般来说，状态码0表示成功，非0表示出错。程序会立刻终止，并且defer的函数不会被执行。
package main
import (
"fmt"
"os"
)
func main() {
// 正常退出
fmt.Println("Exiting the program successfully")
//os.Exit(0)
// 异常退出（不会执行到这里的代码）
fmt.Println("This line will not be executed")
os.Exit(1)
}
5. 获取系统信息
Hostname：返回内核提供的主机名。
Getuid：返回调用者的用户ID。
Getgid：返回调用者的组ID。
Getpid：返回调用者所在进程的进程ID。
package main
import (
"fmt"
"os"
)
func main() {
// 获取主机名
hostname, err := os.Hostname()
if err != nil {
fmt.Println(err)
return
}
fmt.Println("Hostname:", hostname)
// 获取用户ID
uid := os.Getuid()
fmt.Println("User ID:", uid)
// 获取组ID
gid := os.Getgid()
fmt.Println("Group ID:", gid)
// 获取进程ID
pid := os.Getpid()
fmt.Println("Process ID:", pid)
}
6. 执行外部命令
os/exec包用于执行外部命令，它提供了更高级别的接口来创建和管理进程。
1. 执行简单命令
package main
import (
"fmt"
"os/exec"
)
func main() {
//查看系统中是否安装有该命令
cmd_path, err := exec.LookPath("ipconfig")
if err != nil {
fmt.Println(err)
return
}
fmt.Println("系统中有该命令，命令路径为:", cmd_path)
//Command每个参数都不能包含空格。多个参数，用引号包裹，逗号隔开
cmd := exec.Command("ipconfig")
// 如果不需要命令的输出，直接调用cmd.Run()即可
err2 := cmd.Run()
if err2 != nil {
fmt.Println("Error:", err2)
} else {
fmt.Println("Command executed successfully")
}
}
使用exec.Command函数创建一个代表外部命令的Cmd对象，然后使用Run方法执行命令并等待其完成。
我们使用 exec.Command() 创建了一个执行 ipconfig 命令的 *exec.Cmd，然后使用 Run() 方法启动了命令。
请注意，使用 exec.Command() 创建的命令默认不会有任何输出。如果我们想获取命令的标准输出，我们可以使用 Output() 或 CombinedOutput() 方法。如果我们想获取命令的标准错误输出，我们需要单独设置 Cmd.Stderr 字段。
2. 获取命令输出
使用Output方法运行命令并返回其标准输出的内容。
package main
import (
"fmt"
"os/exec"
)
func main() {
//查看系统中是否安装有该命令
cmd_path, err := exec.LookPath("ipconfig")
if err != nil {
fmt.Println(err)
return
}
fmt.Println("系统中有该命令，命令路径为:", cmd_path)
cmd := exec.Command("ipconfig")
output, err := cmd.Output()
if err != nil {
fmt.Println("Error:", err)
} else {
fmt.Println("Command output:", string(output))
}
}
3. 处理标准输出和标准错误
CombinedOutput
func (c *Cmd) CombinedOutput() ([]byte, error)
说明：运行命令并返回组合到一起的标准输出和标准错误
package main
import (
"fmt"
"os/exec"
)
func main() {
//查看系统中是否安装有该命令
cmd_path, err := exec.LookPath("ping")
if err != nil {
fmt.Println(err)
return
}
fmt.Println("系统中有该命令，命令路径为:", cmd_path)
// 创建一个新的命令，例如运行dir命令
cmd := exec.Command("ping", "www.baidu.com")
//获取标准输出
out, err := cmd.CombinedOutput()
if err != nil {
fmt.Println(err)
return
}
fmt.Println("标准输出为:", string(out))
}
三、文件复制
文件复制需要考虑的问题：
1、拿到需求，分析需求，编辑函数 copy()
2、编写参数(源文件，目标地址，缓冲区大小)
3、读取源文件，保存到缓冲区中
4、读取缓冲区的内容，将内容写入到新文件中
5、确保源文件读取完毕了，使用到for()循环。循环结束标志 io.EOF或者读取到的字节数为0
1. 自定义的文件复制实现
我们创建个目录utils。里面创建个copy.go
package utils
import (
"fmt"
"io"
"os"
)
// Copy 方法需要参数为：source 源文件 ,destination 目标文件，缓冲区大小
func Copy(source, destination string, bufferSize int) {
// 读取文件，读文件open方法就可以
sourceFile, err := os.Open(source)
if err != nil {
fmt.Println("Open错误:", err)
}
// 输出文件 O_WRONLY , O_CREATE 如果不不存在，则会创建
destinationFile, err := os.OpenFile(destination, os.O_WRONLY|os.O_CREATE, os.ModePerm)
if err != nil {
fmt.Println("OpenFile错误:", err)
}
// 关闭连接
defer sourceFile.Close()
defer destinationFile.Close()
// 专注业务代码，拷贝
// 创建缓冲区大小
buf := make([]byte, bufferSize)
// 循环读取，写入
for {
//一次读取设置的缓冲区大小的数据
n, err := sourceFile.Read(buf)
// 循环结束标志
if n == 0 || err == io.EOF {
fmt.Println("读取完毕源文件,复制完毕")
break
} else if err != nil {
fmt.Println("读取错误:", err)
return // 错误之后，必须要return终止函数执行。
}
// 将缓冲区的东西写出到目标文件
// 因为循环写入过程中没有关闭文件，所以可以持续写入
//根据实际读取的长度写入，防止最后一次读取时，读取的少于缓冲区长度
_, err = destinationFile.Write(buf[:n])
if err != nil {
fmt.Println("写出错误：", err)
}
}
}
调用我们的copy方法，实现文件copy
package main
import "jingtian/yufa/io操作/utils"
func main() {
source := "C:\\Users\\dell\\Desktop\\appsearch.zip"
dest := "F:\\goworks\\src\\jingtian\\yufa\\io操作\\目标文件.zip"
utils.Copy(source, dest, 1024)
}
运行，可以将源文件复制到目标地址
2. 使用系统自带的io.Copy()方法复制
// Copy2 使用系统自带的io.Copy方法
func Copy2(source, destination string) {
// 读取文件
sourceFile, err := os.Open(source)
if err != nil {
fmt.Println("Open错误:", err)
}
// 输出文件 O_WRONLY , O_CREATE 如果不不存在，则会创建
destinationFile, err := os.OpenFile(destination, os.O_WRONLY|os.O_CREATE, os.ModePerm)
if err != nil {
fmt.Println("OpenFile错误:", err)
}
// 关闭
defer sourceFile.Close()
defer destinationFile.Close()
// 具体的实现
// 不用我们自己写循环读取写入了，直接调用系统的io.Copy()方法
// func Copy(dst Writer, src Reader) (written int64, err error) 返回写入的字节数和错误
written, err := io.Copy(destinationFile, sourceFile)
if err != nil {
fmt.Println("复制时出现错误", err)
return
}
fmt.Println("文件的字节大小:", written)
}
3. 使用os.ReadFile和os.WriteFile来实现读取写入，从而实现复制
// Copy3 使用os.ReadFile和os.WriteFile来实现读取写入，从而实现复制
func Copy3(source, destination string) {
// func ReadFile(name string) ([]byte, error)
fileBuffer, _ := os.ReadFile(source)
//func WriteFile(name string, data []byte, perm FileMode) error
err := os.WriteFile(destination, fileBuffer, 0644)
if err != nil {
fmt.Println(err)
return
}
fmt.Println("文件复制完成")
}
4. io.CopyBuffer() 适合大文件复制
// Copy4 io.CopyBuffer() 适合大文件
func Copy4(source, destination string, bufferSize int) {
// 读取文件
sourceFile, err := os.Open(source)
if err != nil {
fmt.Println("Open错误:", err)
}
// 输出文件 O_WRONLY , O_CREATE 如果不不存在，则会创建
destinationFile, err := os.OpenFile(destination, os.O_WRONLY|os.O_CREATE, os.ModePerm)
if err != nil {
fmt.Println("OpenFile错误:", err)
}
// 关闭
defer sourceFile.Close()
defer destinationFile.Close()
// 具体的实现
// 创建缓冲区大小
buf := make([]byte, bufferSize)
// 不用我们自己写循环读取写入了，直接调用系统的io.Copy()方法
//io.copy这个适合小文件，大文件使用io.CopyBuffer
// func CopyBuffer(dst Writer, src Reader, buf []byte) (written int64, err error)
written, err := io.CopyBuffer(destinationFile, sourceFile, buf)
if err != nil {
fmt.Println("复制时出现错误", err)
return
}
fmt.Println("文件的字节大小:", written)
}
以上就是关于Go语言中的IO操作详解的详细内容，更多关于Go语言IO操作的资料请关注脚本之家其它相关文章！