讀取整個文件 整個文件讀取到記憶體是最基本的文件操作之一。這需要使用 包中的 函數。 以上使用的是運行時指定參數 來指定讀取的文件,也可以使用文件的絕對路徑。 分塊讀取文件 當文件非常大時,尤其在記憶體不足的情況下,把整個文件都讀入記憶體是沒有意義的。 更好的方法是分塊讀取文件。這可以使用 包來完成。 利 ...
讀取整個文件
- 整個文件讀取到記憶體是最基本的文件操作之一。這需要使用
ioutil
包中的ReadFile
函數。
package main
import (
"flag"
"fmt"
"io/ioutil"
)
func main() {
fptr := flag.String("path", "test.txt", "待讀取文件的路徑")
flag.Parse()
data, err := ioutil.ReadFile(*fptr)
if err != nil {
fmt.Println(err)
}
fmt.Println(string(data))
}
- 以上使用的是運行時指定參數
-path=test1.txt
來指定讀取的文件,也可以使用文件的絕對路徑。
分塊讀取文件
- 當文件非常大時,尤其在記憶體不足的情況下,把整個文件都讀入記憶體是沒有意義的。
- 更好的方法是分塊讀取文件。這可以使用
bufio
包來完成。 - 利用
bufio
包的'NewReader()'函數實現,其中Read
方法可以讀取文件內容到緩衝的位元組數組中去。
package main
import (
"bufio"
"fmt"
"io"
"os"
)
func main() {
// 打開文件
f, err := os.Open("test.txt")
if err != nil {
fmt.Println(err)
}
// 延遲關閉文件
defer func() {
if err := f.Close(); err != nil {
fmt.Println(err)
}
}()
// 文件讀取器
r := bufio.NewReader(f)
buf := make([]byte, 3) // 便於演示 設置的比較小
str := ""
for {
_, err := r.Read(buf)
if err != nil {
if err == io.EOF {
fmt.Println("文件讀取完畢!")
} else {
fmt.Println(err)
}
break
}
fmt.Println(string(buf))
str += string(buf)
}
fmt.Println("完整結果如下:\n", str)
}
逐行讀取文件
- 當文件比較大時還可以採用逐行讀取的方式實現。
- 利用
bufio
包的'NewScanner()'函數實現,其中Scan
方法可以判斷是否還有下一行,
Text
方法可以獲取當前行,Err
方法可以獲取錯誤信息。
package main
import (
"bufio"
"fmt"
"os"
)
func main() {
// 打開文件
f, err := os.Open("test.txt")
if err != nil {
fmt.Println(err)
}
// 延遲關閉文件
defer func() {
if err := f.Close(); err != nil {
fmt.Println(err)
}
}()
// 文件讀取器
s := bufio.NewScanner(f)
str := ""
for s.Scan() {
fmt.Println(s.Text(), "\n-------分隔符--------")
str += s.Text()
}
if s.Err() != nil {
fmt.Println(s.Err().Error())
}
fmt.Println("完整結果如下:\n", str)
}
字元串寫入文件
- 使用 create 創建一個指定名字的文件。
如果這個文件已經存在,那麼 create 函數將截斷這個文件。該函數返回一個文件描述符。
package main
import (
"fmt"
"os"
)
func main() {
f, err := os.Create("123.txt")
if err != nil {
fmt.Println(err)
}
defer func() {
if err = f.Close(); err != nil {
fmt.Println(err)
}
}()
_, err = f.WriteString("Hello World!")
if err != nil {
fmt.Println(err)
}
fmt.Println("文件寫入成功!")
}
位元組寫入文件
- 使用
Write
方法可以寫入位元組數組數據。一般為字元對應的UTF-8的10進位編碼。
package main
import (
"fmt"
"os"
)
func main() {
f, err := os.Create("123.txt")
if err != nil {
fmt.Println(err)
}
defer func() {
if err = f.Close(); err != nil {
fmt.Println(err)
}
}()
bytes := []byte{104, 101, 108, 108, 111, 32, 119, 111, 114, 108, 100}
_, err = f.Write(bytes)
if err != nil {
fmt.Println(err)
}
fmt.Println("文件寫入完成")
}
逐行寫入文件
- 使用
fmt.Fprintln
函數實現文件的逐行寫入。
package main
import (
"fmt"
"os"
)
func main() {
f, err := os.Create("123.txt")
if err != nil {
fmt.Println(err)
}
defer func() {
if err = f.Close(); err != nil {
fmt.Println(err)
}
}()
lines := []string{"Hello", "GoLang", "World"}
for _, v := range lines {
_, err = fmt.Fprintln(f, v)
if err != nil {
fmt.Println(err)
}
}
fmt.Println("文件寫入完成.")
}
追加文件內容
os.OpenFile(name string, flag int, perm FileMode) (*File, error)
函數可以指定文件操作方式。
package main
import (
"fmt"
"os"
)
func main() {
f, err := os.OpenFile("123.txt", os.O_APPEND|os.O_RDONLY, 0644)
if err != nil {
fmt.Println(err)
}
defer func() {
if err = f.Close(); err != nil {
fmt.Println(err)
}
}()
lines := []string{"123", "abc", "!@#"}
for _, v := range lines {
_, err = fmt.Fprintln(f, v)
if err != nil {
fmt.Println(err)
}
}
fmt.Println("文件追加寫入完成.")
}
併發寫文件
- 註意文件寫入時的競態條件。
package main
import (
"fmt"
"math/rand"
"os"
"sync"
)
func main() {
chNum := make(chan int)
chExit := make(chan bool)
wg := sync.WaitGroup{}
// 生產者
for i := 0; i < 100; i++ {
wg.Add(1)
go makeRandom(chNum, &wg)
}
// 消費者
go writeFile(chNum, chExit)
wg.Wait()
close(chNum)
if <-chExit {
fmt.Println("文件寫入成功.")
} else {
fmt.Println("文件寫入失敗.")
}
}
func makeRandom(chNum chan int, wg *sync.WaitGroup) {
chNum <- rand.Intn(1000)
wg.Done()
}
func writeFile(chNum chan int, chExit chan bool) {
f, err := os.Create("123.txt")
if err != nil {
fmt.Println(err)
}
defer func() {
if err = f.Close(); err != nil {
fmt.Println(err)
}
}()
for val := range chNum {
_, err = fmt.Fprintln(f, val)
if err != nil {
fmt.Println(err)
}
}
chExit <- true
}