本文主要介紹了Go語(yǔ)言中文件讀寫的相關(guān)操作。
文件是什么？
計(jì)算機(jī)中的文件是存儲(chǔ)在外部介質(zhì)（通常是磁盤）上的數(shù)據(jù)集合，文件分為文本文件和二進(jìn)制文件铜异。
打開(kāi)和關(guān)閉文件
os.Open()函數(shù)能夠打開(kāi)一個(gè)文件，返回一個(gè)*File和一個(gè)err秸架。對(duì)得到的文件實(shí)例調(diào)用close()方法能夠關(guān)閉文件揍庄。
package main
import (
"fmt"
"os"
)
func main() {
// 只讀方式打開(kāi)當(dāng)前目錄下的main.go文件
file, err := os.Open("./main.go")
if err != nil {
fmt.Println("open file failed!, err:", err)
return
}
// 關(guān)閉文件
file.Close()
}
為了防止文件忘記關(guān)閉，我們通常使用defer注冊(cè)文件關(guān)閉語(yǔ)句东抹。
讀取文件
file.Read()
基本使用
Read方法定義如下：
func (f *File) Read(b []byte) (n int, err error)
它接收一個(gè)字節(jié)切片蚂子，返回讀取的字節(jié)數(shù)和可能的具體錯(cuò)誤沃测，讀到文件末尾時(shí)會(huì)返回0和io.EOF。 舉個(gè)例子：
func main() {
// 只讀方式打開(kāi)當(dāng)前目錄下的main.go文件
file, err := os.Open("./main.go")
if err != nil {
fmt.Println("open file failed!, err:", err)
return
}
defer file.Close()
// 使用Read方法讀取數(shù)據(jù)
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字節(jié)數(shù)據(jù)\n", n)
fmt.Println(string(tmp[:n]))
}
循環(huán)讀取
使用for循環(huán)讀取文件中的所有數(shù)據(jù)食茎。
func main() {
// 只讀方式打開(kāi)當(dāng)前目錄下的main.go文件
file, err := os.Open("./main.go")
if err != nil {
fmt.Println("open file failed!, err:", err)
return
}
defer file.Close()
// 循環(huán)讀取文件
var content []byte
var tmp = make([]byte, 128)
for {
n, err := file.Read(tmp)
if err == io.EOF {
fmt.Println("文件讀完了")
break
}
if err != nil {
fmt.Println("read file failed, err:", err)
return
}
content = append(content, tmp[:n]...)
}
fmt.Println(string(content))
}
bufio讀取文件
bufio是在file的基礎(chǔ)上封裝了一層API蒂破，支持更多的功能。
package main
import (
"bufio"
"fmt"
"io"
"os"
)
// bufio按行讀取示例
func main() {
file, err := os.Open("./xx.txt")
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)
}
}
ioutil讀取整個(gè)文件
io/ioutil包的ReadFile方法能夠讀取完整的文件别渔，只需要將文件名作為參數(shù)傳入附迷。
package main
import (
"fmt"
"io/ioutil"
)
// ioutil.ReadFile讀取整個(gè)文件
func main() {
content, err := ioutil.ReadFile("./main.go")
if err != nil {
fmt.Println("read file failed, err:", err)
return
}
fmt.Println(string(content))
}
文件寫入操作
os.OpenFile()函數(shù)能夠以指定模式打開(kāi)文件，從而實(shí)現(xiàn)文件寫入相關(guān)功能钠糊。
func OpenFile(name string, flag int, perm FileMode) (*File, error) {
...
}
其中：
name：要打開(kāi)的文件名 flag：打開(kāi)文件的模式挟秤。 模式有以下幾種：
| 模式 | 含義 |
|---|---|
| os.O_WRONLY | 只寫 |
| os.O_CREATE | 創(chuàng)建文件 |
| os.O_RDONLY | 只讀 |
| os.O_RDWR | 讀寫 |
| os.O_TRUNC | 清空 |
| os.O_APPEND | 追加 |
perm：文件權(quán)限壹哺，一個(gè)八進(jìn)制數(shù)抄伍。r（讀）04，w（寫）02管宵，x（執(zhí)行）01截珍。
Write和WriteString
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()
str := "hello 沙河"
file.Write([]byte(str)) //寫入字節(jié)切片數(shù)據(jù)
file.WriteString("hello 小王子") //直接寫入字符串?dāng)?shù)據(jù)
}
bufio.NewWriter
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沙河\n") //將數(shù)據(jù)先寫入緩存
}
writer.Flush() //將緩存中的內(nèi)容寫入文件
}
ioutil.WriteFile
func main() {
str := "hello 沙河"
err := ioutil.WriteFile("./xx.txt", []byte(str), 0666)
if err != nil {
fmt.Println("write file failed, err:", err)
return
}
}
練習(xí)
copyFile
借助io.Copy()實(shí)現(xiàn)一個(gè)拷貝文件函數(shù)。
// CopyFile 拷貝文件函數(shù)
func CopyFile(dstName, srcName string) (written int64, err error) {
// 以讀方式打開(kāi)源文件
src, err := os.Open(srcName)
if err != nil {
fmt.Printf("open %s failed, err:%v.\n", srcName, err)
return
}
defer src.Close()
// 以寫|創(chuàng)建的方式打開(kāi)目標(biāo)文件
dst, err := os.OpenFile(dstName, os.O_WRONLY|os.O_CREATE, 0644)
if err != nil {
fmt.Printf("open %s failed, err:%v.\n", dstName, err)
return
}
defer dst.Close()
return io.Copy(dst, src) //調(diào)用io.Copy()拷貝內(nèi)容
}
func main() {
_, err := CopyFile("dst.txt", "src.txt")
if err != nil {
fmt.Println("copy file failed, err:", err)
return
}
fmt.Println("copy done!")
}
實(shí)現(xiàn)一個(gè)cat命令
使用文件操作相關(guān)知識(shí)箩朴，模擬實(shí)現(xiàn)linux平臺(tái)cat命令的功能岗喉。
package main
import (
"bufio"
"flag"
"fmt"
"io"
"os"
)
// cat命令實(shí)現(xiàn)
func cat(r *bufio.Reader) {
for {
buf, err := r.ReadBytes('\n') //注意是字符
if err == io.EOF {
break
}
fmt.Fprintf(os.Stdout, "%s", buf)
}
}
func main() {
flag.Parse() // 解析命令行參數(shù)
if flag.NArg() == 0 {
// 如果沒(méi)有參數(shù)默認(rèn)從標(biāo)準(zhǔn)輸入讀取內(nèi)容
cat(bufio.NewReader(os.Stdin))
}
// 依次讀取每個(gè)指定文件的內(nèi)容并打印到終端
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))
}
}
