go语言学习笔记 之 IO

go文档：https://studygolang.com/pkgdoc

读取文件

package main

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

func main() {
	// 获取文件
	// 相对路径并不是在当前文件同级目录下，而是在项目根目录下
	file, err := os.Open("test.txt")
	if err != nil {
		fmt.Println("error", err)
		return
	}
	defer file.Close()

	// 读取文件内数据
	//line := make([]byte, 1024)
	//b, err := file.Read(line)
	//if err == nil {
	//	// 读取一次
	//	fmt.Println("读取了一下",string(line[0:b]))
	//}

	// 带缓冲的reader，一部分一部分读取
	reader := bufio.NewReader(file)
	for {
		line, err := reader.ReadString('\n') // 读取到\n结束
		if err == io.EOF{
			fmt.Print("读取到了最后一行数据",line)
			break
		}
		fmt.Print("读取到了一行数据",line)
	}

	// 使用ioutil包中的方法读取文件所有内容
	fileContent, err := ioutil.ReadFile("test.txt")
	if err == nil {
		fmt.Println(string(fileContent))
	}
}

写文件

package main

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

func main() {
	// 以只写方式打开，如果不存在则创建一个，写入方式为追加写，默认为覆盖写
	file, err := os.OpenFile("testWrite.txt", os.O_WRONLY |os.O_APPEND|os.O_CREATE, 0666)
	if err != nil {
		fmt.Println("error = ", err)
		return
	}
	defer file.Close()

	// 写入数据
	writer := bufio.NewWriter(file)

	for i := 0; i < 10; i++ {
		writeString, err := writer.WriteString(fmt.Sprintf("%d行数据hhhhhh\n", i))
		if err != nil {
			break
		}
		fmt.Printf("写入了%d字节\n", writeString)
	}
	err = writer.Flush()
	if err != nil {
		fmt.Println("写入失败", err)
	}
}