Go语言文件操作

File方法读取文件

文件的打开和关闭
os.Open()函数能够打开一个文件，返回一个*File和一个err。对得到的文件实例调用close()方法能够关闭文件。
为了防止文件忘记关闭，我们通常使用defer注册文件关闭语句。

读取文件
file.Read()
Read方法定义如下：

func (f *File) Read(b []byte) (n int, err error)

它接收一个字节切片，返回读取的字节数和可能的具体错误，读到文件末尾时会返回0和io.EOF。

package main

import (
	"fmt"
	"io"
	"os"
)

func main() {
	file, err := os.Open("test.txt")
	if err != nil {
		fmt.Println("Open file error", err)
		return
	}
	defer file.Close()
	buf := make([]byte, 128)
	var content []byte
	for {
		n, err1 := file.Read(buf)
		if err1 == io.EOF {
			break
		}
		if err1 != nil {
			fmt.Println("Read file error", err)
			return
		}
		content = append(content, buf[0:n]...)
	}
	fmt.Println(string(content))
}

每次读取128byte的数据，成功读取之后添加给切片content，当读到文件末尾时候，跳出循环，打印content。

File方法写入操作

os.OpenFile()函数能够以指定模式打开文件，从而实现文件写入相关功能。

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	追加

Write和WriteString

package main

import (
	"fmt"
	"os"
)

func main() {
	file, err := os.OpenFile("test.txt", os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0755)
	if err != nil {
		fmt.Println("Open file error", err)
		return
	}
	defer file.Close()
	buf := "如果有一天我变得很有钱"
	file.Write([]byte(buf))
	file.WriteString("我就可以吧所以人都留在我身边，每天吃吃喝喝，聊聊天")
}

file.Write和file.WriteString都返回写入的字节数和错误类型两个值，区别在于，Write需要传入[]byte类型，WriteString直接传入字符串类型。

bufio读取文件

bufio是在file的基础上封装了一层API，支持更多的功能。

package main

import (
	"bufio"
	"fmt"
	"io"
	"os"
)

func main() {
	file, err := os.Open("test.txt")
	if err != nil {
		fmt.Println("Open file error", err)
		return
	}
	defer file.Close()
	reader := bufio.NewReader(file)
	for {
		buf, err1 := reader.ReadString('\n') //读到\n
		if err1 == io.EOF {
			fmt.Print(buf)
			break
		}
		if err1 != nil {
			fmt.Println("Read file error", err)
			return
		}
		fmt.Print(buf)
	}
}

bufio写入文件

同file的方法一样，如果需要对文件进行写入操作，那么必须使用OpenFile函数打开文件是否赋予写入权限和打开方式。

package main

import (
	"bufio"
	"fmt"
	"os"
)

func main() {
	file, err := os.OpenFile("test.txt", os.O_APPEND|os.O_WRONLY, 0644)
	if err != nil {
		fmt.Println("Open file error", err)
		return
	}
	defer file.Close()
	writer := bufio.NewWriter(file)
	_, err1 := writer.WriteString("不同担心明天或离别")
	if err1 != nil {
		fmt.Println("WriteString file error", err)
		return
	}
	writer.Flush() //写入存在缓冲区，此操作将缓冲区写入文件
}

ioutil读取整个文件

io/ioutil包的ReadFile方法能够读取完整的文件，只需要将文件名作为参数传入。

package main

import (
	"fmt"
	"io/ioutil"
)

func main() {
	content, err := ioutil.ReadFile("test.txt")
	if err != nil {
		fmt.Println("ioutil.ReadFile error:", err)
	}
	fmt.Println(string(content))
}

ioutil写入文件

package main

import (
	"fmt"
	"io/ioutil"
)

func main() {
	err := ioutil.WriteFile("test.txt", []byte("hello world"), 0644)
	if err != nil {
		fmt.Println("ioutil.WriteFile error:", err)
	}
}