一、打开和关闭文件
Golang标准库的 os 包中 Open() 函数能够打开一个文件，返回一个 *File 和一个 err。
对打开的文件实例 *File 调用 close() 方法能够关闭文件。
为了防止文件忘记关闭，通常使用 defer 注册文件关闭语句:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
package main
import (
"fmt"
"os"
)
func main() {
//只读方式打开当前目录下的main.go文件
file,err:= os.Open("./main.go")
if err !=nil {
fmt.PrintIn("open file failed!,err:",err)
return
}
//关闭文件
file.Close()
}
|
os.OpenFile() 函数能够以指定模式打开文件，从而实现文件写入相关功能，函数签名如下：
1
|
func OpenFile(name string,flag int,perm FileMode) (*File,error)
|
参数说明:
name：要打开的文件名flag：打开文件的模式，模式有以下几种
| 模式 |
含义 |
os.O_WRONLY |
只写 |
OS.O_CREATE |
创建文件 |
os.O_RDONLY |
只读 |
os.O_RDWR |
读写 |
os.O_TRUNC |
清空 |
os.O_APPEND |
追加 |
perm：创建文件时指定创建的文件权限，一个八进制数，r（读）04，w（写）02，x（执行）01。
二、读取文件
2.1 file.Read() 读取指定 size
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
|
package main
import (
"fmt"
"io"
"os"
)
func main() {
// 只读方式打开当前目录下的main.go文件
file, err := os.Open("./main.go")
if err != nil {
fmt.Println("open file failed!, err:", err)
return
}
defer file.Close()
// 使用Read方法读取数据，每次最多只读取128个字节
var tmp = make([]byte, 128)
n, err := file.Read(tmp)
if err == io.EOF {
fmt.Println("文件读完了")
return
}
if err != nil {
fmt.Println("read file failed, err:", err)
return
}
fmt.Printf("读取了%d字节数据\n", n)
fmt.Println(string(tmp[:n]))
}
|
2.2 循环读取
使用 for 循环读取文件中的所有数据
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
|
package main
import (
"fmt"
"io"
"os"
)
func main() {
// 只读方式打开当前目录下的main.go文件
file, err := os.Open("./main.go")
if err != nil {
fmt.Println("open file failed!, err:", err)
return
}
defer file.Close()
// 循环读取文件
var content []byte
var tmp = make([]byte, 128)
for {
n, err :=file.Read(tmp)
if err == io.EOF {
fmt.PrintIn('文件读完了')
break
}
if err != nil {
fmt.Println("read file failed, err:", err)
return
}
content = append(content, tmp[:n]...)
}
fmt.Println(string(content))
}
|
2.3 bufio 按行读取
bufio 是在 file 的基础上封装了一层API，支持更多的功能
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
package main
import (
"bufio"
"fmt"
"io"
"os"
)
// bufio按行读取示例
func main() {
file, err := os.Open("./main.go")
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 {
if len(line) != 0 {
fmt.Println(line)
}
fmt.Println("文件读完了")
break
}
if err != nil {
fmt.Println("read file failed, err:", err)
return
}
fmt.Print(line)
}
}
|
2.4 ioutil 读取整个文件
io/iotil 包的 ReadFile 函数能够读取完整的文件，只需要将文件名作为参数传入
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
package main
import (
"fmt"
"io/ioutil"
)
//ioutil.ReadFile读取整个文件
func main() {
content,err := ioutil.ReadFile("./main.go")
if err != nil{
fmt.PrintIn("read file failed ,err:",err)
return
}
fmt.PrintIn(string(content))
}
|
ioutil.ReadFile 非常适合用于读取配置文件。
三、写入文件
3.1 os.File 的 Write 和 WriteString 方法
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
|
package main
import (
"fmt"
"os"
)
func main() {
file, err := os.OpenFile("test.txt", os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0666)
if err != nil {
fmt.Println("open file failed, err:", err)
return
}
defer file.Close()
str := "Hello golang."
file.Write([]byte(str)) //写入字节切片数据
file.WriteString(str) //直接写入字符串数据
}
|
3.2 bufio.NewWriter
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
package main
import (
"bufio"
"fmt"
"os"
)
func main() {
file, err := os.OpenFile("xx.txt", os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0666)
if err != nil {
fmt.Println("open file failed, err:", err)
return
}
defer file.Close()
writer := bufio.NewWriter(file)
for i := 0; i < 10; i++ {
writer.WriteString("Hello golang.\n") // 将数据先写入缓存
}
writer.Flush() // 将缓存中的内容写入文件
}
|
3.3 ioutil.WriteFile
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
|
package main
import (
"fmt"
"io/ioutil"
)
func main() {
str := "Hello Golang."
err := ioutil.WriteFile("./test.txt", []byte(str), 0666)
if err != nil {
fmt.Println("write file failed, err:", err)
return
}
}
|
四、综合示例
使用文件操作相关知识，模拟实现linux平台 cat 命令的功能
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
|
package main
import (
"bufio"
"flag"
"fmt"
"io"
"os"
)
// cat命令实现
func cat(r *bufio.Reader) {
for {
buf, err := r.ReadBytes('\n') //注意是字符
if err == io.EOF {
// 退出之前将已读到的内容输出
fmt.Fprintf(os.Stdout, "%s", buf)
break
}
fmt.Fprintf(os.Stdout, "%s", buf)
}
}
func main() {
flag.Parse() // 解析命令行参数
if flag.NArg() == 0 {
// 如果没有参数默认从标准输入读取内容
cat(bufio.NewReader(os.Stdin))
}
// 依次读取每个指定文件的内容并打印到终端
for i := 0; i < flag.NArg(); i++ {
f, err := os.Open(flag.Arg(i))
if err != nil {
fmt.Fprintf(os.Stdout, "reading from %s failed, err:%v\n", flag.Arg(i), err)
continue
}
cat(bufio.NewReader(f))
}
}
|