9.2 Go 文件IO操作
9.2 Go 文件IO操作
1.1.1. bufio包
带有缓冲区的IO读写操作，用于读写文件，以及读取键盘输入
func main() { //NewReader返回一个结构体对象指针 reader := bufio.NewReader(os.Stdin) //ReaderLine返回字节切片 data, _, _ := reader.ReadLine() fmt.Printf("data类型：%T\n 数值：%v\n", data, data) //用string()内置函数强转切片为stirng str := string(data) fmt.Printf("data类型：%T\n 数值：%v\n", str, str) }
案例结果
yugoMBP:day01 yuchao$ go run demo01_variable.go 妖怪,放开我爷爷 data类型：[]uint8 数值：[229 166 150 230 148 190 229 188 128 230 136 145 231 136 183 231 136 183] data类型：string 数值：妖放开我爷爷
1.1.2. 文件读写
文件主要用于保存数据，比如图片、视频、音乐，文档等。
程序的输入和输出流。
流： 指数据源(文件)和程序(内存)之间传输的路径 输入流： 数据从文件到内存的路径 输出流： 数据从内存到文件的路径
1.2. os.File
os.File封装了所有文件相关操作。
//File是一个结构体 type File struct{ //内涵隐藏或非导出字段 }
1.3. 打开/关闭文件
使用函数和方法
//打开文件，返回文件句柄指针 func Open(name string)(file *File,err error) func (*File)Close()error //Close关闭文件f，文件此时无法读写。
打开，关闭文件，示例
package main import ( "fmt" "os" ) func main() { //打开文件，获取文件句柄 file, err := os.Open("/Users/yuchao/go/src/gostudy/gobook/mygolang.txt") if err != nil { fmt.Println("打开文件出错了：", err) } //file文件句柄就是一个 指针类型 fmt.Printf("文件句柄类型：%T 值：%v\n ", file, file) //打开文件，注意要释放，关闭文件资源 err = file.Close() if err != nil { fmt.Println("出错了", err) } }
只读取文本
package main import ( "bufio" "fmt" "io" "os" ) func main() { //只读的方式打开文本 file, err := os.Open("filepath") if err != nil { fmt.Println("open file failed, err:", err) return } defer file.Close() reader := bufio.NewReader(file) for { line, err := reader.ReadString('\n') if err == io.EOF { break } if err != nil { fmt.Println("read file failed, err:", err) return } fmt.Println(line) } }
1.4. 写入文件os.OpenFile
Go标准包中os.OpenFile用于打开文件的函数，可以指定参数，文件权限，文件名等。
操作成功返回文件句柄。
func OpenFile(name string, flag int, perm FileMode) (*File, error)
//参数解析
name string 文件名，路径
flag int 文件打开模式
perm FileMode unix文件权限
文件打开模式参数注解( OpenFile函数的参数中的源码)
打开标记： O_RDONLY：只读模式(read-only) O_WRONLY：只写模式(write-only) O_RDWR：读写模式(read-write) O_APPEND：追加模式(append) O_CREATE：文件不存在就创建(create a new file if none exists.) O_EXCL：与 O_CREATE 一起用，构成一个新建文件的功能，它要求文件必须不存在(used with O_CREATE, file must not exist) O_SYNC：同步方式打开，即不使用缓存，直接写入硬盘 O_TRUNC：打开并清空文件 文件权限（unix权限位）：只有在创建文件时才需要，不需要创建文件可以设置为 0。os库虽然提供常量，一般直接写数字，如0664。 如果你需要设置多个打开标记和unix权限位，需要使用位操作符"|"
解析bufio
bufio包实现了有缓冲的I/O。它包装一个io.Reader或io.Writer接口对象，创建另一个也实现了该接口，且同时还提供了缓冲和一些文本I/O的帮助函数的对象。
创建文件实例
package main import ( "bufio" "fmt" "os" ) func main() { filePath := "/Users/yuchao/hello.txt" //O_WRONLY：只写模式(write-only) //O_CREATE：文件不存在就创建 file, err := os.OpenFile(filePath, os.O_WRONLY|os.O_CREATE, 0666) if err != nil { fmt.Printf("打开文件出错了：%v\n", err) return } defer file.Close() //及时关闭文件句柄，防止内存泄露 str := "你好，我是够浪\n" //定义字符串 //写入数据 writer := bufio.NewWriter(file) for i := 0; i < 10; i++ { writer.WriteString(str) } writer.Flush() //将缓冲区的数据，写入到磁盘文件中 }
覆盖写入文件内容
package main import ( "bufio" "fmt" "os" ) func main() { //打开文件，覆盖原有数据 //写入新数据，每天多努力一点，你会发现你更适合加班 filePath := "/Users/yuchao/hello.txt" //O_WRONLY：只写模式(write-only) //O_TRUNC：打开并清空文件 file, err := os.OpenFile(filePath, os.O_WRONLY|os.O_TRUNC, 0666) if err != nil { fmt.Printf("打开文件失败：%v\n", err) return } //关闭文件句柄 defer file.Close() //写入新数据 str := "每天多努力一点，你会发现你更适合加班\n" //写入携带缓存的 *Writer writer := bufio.NewWriter(file) //循环写入5条数据 for i := 0; i < 5; i++ { writer.WriteString(str) } /* writer是携带缓存的，因此调用writerString方法， 其实内容是先写入到缓存中，必须调用flush，将缓冲的数据写入到文件中 */ writer.Flush() }
打开已存在的文件，追加写入内容
package main import ( "bufio" "fmt" "os" ) func main() { filePath := "/Users/yuchao/hello.txt" //O_WRONLY：只写模式(write-only) //O_APPEND：追加模式(append) file, err := os.OpenFile(filePath, os.O_WRONLY|os.O_APPEND, 0666) if err != nil { fmt.Printf("打开文件失败：%v\n", err) return } //关闭文件句柄 defer file.Close() //写入新数据 str := "真的吗？那我一定要好好加班\n" //写入携带缓存的 *Writer writer := bufio.NewWriter(file) //循环写入5条数据 for i := 0; i < 5; i++ { writer.WriteString(str) } /* writer是携带缓存的，因此调用writerString方法， 其实内容是先写入到缓存中，必须调用flush，将缓冲的数据写入到文件中 */ writer.Flush() }
打开一个已有文件，读取出原本的内容显示在终端，并且追加5句"加班万岁"。
package main import ( "bufio" "fmt" "io" "os" ) func main() { //打开一个已有文件，`读取`出原本的内容显示在终端，并且`追加`5句`"加班万岁"`。 filePath := "/Users/yuchao/hello.txt" //O_WRONLY：只写模式(write-only) //O_APPEND：追加模式(append) //OpenFile返回一个文件指针 file, err := os.OpenFile(filePath, os.O_RDWR|os.O_APPEND, 0666) if err != nil { fmt.Printf("打开文件失败：%v\n", err) return } //关闭文件句柄 defer file.Close() //读取原有内容，打印输出 reader := bufio.NewReader(file) for { str, err := reader.ReadString('\n') //如果读取到结尾，结束循环 if err == io.EOF { break } if err != nil { fmt.Println("读取文件出错了：", err) } //显示读取到的文本 fmt.Print(str) } //准备写入新内容 newStr := "加班万岁\n" //使用带缓存的*Writer writer := bufio.NewWriter(file) writer.WriteString(newStr) writer.Flush() }
1.5. 文件拷贝
package main import ( "fmt" "io" "os" ) //拷贝文件 dst目的文件 src源文件 func CopyFile(dstFile string, srcFile string) (written int64, err error) { src, err := os.Open(srcFile) if err != nil { fmt.Println("打开文件出错", err) } defer src.Close() dst, err := os.OpenFile(dstFile, os.O_WRONLY|os.O_CREATE, 0644) if err != nil { fmt.Printf("打开文件%s出错,错误：%v\n", dstFile, err) return } defer dst.Close() return io.Copy(dst, src) } func main() { // _, err := CopyFile("/Users/yuchao/helloCopy.txt", "/Users/yuchao/hello.txt") if err == nil { fmt.Println("拷贝完成") } else { fmt.Println("拷贝出错了，错误：", err) } }
1.6. 获取命令行参数
如linux命令
ls -la
os.args包是一个string的切片，用于存储所有命令行参数。
package main import ( "fmt" "os" ) func main() { fmt.Printf("命令行参数有%d个\n", len(os.Args)) for i, v := range os.Args { fmt.Printf("参数%d 是 %v\n", i, v) } }
对代码编译后，进行传参运行
go build main.go
./main alex peiqi xiaofeng //传入三个参数
1.7. flag包解析命令行参数
package main import ( "flag" "fmt" ) func main() { /* flag包可以解析如下命令 ./main -u root -p 123 -h 127.0.0.1 -port 3306 */ //定义用于接收命令行的参数值 var user string var pwd string var host string var port int flag.StringVar(&user, "u", "", "用户名，默认为空") flag.StringVar(&pwd, "p", "", "密码默认为空") flag.StringVar(&host, "h", "localhost", "主机名，默认localhost") flag.IntVar(&port, "port", 3306, "端口号，默认3306") flag.Parse() fmt.Printf("user=%v pwd=%v host=%v port=%v\n", user, pwd, host, port) }
运行flag包解析
yugoMBP:gobook yuchao$ go run main.go -u root -p 123 -h 127.0.0.1 -port 3306 user=root pwd=123 host=127.0.0.1 port=3306
